use stateFactory in genesis expoert

cosmos · Sep 20, 2024 · 4d1a612 · 4d1a612
1 parent 8e0c4c9
commit 4d1a612
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/runtime/v2/builder.go b/runtime/v2/builder.go
@@ -201,7 +201,12 @@ func (a *AppBuilder[T]) Build(opts ...AppBuilderOption[T]) (*App[T], error) {
 
 			var genesisJson map[string]json.RawMessage
 			_, err = genesisCtx.Run(ctx, func(ctx context.Context) error {
-				genesisJson, err = a.app.moduleManager.ExportGenesisForModules(ctx, a.branch(state))
+				genesisJson, err = a.app.moduleManager.ExportGenesisForModules(
+					ctx,
+					func() store.WriterMap {
+						return a.branch(state)
+					},
+				)
 				return err
 			})
 			if err != nil {

diff --git a/runtime/v2/manager.go b/runtime/v2/manager.go
@@ -195,7 +195,7 @@ func (m *MM[T]) InitGenesisJSON(
 // ExportGenesisForModules performs export genesis functionality for modules
 func (m *MM[T]) ExportGenesisForModules(
 	ctx context.Context,
-	state store.WriterMap,
+	stateFactory func() store.WriterMap,
 	modulesToExport ...string,
 ) (map[string]json.RawMessage, error) {
 	if len(modulesToExport) == 0 {
@@ -229,7 +229,7 @@ func (m *MM[T]) ExportGenesisForModules(
 
 		channels[moduleName] = make(chan genesisResult)
 		go func(moduleI ModuleI, ch chan genesisResult) {
-			genesisCtx := services.NewGenesisContext(state)
+			genesisCtx := services.NewGenesisContext(stateFactory())
 			_, _ = genesisCtx.Run(ctx, func(ctx context.Context) error {
 				jm, err := moduleI.ExportGenesis(ctx)
 				if err != nil {