github.com/Finschia/finschia-sdk@v0.48.1/x/bankplus/module.go (about)

     1  package bankplus
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/Finschia/finschia-sdk/codec"
     7  	"github.com/Finschia/finschia-sdk/types/module"
     8  	accountkeeper "github.com/Finschia/finschia-sdk/x/auth/keeper"
     9  	"github.com/Finschia/finschia-sdk/x/bank"
    10  	bankkeeper "github.com/Finschia/finschia-sdk/x/bank/keeper"
    11  	banktypes "github.com/Finschia/finschia-sdk/x/bank/types"
    12  	"github.com/Finschia/finschia-sdk/x/bankplus/keeper"
    13  )
    14  
    15  var (
    16  	_ module.AppModule           = AppModule{}
    17  	_ module.AppModuleSimulation = AppModule{}
    18  )
    19  
    20  type AppModule struct {
    21  	bank.AppModule
    22  
    23  	bankKeeper bankkeeper.Keeper
    24  }
    25  
    26  func NewAppModule(cdc codec.Codec, keeper bankkeeper.Keeper, accountKeeper accountkeeper.AccountKeeper) AppModule {
    27  	return AppModule{
    28  		AppModule:  bank.NewAppModule(cdc, keeper, accountKeeper),
    29  		bankKeeper: keeper,
    30  	}
    31  }
    32  
    33  func (am AppModule) RegisterServices(cfg module.Configurator) {
    34  	banktypes.RegisterMsgServer(cfg.MsgServer(), bankkeeper.NewMsgServerImpl(am.bankKeeper))
    35  	banktypes.RegisterQueryServer(cfg.QueryServer(), am.bankKeeper)
    36  
    37  	m := bankkeeper.NewMigrator(am.bankKeeper.(keeper.BaseKeeper).BaseKeeper)
    38  	if err := cfg.RegisterMigration(banktypes.ModuleName, 1, m.Migrate1to2); err != nil {
    39  		panic(fmt.Sprintf("failed to migrate x/bank from version 1 to 2: %v", err))
    40  	}
    41  }