WIP: adding quota v2

umee-network · Oct 26, 2023 · 5578d8c · 5578d8c
1 parent ff4a594
commit 5578d8c
Show file tree

Hide file tree

Showing 10 changed files with 355 additions and 87 deletions.
diff --git a/proto/umee/uibc/v1/genesis.proto b/proto/umee/uibc/v1/genesis.proto
@@ -17,7 +17,6 @@ message GenesisState {
     (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.DecCoins",
     (gogoproto.nullable)     = false
   ];
-
   // total_outflow_sum defines the total outflow sum of ibc-transfer in USD.
   string total_outflow_sum = 3 [
     (cosmos_proto.scalar)  = "cosmos.Dec",
@@ -31,4 +30,15 @@ message GenesisState {
     (gogoproto.jsontag)  = "quota_duration,omitempty",
     (gogoproto.moretags) = "yaml:\"quota_expires\""
   ];
+  // inflows defines inflow amount of denoms
+  repeated cosmos.base.v1beta1.DecCoin inflows = 5 [
+    (gogoproto.castrepeated) = "github.com/cosmos/cosmos-sdk/types.DecCoins",
+    (gogoproto.nullable)     = false
+  ];
+  // total_inflow_sum defines the total inflow sum of ibc-transfer in USD.
+  string total_inflow_sum = 6 [
+    (cosmos_proto.scalar)  = "cosmos.Dec",
+    (gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Dec",
+    (gogoproto.nullable)   = false
+  ];
 }
diff --git a/proto/umee/uibc/v1/quota.proto b/proto/umee/uibc/v1/quota.proto
@@ -30,6 +30,12 @@ message Params {
     (gogoproto.jsontag)     = "quota_duration,omitempty",
     (gogoproto.moretags)    = "yaml:\"quota_duration\""
   ];
+  // total_inflow_quota defines the total inflow limit of ibc-transfer in USD
+  string total_inflow_quota = 5 [
+    (cosmos_proto.scalar)  = "cosmos.Dec",
+    (gogoproto.customtype) = "github.com/cosmos/cosmos-sdk/types.Dec",
+    (gogoproto.nullable)   = false
+  ];
 }
 
 // IBCTransferStatus status of ibc-transfer quota check for inflow and outflow

diff --git a/x/uibc/genesis.pb.go b/x/uibc/genesis.pb.go
diff --git a/x/uibc/params.go b/x/uibc/params.go
@@ -10,10 +10,11 @@ import (
 // DefaultParams returns default genesis params
 func DefaultParams() Params {
 	return Params{
-		IbcStatus:     IBCTransferStatus_IBC_TRANSFER_STATUS_QUOTA_ENABLED,
-		TotalQuota:    sdk.NewDec(1_000_000),
-		TokenQuota:    sdk.NewDec(600_000),
-		QuotaDuration: time.Second * 60 * 60 * 24, // 24h
+		IbcStatus:        IBCTransferStatus_IBC_TRANSFER_STATUS_QUOTA_ENABLED,
+		TotalQuota:       sdk.NewDec(1_000_000),
+		TokenQuota:       sdk.NewDec(600_000),
+		QuotaDuration:    time.Second * 60 * 60 * 24, // 24h
+		TotalInflowQuota: sdk.NewDec(1_000_000),
 	}
 }
 
@@ -30,6 +31,9 @@ func (p Params) Validate() error {
 	if err := validateQuota(p.TokenQuota, "quota per token"); err != nil {
 		return err
 	}
+	if err := validateQuota(p.TotalInflowQuota, "total inflow quota"); err != nil {
+		return err
+	}
 	if p.TotalQuota.LT(p.TokenQuota) {
 		return fmt.Errorf("token quota shouldn't be less than quota per denom")
 	}