-
Notifications
You must be signed in to change notification settings - Fork 51
/
setup_handlers.go
59 lines (48 loc) · 1.99 KB
/
setup_handlers.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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package app
import (
"fmt"
storetypes "github.com/cosmos/cosmos-sdk/store/types"
sdk "github.com/cosmos/cosmos-sdk/types"
m "github.com/cosmos/cosmos-sdk/types/module"
"github.com/cosmos/cosmos-sdk/version"
upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
)
func SetupHandlers(app *ElysApp) {
setUpgradeHandler(app)
loadUpgradeStore(app)
}
func setUpgradeHandler(app *ElysApp) {
app.UpgradeKeeper.SetUpgradeHandler(
version.Version,
func(ctx sdk.Context, plan upgradetypes.Plan, vm m.VersionMap) (m.VersionMap, error) {
app.Logger().Info("Running upgrade handler for " + version.Version)
return app.mm.RunMigrations(ctx, app.configurator, vm)
},
)
}
func loadUpgradeStore(app *ElysApp) {
upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
if err != nil {
panic(fmt.Sprintf("Failed to read upgrade info from disk: %v", err))
}
fmt.Printf("Upgrade info: %+v\n", upgradeInfo)
if shouldLoadUpgradeStore(app, upgradeInfo) {
storeUpgrades := storetypes.StoreUpgrades{
// Added: []string{},
// Deleted: []string{},
}
fmt.Printf("Setting store loader with height %d and store upgrades: %+v\n", upgradeInfo.Height, storeUpgrades)
// Use upgrade store loader for the initial loading of all stores when app starts,
// it checks if version == upgradeHeight and applies store upgrades before loading the stores,
// so that new stores start with the correct version (the current height of chain),
// instead the default which is the latest version that store last committed i.e 0 for new stores.
app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &storeUpgrades))
} else {
fmt.Println("No need to load upgrade store.")
}
}
func shouldLoadUpgradeStore(app *ElysApp, upgradeInfo upgradetypes.Plan) bool {
currentHeight := app.LastBlockHeight()
fmt.Printf("Current block height: %d, Upgrade height: %d\n", currentHeight, upgradeInfo.Height)
return upgradeInfo.Name == version.Version && !app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height)
}