Release/v0.50.x cachekv par

store/cachemulti/store.go

@@ -1,8 +1,10 @@
 package cachemulti
 
 import (
+	"errors"
 	"fmt"
 	"io"
+	"sync"
 
 	"cosmossdk.io/store/tracekv"
 	"cosmossdk.io/store/types"
@@ -11,7 +13,11 @@
 
 // storeNameCtxKey is the TraceContext metadata key that identifies
 // the store which emitted a given trace.
-const storeNameCtxKey = "store_name"
+const (
+	// storeNameCtxKey is the TraceContext metadata key that identifies
+	// the store which emitted a given trace.
+	storeNameCtxKey = "store_name"
+)
 
 //----------------------------------------
 // Store
@@ -129,9 +135,41 @@
 	if cms.branched {
 		panic("cannot Write on branched store")
 	}
-	for _, store := range cms.stores {
-		store.Write()
+	if err := cms.writeStoresParallel(); err != nil {
+		panic(err)
+	}
+}
+
+func (cms Store) writeStoresParallel() error {
+	runnerCount := len(cms.stores)
+	errChan := make(chan error, runnerCount) // Channel to collect errors from goroutines
+	var wg sync.WaitGroup
+	for storeKey, store := range cms.stores {
+		wg.Add(1)
+
+		go func(storeKey types.StoreKey, store types.CacheWrap) {
+			defer func() {
+				wg.Done()
+				if r := recover(); r != nil {
+					errChan <- fmt.Errorf("panic in Write for store %s: %v", storeKey.Name(), r)
+				}
+			}()
+			store.Write()
+		}(storeKey, store)
 	}
+
+	go func() {
+		wg.Wait()
+		close(errChan)
+	}()
+
+	// Collect errors from goroutines
+	var allErrors []error
+	for err := range errChan {
+		allErrors = append(allErrors, err)
+	}
+
+	return errors.Join(allErrors...)
 }
 
 func (cms Store) Discard() {