-
Notifications
You must be signed in to change notification settings - Fork 5
/
distributor.go
91 lines (76 loc) · 2.51 KB
/
distributor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package keeper
import (
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/duality-labs/duality/x/incentives/types"
)
type DistributorKeeper interface {
ValueForShares(ctx sdk.Context, coin sdk.Coin, tick int64) (sdk.Int, error)
GetStakesByQueryCondition(ctx sdk.Context, distrTo *types.QueryCondition) types.Stakes
StakeCoinsPassingQueryCondition(ctx sdk.Context, stake *types.Stake, distrTo types.QueryCondition) sdk.Coins
}
type Distributor struct {
keeper DistributorKeeper
}
func NewDistributor(keeper DistributorKeeper) Distributor {
return Distributor{
keeper: keeper,
}
}
func (d Distributor) Distribute(
ctx sdk.Context,
gauge *types.Gauge,
filterStakes types.Stakes,
) (types.DistributionSpec, error) {
if !gauge.IsActiveGauge(ctx.BlockTime()) {
return nil, types.ErrGaugeNotActive
}
distSpec := types.DistributionSpec{}
rewardsNextEpoch := gauge.RewardsNextEpoch()
adjustedGaugeTotal := sdk.ZeroInt()
gaugeStakes := d.keeper.GetStakesByQueryCondition(ctx, &gauge.DistributeTo)
if filterStakes == nil {
filterStakes = gaugeStakes
}
stakeSumCache := make(map[uint64]sdk.Int, len(gaugeStakes))
for _, stake := range gaugeStakes {
stakeCoins := d.keeper.StakeCoinsPassingQueryCondition(ctx, stake, gauge.DistributeTo)
stakeTotal := sdk.ZeroInt()
for _, stakeCoin := range stakeCoins {
adjustedPositionValue, err := d.keeper.ValueForShares(ctx, stakeCoin, gauge.PricingTick)
if err != nil {
return nil, err
}
stakeTotal = stakeTotal.Add(adjustedPositionValue)
}
adjustedGaugeTotal = adjustedGaugeTotal.Add(stakeTotal)
stakeSumCache[stake.ID] = stakeTotal
}
if adjustedGaugeTotal.IsZero() {
return distSpec, nil
}
for _, stake := range filterStakes {
stakeAmt := stakeSumCache[stake.ID]
distCoins := sdk.Coins{}
for _, epochRewards := range rewardsNextEpoch {
// distribution amount = gauge_size * denom_stake_amount / (total_denom_stake_amount * remain_epochs)
amount := sdk.NewDecFromInt(epochRewards.Amount).
Mul(sdk.NewDecFromInt(stakeAmt)).
Quo(sdk.NewDecFromInt(adjustedGaugeTotal)).
TruncateInt()
reward := sdk.Coin{Denom: epochRewards.Denom, Amount: amount}
distCoins = distCoins.Add(reward)
}
// update the amount for that address
if distCoins.Empty() {
continue
}
if spec, ok := distSpec[stake.Owner]; ok {
distSpec[stake.Owner] = spec.Add(distCoins...)
} else {
distSpec[stake.Owner] = distCoins
}
}
gauge.DistributedCoins = gauge.DistributedCoins.Add(rewardsNextEpoch...)
gauge.FilledEpochs++
return distSpec, nil
}