update

apache · Apr 2, 2024 · bc4fce8 · bc4fce8
1 parent 7513106
commit bc4fce8
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/worker/src/main/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManager.java b/worker/src/main/java/org/apache/celeborn/service/deploy/worker/memory/MemoryManager.java
@@ -267,9 +267,15 @@ private MemoryManager(CelebornConf conf, StorageManager storageManager) {
                       || currentServingState() != ServingState.NONE_PAUSED) {
                     try {
                       if (!committedWriters.isEmpty()) {
-                        committedWriters.remove(0).evict();
+                        PartitionDataWriter writer = committedWriters.remove(0);
+                        if (!writer.getMemoryFileInfo().hasReader()) {
+                          writer.evict();
+                        }
                       } else if (!unCommittedWriters.isEmpty()) {
-                        unCommittedWriters.remove(0).evict();
+                        PartitionDataWriter writer = unCommittedWriters.remove(0);
+                        if (!writer.getMemoryFileInfo().hasReader()) {
+                          writer.evict();
+                        }
                       } else {
                         break;
                       }