update calculate delegation rewards

types/staking.go

@@ -71,9 +71,10 @@ func (x BondStatus) String() string {
 
 // DelegationI delegation bond for a delegated proof of stake system
 type DelegationI interface {
-	GetDelegatorAddr() string  // delegator string for the bond
-	GetValidatorAddr() string  // validator operator address
-	GetShares() math.LegacyDec // amount of validator's shares held in this delegation
+	GetDelegatorAddr() string           // delegator string for the bond
+	GetValidatorAddr() string           // validator operator address
+	GetShares() math.LegacyDec          // amount of validator's shares held in this delegation
+	GetEffectiveShares() math.LegacyDec // amount of validator's shares held in this delegation
 }
 
 // ValidatorI expected validator functions

x/distribution/keeper/delegation.go

@@ -41,7 +41,7 @@ func (k Keeper) initializeDelegation(ctx context.Context, val sdk.ValAddress, de
 	// calculate delegation stake in tokens
 	// we don't store directly, so multiply delegation shares * (tokens per share)
 	// note: necessary to truncate so we don't allow withdrawing more rewards than owed
-	stake := validator.TokensFromSharesTruncated(delegation.GetShares())
+	stake := validator.TokensFromSharesTruncated(delegation.GetEffectiveShares())
 	headerinfo := k.HeaderService.HeaderInfo(ctx)
 	return k.DelegatorStartingInfo.Set(ctx, collections.Join(val, del), types.NewDelegatorStartingInfo(previousPeriod, stake, uint64(headerinfo.Height)))
 }

x/staking/types/delegation.go

@@ -55,6 +55,8 @@ func (d Delegation) GetValidatorAddr() string {
 }
 func (d Delegation) GetShares() math.LegacyDec { return d.Shares }
 
+func (d Delegation) GetEffectiveShares() math.LegacyDec { return d.EffectiveShares }
+
 // Delegations is a collection of delegations
 type Delegations []Delegation