mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 20:15:18 +00:00
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
package auction
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/0glabs/0g-chain/x/auction/keeper"
|
|
"github.com/0glabs/0g-chain/x/auction/types"
|
|
)
|
|
|
|
// InitGenesis initializes the store state from a genesis state.
|
|
func InitGenesis(ctx sdk.Context, keeper keeper.Keeper, bankKeeper types.BankKeeper, 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))
|
|
}
|
|
|
|
keeper.SetNextAuctionID(ctx, gs.NextAuctionId)
|
|
|
|
keeper.SetParams(ctx, gs.Params)
|
|
|
|
totalAuctionCoins := sdk.NewCoins()
|
|
|
|
auctions, err := types.UnpackGenesisAuctions(gs.Auctions)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to unpack genesis auctions: %s", err))
|
|
}
|
|
for _, a := range auctions {
|
|
keeper.SetAuction(ctx, a)
|
|
// find the total coins that should be present in the module account
|
|
totalAuctionCoins = totalAuctionCoins.Add(a.GetModuleAccountCoins()...)
|
|
}
|
|
|
|
// check if the module account exists
|
|
moduleAcc := accountKeeper.GetModuleAccount(ctx, types.ModuleName)
|
|
if moduleAcc == nil {
|
|
panic(fmt.Sprintf("%s module account has not been set", types.ModuleName))
|
|
}
|
|
|
|
maccCoins := bankKeeper.GetAllBalances(ctx, moduleAcc.GetAddress())
|
|
|
|
// check module coins match auction coins
|
|
// Note: Other sdk modules do not check this, instead just using the existing module account coins, or if zero, setting them.
|
|
if !maccCoins.IsEqual(totalAuctionCoins) {
|
|
panic(fmt.Sprintf("total auction coins (%s) do not equal (%s) module account (%s) ", maccCoins, types.ModuleName, totalAuctionCoins))
|
|
}
|
|
}
|
|
|
|
// ExportGenesis returns a GenesisState for a given context and keeper.
|
|
func ExportGenesis(ctx sdk.Context, keeper keeper.Keeper) *types.GenesisState {
|
|
nextAuctionID, err := keeper.GetNextAuctionID(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
params := keeper.GetParams(ctx)
|
|
|
|
genAuctions := []types.GenesisAuction{} // return empty list instead of nil if no auctions
|
|
keeper.IterateAuctions(ctx, func(a types.Auction) bool {
|
|
ga, ok := a.(types.GenesisAuction)
|
|
if !ok {
|
|
panic("could not convert stored auction to GenesisAuction type")
|
|
}
|
|
genAuctions = append(genAuctions, ga)
|
|
return false
|
|
})
|
|
|
|
gs, err := types.NewGenesisState(nextAuctionID, params, genAuctions)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return gs
|
|
}
|