Add new query

osmosis-labs · Nov 9, 2023 · bb24d9f · bb24d9f
1 parent 7c5bc56
commit bb24d9f
Show file tree

Hide file tree

Showing 6 changed files with 621 additions and 130 deletions.
diff --git a/go.mod b/go.mod
@@ -8,7 +8,7 @@ require (
 	github.com/CosmWasm/wasmd v0.31.0
 	github.com/cometbft/cometbft v0.38.0
 	github.com/cosmos/cosmos-proto v1.0.0-beta.3
-	github.com/cosmos/cosmos-sdk v0.47.5
+	github.com/cosmos/cosmos-sdk v0.50.1
 	github.com/cosmos/go-bip39 v1.0.0
 	github.com/cosmos/ibc-apps/middleware/packet-forward-middleware/v4 v4.1.0
 	github.com/cosmos/ibc-apps/modules/async-icq/v4 v4.1.0
@@ -43,6 +43,7 @@ require (
 	google.golang.org/genproto/googleapis/api v0.0.0-20230913181813-007df8e322eb
 	google.golang.org/grpc v1.58.2
 	gopkg.in/yaml.v2 v2.4.0
+	gotest.tools v2.2.0+incompatible
 	mvdan.cc/gofumpt v0.5.0
 )
 

diff --git a/proto/osmosis/superfluid/query.proto b/proto/osmosis/superfluid/query.proto
@@ -141,6 +141,10 @@ service Query {
                                    "account_undelegating_cl_positions/"
                                    "{delegator_address}";
   }
+
+  rpc RestSupply(QueryRestSupplyRequest) returns (QueryRestSupplyResponse) {
+    option (google.api.http).get = "/osmosis/superfluid/v1beta1/supply";
+  }
 }
 
 message QueryParamsRequest {}
@@ -320,3 +324,11 @@ message UserConcentratedSuperfluidPositionsUndelegatingResponse {
   repeated ConcentratedPoolUserPositionRecord cl_pool_user_position_records = 1
       [ (gogoproto.nullable) = false ];
 }
+
+// THIS QUERY IS TEMPORARY
+message QueryRestSupplyRequest { string denom = 1; }
+
+message QueryRestSupplyResponse {
+  // amount is the supply of the coin.
+  cosmos.base.v1beta1.Coin amount = 1 [ (gogoproto.nullable) = false ];
+}
diff --git a/x/superfluid/keeper/grpc_query.go b/x/superfluid/keeper/grpc_query.go
@@ -704,3 +704,13 @@ func (q Querier) filterConcentratedPositionLocks(ctx sdk.Context, positions []mo
 	}
 	return clPoolUserPositionRecords, nil
 }
+
+// TEMPORARY CODE
+func (q Querier) RestSupply(goCtx context.Context, req *types.QueryRestSupplyRequest) (*types.QueryRestSupplyResponse, error) {
+	if req == nil {
+		return nil, status.Error(codes.InvalidArgument, "empty request")
+	}
+
+	supply := q.bk.GetSupply(sdk.UnwrapSDKContext(goCtx), req.Denom)
+	return &types.QueryRestSupplyResponse{Amount: supply}, nil
+}
diff --git a/x/superfluid/types/expected_keepers.go b/x/superfluid/types/expected_keepers.go
@@ -67,6 +67,7 @@ type BankKeeper interface {
 	AddSupplyOffset(ctx sdk.Context, denom string, offsetAmount osmomath.Int)
 	SendCoinsFromAccountToModule(ctx sdk.Context, senderAddr sdk.AccAddress, recipientModule string, amt sdk.Coins) error
 	SendCoinsFromModuleToAccount(ctx sdk.Context, senderModule string, recipientAddr sdk.AccAddress, amt sdk.Coins) error
+	GetSupply(ctx sdk.Context, denom string) sdk.Coin
 }
 
 // StakingKeeper expected staking keeper.