Skip to content

Commit

Permalink
add store upgrade for incentive module
Browse files Browse the repository at this point in the history
  • Loading branch information
keithsue committed Jan 20, 2025
1 parent 20fc247 commit 22b44e0
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 0 deletions.
14 changes: 14 additions & 0 deletions app/app.go
Original file line number Diff line number Diff line change
Expand Up @@ -1142,6 +1142,20 @@ func BlockedAddresses() map[string]bool {
// SetUpgradeHandlers sets the upgrade handlers
func (app *App) SetUpgradeHandlers() {
app.UpgradeKeeper.SetUpgradeHandler(upgradev2.UpgradeName, upgradev2.CreateUpgradeHandler(app.ModuleManager, app.configurator))

upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
if err != nil {
panic(fmt.Sprintf("failed to read upgrade info from disk: %v", err))
}

if app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
return
}

// register store loader for current upgrade
if upgradeInfo.Name == upgradev2.UpgradeName {
app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &upgradev2.StoreUpgrades))
}
}

func GetWasmOpts(appOpts servertypes.AppOptions) []wasmkeeper.Option {
Expand Down
9 changes: 9 additions & 0 deletions app/upgrades/v2/upgrade.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,22 @@ package v2
import (
"context"

storetypes "cosmossdk.io/store/types"
upgradetypes "cosmossdk.io/x/upgrade/types"
"github.com/cosmos/cosmos-sdk/types/module"

incentivetypes "github.com/sideprotocol/side/x/incentive/types"
)

// UpgradeName is the upgrade version name
const UpgradeName = "v2"

var StoreUpgrades = storetypes.StoreUpgrades{
Added: []string{
incentivetypes.ModuleName,
},
}

// CreateUpgradeHandler creates the upgrade handler
func CreateUpgradeHandler(
mm *module.Manager,
Expand Down

0 comments on commit 22b44e0

Please sign in to comment.