305 lines
10 KiB
Go
305 lines
10 KiB
Go
package querier
|
|
|
|
import (
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
keep "github.com/cosmos/cosmos-sdk/x/stake/keeper"
|
|
"github.com/cosmos/cosmos-sdk/x/stake/types"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
)
|
|
|
|
// query endpoints supported by the staking Querier
|
|
const (
|
|
QueryValidators = "validators"
|
|
QueryValidator = "validator"
|
|
QueryDelegatorDelegations = "delegatorDelegations"
|
|
QueryDelegatorUnbondingDelegations = "delegatorUnbondingDelegations"
|
|
QueryDelegatorRedelegations = "delegatorRedelegations"
|
|
QueryValidatorUnbondingDelegations = "validatorUnbondingDelegations"
|
|
QueryValidatorRedelegations = "validatorRedelegations"
|
|
QueryDelegator = "delegator"
|
|
QueryDelegation = "delegation"
|
|
QueryUnbondingDelegation = "unbondingDelegation"
|
|
QueryDelegatorValidators = "delegatorValidators"
|
|
QueryDelegatorValidator = "delegatorValidator"
|
|
QueryPool = "pool"
|
|
QueryParameters = "parameters"
|
|
)
|
|
|
|
// creates a querier for staking REST endpoints
|
|
func NewQuerier(k keep.Keeper, cdc *codec.Codec) sdk.Querier {
|
|
return func(ctx sdk.Context, path []string, req abci.RequestQuery) (res []byte, err sdk.Error) {
|
|
switch path[0] {
|
|
case QueryValidators:
|
|
return queryValidators(ctx, cdc, k)
|
|
case QueryValidator:
|
|
return queryValidator(ctx, cdc, req, k)
|
|
case QueryValidatorUnbondingDelegations:
|
|
return queryValidatorUnbondingDelegations(ctx, cdc, req, k)
|
|
case QueryValidatorRedelegations:
|
|
return queryValidatorRedelegations(ctx, cdc, req, k)
|
|
case QueryDelegation:
|
|
return queryDelegation(ctx, cdc, req, k)
|
|
case QueryUnbondingDelegation:
|
|
return queryUnbondingDelegation(ctx, cdc, req, k)
|
|
case QueryDelegatorDelegations:
|
|
return queryDelegatorDelegations(ctx, cdc, req, k)
|
|
case QueryDelegatorUnbondingDelegations:
|
|
return queryDelegatorUnbondingDelegations(ctx, cdc, req, k)
|
|
case QueryDelegatorRedelegations:
|
|
return queryDelegatorRedelegations(ctx, cdc, req, k)
|
|
case QueryDelegatorValidators:
|
|
return queryDelegatorValidators(ctx, cdc, req, k)
|
|
case QueryDelegatorValidator:
|
|
return queryDelegatorValidator(ctx, cdc, req, k)
|
|
case QueryPool:
|
|
return queryPool(ctx, cdc, k)
|
|
case QueryParameters:
|
|
return queryParameters(ctx, cdc, k)
|
|
default:
|
|
return nil, sdk.ErrUnknownRequest("unknown stake query endpoint")
|
|
}
|
|
}
|
|
}
|
|
|
|
// defines the params for the following queries:
|
|
// - 'custom/stake/delegatorDelegations'
|
|
// - 'custom/stake/delegatorUnbondingDelegations'
|
|
// - 'custom/stake/delegatorRedelegations'
|
|
// - 'custom/stake/delegatorValidators'
|
|
type QueryDelegatorParams struct {
|
|
DelegatorAddr sdk.AccAddress
|
|
}
|
|
|
|
// defines the params for the following queries:
|
|
// - 'custom/stake/validator'
|
|
// - 'custom/stake/validatorUnbondingDelegations'
|
|
// - 'custom/stake/validatorRedelegations'
|
|
type QueryValidatorParams struct {
|
|
ValidatorAddr sdk.ValAddress
|
|
}
|
|
|
|
// defines the params for the following queries:
|
|
// - 'custom/stake/delegation'
|
|
// - 'custom/stake/unbondingDelegation'
|
|
// - 'custom/stake/delegatorValidator'
|
|
type QueryBondsParams struct {
|
|
DelegatorAddr sdk.AccAddress
|
|
ValidatorAddr sdk.ValAddress
|
|
}
|
|
|
|
func queryValidators(ctx sdk.Context, cdc *codec.Codec, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
stakeParams := k.GetParams(ctx)
|
|
validators := k.GetValidators(ctx, stakeParams.MaxValidators)
|
|
|
|
res, errRes := codec.MarshalJSONIndent(cdc, validators)
|
|
if err != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryValidator(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryValidatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
validator, found := k.GetValidator(ctx, params.ValidatorAddr)
|
|
if !found {
|
|
return []byte{}, types.ErrNoValidatorFound(types.DefaultCodespace)
|
|
}
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, validator)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryValidatorUnbondingDelegations(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryValidatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
unbonds := k.GetUnbondingDelegationsFromValidator(ctx, params.ValidatorAddr)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, unbonds)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryValidatorRedelegations(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryValidatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
redelegations := k.GetRedelegationsFromValidator(ctx, params.ValidatorAddr)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, redelegations)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegatorDelegations(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryDelegatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
delegations := k.GetAllDelegatorDelegations(ctx, params.DelegatorAddr)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, delegations)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegatorUnbondingDelegations(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryDelegatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
unbondingDelegations := k.GetAllUnbondingDelegations(ctx, params.DelegatorAddr)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, unbondingDelegations)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegatorRedelegations(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryDelegatorParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
redelegations := k.GetAllRedelegations(ctx, params.DelegatorAddr)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, redelegations)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegatorValidators(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryDelegatorParams
|
|
|
|
stakeParams := k.GetParams(ctx)
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
validators := k.GetDelegatorValidators(ctx, params.DelegatorAddr, stakeParams.MaxValidators)
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, validators)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegatorValidator(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryBondsParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
validator, err := k.GetDelegatorValidator(ctx, params.DelegatorAddr, params.ValidatorAddr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, validator)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryDelegation(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryBondsParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
delegation, found := k.GetDelegation(ctx, params.DelegatorAddr, params.ValidatorAddr)
|
|
if !found {
|
|
return []byte{}, types.ErrNoDelegation(types.DefaultCodespace)
|
|
}
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, delegation)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryUnbondingDelegation(ctx sdk.Context, cdc *codec.Codec, req abci.RequestQuery, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
var params QueryBondsParams
|
|
|
|
errRes := cdc.UnmarshalJSON(req.Data, ¶ms)
|
|
if errRes != nil {
|
|
return []byte{}, sdk.ErrUnknownAddress("")
|
|
}
|
|
|
|
unbond, found := k.GetUnbondingDelegation(ctx, params.DelegatorAddr, params.ValidatorAddr)
|
|
if !found {
|
|
return []byte{}, types.ErrNoUnbondingDelegation(types.DefaultCodespace)
|
|
}
|
|
|
|
res, errRes = codec.MarshalJSONIndent(cdc, unbond)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryPool(ctx sdk.Context, cdc *codec.Codec, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
pool := k.GetPool(ctx)
|
|
|
|
res, errRes := codec.MarshalJSONIndent(cdc, pool)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func queryParameters(ctx sdk.Context, cdc *codec.Codec, k keep.Keeper) (res []byte, err sdk.Error) {
|
|
params := k.GetParams(ctx)
|
|
|
|
res, errRes := codec.MarshalJSONIndent(cdc, params)
|
|
if errRes != nil {
|
|
return nil, sdk.ErrInternal(sdk.AppendMsgToErr("could not marshal result to JSON", errRes.Error()))
|
|
}
|
|
return res, nil
|
|
}
|