-
Notifications
You must be signed in to change notification settings - Fork 6
/
migrations.go
32 lines (27 loc) · 1.01 KB
/
migrations.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package keeper
import (
"github.com/chain4energy/c4e-chain/types/subspace"
v2 "github.com/chain4energy/c4e-chain/x/cfevesting/migrations/v2"
v3 "github.com/chain4energy/c4e-chain/x/cfevesting/migrations/v3"
sdk "github.com/cosmos/cosmos-sdk/types"
)
// Migrator is a struct for handling in-place store migrations.
type Migrator struct {
keeper Keeper
legacySubspace subspace.Subspace
}
// NewMigrator returns a new Migrator.
func NewMigrator(keeper Keeper, legacySubspace subspace.Subspace) Migrator {
return Migrator{keeper: keeper, legacySubspace: legacySubspace}
}
// Migrate2to3 migrates from version 1 to 2.
func (m Migrator) Migrate1to2(ctx sdk.Context) error {
return v2.MigrateStore(ctx, m.keeper.storeKey, m.keeper.cdc)
}
// Migrate2to3 migrates from version 2 to 3.
func (m Migrator) Migrate2to3(ctx sdk.Context) error {
if err := v3.MigrateParams(ctx, m.keeper.storeKey, m.legacySubspace, m.keeper.cdc); err != nil {
return err
}
return v3.MigrateStore(ctx, m.keeper.storeKey, m.keeper.cdc)
}