Add TotalPools() query

proto/osmosis/gamm/v1beta1/query.proto

@@ -19,6 +19,9 @@ service Query {
   rpc Pools(QueryPoolsRequest) returns (QueryPoolsResponse) {
     option (google.api.http).get = "/osmosis/gamm/v1beta1/pools/all";
   }
+  rpc TotalPools(QueryTotalPoolsRequest) returns (QueryTotalPoolsResponse) {
+    option (google.api.http).get = "/osmosis/gamm/v1beta1/pools/total_pools";
+  }
   rpc PoolParams(QueryPoolParamsRequest) returns (QueryPoolParamsResponse) {
     option (google.api.http).get = "/osmosis/gamm/v1beta1/{poolId}/params";
   }
@@ -67,6 +70,13 @@ message QueryPoolsResponse {
   cosmos.base.query.v1beta1.PageResponse pagination = 2;
 }
 
+//=============================== TotalPools
+message QueryTotalPoolsRequest {
+}
+message QueryTotalPoolsResponse {
+  uint64 totalPools = 1 [ (gogoproto.moretags) = "yaml:\"total_pools\"" ];
+}
+
 //=============================== PoolParams
 message QueryPoolParamsRequest {
   uint64 poolId = 1 [ (gogoproto.moretags) = "yaml:\"pool_id\"" ];

x/gamm/keeper/grpc_query.go

@@ -104,6 +104,21 @@ func (k Keeper) Pools(
 	}, nil
 }
 
+func (k Keeper) TotalPools(
+	ctx context.Context,
+	req *types.QueryTotalPoolsRequest,
+) (*types.QueryTotalPoolsResponse, error) {
+	if req == nil {
+		return nil, status.Error(codes.InvalidArgument, "empty request")
+	}
+
+	sdkCtx := sdk.UnwrapSDKContext(ctx)
+
+	return &types.QueryTotalPoolsResponse{
+		TotalPools: k.getNextPoolNumber(sdkCtx) - 1,
+	}, nil
+}
+
 func (k Keeper) PoolParams(ctx context.Context, req *types.QueryPoolParamsRequest) (*types.QueryPoolParamsResponse, error) {
 	if req == nil {
 		return nil, status.Error(codes.InvalidArgument, "empty request")

x/gamm/keeper/grpc_query_test.go

@@ -88,6 +88,22 @@ func (suite *KeeperTestSuite) TestQueryPools() {
 	}
 }
 
+func (suite *KeeperTestSuite) TestQueryTotalPools1() {
+	res, err := suite.queryClient.TotalPools(gocontext.Background(), &types.QueryTotalPoolsRequest{})
+	suite.Require().NoError(err)
+	suite.Require().Equal(uint64(0), res.TotalPools)
+}
+
+func (suite *KeeperTestSuite) TestQueryTotalPools2() {
+	for i := 0; i < 10; i++ {
+		suite.preparePool()
+	}
+
+	res, err := suite.queryClient.TotalPools(gocontext.Background(), &types.QueryTotalPoolsRequest{})
+	suite.Require().NoError(err)
+	suite.Require().Equal(uint64(10), res.TotalPools)
+}
+
 func (suite *KeeperTestSuite) TestQueryPoolParams() {
 	queryClient := suite.queryClient