Skip to content

Commit 390ea25

Browse files
committed
fix
1 parent 75c5462 commit 390ea25

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

client/src/main/java/org/apache/uniffle/client/impl/DecompressionWorker.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -159,9 +159,7 @@ public DecompressedShuffleBlock get(int batchIndex, int segmentIndex) {
159159
for (int i = 0; i < batchIndex; i++) {
160160
ConcurrentHashMap<Integer, DecompressedShuffleBlock> prevBlocks = tasks.remove(i);
161161
if (prevBlocks != null) {
162-
for (DecompressedShuffleBlock block : prevBlocks.values()) {
163-
segmentPermits.ifPresent(x -> x.release());
164-
}
162+
segmentPermits.ifPresent(x -> x.release(prevBlocks.values().size()));
165163
}
166164
}
167165

0 commit comments

Comments
 (0)