From a6ef3c42aa007f54f58e613b2511970309e1012a Mon Sep 17 00:00:00 2001 From: rigelrozanski Date: Fri, 19 Oct 2018 17:41:39 -0400 Subject: [PATCH] use power instead of total tokens --- x/distribution/keeper/delegation.go | 4 ++-- x/distribution/keeper/validator.go | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/x/distribution/keeper/delegation.go b/x/distribution/keeper/delegation.go index 7829815798..8183d95b1a 100644 --- a/x/distribution/keeper/delegation.go +++ b/x/distribution/keeper/delegation.go @@ -86,7 +86,7 @@ func (k Keeper) WithdrawDelegationReward(ctx sdk.Context, delegatorAddr sdk.AccA delegation := k.stakeKeeper.Delegation(ctx, delegatorAddr, validatorAddr) delInfo, valInfo, feePool, withdraw := delInfo.WithdrawRewards(feePool, valInfo, height, bondedTokens, - validator.GetTokens(), validator.GetDelegatorShares(), delegation.GetShares(), validator.GetCommission()) + validator.GetPower(), validator.GetDelegatorShares(), delegation.GetShares(), validator.GetCommission()) k.SetValidatorDistInfo(ctx, valInfo) k.SetDelegationDistInfo(ctx, delInfo) @@ -134,7 +134,7 @@ func (k Keeper) getDelegatorRewardsAll(ctx sdk.Context, delAddr sdk.AccAddress, delegation := k.stakeKeeper.Delegation(ctx, delAddr, valAddr) delInfo, valInfo, feePool, diWithdraw := delInfo.WithdrawRewards(feePool, valInfo, height, bondedTokens, - validator.GetTokens(), validator.GetDelegatorShares(), delegation.GetShares(), validator.GetCommission()) + validator.GetPower(), validator.GetDelegatorShares(), delegation.GetShares(), validator.GetCommission()) withdraw = withdraw.Plus(diWithdraw) k.SetFeePool(ctx, feePool) k.SetValidatorDistInfo(ctx, valInfo) diff --git a/x/distribution/keeper/validator.go b/x/distribution/keeper/validator.go index 8b38b0f6a0..0d34533874 100644 --- a/x/distribution/keeper/validator.go +++ b/x/distribution/keeper/validator.go @@ -58,7 +58,7 @@ func (k Keeper) WithdrawValidatorRewardsAll(ctx sdk.Context, operatorAddr sdk.Va valInfo := k.GetValidatorDistInfo(ctx, operatorAddr) feePool := k.GetFeePool(ctx) valInfo, feePool, commission := valInfo.WithdrawCommission(feePool, height, bondedTokens, - validator.GetTokens(), validator.GetCommission()) + validator.GetPower(), validator.GetCommission()) withdraw = withdraw.Plus(commission) k.SetValidatorDistInfo(ctx, valInfo)