Skip to content

Commit 288e920

Browse files
committed
fix(upgrade): use preupgrade store prefixes from migrate package
Signed-off-by: Artur Troian <[email protected]>
1 parent 70aa0ec commit 288e920

File tree

3 files changed

+9
-11
lines changed

3 files changed

+9
-11
lines changed

upgrades/software/v1.0.0/deployment.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ func (m deploymentsMigrations) GetHandler() sdkmodule.MigrationHandler {
3232
func (m deploymentsMigrations) handler(ctx sdk.Context) error {
3333
store := ctx.KVStore(m.StoreKey())
3434

35-
oStore := prefix.NewStore(store, dkeeper.DeploymentPrefix)
35+
oStore := prefix.NewStore(store, migrate.DeploymentV1beta3Prefix())
3636

3737
iter := oStore.Iterator(nil, nil)
3838
defer func() {
@@ -66,7 +66,7 @@ func (m deploymentsMigrations) handler(ctx sdk.Context) error {
6666
store.Set(key, bz)
6767
}
6868

69-
oStore = prefix.NewStore(store, dkeeper.GroupPrefix)
69+
oStore = prefix.NewStore(store, migrate.GroupV1beta3Prefix())
7070

7171
iter = oStore.Iterator(nil, nil)
7272
defer func() {

upgrades/software/v1.0.0/market.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ func (m marketMigrations) handler(ctx sdk.Context) error {
3535

3636
cdc := m.Codec()
3737

38-
oiter := storetypes.KVStorePrefixIterator(store, mkeys.OrderPrefix)
38+
oiter := storetypes.KVStorePrefixIterator(store, migrate.OrderV1beta4Prefix())
3939
defer func() {
4040
_ = oiter.Close()
4141
}()
@@ -70,8 +70,7 @@ func (m marketMigrations) handler(ctx sdk.Context) error {
7070
}
7171

7272
store.Delete(mkeys.BidPrefixReverse)
73-
74-
biter := storetypes.KVStorePrefixIterator(store, mkeys.BidPrefix)
73+
biter := storetypes.KVStorePrefixIterator(store, migrate.BidV1beta4Prefix())
7574
defer func() {
7675
_ = biter.Close()
7776
}()
@@ -125,7 +124,7 @@ func (m marketMigrations) handler(ctx sdk.Context) error {
125124
}
126125

127126
store.Delete(mkeys.LeasePrefixReverse)
128-
liter := storetypes.KVStorePrefixIterator(store, mkeys.LeasePrefix)
127+
liter := storetypes.KVStorePrefixIterator(store, migrate.LeaseV1beta4Prefix())
129128
defer func() {
130129
_ = liter.Close()
131130
}()

upgrades/software/v1.0.0/provider.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ import (
77
sdk "github.com/cosmos/cosmos-sdk/types"
88
sdkmodule "github.com/cosmos/cosmos-sdk/types/module"
99
"pkg.akt.dev/go/node/migrate"
10-
types "pkg.akt.dev/go/node/provider/v1beta4"
1110
"pkg.akt.dev/go/sdkutil"
1211

1312
utypes "pkg.akt.dev/node/upgrades/types"
@@ -29,9 +28,9 @@ func (m providerMigrations) GetHandler() sdkmodule.MigrationHandler {
2928
// handler migrates provider store from version 2 to 3.
3029
func (m providerMigrations) handler(ctx sdk.Context) (err error) {
3130
store := ctx.KVStore(m.StoreKey())
32-
pstore := prefix.NewStore(store, types.ProviderPrefix())
31+
oStore := prefix.NewStore(store, migrate.PaymentV1beta3Prefix())
3332

34-
iter := store.Iterator(nil, nil)
33+
iter := oStore.Iterator(nil, nil)
3534
defer func() {
3635
err = iter.Close()
3736
}()
@@ -44,8 +43,8 @@ func (m providerMigrations) handler(ctx sdk.Context) (err error) {
4443
id := sdkutil.MustAccAddressFromBech32(to.Owner)
4544
bz := cdc.MustMarshal(&to)
4645

47-
store.Delete(iter.Key())
48-
pstore.Set(pkeeper.ProviderKey(id), bz)
46+
oStore.Delete(iter.Key())
47+
store.Set(pkeeper.ProviderKey(id), bz)
4948
}
5049

5150
return nil

0 commit comments

Comments
 (0)