Skip to content

Commit 22b44e0

Browse files
committed
add store upgrade for incentive module
1 parent 20fc247 commit 22b44e0

File tree

2 files changed

+23
-0
lines changed

2 files changed

+23
-0
lines changed

app/app.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1142,6 +1142,20 @@ func BlockedAddresses() map[string]bool {
11421142
// SetUpgradeHandlers sets the upgrade handlers
11431143
func (app *App) SetUpgradeHandlers() {
11441144
app.UpgradeKeeper.SetUpgradeHandler(upgradev2.UpgradeName, upgradev2.CreateUpgradeHandler(app.ModuleManager, app.configurator))
1145+
1146+
upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
1147+
if err != nil {
1148+
panic(fmt.Sprintf("failed to read upgrade info from disk: %v", err))
1149+
}
1150+
1151+
if app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
1152+
return
1153+
}
1154+
1155+
// register store loader for current upgrade
1156+
if upgradeInfo.Name == upgradev2.UpgradeName {
1157+
app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &upgradev2.StoreUpgrades))
1158+
}
11451159
}
11461160

11471161
func GetWasmOpts(appOpts servertypes.AppOptions) []wasmkeeper.Option {

app/upgrades/v2/upgrade.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,22 @@ package v2
33
import (
44
"context"
55

6+
storetypes "cosmossdk.io/store/types"
67
upgradetypes "cosmossdk.io/x/upgrade/types"
78
"github.com/cosmos/cosmos-sdk/types/module"
9+
10+
incentivetypes "github.com/sideprotocol/side/x/incentive/types"
811
)
912

1013
// UpgradeName is the upgrade version name
1114
const UpgradeName = "v2"
1215

16+
var StoreUpgrades = storetypes.StoreUpgrades{
17+
Added: []string{
18+
incentivetypes.ModuleName,
19+
},
20+
}
21+
1322
// CreateUpgradeHandler creates the upgrade handler
1423
func CreateUpgradeHandler(
1524
mm *module.Manager,

0 commit comments

Comments
 (0)