Fix memory leaking when loading data

Alluxio · Jul 11, 2023 · 943518c · 943518c
1 parent c237203
commit 943518c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/core/server/worker/src/main/java/alluxio/worker/block/MonoBlockStore.java b/core/server/worker/src/main/java/alluxio/worker/block/MonoBlockStore.java
@@ -347,12 +347,11 @@ public CompletableFuture<List<BlockStatus>> load(List<Block> blocks, UfsReadOpti
               blockWriter.close();
             } catch (IOException e) {
               throw AlluxioRuntimeException.from(e);
-            } finally {
-              NioDirectBufferPool.release(buf);
             }
           })
           .thenRun(() -> commitBlock(sessionId, blockId, false))
           .exceptionally(t -> {
+            NioDirectBufferPool.release(buf);
             handleException(t.getCause(), block, errors, sessionId);
             return null;
           });