-
Notifications
You must be signed in to change notification settings - Fork 126
/
module.go
53 lines (43 loc) · 1.51 KB
/
module.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
50
51
52
53
package transfer
import (
"fmt"
"github.com/cosmos/cosmos-sdk/types/module"
ibctransfer "github.com/cosmos/ibc-go/v7/modules/apps/transfer"
ibctransferkeeper "github.com/cosmos/ibc-go/v7/modules/apps/transfer/keeper"
"github.com/cosmos/ibc-go/v7/modules/apps/transfer/types"
"github.com/haqq-network/haqq/x/ibc/transfer/keeper"
)
var (
_ module.AppModule = AppModule{}
_ module.AppModuleBasic = AppModuleBasic{}
)
// AppModuleBasic embeds the IBC Transfer AppModuleBasic
type AppModuleBasic struct {
*ibctransfer.AppModuleBasic
}
// AppModule represents the AppModule for this module
type AppModule struct {
*ibctransfer.AppModule
keeper keeper.Keeper
}
// NewAppModule creates a new 20-transfer module
func NewAppModule(k keeper.Keeper) AppModule {
am := ibctransfer.NewAppModule(*k.Keeper)
return AppModule{
AppModule: &am,
keeper: k,
}
}
// RegisterServices registers module services.
func (am AppModule) RegisterServices(cfg module.Configurator) {
// Override Transfer Msg Server
types.RegisterMsgServer(cfg.MsgServer(), am.keeper)
types.RegisterQueryServer(cfg.QueryServer(), am.keeper)
m := ibctransferkeeper.NewMigrator(*am.keeper.Keeper)
if err := cfg.RegisterMigration(types.ModuleName, 1, m.MigrateTraces); err != nil {
panic(fmt.Sprintf("failed to migrate transfer app from version 1 to 2: %v", err))
}
if err := cfg.RegisterMigration(types.ModuleName, 2, m.MigrateTotalEscrowForDenom); err != nil {
panic(fmt.Sprintf("failed to migrate transfer app from version 2 to 3: %v", err))
}
}