Fix compile error

apache · Sep 18, 2024 · 610e096 · 610e096
1 parent e83bd92
commit 610e096
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/...c/main/java/org/apache/celeborn/service/deploy/worker/storage/MapPartitionDataWriter.java b/...c/main/java/org/apache/celeborn/service/deploy/worker/storage/MapPartitionDataWriter.java
@@ -120,8 +120,8 @@ public void write(ByteBuf data) throws IOException {
   }
 
   @Override
-  public void setHasWriteFinished() {
-    getFileMeta().setHasWriteFinished(true);
+  public void setWriterClosed() {
+    getFileMeta().setIsWriterClosed(true);
   }
 
   @Override

diff --git a/.../src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java b/.../src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java
@@ -419,7 +419,7 @@ public boolean isClosed() {
     return closed;
   }
 
-  public void setHasWriteFinished() {}
+  public void setWriterClosed() {}
 
   protected synchronized long close(
       RunnableWithIOException tryClose,

diff --git a/.../apache/celeborn/service/deploy/worker/storage/segment/SegmentMapPartitionFileWriter.java b/.../apache/celeborn/service/deploy/worker/storage/segment/SegmentMapPartitionFileWriter.java
@@ -77,7 +77,7 @@ public void pushDataHandShake(int numSubpartitions, int bufferSize) {
     super.pushDataHandShake(numSubpartitions, bufferSize);
     subPartitionBufferIndex = new int[numSubpartitions];
     Arrays.fill(subPartitionBufferIndex, 0);
-    getFileMeta().setHasWriteFinished(false);
+    getFileMeta().setIsWriterClosed(false);
     getFileMeta().setSegmentGranularityVisible(true);
   }
 
@@ -149,7 +149,7 @@ public void write(ByteBuf data) throws IOException {
   public synchronized long close() throws IOException {
     long fileLength = super.close();
     logger.debug("Close {} for file {}", this, getFile());
-    getFileMeta().setHasWriteFinished(true);
+    getFileMeta().setIsWriterClosed(true);
     return fileLength;
   }
 

diff --git a/worker/src/main/scala/org/apache/celeborn/service/deploy/worker/PushDataHandler.scala b/worker/src/main/scala/org/apache/celeborn/service/deploy/worker/PushDataHandler.scala
@@ -1265,7 +1265,7 @@ class PushDataHandler(val workerSource: WorkerSource) extends BaseMessageHandler
                |fileLength:${diskFileInfo.getFileLength},
                |fileName:${diskFileInfo.getFilePath}
                |""".stripMargin)
-          fileWriter.setHasWriteFinished()
+          fileWriter.setWriterClosed()
           return true
         }
       }