-
Notifications
You must be signed in to change notification settings - Fork 48
/
query_pool.go
67 lines (55 loc) · 1.87 KB
/
query_pool.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
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"
"github.com/elys-network/elys/x/amm/types"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
func (k Keeper) PoolExtraInfo(ctx sdk.Context, pool types.Pool) types.PoolExtraInfo {
tvl, _ := pool.TVL(ctx, k.oracleKeeper)
lpTokenPrice, _ := pool.LpTokenPrice(ctx, k.oracleKeeper)
return types.PoolExtraInfo{
Tvl: tvl,
LpTokenPrice: lpTokenPrice,
}
}
func (k Keeper) PoolAll(goCtx context.Context, req *types.QueryAllPoolRequest) (*types.QueryAllPoolResponse, error) {
if req == nil {
return nil, status.Error(codes.InvalidArgument, "invalid request")
}
var pools []types.Pool
var extraInfos []types.PoolExtraInfo
ctx := sdk.UnwrapSDKContext(goCtx)
store := ctx.KVStore(k.storeKey)
poolStore := prefix.NewStore(store, types.KeyPrefix(types.PoolKeyPrefix))
pageRes, err := query.Paginate(poolStore, req.Pagination, func(key []byte, value []byte) error {
var pool types.Pool
if err := k.cdc.Unmarshal(value, &pool); err != nil {
return err
}
pools = append(pools, pool)
extraInfos = append(extraInfos, k.PoolExtraInfo(ctx, pool))
return nil
})
if err != nil {
return nil, status.Error(codes.Internal, err.Error())
}
return &types.QueryAllPoolResponse{Pool: pools, ExtraInfos: extraInfos, Pagination: pageRes}, nil
}
func (k Keeper) Pool(goCtx context.Context, req *types.QueryGetPoolRequest) (*types.QueryGetPoolResponse, error) {
if req == nil {
return nil, status.Error(codes.InvalidArgument, "invalid request")
}
ctx := sdk.UnwrapSDKContext(goCtx)
pool, found := k.GetPool(ctx, req.PoolId)
if !found {
return nil, status.Error(codes.NotFound, "not found")
}
return &types.QueryGetPoolResponse{
Pool: pool,
ExtraInfo: k.PoolExtraInfo(ctx, pool),
}, nil
}