fix after review

multiversx · Dec 11, 2024 · fee796f · fee796f
1 parent 0ec0695
commit fee796f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/storage/pruning/pruningStorer.go b/storage/pruning/pruningStorer.go
@@ -779,7 +779,7 @@ func (ps *PruningStorer) changeEpoch(header data.HeaderHandler) error {
 		}
 		log.Debug("change epoch pruning storer success", "persister", ps.identifier, "epoch", epoch)
 
-		return ps.removeOldPersistersIfNeeded(header, epoch)
+		return ps.removeOldPersistersIfNeeded(header)
 	}
 
 	shardID := core.GetShardIDString(ps.shardCoordinator.SelfId())
@@ -802,10 +802,11 @@ func (ps *PruningStorer) changeEpoch(header data.HeaderHandler) error {
 	ps.activePersisters = append(singleItemPersisters, ps.activePersisters...)
 	ps.persistersMapByEpoch[epoch] = newPersister
 
-	return ps.removeOldPersistersIfNeeded(header, epoch)
+	return ps.removeOldPersistersIfNeeded(header)
 }
 
-func (ps *PruningStorer) removeOldPersistersIfNeeded(header data.HeaderHandler, epoch uint32) error {
+func (ps *PruningStorer) removeOldPersistersIfNeeded(header data.HeaderHandler) error {
+	epoch := header.GetEpoch()
 	wasExtended := ps.extendSavedEpochsIfNeeded(header)
 	if wasExtended {
 		if len(ps.activePersisters) > int(ps.numOfActivePersisters) {