fix create prunning storer maps epoch

multiversx · Nov 6, 2023 · d2f3f82 · d2f3f82
1 parent c192ba4
commit d2f3f82
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/storage/factory/storageServiceFactory.go b/storage/factory/storageServiceFactory.go
@@ -633,8 +633,6 @@ func (psf *StorageServiceFactory) createTriePersister(
 }
 
 func (psf *StorageServiceFactory) createTriePruningPersister(arg pruning.StorerArgs) (storage.Storer, error) {
-	arg.StateStatsHandler = psf.stateStatsHandler
-
 	isFullArchive := psf.prefsConfig.FullArchive
 	isDBLookupExtension := psf.generalConfig.DbLookupExtensions.Enabled
 	if !isFullArchive && !isDBLookupExtension {

diff --git a/storage/pruning/pruningStorer.go b/storage/pruning/pruningStorer.go
@@ -263,11 +263,13 @@ func createPersisterIfPruningDisabled(
 	var persisters []*persisterData
 	persistersMapByEpoch := make(map[uint32]*persisterData)
 
-	p, err := createPersisterDataForEpoch(args, 0, shardIDStr)
+	epoch := uint32(0)
+	p, err := createPersisterDataForEpoch(args, epoch, shardIDStr)
 	if err != nil {
 		return nil, nil, err
 	}
 	persisters = append(persisters, p)
+	persistersMapByEpoch[epoch] = p
 
 	return persisters, persistersMapByEpoch, nil
 }