Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| package app | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "path" | ||
| "strconv" | ||
|
|
||
| storetypes "github.com/cosmos/cosmos-sdk/store/types" | ||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| "github.com/cosmos/cosmos-sdk/types/module" | ||
| ) | ||
|
|
||
| type upgradeHandlerFn func(ctx sdk.Context, vm module.VersionMap) (module.VersionMap, error) | ||
|
|
||
| type upgradeTrackerItem struct { | ||
| // Monotonically increasing index to order and track migrations. Typically the current unix epoch timestamp. | ||
| index int64 | ||
| // Function that will run during the SetUpgradeHandler callback. The VersionMap must always be returned. | ||
| upgradeHandler upgradeHandlerFn | ||
| // StoreUpgrades that will be provided to UpgradeStoreLoader | ||
| storeUpgrade *storetypes.StoreUpgrades | ||
| } | ||
|
|
||
| // upgradeTracker allows us to track needed upgrades/migrations across both release and develop builds | ||
| type upgradeTracker struct { | ||
| upgrades []upgradeTrackerItem | ||
| stateFileDir string | ||
| } | ||
|
|
||
| func (t upgradeTracker) getDevelopUpgrades() ([]upgradeHandlerFn, *storetypes.StoreUpgrades, error) { | ||
| neededUpgrades := &storetypes.StoreUpgrades{} | ||
| neededUpgradeHandlers := []upgradeHandlerFn{} | ||
| stateFilePath := path.Join(t.stateFileDir, "developupgradetracker") | ||
|
|
||
| currentIndex := int64(0) | ||
| // #nosec G304 stateFilePath is not user controllable | ||
| if stateFileContents, err := os.ReadFile(stateFilePath); err == nil { | ||
| currentIndex, err = strconv.ParseInt(string(stateFileContents), 10, 64) | ||
| if err != nil { | ||
| return nil, nil, fmt.Errorf("unable to decode upgrade tracker: %w", err) | ||
| } | ||
| } else { | ||
| fmt.Printf("unable to load upgrade tracker: %v\n", err) | ||
| } | ||
|
|
||
| maxIndex := currentIndex | ||
| for _, item := range t.upgrades { | ||
| index := item.index | ||
| upgrade := item.storeUpgrade | ||
| versionModifier := item.upgradeHandler | ||
| if index <= currentIndex { | ||
| continue | ||
| } | ||
| if versionModifier != nil { | ||
| neededUpgradeHandlers = append(neededUpgradeHandlers, versionModifier) | ||
| } | ||
| if upgrade != nil { | ||
| neededUpgrades.Added = append(neededUpgrades.Added, upgrade.Added...) | ||
| neededUpgrades.Deleted = append(neededUpgrades.Deleted, upgrade.Deleted...) | ||
| neededUpgrades.Renamed = append(neededUpgrades.Renamed, upgrade.Renamed...) | ||
| } | ||
| maxIndex = index | ||
| } | ||
| err := os.WriteFile(stateFilePath, []byte(strconv.FormatInt(maxIndex, 10)), 0o600) | ||
| if err != nil { | ||
| return nil, nil, fmt.Errorf("unable to write upgrade state file: %w", err) | ||
| } | ||
| return neededUpgradeHandlers, neededUpgrades, nil | ||
| } | ||
|
|
||
| func (t upgradeTracker) mergeAllUpgrades() ([]upgradeHandlerFn, *storetypes.StoreUpgrades, error) { | ||
| upgrades := &storetypes.StoreUpgrades{} | ||
| upgradeHandlers := []upgradeHandlerFn{} | ||
| for _, item := range t.upgrades { | ||
| upgrade := item.storeUpgrade | ||
| versionModifier := item.upgradeHandler | ||
| if versionModifier != nil { | ||
| upgradeHandlers = append(upgradeHandlers, versionModifier) | ||
| } | ||
| if upgrade != nil { | ||
| upgrades.Added = append(upgrades.Added, upgrade.Added...) | ||
| upgrades.Deleted = append(upgrades.Deleted, upgrade.Deleted...) | ||
| upgrades.Renamed = append(upgrades.Renamed, upgrade.Renamed...) | ||
| } | ||
| } | ||
| return upgradeHandlers, upgrades, nil | ||
| } | ||
|
|
||
| func (t upgradeTracker) getUpgrades(isDevelop bool) ([]upgradeHandlerFn, *storetypes.StoreUpgrades, error) { | ||
| if isDevelop { | ||
| return t.getDevelopUpgrades() | ||
| } | ||
| return t.mergeAllUpgrades() | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters