Add upgrade logic

Author: philipsu522

app/app.go

@@ -245,6 +245,8 @@ type App struct {
 	// sm is the simulation manager
 	sm *module.SimulationManager
 
+	configurator module.Configurator
+
 	tracingInfo *tracing.TracingInfo
 }
 
@@ -571,7 +573,8 @@ func New(
 
 	app.mm.RegisterInvariants(&app.CrisisKeeper)
 	app.mm.RegisterRoutes(app.Router(), app.QueryRouter(), encodingConfig.Amino)
-	app.mm.RegisterServices(module.NewConfigurator(app.appCodec, app.MsgServiceRouter(), app.GRPCQueryRouter()))
+	app.configurator = module.NewConfigurator(app.appCodec, app.MsgServiceRouter(), app.GRPCQueryRouter())
+	app.mm.RegisterServices(app.configurator)
 
 	// create the simulation manager and define the order of the modules for deterministic simulations
 	app.sm = module.NewSimulationManager(
@@ -595,6 +598,7 @@ func New(
 	)
 	app.sm.RegisterStoreDecoders()
 
+	app.RegisterUpgradeHandlers()
 	// initialize stores
 	app.MountKVStores(keys)
 	app.MountTransientStores(tkeys)

app/upgrades.go

@@ -0,0 +1,21 @@
+package app
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+)
+
+const UPGRADE_1_0_1_BETA = "upgrade-1.0.1beta"
+
+func (app App) RegisterUpgradeHandlers() {
+	app.UpgradeKeeper.SetUpgradeHandler(UPGRADE_1_0_1_BETA, func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+		// Upgrade specific logic goes here
+		// For now, remove dex, epoch and oracle from the version map since
+		// they do not yet have upgrade logic
+		delete(fromVM, "dex")
+		delete(fromVM, "epoch")
+		delete(fromVM, "oracle")
+		return app.mm.RunMigrations(ctx, app.configurator, fromVM)
+	})
+}