add lock for ModificationFile.remove

apache · Jan 23, 2025 · 74974dd · 74974dd
1 parent c05079b
commit 74974dd
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/...main/java/org/apache/iotdb/db/storageengine/dataregion/modification/ModificationFile.java b/...main/java/org/apache/iotdb/db/storageengine/dataregion/modification/ModificationFile.java
@@ -336,13 +336,18 @@ public boolean exists() {
   }
 
   public void remove() throws IOException {
-    close();
-    FileUtils.deleteFileOrDirectory(file);
-    if (fileExists) {
-      updateModFileMetric(-1, -getFileLength());
+    lock.writeLock().lock();
+    try {
+      close();
+      FileUtils.deleteFileOrDirectory(file);
+      if (fileExists) {
+        updateModFileMetric(-1, -getFileLength());
+      }
+      fileExists = false;
+      removed = true;
+    } finally {
+    	lock.writeLock().unlock();
     }
-    fileExists = false;
-    removed = true;
   }
 
   public static ModificationFile getExclusiveMods(TsFileResource tsFileResource) {