temp

apache · Dec 19, 2024 · 8303f58 · 8303f58
1 parent be09397
commit 8303f58
Showing 1 changed file with 3 additions and 3 deletions.
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
@@ -613,9 +613,9 @@ class PushDataHandler(val workerSource: WorkerSource) extends BaseMessageHandler
         peer.getReplicatePort)
       if (unavailablePeers.containsKey(peerWorker)) {
         for (fileWriterIndex <- 0 until totalFileWriters) {
-          if (fileWriters(fileWriterIndex) != null &&
-            !pushMergedDataCallback.isHardSplitPartition(fileWriterIndex)) {
-            fileWriters(fileWriterIndex).decrementPendingWrites()
+          val fileWriter = fileWriters(fileWriterIndex)
+          if (fileWriter != null && !pushMergedDataCallback.isHardSplitPartition(fileWriterIndex)) {
+            fileWriter.decrementPendingWrites()
           }
         }
         handlePushMergedDataConnectionFail(pushMergedDataCallback, location)