fix(staking): add remove old UnbondingID func (#29)

piplabs · Dec 5, 2024 · bf84b22 · bf84b22
1 parent b7c7e97
commit bf84b22
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/x/staking/keeper/unbonding.go b/x/staking/keeper/unbonding.go
@@ -25,6 +25,12 @@ func (k Keeper) GetOldUnbondingID(ctx context.Context) (unbondingID uint64, err
 	return unbondingID, err
 }
 
+func (k Keeper) RemoveOldUnbondingID(ctx context.Context) error {
+	store := k.storeService.OpenKVStore(ctx)
+	// Old UnbondingID used same key as period delegation, which is not used independently.
+	return store.Delete(types.PeriodDelegationKey)
+}
+
 func (k Keeper) GetUnbondingID(ctx context.Context) (unbondingID uint64, err error) {
 	store := k.storeService.OpenKVStore(ctx)
 	bz, err := store.Get(types.UnbondingIDKey)