From 610e0964568d97833a1076b9586b2396c4f93a6d Mon Sep 17 00:00:00 2001
From: codenohup <huangxu.walker@gmail.com>
Date: Wed, 18 Sep 2024 17:46:10 +0800
Subject: [PATCH] Fix compile error

---
 .../service/deploy/worker/storage/MapPartitionDataWriter.java | 4 ++--
 .../service/deploy/worker/storage/PartitionDataWriter.java    | 2 +-
 .../worker/storage/segment/SegmentMapPartitionFileWriter.java | 4 ++--
 .../celeborn/service/deploy/worker/PushDataHandler.scala      | 2 +-
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/MapPartitionDataWriter.java b/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/MapPartitionDataWriter.java
index 5c787df60d..1b6d9c05b7 100644
--- a/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/MapPartitionDataWriter.java
+++ b/worker/src/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/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java b/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java
index 6e9a518e3f..ccb8ef200b 100644
--- a/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java
+++ b/worker/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/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/segment/SegmentMapPartitionFileWriter.java b/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/segment/SegmentMapPartitionFileWriter.java
index 154c2cd6dd..b91679598a 100644
--- a/worker/src/main/java/org/apache/celeborn/service/deploy/worker/storage/segment/SegmentMapPartitionFileWriter.java
+++ b/worker/src/main/java/org/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
index 5b5409664b..0c614f5f34 100644
--- 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
         }
       }