[incubator-celeborn] branch main updated: [CELEBORN-317] add REGISTER_MAP_PARTITION_TASK message type (#1250)

[ethanfeng]

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 a6103e4b [CELEBORN-317] add REGISTER_MAP_PARTITION_TASK message type 
(#1250)
a6103e4b is described below

commit a6103e4bf890abf3fce4957a085510816126c63b
Author: Shuang <lvshuang...@gmail.com>
AuthorDate: Mon Feb 20 22:01:35 2023 +0800

    [CELEBORN-317] add REGISTER_MAP_PARTITION_TASK message type (#1250)
---
 common/src/main/proto/TransportMessages.proto                       | 1 +
 .../apache/celeborn/common/protocol/message/ControlMessages.scala   | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/common/src/main/proto/TransportMessages.proto 
b/common/src/main/proto/TransportMessages.proto
index 3132dc7e..a9785590 100644
--- a/common/src/main/proto/TransportMessages.proto
+++ b/common/src/main/proto/TransportMessages.proto
@@ -66,6 +66,7 @@ enum MessageType {
   STAGE_END = 45;
   STAGE_END_RESPONSE = 46;
   PARTITION_SPLIT = 47;
+  REGISTER_MAP_PARTITION_TASK = 48;
 }
 
 message PbStorageInfo {
diff --git 
a/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala
 
b/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala
index 5b65c0d5..664c4253 100644
--- 
a/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala
+++ 
b/common/src/main/scala/org/apache/celeborn/common/protocol/message/ControlMessages.scala
@@ -468,6 +468,9 @@ object ControlMessages extends Logging {
     case pb: PbRegisterShuffle =>
       new TransportMessage(MessageType.REGISTER_SHUFFLE, pb.toByteArray)
 
+    case pb: PbRegisterMapPartitionTask =>
+      new TransportMessage(MessageType.REGISTER_MAP_PARTITION_TASK, 
pb.toByteArray)
+
     case pb: PbRegisterShuffleResponse =>
       new TransportMessage(MessageType.REGISTER_SHUFFLE_RESPONSE, 
pb.toByteArray)
 
@@ -831,6 +834,9 @@ object ControlMessages extends Logging {
       case REGISTER_SHUFFLE =>
         PbRegisterShuffle.parseFrom(message.getPayload)
 
+      case REGISTER_MAP_PARTITION_TASK =>
+        PbRegisterMapPartitionTask.parseFrom(message.getPayload)
+
       case REGISTER_SHUFFLE_RESPONSE =>
         PbRegisterShuffleResponse.parseFrom(message.getPayload)