0g-chain/x/committee/keeper/querier.go

175 lines
5.3 KiB
Go
Raw Normal View History

2020-03-13 15:11:31 +00:00
package keeper
import (
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
2020-04-27 18:19:05 +00:00
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
2020-04-30 14:23:41 +00:00
2020-03-13 15:11:31 +00:00
abci "github.com/tendermint/tendermint/abci/types"
"github.com/kava-labs/kava/x/committee/types"
)
// NewQuerier creates a new gov Querier instance
func NewQuerier(keeper Keeper) sdk.Querier {
2020-04-27 18:19:05 +00:00
return func(ctx sdk.Context, path []string, req abci.RequestQuery) ([]byte, error) {
2020-03-13 15:11:31 +00:00
switch path[0] {
case types.QueryCommittees:
return queryCommittees(ctx, path[1:], req, keeper)
case types.QueryCommittee:
return queryCommittee(ctx, path[1:], req, keeper)
case types.QueryProposals:
return queryProposals(ctx, path[1:], req, keeper)
case types.QueryProposal:
return queryProposal(ctx, path[1:], req, keeper)
case types.QueryVotes:
return queryVotes(ctx, path[1:], req, keeper)
case types.QueryVote:
return queryVote(ctx, path[1:], req, keeper)
case types.QueryTally:
return queryTally(ctx, path[1:], req, keeper)
default:
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrapf(sdkerrors.ErrUnknownRequest, "unknown %s query endpoint", types.ModuleName)
2020-03-13 15:11:31 +00:00
}
}
}
2020-03-30 13:06:31 +00:00
// ------------------------------------------
// Committees
// ------------------------------------------
2020-03-13 15:11:31 +00:00
2020-04-27 18:19:05 +00:00
func queryCommittees(ctx sdk.Context, path []string, _ abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
2020-04-27 14:04:47 +00:00
committees := keeper.GetCommittees(ctx)
2020-03-13 15:11:31 +00:00
bz, err := codec.MarshalJSONIndent(keeper.cdc, committees)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-04-27 18:19:05 +00:00
func queryCommittee(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryCommitteeParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
committee, found := keeper.GetCommittee(ctx, params.CommitteeID)
if !found {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrapf(types.ErrUnknownCommittee, "%d", params.CommitteeID)
2020-03-13 15:11:31 +00:00
}
bz, err := codec.MarshalJSONIndent(keeper.cdc, committee)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-03-30 13:06:31 +00:00
// ------------------------------------------
// Proposals
// ------------------------------------------
2020-03-13 15:11:31 +00:00
2020-04-27 18:19:05 +00:00
func queryProposals(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryCommitteeParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
2020-04-27 14:04:47 +00:00
proposals := keeper.GetProposalsByCommittee(ctx, params.CommitteeID)
2020-03-13 15:11:31 +00:00
bz, err := codec.MarshalJSONIndent(keeper.cdc, proposals)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-04-27 18:19:05 +00:00
func queryProposal(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryProposalParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
proposal, found := keeper.GetProposal(ctx, params.ProposalID)
if !found {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrapf(types.ErrUnknownProposal, "%d", params.ProposalID)
2020-03-13 15:11:31 +00:00
}
bz, err := codec.MarshalJSONIndent(keeper.cdc, proposal)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-03-30 13:06:31 +00:00
// ------------------------------------------
// Votes
// ------------------------------------------
2020-03-13 15:11:31 +00:00
2020-04-27 18:19:05 +00:00
func queryVotes(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryProposalParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
2020-04-27 14:04:47 +00:00
votes := keeper.GetVotesByProposal(ctx, params.ProposalID)
2020-03-13 15:11:31 +00:00
bz, err := codec.MarshalJSONIndent(keeper.cdc, votes)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-04-27 18:19:05 +00:00
func queryVote(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryVoteParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
vote, found := keeper.GetVote(ctx, params.ProposalID, params.Voter)
if !found {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrapf(types.ErrUnknownVote, "proposal id: %d, voter: %s", params.ProposalID, params.Voter)
2020-03-13 15:11:31 +00:00
}
bz, err := codec.MarshalJSONIndent(keeper.cdc, vote)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}
2020-03-30 13:06:31 +00:00
// ------------------------------------------
// Tally
// ------------------------------------------
2020-03-13 15:11:31 +00:00
2020-04-27 18:19:05 +00:00
func queryTally(ctx sdk.Context, path []string, req abci.RequestQuery, keeper Keeper) ([]byte, error) {
2020-03-13 15:11:31 +00:00
var params types.QueryProposalParams
err := keeper.cdc.UnmarshalJSON(req.Data, &params)
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONUnmarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
2020-03-26 20:17:49 +00:00
_, found := keeper.GetProposal(ctx, params.ProposalID)
2020-03-13 15:11:31 +00:00
if !found {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrapf(types.ErrUnknownProposal, "%d", params.ProposalID)
2020-03-13 15:11:31 +00:00
}
2020-03-26 20:17:49 +00:00
numVotes := keeper.TallyVotes(ctx, params.ProposalID)
2020-03-13 15:11:31 +00:00
2020-03-26 20:17:49 +00:00
bz, err := codec.MarshalJSONIndent(keeper.cdc, numVotes)
2020-03-13 15:11:31 +00:00
if err != nil {
2020-04-27 18:19:05 +00:00
return nil, sdkerrors.Wrap(sdkerrors.ErrJSONMarshal, err.Error())
2020-03-13 15:11:31 +00:00
}
return bz, nil
}