Release BlockReader in CRC check

Alluxio · Sep 12, 2024 · 38819af · 38819af
1 parent ae64e71
commit 38819af
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/core/common/src/main/java/alluxio/Sessions.java b/core/common/src/main/java/alluxio/Sessions.java
@@ -36,9 +36,10 @@ public class Sessions {
   public static final int CACHE_UFS_SESSION_ID = -8;
 
   public static final int LOAD_SESSION_ID = -9;
+  public static final int WORKER_CHECKSUM_CHECK_SESSION_ID = -10;
 
   // internal session id base should be smaller than all predefined session ids
-  public static final long INTERNAL_SESSION_ID_BASE = -8;
+  public static final long INTERNAL_SESSION_ID_BASE = -11;
 
   /** Map from SessionId to {@link alluxio.SessionInfo} object. */
   private final Map<Long, SessionInfo> mSessions;

diff --git a/core/server/worker/src/main/java/alluxio/worker/block/DefaultBlockWorker.java b/core/server/worker/src/main/java/alluxio/worker/block/DefaultBlockWorker.java
@@ -637,10 +637,10 @@ public Map<Long, BlockChecksum> calculateBlockChecksum(List<Long> blockIds) {
     for (long blockId : blockIds) {
       Future<?> future = mChecksumCalculationThreadPool.submit(() -> {
         ByteBuffer bf = null;
-        try {
+        try (BlockReader br = mBlockStore.createBlockReader(
+            Sessions.WORKER_CHECKSUM_CHECK_SESSION_ID,
+            blockId, 0, false, Protocol.OpenUfsBlockOptions.getDefaultInstance())) {
           CRC64 crc64 = new CRC64();
-          BlockReader br = mBlockStore.createBlockReader(
-              -1, blockId, 0, false, Protocol.OpenUfsBlockOptions.getDefaultInstance());
           if (mChecksumCalculationUsingBufferPool) {
             bf = NioHeapBufferPool.acquire(chunkSize);
           } else {