Preload process stream in parallel

Signed-off-by: Gabriel Fukushima <[email protected]>
gfukushima · Dec 15, 2023 · 98991c2 · 98991c2
1 parent e70e21e
commit 98991c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ethereum/core/src/main/java/org/hyperledger/besu/ethereum/bonsai/trielog/TrieLogPruner.java b/ethereum/core/src/main/java/org/hyperledger/besu/ethereum/bonsai/trielog/TrieLogPruner.java
@@ -73,12 +73,12 @@ private int preloadQueue() {
     try (final Stream<byte[]> trieLogKeys = rootWorldStateStorage.streamTrieLogKeys(loadingLimit)) {
       final AtomicLong count = new AtomicLong();
       final AtomicLong orphansPruned = new AtomicLong();
-      trieLogKeys.forEach(
+      trieLogKeys.parallel().forEach(
           blockHashAsBytes -> {
             final Hash blockHash = Hash.wrap(Bytes32.wrap(blockHashAsBytes));
             final Optional<BlockHeader> header = blockchain.getBlockHeader(blockHash);
             if (header.isPresent()) {
-              trieLogBlocksAndForksByDescendingBlockNumber.put(header.get().getNumber(), blockHash);
+              addToPruneQueue(header.get().getNumber(), blockHash);
               count.getAndIncrement();
             } else {
               // prune orphaned blocks (sometimes created during block production)