fix(staking): temporary fix for key duplicated issue (#28)

* fix(staking): temporary solution for key duplicated issue

* fix(staking): add GetUnbondingID func

Author: gezerui0124

x/staking/keeper/unbonding.go

@@ -10,6 +10,48 @@ import (
 	"github.com/cosmos/cosmos-sdk/x/staking/types"
 )
 
+func (k Keeper) GetOldUnbondingID(ctx context.Context) (unbondingID uint64, err error) {
+	store := k.storeService.OpenKVStore(ctx)
+	// Old UnbondingID used same key as period delegation
+	bz, err := store.Get(types.PeriodDelegationKey)
+	if err != nil {
+		return 0, err
+	}
+
+	if bz != nil {
+		unbondingID = binary.BigEndian.Uint64(bz)
+	}
+
+	return unbondingID, err
+}
+
+func (k Keeper) GetUnbondingID(ctx context.Context) (unbondingID uint64, err error) {
+	store := k.storeService.OpenKVStore(ctx)
+	bz, err := store.Get(types.UnbondingIDKey)
+	if err != nil {
+		return 0, err
+	}
+
+	if bz != nil {
+		unbondingID = binary.BigEndian.Uint64(bz)
+	}
+
+	return unbondingID, err
+}
+
+func (k Keeper) SetUnbondingID(ctx context.Context, unbondingID uint64) error {
+	store := k.storeService.OpenKVStore(ctx)
+	// Convert into bytes for storage
+	bz := make([]byte, 8)
+	binary.BigEndian.PutUint64(bz, unbondingID)
+
+	if err := store.Set(types.UnbondingIDKey, bz); err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // IncrementUnbondingID increments and returns a unique ID for an unbonding operation
 func (k Keeper) IncrementUnbondingID(ctx context.Context) (unbondingID uint64, err error) {
 	store := k.storeService.OpenKVStore(ctx)

x/staking/types/keys.go

@@ -43,9 +43,9 @@ var (
 	RedelegationByValDstIndexKey     = []byte{0x36} // prefix for each key for an redelegation, by destination validator operator
 	PeriodDelegationKey              = []byte{0x37} // key for a period delegation
 
-	UnbondingIDKey    = []byte{0x37} // key for the counter for the incrementing id for UnbondingOperations
 	UnbondingIndexKey = []byte{0x38} // prefix for an index for looking up unbonding operations by their IDs
 	UnbondingTypeKey  = []byte{0x39} // prefix for an index containing the type of unbonding operations
+	UnbondingIDKey    = []byte{0x40} // key for the counter for the incrementing id for UnbondingOperations
 
 	UnbondingQueueKey    = []byte{0x41} // prefix for the timestamps in unbonding queue
 	RedelegationQueueKey = []byte{0x42} // prefix for the timestamps in redelegations queue