diff --git a/app/upgrade.go b/app/upgrade.go index 5ebb20570..f3e3ef774 100644 --- a/app/upgrade.go +++ b/app/upgrade.go @@ -34,6 +34,7 @@ func (app *App) RegisterUpgradeHandlers(chainID string, serverCfg *serverconfig. app.registerPawneeUpgradeHandler() app.registerSerengetiUpgradeHandler() app.registerErdosUpgradeHandler() + app.registerVeldUpgradeHandler() // app.register...() // ... return nil @@ -262,3 +263,19 @@ func (app *App) registerErdosUpgradeHandler() { return nil }) } + +func (app *App) registerVeldUpgradeHandler() { + // Register the upgrade handler + app.UpgradeKeeper.SetUpgradeHandler(upgradetypes.Veld, + func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) { + app.Logger().Info("upgrade to ", plan.Name) + return app.mm.RunMigrations(ctx, app.configurator, fromVM) + }) + + // Register the upgrade initializer + app.UpgradeKeeper.SetUpgradeInitializer(upgradetypes.Veld, + func() error { + app.Logger().Info("Init Veld upgrade") + return nil + }) +}