Remove parallel from withdrawal warmup (#7153)

NethermindEth · Jun 10, 2024 · 7aa575a · 7aa575a
1 parent 752cf12
commit 7aa575a
Showing 1 changed file with 21 additions and 18 deletions.
diff --git a/src/Nethermind/Nethermind.Consensus/Processing/BlockCachePreWarmer.cs b/src/Nethermind/Nethermind.Consensus/Processing/BlockCachePreWarmer.cs
@@ -72,30 +72,33 @@ void WarmupWithdrawals(ParallelOptions parallelOptions, IReleaseSpec spec, Block
         {
             if (spec.WithdrawalsEnabled && block.Withdrawals is not null)
             {
-                Parallel.For(0, block.Withdrawals.Length, parallelOptions,
-                    i =>
+                ReadOnlyTxProcessingEnv env = _envPool.Get();
+                try
+                {
+                    foreach (Withdrawal withdrawal in block.Withdrawals)
                     {
-                        ReadOnlyTxProcessingEnv env = _envPool.Get();
-                        try
-                        {
-                            env.StateProvider.WarmUp(block.Withdrawals[i].Address);
-                        }
-                        catch (Exception ex)
-                        {
-                            if (_logger.IsDebug) _logger.Error($"Error pre-warming withdrawal {i}", ex);
-                        }
-                        finally
-                        {
-                            env.Reset();
-                            _envPool.Return(env);
-                        }
-                    });
+                        if (parallelOptions.CancellationToken.IsCancellationRequested) return;
+
+                        env.StateProvider.WarmUp(withdrawal.Address);
+                    }
+                }
+                catch (Exception ex)
+                {
+                    if (_logger.IsDebug) _logger.Error($"Error pre-warming withdrawals", ex);
+                }
+                finally
+                {
+                    env.Reset();
+                    _envPool.Return(env);
+                }
             }
         }
 
         void WarmupTransactions(ParallelOptions parallelOptions, IReleaseSpec spec, Block block, Hash256 stateRoot)
         {
-            Parallel.For(0, block.Transactions.Length, parallelOptions, i =>
+            if (parallelOptions.CancellationToken.IsCancellationRequested) return;
+
+            Parallel.For(0, block.Transactions.Length, parallelOptions, (i) =>
             {
                 // If the transaction has already been processed or being processed, exit early
                 if (block.TransactionProcessed >= i) return;