0g-chain/x/issuance/genesis.go

46 lines
1.2 KiB
Go
Raw Normal View History

package issuance
import (
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/kava-labs/kava/x/issuance/keeper"
"github.com/kava-labs/kava/x/issuance/types"
)
// InitGenesis initializes the store state from a genesis state.
func InitGenesis(ctx sdk.Context, k keeper.Keeper, accountKeeper types.AccountKeeper, gs types.GenesisState) {
if err := gs.Validate(); err != nil {
panic(fmt.Sprintf("failed to validate %s genesis state: %s", types.ModuleName, err))
}
// check if the module account exists
moduleAcc := accountKeeper.GetModuleAccount(ctx, types.ModuleAccountName)
if moduleAcc == nil {
panic(fmt.Sprintf("%s module account has not been set", types.ModuleAccountName))
}
k.SetParams(ctx, gs.Params)
for _, supply := range gs.Supplies {
k.SetAssetSupply(ctx, supply, supply.GetDenom())
}
for _, asset := range gs.Params.Assets {
if asset.RateLimit.Active {
_, found := k.GetAssetSupply(ctx, asset.Denom)
if !found {
k.CreateNewAssetSupply(ctx, asset.Denom)
}
}
}
}
// ExportGenesis export genesis state for issuance module
func ExportGenesis(ctx sdk.Context, k keeper.Keeper) types.GenesisState {
params := k.GetParams(ctx)
supplies := k.GetAllAssetSupplies(ctx)
return types.NewGenesisState(params, supplies)
}