Run account plus migration in v8 upgrade (backport #2628) (#2629)

Co-authored-by: jayy04 <[email protected]>
dydxprotocol · Dec 6, 2024 · c0a8826 · c0a8826
1 parent 1492a32
commit c0a8826
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/protocol/app/upgrades.go b/protocol/app/upgrades.go
@@ -33,6 +33,7 @@ func (app *App) setupUpgradeHandlers() {
 			app.PricesKeeper,
 			app.PerpetualsKeeper,
 			app.ClobKeeper,
+			app.AccountPlusKeeper,
 		),
 	)
 }

diff --git a/protocol/app/upgrades/v8.0/upgrade.go b/protocol/app/upgrades/v8.0/upgrade.go
@@ -94,11 +94,14 @@ func CreateUpgradeHandler(
 	pricesKeeper pricestypes.PricesKeeper,
 	perpetualsKeeper perptypes.PerpetualsKeeper,
 	clobKeeper clobtypes.ClobKeeper,
+	accountplusKeeper accountpluskeeper.Keeper,
 ) upgradetypes.UpgradeHandler {
 	return func(ctx context.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
 		sdkCtx := lib.UnwrapSDKContext(ctx, "app/upgrades")
 		sdkCtx.Logger().Info(fmt.Sprintf("Running %s Upgrade...", UpgradeName))
 
+		MigrateAccountplusAccountState(sdkCtx, accountplusKeeper)
+
 		// Set market, perpetual, and clob ids to a set number
 		setMarketListingBaseIds(sdkCtx, pricesKeeper, perpetualsKeeper, clobKeeper)