Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

x/upgrade: added consensus version tracking (part of ADR-041) #8743

Merged
merged 42 commits into from
Mar 19, 2021

Conversation

technicallyty
Copy link
Contributor

@technicallyty technicallyty commented Mar 2, 2021

Description

This PR is part of ADR 041 - In-Place Store Migrations

-Adds a new method to module.Manager to get a VersionMap
-Adds a new store prefix 0x2 to x/upgrade to store a VersionMap
-Adds new functionality to x/upgrade to get a VersionMap from state
-Saves a VersionMap on InitGenesis in app.go

You may notice some things in simapp had to be added - this is to avoid circular dependancy. UpgradeKeeper needs a reference to Module Manager, so after the manager is initialized, we give the upgradekeeper a reference to it, and re-set the value of the upgrade keeper in the manager's module map.

closes: #8514


Before we can merge this PR, please make sure that all the following items have been
checked off. If any of the checklist items are not applicable, please leave them but
write a little note why.

  • Targeted PR against correct branch (see CONTRIBUTING.md)
  • Linked to Github issue with discussion and accepted design OR link to spec that describes this work.
  • Code follows the module structure standards.
  • Wrote unit and integration tests
  • Updated relevant documentation (docs/) or specification (x/<module>/spec/)
  • Added relevant godoc comments.
  • Added a relevant changelog entry to the Unreleased section in CHANGELOG.md
  • Re-reviewed Files changed in the Github PR explorer
  • Review Codecov Report in the comment section below once CI passes

@technicallyty technicallyty marked this pull request as ready for review March 2, 2021 04:43
Copy link
Contributor

@amaury1093 amaury1093 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

x/upgrade's store wrt saving & retrieving module's ConsensusVersions looks good 👍

I'd like to see a test with applying an upgrade handler, and checking the store before and after the upgrade.

x/upgrade/types/keys.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
@@ -101,7 +101,8 @@ func (am AppModule) RegisterServices(cfg module.Configurator) {
}

// InitGenesis is ignored, no sense in serializing future upgrades
func (am AppModule) InitGenesis(_ sdk.Context, _ codec.JSONMarshaler, _ json.RawMessage) []abci.ValidatorUpdate {
func (am AppModule) InitGenesis(ctx sdk.Context, _ codec.JSONMarshaler, _ json.RawMessage) []abci.ValidatorUpdate {
InitChainer(am.keeper, ctx)
Copy link
Contributor

@amaury1093 amaury1093 Mar 2, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@tyler I think i might have been mistaken as to when we should call keeper.SetConsensusVersions. We actually don't need to call it in InitChain, just leave it blank on a fresh chain, as per #8504 (comment), I believe it should be called AFTER the handler has run (so here), to say "ok I've ran the handler (e.g. migration scripts), now I'll update the store with the newest modules' ConsensusVersions".

I think the ultimate E2E test would be with 2 binaries, but since that's hard, let's try to simulate that in a test here:

  • create a mock module.VersionManager, say with bank set to 1, and everything else set to latest ConsensusVersion.
  • upgradeKeeper.SetModuleManager(mockVersionMgr). This simulates saving the "old" modules version to state.
  • add an upgrade handler to keeper.
  • call ApplyUpgrade on keeper
  • after ApplyUpgrade, call keeper.GetConsensusVersions, test that the return value's bank key should be 2, not 1.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

test added and initchainer removed

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If it's not set in IntiChain then how do initial versions get set?? I don't see a good alternative and it's not a safe assumption that all chains start at module version 1

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we need to add this back

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

simapp/app.go Outdated Show resolved Hide resolved
Copy link
Member

@aaronc aaronc left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks like a great start here @technicallyty 👍. Would be nice to see some grpc query methods in a follow up PR.

…ng consensus version store in x/upgrade -cleaned up x/upgrade Keeper -handler in apply upgrade now handles errors and setting consensus versions -cleaned up migration map keys -removed init chainer method -simapp now implements GetConsensusVersions to assist with testing
@codecov
Copy link

codecov bot commented Mar 3, 2021

Codecov Report

Merging #8743 (5d6e4ab) into master (deaee53) will increase coverage by 0.04%.
The diff coverage is 61.76%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master    #8743      +/-   ##
==========================================
+ Coverage   59.23%   59.27%   +0.04%     
==========================================
  Files         571      571              
  Lines       31800    31827      +27     
==========================================
+ Hits        18837    18867      +30     
+ Misses      10761    10757       -4     
- Partials     2202     2203       +1     
Impacted Files Coverage Δ
types/module/module.go 76.13% <0.00%> (-6.58%) ⬇️
x/upgrade/types/keys.go 0.00% <ø> (ø)
x/upgrade/keeper/keeper.go 76.72% <90.00%> (+2.49%) ⬆️
simapp/app.go 83.33% <100.00%> (+0.08%) ⬆️
testutil/network/util.go 71.02% <100.00%> (ø)
x/feegrant/module.go 47.72% <0.00%> (+2.27%) ⬆️
x/slashing/module.go 59.09% <0.00%> (+2.27%) ⬆️
x/distribution/module.go 59.09% <0.00%> (+2.27%) ⬆️
x/evidence/module.go 46.51% <0.00%> (+2.32%) ⬆️
x/bank/module.go 58.13% <0.00%> (+2.32%) ⬆️
... and 7 more

Copy link
Contributor

@amaury1093 amaury1093 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice. ADR-041 has been updated with a bunch of comments from various people. Could you make sure the variable names etc match? 🙏

Two additional things which would be nice to have in this PR:

  • a changelog (UpgradeHandler is API-breaking)
  • update x/upgrade/spec docs. Most importantly the state.md one. A link to ADR-041 from one of the other mds would be good too.

x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper_test.go Outdated Show resolved Hide resolved
x/upgrade/module.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Show resolved Hide resolved
x/upgrade/module_test.go Outdated Show resolved Hide resolved
@technicallyty technicallyty requested a review from amaury1093 March 8, 2021 17:30
Copy link
Contributor

@amaury1093 amaury1093 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

getting close! Left some nits

x/upgrade/spec/02_state.md Outdated Show resolved Hide resolved
simapp/app.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
simapp/app.go Outdated Show resolved Hide resolved
simapp/app.go Outdated Show resolved Hide resolved
@technicallyty technicallyty requested a review from amaury1093 March 8, 2021 19:43
Copy link
Contributor

@amaury1093 amaury1093 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm, thanks @technicallyty 🎉

pinging @aaronc and @robert-zaremba for a review as you both had ideas on ADR-041.

docs/architecture/adr-041-in-place-store-migrations.md Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/module.go Outdated Show resolved Hide resolved
x/upgrade/module_test.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
@technicallyty
Copy link
Contributor Author

Changes should be aligned with @aaronc 's suggestions now. Additionally updated identifiers to be more go-like and changed the store get/set methods for VersionMap to align with the return type per @robert-zaremba 's suggestion.

x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
x/upgrade/keeper/keeper.go Outdated Show resolved Hide resolved
Comment on lines +54 to +63
if len(vm) > 0 {
store := ctx.KVStore(k.storeKey)
versionStore := prefix.NewStore(store, []byte{types.VersionMapByte})
for modName, ver := range vm {
nameBytes := []byte(modName)
verBytes := make([]byte, 8)
binary.BigEndian.PutUint64(verBytes, ver)
versionStore.Set(nameBytes, verBytes)
}
}
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why do these not happen in InitGenesis? I thought we agreed upon that.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this seemed like correct spot based on the suggestions, did you have somewhere else in mind?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe you're right. I guess I was thinking it could be in InitGenesis but maybe it doesn't matter

@aaronc
Copy link
Member

aaronc commented Mar 18, 2021

Mostly looks good now just one question and a couple suggested renamings.

Copy link
Member

@aaronc aaronc left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ACK. But I would like to rename those functions -> Get/SetModuleVersions

x/upgrade/keeper/keeper_test.go Outdated Show resolved Hide resolved
@amaury1093 amaury1093 added the A:automerge Automatically merge PR once all prerequisites pass. label Mar 19, 2021
@aaronc
Copy link
Member

aaronc commented Mar 19, 2021

Some tests are failing here @technicallyty

@mergify mergify bot merged commit 5bd93bf into cosmos:master Mar 19, 2021
@orijbot
Copy link

orijbot commented Mar 19, 2021

larry0x pushed a commit to larry0x/cosmos-sdk that referenced this pull request May 22, 2023
…#8743)

* -added consensus version tracking to x/upgrade

* -added interface to module manager -added e2e test for migrations using consensus version store in x/upgrade -cleaned up x/upgrade Keeper -handler in apply upgrade now handles errors and setting consensus versions -cleaned up migration map keys -removed init chainer method -simapp now implements GetConsensusVersions to assist with testing

* Changed MigrationMap identifier to VersionMap
removed module_test

* updated docs

* forgot this

* added line to changelog for this PR

* Change set consensus version function to match adr 041 spec

* add documentation

* remove newline from changelog

unnecessary newline removed

* updated example in simapp for RunMigrations, SetCurrentConsensusVersions now returns an error

* switch TestMigrations to use Require instead of t.Fatal

* Update CHANGELOG.md

Co-authored-by: Aaron Craelius <[email protected]>

* docs for SetVersionManager

* -init genesis method added -removed panics/fails from setting consensus versions

* update identifiers to be more go-like

* update docs and UpgradeHandler fnc sig

* Upgrade Keeper now takes a VersionMap instead of a VersionManager interface

* upgrade keeper transition to Version Map

* cleanup, added versionmap return to RunMigrations

* quick fix

* Update docs/architecture/adr-041-in-place-store-migrations.md

Co-authored-by: Amaury <[email protected]>

* remove support for versionmap field on upgrade keeper

* cleanup

* rename get/set version map keeper functions

* update adr doc to match name changes

* remove redudant line

Co-authored-by: technicallyty <[email protected]>
Co-authored-by: Aaron Craelius <[email protected]>
Co-authored-by: Amaury <[email protected]>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
A:automerge Automatically merge PR once all prerequisites pass. C:x/upgrade
Projects
None yet
Development

Successfully merging this pull request may close these issues.

x/upgrade should track module versions
6 participants