space-pussy/x/bandwidth/abci.go

package bandwidth

import (
	"github.com/cosmos/cosmos-sdk/telemetry"
	sdk "github.com/cosmos/cosmos-sdk/types"

	"time"

	"github.com/joinresistance/space-pussy/x/bandwidth/keeper"
	"github.com/joinresistance/space-pussy/x/bandwidth/types"
)

var accountsToUpdate = make([]sdk.AccAddress, 0)

// recover and update bandwidth for accounts with changed stake
func updateAccountsMaxBandwidth(ctx sdk.Context, meter *keeper.BandwidthMeter) {
	for _, addr := range accountsToUpdate {
		meter.UpdateAccountMaxBandwidth(ctx, addr)
	}
	accountsToUpdate = make([]sdk.AccAddress, 0)
}

// collect all addresses with updated stake
func CollectAddressesWithStakeChange() func(ctx sdk.Context, from sdk.AccAddress, to sdk.AccAddress) {
	return func(ctx sdk.Context, from sdk.AccAddress, to sdk.AccAddress) {
		if ctx.IsCheckTx() {
			return
		}
		if from != nil {
			accountsToUpdate = append(accountsToUpdate, from)
		}
		if to != nil {
			accountsToUpdate = append(accountsToUpdate, to)
		}
	}
}

func EndBlocker(ctx sdk.Context, bm *keeper.BandwidthMeter) {
	defer telemetry.ModuleMeasureSince(types.ModuleName, time.Now(), telemetry.MetricKeyEndBlocker)

	params := bm.GetParams(ctx)
	if ctx.BlockHeight() != 0 && uint64(ctx.BlockHeight())%params.AdjustPricePeriod == 0 {
		bm.AdjustPrice(ctx) // TODO Add block event for price and load
	}

	bm.CommitBlockBandwidth(ctx) // TODO add block event for committed bandwidth

	updateAccountsMaxBandwidth(ctx, bm)
}