mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-10-04 16:35:18 +00:00
63 lines
2.2 KiB
Go
63 lines
2.2 KiB
Go
package keeper
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authexported "github.com/cosmos/cosmos-sdk/x/auth/exported"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/vesting"
|
|
"github.com/kava-labs/kava/x/validator-vesting/internal/types"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
)
|
|
|
|
// NewQuerier returns a new querier function
|
|
func NewQuerier(keeper Keeper) sdk.Querier {
|
|
return func(ctx sdk.Context, path []string, req abci.RequestQuery) (res []byte, err sdk.Error) {
|
|
switch path[0] {
|
|
case types.QueryCirculatingSupply:
|
|
return queryGetCirculatingSupply(ctx, req, keeper)
|
|
case types.QueryTotalSupply:
|
|
return queryGetTotalSupply(ctx, req, keeper)
|
|
default:
|
|
return nil, sdk.ErrUnknownRequest("unknown cdp query endpoint")
|
|
}
|
|
}
|
|
}
|
|
|
|
func queryGetTotalSupply(ctx sdk.Context, req abci.RequestQuery, keeper Keeper) ([]byte, sdk.Error) {
|
|
totalSupply := keeper.supplyKeeper.GetSupply(ctx).GetTotal().AmountOf("ukava")
|
|
supplyInt := sdk.NewDecFromInt(totalSupply).Mul(sdk.MustNewDecFromStr("0.000001")).TruncateInt64()
|
|
bz, err := keeper.cdc.MarshalJSON(supplyInt)
|
|
if err != nil {
|
|
return nil, sdk.ErrInternal(err.Error())
|
|
}
|
|
return bz, nil
|
|
}
|
|
|
|
func queryGetCirculatingSupply(ctx sdk.Context, req abci.RequestQuery, keeper Keeper) ([]byte, sdk.Error) {
|
|
circulatingSupply := keeper.supplyKeeper.GetSupply(ctx).GetTotal().AmountOf("ukava")
|
|
keeper.ak.IterateAccounts(ctx,
|
|
func(acc authexported.Account) (stop bool) {
|
|
|
|
// validator vesting account
|
|
vvacc, ok := acc.(*types.ValidatorVestingAccount)
|
|
if ok {
|
|
vestedBalance := vvacc.GetVestingCoins(ctx.BlockTime()).AmountOf("ukava")
|
|
circulatingSupply = circulatingSupply.Sub(vestedBalance)
|
|
return false
|
|
}
|
|
// periodic vesting account
|
|
pvacc, ok := acc.(*vesting.PeriodicVestingAccount)
|
|
if ok {
|
|
vestedBalance := pvacc.GetVestingCoins(ctx.BlockTime()).AmountOf("ukava")
|
|
circulatingSupply = circulatingSupply.Sub(vestedBalance)
|
|
return false
|
|
}
|
|
return false
|
|
})
|
|
supplyInt := sdk.NewDecFromInt(circulatingSupply).Mul(sdk.MustNewDecFromStr("0.000001")).TruncateInt64()
|
|
bz, err := keeper.cdc.MarshalJSON(supplyInt)
|
|
if err != nil {
|
|
return nil, sdk.ErrInternal(err.Error())
|
|
}
|
|
return bz, nil
|
|
}
|