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

     1  package v2
     2  
     3  import (
     4  	"cosmossdk.io/store/prefix"
     5  	storetypes "cosmossdk.io/store/types"
     6  
     7  	"github.com/cosmos/cosmos-sdk/types/address"
     8  	v1auth "github.com/cosmos/cosmos-sdk/x/auth/migrations/v1"
     9  )
    10  
    11  // MigratePrefixAddress is a helper function that migrates all keys of format:
    12  // prefix_bytes | address_bytes
    13  // into format:
    14  // prefix_bytes | address_len (1 byte) | address_bytes
    15  func MigratePrefixAddress(store storetypes.KVStore, prefixBz []byte) {
    16  	oldStore := prefix.NewStore(store, prefixBz)
    17  
    18  	oldStoreIter := oldStore.Iterator(nil, nil)
    19  	defer oldStoreIter.Close()
    20  
    21  	for ; oldStoreIter.Valid(); oldStoreIter.Next() {
    22  		addr := oldStoreIter.Key()
    23  		newStoreKey := prefixBz
    24  		newStoreKey = append(newStoreKey, address.MustLengthPrefix(addr)...)
    25  
    26  		// Set new key on store. Values don't change.
    27  		store.Set(newStoreKey, oldStoreIter.Value())
    28  		oldStore.Delete(oldStoreIter.Key())
    29  	}
    30  }
    31  
    32  // MigratePrefixAddressBytes is a helper function that migrates all keys of format:
    33  // prefix_bytes | address_bytes | arbitrary_bytes
    34  // into format:
    35  // prefix_bytes | address_len (1 byte) | address_bytes | arbitrary_bytes
    36  func MigratePrefixAddressBytes(store storetypes.KVStore, prefixBz []byte) {
    37  	oldStore := prefix.NewStore(store, prefixBz)
    38  
    39  	oldStoreIter := oldStore.Iterator(nil, nil)
    40  	defer oldStoreIter.Close()
    41  
    42  	for ; oldStoreIter.Valid(); oldStoreIter.Next() {
    43  		addr := oldStoreIter.Key()[:v1auth.AddrLen]
    44  		endBz := oldStoreIter.Key()[v1auth.AddrLen:]
    45  		newStoreKey := append(append(prefixBz, address.MustLengthPrefix(addr)...), endBz...)
    46  
    47  		// Set new key on store. Values don't change.
    48  		store.Set(newStoreKey, oldStoreIter.Value())
    49  		oldStore.Delete(oldStoreIter.Key())
    50  	}
    51  }
    52  
    53  // MigratePrefixAddressAddress is a helper function that migrates all keys of format:
    54  // prefix_bytes | address_1_bytes | address_2_bytes
    55  // into format:
    56  // prefix_bytes | address_1_len (1 byte) | address_1_bytes | address_2_len (1 byte) | address_2_bytes
    57  func MigratePrefixAddressAddress(store storetypes.KVStore, prefixBz []byte) {
    58  	oldStore := prefix.NewStore(store, prefixBz)
    59  
    60  	oldStoreIter := oldStore.Iterator(nil, nil)
    61  	defer oldStoreIter.Close()
    62  
    63  	for ; oldStoreIter.Valid(); oldStoreIter.Next() {
    64  		addr1 := oldStoreIter.Key()[:v1auth.AddrLen]
    65  		addr2 := oldStoreIter.Key()[v1auth.AddrLen:]
    66  		newStoreKey := append(append(prefixBz, address.MustLengthPrefix(addr1)...), address.MustLengthPrefix(addr2)...)
    67  
    68  		// Set new key on store. Values don't change.
    69  		store.Set(newStoreKey, oldStoreIter.Value())
    70  		oldStore.Delete(oldStoreIter.Key())
    71  	}
    72  }