69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
|
|
v5 "cosmossdk.io/x/auth/migrations/v5"
|
|
"cosmossdk.io/x/auth/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// Migrator is a struct for handling in-place store migrations.
|
|
type Migrator struct {
|
|
keeper AccountKeeper
|
|
}
|
|
|
|
// NewMigrator returns a new Migrator.
|
|
func NewMigrator(keeper AccountKeeper) Migrator {
|
|
return Migrator{keeper: keeper}
|
|
}
|
|
|
|
// Migrate1to2 migrates from version 1 to 2.
|
|
func (m Migrator) Migrate1to2(ctx context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
// Migrate2to3 migrates from consensus version 2 to version 3. Specifically, for each account
|
|
// we index the account's ID to their address.
|
|
func (m Migrator) Migrate2to3(ctx context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
// Migrate3to4 migrates the x/auth module state from the consensus version 3 to
|
|
// version 4. Specifically, it takes the parameters that are currently stored
|
|
// and managed by the x/params modules and stores them directly into the x/auth
|
|
// module state.
|
|
func (m Migrator) Migrate3to4(ctx context.Context) error {
|
|
return nil
|
|
}
|
|
|
|
// Migrate4To5 migrates the x/auth module state from the consensus version 4 to 5.
|
|
// It migrates the GlobalAccountNumber from being a protobuf defined value to a
|
|
// big-endian encoded uint64, it also migrates it to use a more canonical prefix.
|
|
func (m Migrator) Migrate4To5(ctx context.Context) error {
|
|
return v5.Migrate(ctx, m.keeper.KVStoreService, m.keeper.accountNumber)
|
|
}
|
|
|
|
// V45SetAccount implements V45_SetAccount
|
|
// set the account without map to accAddr to accNumber.
|
|
//
|
|
// NOTE: This is used for testing purposes only.
|
|
func (m Migrator) V45SetAccount(ctx context.Context, acc sdk.AccountI) error {
|
|
addr := acc.GetAddress()
|
|
store := m.keeper.KVStoreService.OpenKVStore(ctx)
|
|
|
|
bz, err := m.keeper.Accounts.ValueCodec().Encode(acc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return store.Set(addressStoreKey(addr), bz)
|
|
}
|
|
|
|
// addressStoreKey turn an address to key used to get it from the account store
|
|
// NOTE(tip): exists for legacy compatibility
|
|
func addressStoreKey(addr sdk.AccAddress) []byte {
|
|
return append(types.AddressStoreKeyPrefix, addr.Bytes()...)
|
|
}
|