-
Notifications
You must be signed in to change notification settings - Fork 0
/
genesis.go
49 lines (40 loc) · 1.25 KB
/
genesis.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package bridge
import (
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/fury-labs/fury-bridge/x/bridge/keeper"
"github.com/fury-labs/fury-bridge/x/bridge/types"
)
// InitGenesis initializes genesis state based on exported genesis
func InitGenesis(
ctx sdk.Context,
k keeper.Keeper,
accountKeeper types.AccountKeeper,
data types.GenesisState,
) {
k.SetParams(ctx, data.Params)
// Ensure bridge module account is set
if moduleAcc := accountKeeper.GetModuleAccount(ctx, types.ModuleName); moduleAcc == nil {
panic("the bridge module account has not been set")
}
for _, pair := range data.ERC20BridgePairs {
k.RegisterBridgePair(ctx, pair)
}
k.SetNextWithdrawSequence(ctx, data.NextWithdrawSequence)
}
// ExportGenesis exports genesis state of the bridge module
func ExportGenesis(ctx sdk.Context, k keeper.Keeper, ak types.AccountKeeper) *types.GenesisState {
var bridgePairs types.ERC20BridgePairs
k.IterateBridgePairs(ctx, func(pair types.ERC20BridgePair) bool {
bridgePairs = append(bridgePairs, pair)
return false
})
nextSeq, err := k.GetNextWithdrawSequence(ctx)
if err != nil {
panic(err)
}
return &types.GenesisState{
Params: k.GetParams(ctx),
ERC20BridgePairs: bridgePairs,
NextWithdrawSequence: nextSeq,
}
}