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)) }