/
genesis.go
43 lines (36 loc) · 1.01 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
package revenue
import (
errorsmod "cosmossdk.io/errors"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cvn-network/cvn/v2/x/revenue/v1/keeper"
"github.com/cvn-network/cvn/v2/x/revenue/v1/types"
)
// InitGenesis import module genesis
func InitGenesis(
ctx sdk.Context,
k keeper.Keeper,
data types.GenesisState,
) {
err := k.SetParams(ctx, data.Params)
if err != nil {
panic(errorsmod.Wrapf(err, "failed setting params"))
}
for _, revenue := range data.Revenues {
contract := revenue.GetContractAddr()
deployer := revenue.GetDeployerAddr()
withdrawer := revenue.GetWithdrawerAddr()
// Set initial contracts receiving transaction fees
k.SetRevenue(ctx, revenue)
k.SetDeployerMap(ctx, deployer, contract)
if len(withdrawer) != 0 {
k.SetWithdrawerMap(ctx, withdrawer, contract)
}
}
}
// ExportGenesis export module state
func ExportGenesis(ctx sdk.Context, k keeper.Keeper) *types.GenesisState {
return &types.GenesisState{
Params: k.GetParams(ctx),
Revenues: k.GetRevenues(ctx),
}
}