github.com/cosmos/cosmos-sdk@v0.50.10/x/staking/migrations/v2/store.go (about)

     1  package v2
     2  
     3  import (
     4  	"cosmossdk.io/store/prefix"
     5  	storetypes "cosmossdk.io/store/types"
     6  
     7  	sdk "github.com/cosmos/cosmos-sdk/types"
     8  	"github.com/cosmos/cosmos-sdk/types/address"
     9  	v1auth "github.com/cosmos/cosmos-sdk/x/auth/migrations/v1"
    10  	v2distribution "github.com/cosmos/cosmos-sdk/x/distribution/migrations/v2"
    11  	v1 "github.com/cosmos/cosmos-sdk/x/staking/migrations/v1"
    12  	"github.com/cosmos/cosmos-sdk/x/staking/types"
    13  )
    14  
    15  // migratePrefixAddressAddressAddress is a helper function that migrates all keys of format:
    16  // prefix_bytes | address_1_bytes | address_2_bytes | address_3_bytes
    17  // into format:
    18  // prefix_bytes | address_1_len (1 byte) | address_1_bytes | address_2_len (1 byte) | address_2_bytes | address_3_len (1 byte) | address_3_bytes
    19  func migratePrefixAddressAddressAddress(store storetypes.KVStore, prefixBz []byte) {
    20  	oldStore := prefix.NewStore(store, prefixBz)
    21  
    22  	oldStoreIter := oldStore.Iterator(nil, nil)
    23  	defer oldStoreIter.Close()
    24  
    25  	for ; oldStoreIter.Valid(); oldStoreIter.Next() {
    26  		addr1 := oldStoreIter.Key()[:v1auth.AddrLen]
    27  		addr2 := oldStoreIter.Key()[v1auth.AddrLen : 2*v1auth.AddrLen]
    28  		addr3 := oldStoreIter.Key()[2*v1auth.AddrLen:]
    29  		newStoreKey := append(append(append(
    30  			prefixBz,
    31  			address.MustLengthPrefix(addr1)...), address.MustLengthPrefix(addr2)...), address.MustLengthPrefix(addr3)...,
    32  		)
    33  
    34  		// Set new key on store. Values don't change.
    35  		store.Set(newStoreKey, oldStoreIter.Value())
    36  		oldStore.Delete(oldStoreIter.Key())
    37  	}
    38  }
    39  
    40  const powerBytesLen = 8
    41  
    42  func migrateValidatorsByPowerIndexKey(store storetypes.KVStore) {
    43  	oldStore := prefix.NewStore(store, v1.ValidatorsByPowerIndexKey)
    44  
    45  	oldStoreIter := oldStore.Iterator(nil, nil)
    46  	defer oldStoreIter.Close()
    47  
    48  	for ; oldStoreIter.Valid(); oldStoreIter.Next() {
    49  		powerBytes := oldStoreIter.Key()[:powerBytesLen]
    50  		valAddr := oldStoreIter.Key()[powerBytesLen:]
    51  		newStoreKey := append(append(types.ValidatorsByPowerIndexKey, powerBytes...), address.MustLengthPrefix(valAddr)...)
    52  
    53  		// Set new key on store. Values don't change.
    54  		store.Set(newStoreKey, oldStoreIter.Value())
    55  		oldStore.Delete(oldStoreIter.Key())
    56  	}
    57  }
    58  
    59  // MigrateStore performs in-place store migrations from v0.40 to v0.43. The
    60  // migration includes:
    61  //
    62  // - Setting the Power Reduction param in the paramstore
    63  func MigrateStore(ctx sdk.Context, store storetypes.KVStore) error {
    64  	v2distribution.MigratePrefixAddress(store, v1.LastValidatorPowerKey)
    65  	v2distribution.MigratePrefixAddress(store, v1.ValidatorsKey)
    66  	v2distribution.MigratePrefixAddress(store, v1.ValidatorsByConsAddrKey)
    67  	migrateValidatorsByPowerIndexKey(store)
    68  
    69  	v2distribution.MigratePrefixAddressAddress(store, v1.DelegationKey)
    70  	v2distribution.MigratePrefixAddressAddress(store, v1.UnbondingDelegationKey)
    71  	v2distribution.MigratePrefixAddressAddress(store, v1.UnbondingDelegationByValIndexKey)
    72  	migratePrefixAddressAddressAddress(store, v1.RedelegationKey)
    73  	migratePrefixAddressAddressAddress(store, v1.RedelegationByValSrcIndexKey)
    74  	migratePrefixAddressAddressAddress(store, v1.RedelegationByValDstIndexKey)
    75  
    76  	return nil
    77  }