69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package keeper
|
|
|
|
import (
|
|
"github.com/gogo/protobuf/grpc"
|
|
|
|
v043 "github.com/cosmos/cosmos-sdk/x/auth/migrations/v043"
|
|
v046 "github.com/cosmos/cosmos-sdk/x/auth/migrations/v046"
|
|
|
|
"github.com/cosmos/cosmos-sdk/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
|
|
queryServer grpc.Server
|
|
}
|
|
|
|
// NewMigrator returns a new Migrator.
|
|
func NewMigrator(keeper AccountKeeper, queryServer grpc.Server) Migrator {
|
|
return Migrator{keeper: keeper, queryServer: queryServer}
|
|
}
|
|
|
|
// Migrate1to2 migrates from version 1 to 2.
|
|
func (m Migrator) Migrate1to2(ctx sdk.Context) error {
|
|
var iterErr error
|
|
|
|
m.keeper.IterateAccounts(ctx, func(account types.AccountI) (stop bool) {
|
|
wb, err := v043.MigrateAccount(ctx, account, m.queryServer)
|
|
if err != nil {
|
|
iterErr = err
|
|
return true
|
|
}
|
|
|
|
if wb == nil {
|
|
return false
|
|
}
|
|
|
|
m.keeper.SetAccount(ctx, wb)
|
|
return false
|
|
})
|
|
|
|
return iterErr
|
|
}
|
|
|
|
// 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 sdk.Context) error {
|
|
return v046.MigrateStore(ctx, m.keeper.key, m.keeper.cdc)
|
|
}
|
|
|
|
// V45_SetAccount implements V45_SetAccount
|
|
// set the account without map to accAddr to accNumber.
|
|
//
|
|
// NOTE: This is used for testing purposes only.
|
|
func (m Migrator) V45_SetAccount(ctx sdk.Context, acc types.AccountI) error {
|
|
addr := acc.GetAddress()
|
|
store := ctx.KVStore(m.keeper.key)
|
|
|
|
bz, err := m.keeper.MarshalAccount(acc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
store.Set(types.AddressStoreKey(addr), bz)
|
|
return nil
|
|
}
|