mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-25 15:55:18 +00:00
0efe7f2281
* add annualized_reward query proto * use sdkmath.LegacyDec to match RPS param... * add AnnualizedRewards grpc query * add changelog entry * simplify calculation & expand test cases
100 lines
3.4 KiB
Go
100 lines
3.4 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/kava-labs/kava/x/community/types"
|
|
)
|
|
|
|
type queryServer struct {
|
|
keeper Keeper
|
|
}
|
|
|
|
var _ types.QueryServer = queryServer{}
|
|
|
|
// NewQueryServerImpl creates a new server for handling gRPC queries.
|
|
func NewQueryServerImpl(k Keeper) types.QueryServer {
|
|
return &queryServer{keeper: k}
|
|
}
|
|
|
|
// Params implements the gRPC service handler for querying x/community params.
|
|
func (s queryServer) Params(c context.Context, req *types.QueryParamsRequest) (*types.QueryParamsResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
params, found := s.keeper.GetParams(ctx)
|
|
if !found {
|
|
return nil, status.Error(codes.NotFound, "params not found")
|
|
}
|
|
|
|
return &types.QueryParamsResponse{
|
|
Params: params,
|
|
}, nil
|
|
}
|
|
|
|
// Balance implements the gRPC service handler for querying x/community balance.
|
|
func (s queryServer) Balance(c context.Context, _ *types.QueryBalanceRequest) (*types.QueryBalanceResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
return &types.QueryBalanceResponse{
|
|
Coins: s.keeper.GetModuleAccountBalance(ctx),
|
|
}, nil
|
|
}
|
|
|
|
// CommunityPool queries the community pool coins
|
|
func (s queryServer) TotalBalance(
|
|
c context.Context,
|
|
req *types.QueryTotalBalanceRequest,
|
|
) (*types.QueryTotalBalanceResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
// x/distribution community pool balance
|
|
nativePoolBalance := s.keeper.distrKeeper.GetFeePoolCommunityCoins(ctx)
|
|
|
|
// x/community pool balance
|
|
communityPoolBalance := s.keeper.GetModuleAccountBalance(ctx)
|
|
|
|
totalBalance := nativePoolBalance.Add(sdk.NewDecCoinsFromCoins(communityPoolBalance...)...)
|
|
|
|
return &types.QueryTotalBalanceResponse{
|
|
Pool: totalBalance,
|
|
}, nil
|
|
}
|
|
|
|
// AnnualizedRewards calculates the annualized rewards for the chain.
|
|
func (s queryServer) AnnualizedRewards(
|
|
c context.Context,
|
|
req *types.QueryAnnualizedRewardsRequest,
|
|
) (*types.QueryAnnualizedRewardsResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
// staking rewards come from one of two sources depending on if inflation is enabled or not.
|
|
// at any given time, only one source will contribute to the staking rewards. the other will be zero.
|
|
// this method adds both sources together so it is accurate in both cases.
|
|
|
|
params := s.keeper.mustGetParams(ctx)
|
|
bondDenom := s.keeper.stakingKeeper.BondDenom(ctx)
|
|
|
|
totalSupply := s.keeper.bankKeeper.GetSupply(ctx, bondDenom).Amount
|
|
totalBonded := s.keeper.stakingKeeper.TotalBondedTokens(ctx)
|
|
rewardsPerSecond := params.StakingRewardsPerSecond
|
|
// need to convert these from sdk.Dec to sdkmath.LegacyDec
|
|
inflationRate := convertDecToLegacyDec(s.keeper.mintKeeper.GetMinter(ctx).Inflation)
|
|
communityTax := convertDecToLegacyDec(s.keeper.distrKeeper.GetCommunityTax(ctx))
|
|
|
|
return &types.QueryAnnualizedRewardsResponse{
|
|
StakingRewards: CalculateStakingAnnualPercentage(totalSupply, totalBonded, inflationRate, communityTax, rewardsPerSecond),
|
|
}, nil
|
|
}
|
|
|
|
// convertDecToLegacyDec is a helper method for converting between new and old Dec types
|
|
// current version of cosmos-sdk in this repo uses sdk.Dec
|
|
// this module uses sdkmath.LegacyDec in its parameters
|
|
// TODO: remove me after upgrade to cosmos-sdk v50 (LegacyDec is everywhere)
|
|
func convertDecToLegacyDec(in sdk.Dec) sdkmath.LegacyDec {
|
|
return sdkmath.LegacyNewDecFromBigIntWithPrec(in.BigInt(), sdk.Precision)
|
|
}
|