mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-15 04:25:27 +00:00
102 lines
3.4 KiB
Go
102 lines
3.4 KiB
Go
package keeper
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/0glabs/0g-chain/x/kavadist/types"
|
|
)
|
|
|
|
func (k Keeper) mintInfrastructurePeriods(ctx sdk.Context, periods types.Periods, previousBlockTime time.Time) (sdk.Coin, sdkmath.Int, error) {
|
|
var err error
|
|
coinsMinted := sdk.NewCoin(types.GovDenom, sdk.ZeroInt())
|
|
timeElapsed := sdk.ZeroInt()
|
|
for _, period := range periods {
|
|
switch {
|
|
// Case 1 - period is fully expired
|
|
case period.End.Before(previousBlockTime):
|
|
continue
|
|
|
|
// Case 2 - period has ended since the previous block time
|
|
case period.End.After(previousBlockTime) && (period.End.Before(ctx.BlockTime()) || period.End.Equal(ctx.BlockTime())):
|
|
// calculate time elapsed relative to the periods end time
|
|
timeElapsed = sdkmath.NewInt(period.End.Unix() - previousBlockTime.Unix())
|
|
coins, errI := k.mintInflationaryCoins(ctx, period.Inflation, timeElapsed, types.GovDenom)
|
|
err = errI
|
|
if !coins.IsZero() {
|
|
coinsMinted = coinsMinted.Add(coins)
|
|
}
|
|
// update the value of previousBlockTime so that the next period starts from the end of the last
|
|
// period and not the original value of previousBlockTime
|
|
previousBlockTime = period.End
|
|
|
|
// Case 3 - period is ongoing
|
|
case (period.Start.Before(previousBlockTime) || period.Start.Equal(previousBlockTime)) && period.End.After(ctx.BlockTime()):
|
|
// calculate time elapsed relative to the current block time
|
|
timeElapsed = sdkmath.NewInt(ctx.BlockTime().Unix() - previousBlockTime.Unix())
|
|
coins, errI := k.mintInflationaryCoins(ctx, period.Inflation, timeElapsed, types.GovDenom)
|
|
if !coins.IsZero() {
|
|
coinsMinted = coinsMinted.Add(coins)
|
|
}
|
|
err = errI
|
|
|
|
// Case 4 - period hasn't started
|
|
case period.Start.After(ctx.BlockTime()) || period.Start.Equal(ctx.BlockTime()):
|
|
timeElapsed = sdkmath.NewInt(ctx.BlockTime().Unix() - previousBlockTime.Unix())
|
|
continue
|
|
}
|
|
|
|
if err != nil {
|
|
return sdk.Coin{}, sdkmath.Int{}, err
|
|
}
|
|
}
|
|
return coinsMinted, timeElapsed, nil
|
|
}
|
|
|
|
func (k Keeper) distributeInfrastructureCoins(ctx sdk.Context, partnerRewards types.PartnerRewards, coreRewards types.CoreRewards, timeElapsed sdkmath.Int, coinsToDistribute sdk.Coin) error {
|
|
if timeElapsed.IsZero() {
|
|
return nil
|
|
}
|
|
if coinsToDistribute.IsZero() {
|
|
return nil
|
|
}
|
|
for _, pr := range partnerRewards {
|
|
coinsToSend := sdk.NewCoin(types.GovDenom, pr.RewardsPerSecond.Amount.Mul(timeElapsed))
|
|
// TODO check balance, log if insufficient and return rather than error
|
|
err := k.bankKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleName, pr.Address, sdk.NewCoins(coinsToSend))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
neg, updatedCoins := safeSub(coinsToDistribute, coinsToSend)
|
|
if neg {
|
|
return fmt.Errorf("negative coins")
|
|
}
|
|
coinsToDistribute = updatedCoins
|
|
}
|
|
for _, cr := range coreRewards {
|
|
coinsToSend := sdk.NewCoin(types.GovDenom, sdk.NewDecFromInt(coinsToDistribute.Amount).Mul(cr.Weight).RoundInt())
|
|
// TODO check balance, log if insufficient and return rather than error
|
|
err := k.bankKeeper.SendCoinsFromModuleToAccount(ctx, types.ModuleName, cr.Address, sdk.NewCoins(coinsToSend))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
neg, updatedCoins := safeSub(coinsToDistribute, coinsToSend)
|
|
if neg {
|
|
return fmt.Errorf("negative coins")
|
|
}
|
|
coinsToDistribute = updatedCoins
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func safeSub(a, b sdk.Coin) (bool, sdk.Coin) {
|
|
isNeg := a.IsLT(b)
|
|
if isNeg {
|
|
return true, sdk.Coin{}
|
|
}
|
|
return false, a.Sub(b)
|
|
}
|