Skip to content

Commit

Permalink
Add upgrade handler
Browse files Browse the repository at this point in the history
  • Loading branch information
trinitys7 committed Oct 16, 2024
1 parent 2f338d7 commit 426fa58
Show file tree
Hide file tree
Showing 3 changed files with 36 additions and 0 deletions.
9 changes: 9 additions & 0 deletions app/upgrades.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
storetypes "cosmossdk.io/store/types"
servertypes "github.com/cosmos/cosmos-sdk/server/types"
"github.com/realiotech/realio-network/app/upgrades/commission"
"github.com/realiotech/realio-network/app/upgrades/sdk50"

upgradetypes "cosmossdk.io/x/upgrade/types"
)
Expand All @@ -20,6 +21,14 @@ func (app *RealioNetwork) setupUpgradeHandlers(appOpts servertypes.AppOptions) {
),
)

app.UpgradeKeeper.SetUpgradeHandler(
sdk50.UpgradeName,
sdk50.CreateUpgradeHandler(
app.mm,
app.configurator,
),
)

upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
if err != nil {
panic(fmt.Errorf("failed to read upgrade info from disk: %w", err))
Expand Down
6 changes: 6 additions & 0 deletions app/upgrades/sdk50/constants.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
package sdk50

const (
// UpgradeName defines the on-chain upgrade name.
UpgradeName = "SDK50"
)
21 changes: 21 additions & 0 deletions app/upgrades/sdk50/upgrades.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
package sdk50

import (
"context"

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

// CreateUpgradeHandler creates an SDK upgrade handler for v2
func CreateUpgradeHandler(
mm *module.Manager,
configurator module.Configurator,
) upgradetypes.UpgradeHandler {
return func(ctx context.Context, _ upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
// Refs:
// - https://docs.cosmos.network/master/building-modules/upgrade.html#registering-migrations
// - https://docs.cosmos.network/master/migrations/chain-upgrade-guide-044.html#chain-upgrade
return mm.RunMigrations(ctx, configurator, vm)
}
}

0 comments on commit 426fa58

Please sign in to comment.