This is an automated email from the ASF dual-hosted git repository.

rong pushed a commit to branch load-npe
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit fd81603b3ad9d6747ae63b33e30ecf6b1ed653b8
Author: Steve Yurong Su <[email protected]>
AuthorDate: Mon May 6 21:05:51 2024 +0800

    Load: Avoid NPE when load task ends after clean task's execution
---
 .../iotdb/db/queryengine/execution/load/LoadTsFileManager.java      | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/load/LoadTsFileManager.java
 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/load/LoadTsFileManager.java
index 1ef706a8bdb..ef4ed7f16b9 100644
--- 
a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/load/LoadTsFileManager.java
+++ 
b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/load/LoadTsFileManager.java
@@ -201,8 +201,10 @@ public class LoadTsFileManager {
   }
 
   private void forceCloseWriterManager(String uuid) {
-    uuid2WriterManager.get(uuid).close();
-    uuid2WriterManager.remove(uuid);
+    final TsFileWriterManager writerManager = uuid2WriterManager.remove(uuid);
+    if (Objects.nonNull(writerManager)) {
+      writerManager.close();
+    }
 
     final Path loadDirPath = loadDir.toPath();
     if (!Files.exists(loadDirPath)) {

Reply via email to