mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 12:05:18 +00:00
b356309d90
* add rate-limiting and optional blocklists * fix: check account is not nil * add tests for rate-limiting * update simulations * fix typos * remove unsued function arg
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
package keeper
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/store/prefix"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/params/subspace"
|
|
|
|
"github.com/kava-labs/kava/x/issuance/types"
|
|
)
|
|
|
|
// Keeper keeper for the issuance module
|
|
type Keeper struct {
|
|
key sdk.StoreKey
|
|
cdc *codec.Codec
|
|
paramSubspace subspace.Subspace
|
|
accountKeeper types.AccountKeeper
|
|
supplyKeeper types.SupplyKeeper
|
|
}
|
|
|
|
// NewKeeper returns a new keeper
|
|
func NewKeeper(cdc *codec.Codec, key sdk.StoreKey, paramstore subspace.Subspace, ak types.AccountKeeper, sk types.SupplyKeeper) Keeper {
|
|
if !paramstore.HasKeyTable() {
|
|
paramstore = paramstore.WithKeyTable(types.ParamKeyTable())
|
|
}
|
|
|
|
return Keeper{
|
|
key: key,
|
|
cdc: cdc,
|
|
paramSubspace: paramstore,
|
|
accountKeeper: ak,
|
|
supplyKeeper: sk,
|
|
}
|
|
}
|
|
|
|
// GetAssetSupply gets an asset's current supply from the store.
|
|
func (k Keeper) GetAssetSupply(ctx sdk.Context, denom string) (types.AssetSupply, bool) {
|
|
var assetSupply types.AssetSupply
|
|
store := prefix.NewStore(ctx.KVStore(k.key), types.AssetSupplyPrefix)
|
|
bz := store.Get([]byte(denom))
|
|
if bz == nil {
|
|
return types.AssetSupply{}, false
|
|
}
|
|
k.cdc.MustUnmarshalBinaryBare(bz, &assetSupply)
|
|
return assetSupply, true
|
|
}
|
|
|
|
// SetAssetSupply updates an asset's supply
|
|
func (k Keeper) SetAssetSupply(ctx sdk.Context, supply types.AssetSupply, denom string) {
|
|
store := prefix.NewStore(ctx.KVStore(k.key), types.AssetSupplyPrefix)
|
|
store.Set([]byte(denom), k.cdc.MustMarshalBinaryBare(supply))
|
|
}
|
|
|
|
// IterateAssetSupplies provides an iterator over all stored AssetSupplies.
|
|
func (k Keeper) IterateAssetSupplies(ctx sdk.Context, cb func(supply types.AssetSupply) (stop bool)) {
|
|
iterator := sdk.KVStorePrefixIterator(ctx.KVStore(k.key), types.AssetSupplyPrefix)
|
|
|
|
defer iterator.Close()
|
|
for ; iterator.Valid(); iterator.Next() {
|
|
var supply types.AssetSupply
|
|
k.cdc.MustUnmarshalBinaryBare(iterator.Value(), &supply)
|
|
|
|
if cb(supply) {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// GetAllAssetSupplies returns all asset supplies from the store
|
|
func (k Keeper) GetAllAssetSupplies(ctx sdk.Context) (supplies types.AssetSupplies) {
|
|
k.IterateAssetSupplies(ctx, func(supply types.AssetSupply) bool {
|
|
supplies = append(supplies, supply)
|
|
return false
|
|
})
|
|
return
|
|
}
|
|
|
|
// GetPreviousBlockTime get the blocktime for the previous block
|
|
func (k Keeper) GetPreviousBlockTime(ctx sdk.Context) (blockTime time.Time, found bool) {
|
|
store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousBlockTimeKey)
|
|
b := store.Get([]byte{})
|
|
if b == nil {
|
|
return time.Time{}, false
|
|
}
|
|
k.cdc.MustUnmarshalBinaryLengthPrefixed(b, &blockTime)
|
|
return blockTime, true
|
|
}
|
|
|
|
// SetPreviousBlockTime set the time of the previous block
|
|
func (k Keeper) SetPreviousBlockTime(ctx sdk.Context, blockTime time.Time) {
|
|
store := prefix.NewStore(ctx.KVStore(k.key), types.PreviousBlockTimeKey)
|
|
store.Set([]byte{}, k.cdc.MustMarshalBinaryLengthPrefixed(blockTime))
|
|
}
|