/
grpc_query_pool_reserves.go
88 lines (73 loc) · 2.32 KB
/
grpc_query_pool_reserves.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package keeper
import (
"context"
"github.com/cosmos/cosmos-sdk/store/prefix"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/types/query"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"github.com/neutron-org/neutron/v2/x/dex/types"
)
func (k Keeper) PoolReservesAll(
goCtx context.Context,
req *types.QueryAllPoolReservesRequest,
) (*types.QueryAllPoolReservesResponse, error) {
if req == nil {
return nil, status.Error(codes.InvalidArgument, "invalid request")
}
ctx := sdk.UnwrapSDKContext(goCtx)
pairID, err := types.NewPairIDFromCanonicalString(req.PairId)
if err != nil {
return nil, err
}
tradePairID := types.NewTradePairIDFromMaker(pairID, req.TokenIn)
store := ctx.KVStore(k.storeKey)
PoolReservesStore := prefix.NewStore(store, types.TickLiquidityPrefix(tradePairID))
var poolReserves []*types.PoolReserves
pageRes, err := query.FilteredPaginate(
PoolReservesStore,
req.Pagination,
func(key, value []byte, accum bool) (hit bool, err error) {
var tick types.TickLiquidity
if err := k.cdc.Unmarshal(value, &tick); err != nil {
return false, err
}
reserves := tick.GetPoolReserves()
// Check if this is a LimitOrderTranche and not PoolReserves
if reserves != nil {
if accum {
poolReserves = append(poolReserves, reserves)
}
return true, nil
}
return false, nil
})
if err != nil {
return nil, status.Error(codes.Internal, err.Error())
}
return &types.QueryAllPoolReservesResponse{PoolReserves: poolReserves, Pagination: pageRes}, nil
}
func (k Keeper) PoolReserves(
goCtx context.Context,
req *types.QueryGetPoolReservesRequest,
) (*types.QueryGetPoolReservesResponse, error) {
if req == nil {
return nil, status.Error(codes.InvalidArgument, "invalid request")
}
ctx := sdk.UnwrapSDKContext(goCtx)
pairID, err := types.NewPairIDFromCanonicalString(req.PairId)
if err != nil {
return nil, err
}
tradePairID := types.NewTradePairIDFromMaker(pairID, req.TokenIn)
poolReservesID := &types.PoolReservesKey{
TradePairId: tradePairID,
TickIndexTakerToMaker: req.TickIndex,
Fee: req.Fee,
}
val, found := k.GetPoolReserves(ctx, poolReservesID)
if !found {
return nil, status.Error(codes.NotFound, "not found")
}
return &types.QueryGetPoolReservesResponse{PoolReserves: val}, nil
}