-
Notifications
You must be signed in to change notification settings - Fork 557
/
upgrades.go
98 lines (85 loc) · 3 KB
/
upgrades.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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package simapp
import (
storetypes "cosmossdk.io/store/types"
circuittypes "cosmossdk.io/x/circuit/types"
upgradetypes "cosmossdk.io/x/upgrade/types"
consensusparamtypes "github.com/cosmos/cosmos-sdk/x/consensus/types"
crisistypes "github.com/cosmos/cosmos-sdk/x/crisis/types"
capabilitytypes "github.com/cosmos/ibc-go/modules/capability/types"
icacontrollertypes "github.com/cosmos/ibc-go/v8/modules/apps/27-interchain-accounts/controller/types"
ibcmock "github.com/cosmos/ibc-go/v8/testing/mock"
"github.com/cosmos/ibc-go/v8/testing/simapp/upgrades"
)
// registerUpgradeHandlers registers all supported upgrade handlers
func (app *SimApp) registerUpgradeHandlers() {
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V5,
upgrades.CreateDefaultUpgradeHandler(app.ModuleManager, app.configurator),
)
// NOTE: The moduleName arg of v6.CreateUpgradeHandler refers to the auth module ScopedKeeper name to which the channel capability should be migrated from.
// This should be the same string value provided upon instantiation of the ScopedKeeper with app.CapabilityKeeper.ScopeToModule()
// See: https://github.com/cosmos/ibc-go/blob/v6.1.0/testing/simapp/app.go#L310
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V6,
upgrades.CreateV6UpgradeHandler(
app.ModuleManager,
app.configurator,
app.appCodec,
app.keys[capabilitytypes.ModuleName],
app.CapabilityKeeper,
ibcmock.ModuleName+icacontrollertypes.SubModuleName,
),
)
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V7,
upgrades.CreateV7UpgradeHandler(
app.ModuleManager,
app.configurator,
app.appCodec,
app.IBCKeeper.ClientKeeper,
app.ConsensusParamsKeeper,
app.ParamsKeeper,
),
)
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V7_1,
upgrades.CreateV7LocalhostUpgradeHandler(app.ModuleManager, app.configurator, app.IBCKeeper.ClientKeeper),
)
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V8,
upgrades.CreateDefaultUpgradeHandler(
app.ModuleManager,
app.configurator,
),
)
app.UpgradeKeeper.SetUpgradeHandler(
upgrades.V8_1,
upgrades.CreateDefaultUpgradeHandler(
app.ModuleManager,
app.configurator,
),
)
upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
if err != nil {
panic(err)
}
if upgradeInfo.Name == upgrades.V7 && !app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
storeUpgrades := storetypes.StoreUpgrades{
Added: []string{
consensusparamtypes.StoreKey,
crisistypes.StoreKey,
},
}
// configure store loader that checks if version == upgradeHeight and applies store upgrades
app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &storeUpgrades))
}
if upgradeInfo.Name == upgrades.V8 && !app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
storeUpgrades := storetypes.StoreUpgrades{
Added: []string{
circuittypes.ModuleName,
},
}
// configure store loader that checks if version == upgradeHeight and applies store upgrades
app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &storeUpgrades))
}
}