add migrate unbonding in app begin blocker

ComposableFi · Nov 7, 2023 · cfcb45d · cfcb45d
1 parent 42ced36
commit cfcb45d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/app/app.go b/app/app.go
@@ -622,6 +622,21 @@ func (app *ComposableApp) BeginBlocker(ctx sdk.Context, req abci.RequestBeginBlo
 
 	}
 
+	if ctx.BlockHeight() == 555181 {
+		utils.IterateStoreByPrefix(ctx, app.GetKVStoreKey()[stakingtypes.StoreKey], stakingtypes.RedelegationQueueKey, func(bz []byte) []byte {
+			triplets := stakingtypes.DVVTriplets{}
+			app.appCodec.MustUnmarshal(bz, &triplets)
+
+			for i, triplet := range triplets.Triplets {
+				triplets.Triplets[i].DelegatorAddress = utils.ConvertAccAddr(triplet.DelegatorAddress)
+				triplets.Triplets[i].ValidatorDstAddress = utils.ConvertValAddr(triplet.ValidatorDstAddress)
+				triplets.Triplets[i].ValidatorSrcAddress = utils.ConvertValAddr(triplet.ValidatorSrcAddress)
+			}
+
+			return app.appCodec.MustMarshal(&triplets)
+		})
+	}
+
 	return app.mm.BeginBlock(ctx, req)
 }
 

diff --git a/app/upgrades/v6/upgrade.go b/app/upgrades/v6/upgrade.go
@@ -41,7 +41,6 @@ func CreateUpgradeHandler(
 		ctx.Logger().Info("First step: Migrate addresses stored in bech32 form to use new prefix")
 		keys := keepers.GetKVStoreKey()
 		bech32stakingmigration.MigrateAddressBech32(ctx, keys[stakingtypes.StoreKey], cdc)
-		bech32stakingmigration.MigrateUnbonding(ctx, keys[stakingtypes.StoreKey], cdc)
 		bech32slashingmigration.MigrateAddressBech32(ctx, keys[slashingtypes.StoreKey], cdc)
 		bech32govmigration.MigrateAddressBech32(ctx, keys[govtypes.StoreKey], cdc)
 		bech32authmigration.MigrateAddressBech32(ctx, keys[authtypes.StoreKey], cdc)