/
querier.go
78 lines (69 loc) · 2.5 KB
/
querier.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
package wasm
import (
"encoding/json"
"cosmossdk.io/errors"
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
"github.com/desmos-labs/desmos/v6/cosmwasm"
relationshipskeeper "github.com/desmos-labs/desmos/v6/x/relationships/keeper"
"github.com/desmos-labs/desmos/v6/x/relationships/types"
)
var _ cosmwasm.Querier = RelationshipsWasmQuerier{}
type RelationshipsWasmQuerier struct {
relationshipsKeeper relationshipskeeper.Keeper
cdc codec.Codec
}
func NewRelationshipsWasmQuerier(relationshipsKeeper relationshipskeeper.Keeper, cdc codec.Codec) RelationshipsWasmQuerier {
return RelationshipsWasmQuerier{
relationshipsKeeper: relationshipsKeeper,
cdc: cdc,
}
}
func (querier RelationshipsWasmQuerier) QueryCustom(ctx sdk.Context, data json.RawMessage) ([]byte, error) {
var query types.RelationshipsQuery
err := json.Unmarshal(data, &query)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
}
switch {
case query.Relationships != nil:
return querier.handleRelationshipsRequest(ctx, *query.Relationships)
case query.Blocks != nil:
return querier.handleBlocksRequest(ctx, *query.Blocks)
default:
return nil, sdkerrors.ErrInvalidRequest
}
}
func (querier RelationshipsWasmQuerier) handleRelationshipsRequest(ctx sdk.Context, request json.RawMessage) (bz []byte, err error) {
var req types.QueryRelationshipsRequest
err = querier.cdc.UnmarshalJSON(request, &req)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
}
res, err := querier.relationshipsKeeper.Relationships(sdk.WrapSDKContext(ctx), &req)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrInvalidRequest, err.Error())
}
bz, err = querier.cdc.MarshalJSON(res)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
}
return bz, nil
}
func (querier RelationshipsWasmQuerier) handleBlocksRequest(ctx sdk.Context, request json.RawMessage) (bz []byte, err error) {
var req types.QueryBlocksRequest
err = querier.cdc.UnmarshalJSON(request, &req)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
}
res, err := querier.relationshipsKeeper.Blocks(sdk.WrapSDKContext(ctx), &req)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrInvalidRequest, err.Error())
}
bz, err = querier.cdc.MarshalJSON(res)
if err != nil {
return nil, errors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
}
return bz, nil
}