48 lines
1.6 KiB
Go
48 lines
1.6 KiB
Go
package keeper
|
|
|
|
import (
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/slashing/exported"
|
|
v2 "github.com/cosmos/cosmos-sdk/x/slashing/migrations/v2"
|
|
v3 "github.com/cosmos/cosmos-sdk/x/slashing/migrations/v3"
|
|
v4 "github.com/cosmos/cosmos-sdk/x/slashing/migrations/v4"
|
|
)
|
|
|
|
// Migrator is a struct for handling in-place store migrations.
|
|
type Migrator struct {
|
|
keeper Keeper
|
|
legacySubspace exported.Subspace
|
|
}
|
|
|
|
// NewMigrator returns a new Migrator.
|
|
func NewMigrator(keeper Keeper, ss exported.Subspace) Migrator {
|
|
return Migrator{keeper: keeper, legacySubspace: ss}
|
|
}
|
|
|
|
// Migrate1to2 migrates from version 1 to 2.
|
|
func (m Migrator) Migrate1to2(ctx sdk.Context) error {
|
|
return v2.MigrateStore(ctx, m.keeper.storeService)
|
|
}
|
|
|
|
// Migrate2to3 migrates the x/slashing module state from the consensus
|
|
// version 2 to version 3. Specifically, it takes the parameters that are currently stored
|
|
// and managed by the x/params modules and stores them directly into the x/slashing
|
|
// module state.
|
|
func (m Migrator) Migrate2to3(ctx sdk.Context) error {
|
|
store := runtime.KVStoreAdapter(m.keeper.storeService.OpenKVStore(ctx))
|
|
return v3.Migrate(ctx, store, m.legacySubspace, m.keeper.cdc)
|
|
}
|
|
|
|
// Migrate3to4 migrates the x/slashing module state from the consensus
|
|
// version 3 to version 4. Specifically, it migrates the validator missed block
|
|
// bitmap.
|
|
func (m Migrator) Migrate3to4(ctx sdk.Context) error {
|
|
store := runtime.KVStoreAdapter(m.keeper.storeService.OpenKVStore(ctx))
|
|
params, err := m.keeper.Params.Get(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return v4.Migrate(ctx, m.keeper.cdc, store, params)
|
|
}
|