feat(stf): port stf's simappv2 changes. (#20611)

Co-authored-by: unknown unknown <unknown@unknown>
cosmos · Jun 18, 2024 · 55821b3 · 55821b3
1 parent 4bd73b6
commit 55821b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/runtime/v2/module.go b/runtime/v2/module.go
@@ -161,11 +161,10 @@ func SetupAppBuilder(inputs AppInputs) {
 	app.moduleManager.RegisterInterfaces(inputs.InterfaceRegistrar)
 	app.moduleManager.RegisterLegacyAminoCodec(inputs.LegacyAmino)
 
-	// TODO: this is a bit of a hack, but it's the only way to get the store keys into the app
-	// registerStoreKey could instead set this on StoreOptions directly
 	if inputs.StoreOptions != nil {
 		inputs.AppBuilder.storeOptions = inputs.StoreOptions
 		inputs.AppBuilder.storeOptions.StoreKeys = inputs.AppBuilder.app.storeKeys
+		inputs.AppBuilder.storeOptions.StoreKeys = append(inputs.AppBuilder.storeOptions.StoreKeys, "stf")
 	}
 }
 

diff --git a/server/v2/stf/branch/writer_map.go b/server/v2/stf/branch/writer_map.go
@@ -77,6 +77,8 @@ func (b WriterMap) GetStateChanges() ([]store.StateChanges, error) {
 	return sc, nil
 }
 
+// recurseStateChanges will recursively collect state changes from the tree of
+// WriterMap's and write them to the `changes` map.
 func (b WriterMap) recurseStateChanges(changes map[string][]store.KVPair) error {
 	// depth first
 	if wr, ok := b.state.(WriterMap); ok {