This is an automated email from the ASF dual-hosted git repository.

ethanfeng pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-celeborn.git


The following commit(s) were added to refs/heads/main by this push:
     new edf85de8 [CELEBORN-123] PushDataHandler handleRpcRequestCore fix bug 
about val isMaster (#1063)
edf85de8 is described below

commit edf85de8f609aa587dc6d30460374233b13aeaa6
Author: zhongqiangczq <[email protected]>
AuthorDate: Mon Dec 12 15:51:58 2022 +0800

    [CELEBORN-123] PushDataHandler handleRpcRequestCore fix bug about val 
isMaster (#1063)
---
 .../org/apache/celeborn/service/deploy/worker/PushDataHandler.scala     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 b6b729dd..020e2eec 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
@@ -660,7 +660,7 @@ class PushDataHandler extends BaseMessageHandler with 
Logging {
       requestId: Long,
       isCheckSplit: Boolean,
       callback: RpcResponseCallback): Unit = {
-    val isMaster = mode == PartitionLocation.Mode.MASTER
+    val isMaster = PartitionLocation.getMode(mode) == 
PartitionLocation.Mode.MASTER
     val messageType = message.`type`()
     log.info(s"requestId:$requestId, pushdata rpc:$messageType, mode:$mode, 
shuffleKey:$shuffleKey, partitionUniqueId:$partitionUniqueId")
     val (workerSourceMaster, workerSourceSlave) =

Reply via email to