Module Manager
Cosmos SDK modules need to implement the AppModule interfaces, in order to be managed by the application's module manager. The module manager plays an important role in message and query routing, and allows application developers to set the order of execution of a variety of functions like BeginBlocker and EndBlocker.
Pre-requisite Readings
Application Module Interfaces
Application module interfaces exist to facilitate the composition of modules together to form a functional Cosmos SDK application. There are 4 main application module interfaces:
AppModuleBasicfor independent module functionalities.AppModulefor inter-dependent module functionalities (except genesis-related functionalities).AppModuleGenesisfor inter-dependent genesis-related module functionalities.GenesisOnlyAppModule: Defines anAppModulethat only has import/export functionality
The above interfaces are mostly embedding smaller interfaces (extension interfaces), that defines specific functionalities:
HasName: Allows the module to provide its own name for legacy purposes.HasGenesisBasics: The legacy interface for stateless genesis methods.HasGenesis: The extension interface for stateful genesis methods.HasInvariants: The extension interface for registering invariants.HasServices: The extension interface for modules to register services.HasConsensusVersion: The extension interface for declaring a module consensus version.BeginBlockAppModule: The extension interface that contains information about theAppModuleandBeginBlock.EndBlockAppModule: The extension interface that contains information about theAppModuleandEndBlock.HasPrecommit: The extension interface that contains information about theAppModuleandPrecommit.HasPrepareCheckState: The extension interface that contains information about theAppModuleandPrepareCheckState.
The AppModuleBasic interface exists to define independent methods of the module, i.e. those that do not depend on other modules in the application. This allows for the construction of the basic application structure early in the application definition, generally in the init() function of the main application file.
The AppModule interface exists to define inter-dependent module methods. Many modules need to interact with other modules, typically through keepers, which means there is a need for an interface where modules list their keepers and other methods that require a reference to another module's object. AppModule interface extension, such as BeginBlockAppModule and EndBlockAppModule, also enables the module manager to set the order of execution between module's methods like BeginBlock and EndBlock, which is important in cases where the order of execution between modules matters in the context of the application.
The usage of extension interfaces allows modules to define only the functionalities they need. For example, a module that does not need an EndBlock does not need to define the EndBlockAppModule interface and thus the EndBlock method. AppModule and AppModuleGenesis are voluntarily small interfaces, that can take advantage of the Module patterns without having to define many placeholder functions.
AppModuleBasic
The AppModuleBasic interface defines the independent methods modules need to implement.
loading...
Let us go through the methods:
RegisterLegacyAminoCodec(*codec.LegacyAmino): Registers theaminocodec for the module, which is used to marshal and unmarshal structs to/from[]bytein order to persist them in the module'sKVStore.RegisterInterfaces(codectypes.InterfaceRegistry): Registers a module's interface types and their concrete implementations asproto.Message.RegisterGRPCGatewayRoutes(client.Context, *runtime.ServeMux): Registers gRPC routes for the module.GetTxCmd(): Returns the rootTxcommand for the module. The subcommands of this root command are used by end-users to generate new transactions containingmessages defined in the module.GetQueryCmd(): Return the rootquerycommand for the module. The subcommands of this root command are used by end-users to generate new queries to the subset of the state defined by the module.
All the AppModuleBasic of an application are managed by the BasicManager.
HasName
loading...
HasNameis an interface that has a methodName(). This method returns the name of the module as astring.
HasGenesisBasics
loading...
Let us go through the methods:
DefaultGenesis(codec.JSONCodec): Returns a defaultGenesisStatefor the module, marshalled tojson.RawMessage. The defaultGenesisStateneed to be defined by the module developer and is primarily used for testing.ValidateGenesis(codec.JSONCodec, client.TxEncodingConfig, json.RawMessage): Used to validate theGenesisStatedefined by a module, given in itsjson.RawMessageform. It will usually unmarshall thejsonbefore running a customValidateGenesisfunction defined by the module developer.
AppModuleGenesis
The AppModuleGenesis interface is a simple embedding of the AppModuleBasic and HasGenesis interfaces.
loading...
It does not have its own manager, and exists separately from AppModule only for modules that exist only to implement genesis functionalities, so that they can be managed without having to implement all of AppModule's methods.
HasGenesis
The HasGenesis interface is an extension interface of HasGenesisBasics.
loading...
Let us go through the two added methods:
InitGenesis(sdk.Context, codec.JSONCodec, json.RawMessage): Initializes the subset of the state managed by the module. It is called at genesis (i.e. when the chain is first started).ExportGenesis(sdk.Context, codec.JSONCodec): Exports the latest subset of the state managed by the module to be used in a new genesis file.ExportGenesisis called for each module when a new chain is started from the state of an existing chain.
AppModule
The AppModule interface defines a module. Modules can declare their functionalities by implementing extensions interfaces.
loading...
AppModules are managed by the module manager, which checks which extension interfaces are implemented by the module.
Previously the AppModule interface was containing all the methods that are defined in the extensions interfaces. This was leading to much boilerplate for modules that did not need all the functionalities.
HasInvariants
This interface defines one method. It allows to checks if a module can register invariants.
loading...
RegisterInvariants(sdk.InvariantRegistry): Registers theinvariantsof the module. If an invariant deviates from its predicted value, theInvariantRegistrytriggers appropriate logic (most often the chain will be halted).
HasServices
This interface defines one method. It allows to checks if a module can register invariants.
loading...
RegisterServices(Configurator): Allows a module to register services.
HasConsensusVersion
This interface defines one method for checking a module consensus version.
loading...
ConsensusVersion() uint64: Returns the consensus version of the module.
BeginBlockAppModule
The BeginBlockAppModule is an extension interface from AppModule. All modules that have an BeginBlock method implement this interface.
loading...
BeginBlock(sdk.Context, abci.RequestBeginBlock): This method gives module developers the option to implement logic that is automatically triggered at the beginning of each block. Implement empty if no logic needs to be triggered at the beginning of each block for this module.
EndBlockAppModule
The EndBlockAppModule is an extension interface from AppModule. All modules that have an EndBlock method implement this interface.
loading...
EndBlock(sdk.Context, abci.RequestEndBlock): This method gives module developers the option to implement logic that is automatically triggered at the end of each block. This is also where the module can inform the underlying consensus engine of validator set changes (e.g. thestakingmodule). Implement empty if no logic needs to be triggered at the end of each block for this module.
HasPrecommit
HasPrecommit is an extension interface from AppModule. All modules that have a Precommit method implement this interface.
Precommit(sdk.Context): This method gives module developers the option to implement logic that is automatically triggered during `Commit' of each block using thedeliverStateof the block to be committed. Implement empty if no logic needs to be triggered duringCommitof each block for this module.
HasPrepareCheckState
HasPrepareCheckState is an extension interface from AppModule. All modules that have a PrepareCheckState method implement this interface.
PrepareCheckState(sdk.Context): This method gives module developers the option to implement logic that is automatically triggered during `Commit' of each block using thecheckStateof the next block. Implement empty if no logic needs to be triggered duringCommitof each block for this module.
Implementing the Application Module Interfaces
Typically, the various application module interfaces are implemented in a file called module.go, located in the module's folder (e.g. ./x/module/module.go).
Almost every module needs to implement the AppModuleBasic and AppModule interfaces. If the module is only used for genesis, it will implement AppModuleGenesis instead of AppModule. The concrete type that implements the interface can add parameters that are required for the implementation of the various methods of the interface. For example, the Route() function often calls a NewMsgServerImpl(k keeper) function defined in keeper/msg_server.go and therefore needs to pass the module's keeper as a parameter.
// example
type AppModule struct {
AppModuleBasic
keeper Keeper
}
In the example above, you can see that the AppModule concrete type references an AppModuleBasic, and not an AppModuleGenesis. That is because AppModuleGenesis only needs to be implemented in modules that focus on genesis-related functionalities. In most modules, the concrete AppModule type will have a reference to an AppModuleBasic and implement the two added methods of AppModuleGenesis directly in the AppModule type.
If no parameter is required (which is often the case for AppModuleBasic), just declare an empty concrete type like so:
type AppModuleBasic struct{}
Module Managers
Module managers are used to manage collections of AppModuleBasic and AppModule.
BasicManager
The BasicManager is a structure that lists all the AppModuleBasic of an application:
loading...
It implements the following methods:
NewBasicManager(modules ...AppModuleBasic): Constructor function. It takes a list of the application'sAppModuleBasicand builds a newBasicManager. This function is generally called in theinit()function ofapp.goto quickly initialize the independent elements of the application's modules (click here to see an example).RegisterLegacyAminoCodec(cdc *codec.LegacyAmino): Registers thecodec.LegacyAminos of each of the application'sAppModuleBasic. This function is usually called early on in the application's construction.RegisterInterfaces(registry codectypes.InterfaceRegistry): Registers interface types and implementations of each of the application'sAppModuleBasic.DefaultGenesis(cdc codec.JSONCodec): Provides default genesis information for modules in the application by calling theDefaultGenesis(cdc codec.JSONCodec)function of each module. It only calls the modules that implements theHasGenesisBasicsinterfaces.ValidateGenesis(cdc codec.JSONCodec, txEncCfg client.TxEncodingConfig, genesis map[string]json.RawMessage): Validates the genesis information modules by calling theValidateGenesis(codec.JSONCodec, client.TxEncodingConfig, json.RawMessage)function of modules implementing theHasGenesisBasicsinterface.RegisterGRPCGatewayRoutes(clientCtx client.Context, rtr *runtime.ServeMux): Registers gRPC routes for modules.AddTxCommands(rootTxCmd *cobra.Command): Adds modules' transaction commands to the application'srootTxCommand. This function is usually called function from themain.gofunction of the application's command-line interface.AddQueryCommands(rootQueryCmd *cobra.Command): Adds modules' query commands to the application'srootQueryCommand. This function is usually called function from themain.gofunction of the application's command-line interface.
Manager
The Manager is a structure that holds all the AppModule of an application, and defines the order of execution between several key components of these modules:
loading...
The module manager is used throughout the application whenever an action on a collection of modules is required. It implements the following methods:
NewManager(modules ...AppModule): Constructor function. It takes a list of the application'sAppModules and builds a newManager. It is generally called from the application's main constructor function.SetOrderInitGenesis(moduleNames ...string): Sets the order in which theInitGenesisfunction of each module will be called when the application is first started. This function is generally called from the application's main constructor function. To initialize modules successfully, module dependencies should be considered. For example, thegenutilmodule must occur afterstakingmodule so that the pools are properly initialized with tokens from genesis accounts, thegenutilsmodule must also occur afterauthso that it can access the params from auth, IBC'scapabilitymodule should be initialized before all other modules so that it can initialize any capabilities.SetOrderExportGenesis(moduleNames ...string): Sets the order in which theExportGenesisfunction of each module will be called in case of an export. This function is generally called from the application's main constructor function.SetOrderBeginBlockers(moduleNames ...string): Sets the order in which theBeginBlock()function of each module will be called at the beginning of each block. This function is generally called from the application's main constructor function.SetOrderEndBlockers(moduleNames ...string): Sets the order in which theEndBlock()function of each module will be called at the end of each block. This function is generally called from the application's main constructor function.SetOrderPrecommiters(moduleNames ...string): Sets the order in which thePrecommit()function of each module will be called during commit of each block. This function is generally called from the application's main constructor function.SetOrderPrepareCheckStaters(moduleNames ...string): Sets the order in which thePrepareCheckState()function of each module will be called during commit of each block. This function is generally called from the application's main constructor function.SetOrderMigrations(moduleNames ...string): Sets the order of migrations to be run. If not set then migrations will be run with an order defined inDefaultMigrationsOrder.RegisterInvariants(ir sdk.InvariantRegistry): Registers the invariants of module implementing theHasInvariantsinterface.RegisterRoutes(router sdk.Router, queryRouter sdk.QueryRouter, legacyQuerierCdc *codec.LegacyAmino): Registers legacyMsgandquerierroutes.RegisterServices(cfg Configurator): Registers the services of modules implementing theHasServicesinterface.InitGenesis(ctx sdk.Context, cdc codec.JSONCodec, genesisData map[string]json.RawMessage): Calls theInitGenesisfunction of each module when the application is first started, in the order defined inOrderInitGenesis. Returns anabci.ResponseInitChainto the underlying consensus engine, which can contain validator updates.ExportGenesis(ctx sdk.Context, cdc codec.JSONCodec): Calls theExportGenesisfunction of each module, in the order defined inOrderExportGenesis. The export constructs a genesis file from a previously existing state, and is mainly used when a hard-fork upgrade of the chain is required.ExportGenesisForModules(ctx sdk.Context, cdc codec.JSONCodec, modulesToExport []string): Behaves the same asExportGenesis, except takes a list of modules to export.BeginBlock(ctx sdk.Context, req abci.RequestBeginBlock): At the beginning of each block, this function is called fromBaseAppand, in turn, calls theBeginBlockfunction of each modules implementing theBeginBlockAppModuleinterface, in the order defined inOrderBeginBlockers. It creates a child context with an event manager to aggregate events emitted from all modules. The function returns anabci.ResponseBeginBlockwhich contains the aforementioned events.EndBlock(ctx sdk.Context, req abci.RequestEndBlock): At the end of each block, this function is called fromBaseAppand, in turn, calls theEndBlockfunction of each modules implementing theEndBlockAppModuleinterface, in the order defined inOrderEndBlockers. It creates a child context with an event manager to aggregate events emitted from all modules. The function returns anabci.ResponseEndBlockwhich contains the aforementioned events, as well as validator set updates (if any).Precommit(ctx sdk.Context): DuringCommit, this function is called fromBaseAppimmediately before thedeliverStateis written to the underlyingrootMultiStoreand, in turn calls thePrecommitfunction of each modules implementing theHasPrecommitinterface, in the order defined inOrderPrecommiters. It creates a child context where the underlyingCacheMultiStoreis that of the newly committed block'sdeliverState.PrepareCheckState(ctx sdk.Context): DuringCommit, this function is called fromBaseAppimmediately after thedeliverStateis written to the underlyingrootMultiStoreand, in turn calls thePrepareCheckStatefunction of each module implementing theHasPrepareCheckStateinterface, in the order defined inOrderPrepareCheckStaters. It creates a child context where the underlyingCacheMultiStoreis that of the next block'scheckState. Writes to this state will be present in thecheckStateof the next block, and therefore this method can be used to prepare thecheckStatefor the next block.
Here's an example of a concrete integration within an simapp:
loading...
This is the same example from runtime (the package that powers app v2):
loading...
loading...