Browse Source

fixed message header

tags/v1.2.0-rc1
chendongsheng 4 years ago
parent
commit
a43d0f8767
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      mindspore/ccsrc/ps/core/tcp_client.cc
  2. +1
    -1
      mindspore/ccsrc/ps/core/tcp_server.cc

+ 2
- 2
mindspore/ccsrc/ps/core/tcp_client.cc View File

@@ -268,7 +268,7 @@ bool TcpClient::SendMessage(const CommMessage &message) const {
bool res = true; bool res = true;
size_t buf_size = IntToUint(message.ByteSizeLong()); size_t buf_size = IntToUint(message.ByteSizeLong());
uint32_t meta_size = SizeToUint(message.pb_meta().ByteSizeLong()); uint32_t meta_size = SizeToUint(message.pb_meta().ByteSizeLong());
Messageheader header;
MessageHeader header;
header.message_proto_ = Protos::PROTOBUF; header.message_proto_ = Protos::PROTOBUF;
header.message_length_ = buf_size; header.message_length_ = buf_size;
header.message_meta_length_ = meta_size; header.message_meta_length_ = meta_size;
@@ -295,7 +295,7 @@ bool TcpClient::SendMessage(std::shared_ptr<MessageMeta> meta, const Protos &pro
bufferevent_lock(buffer_event_); bufferevent_lock(buffer_event_);
bool res = true; bool res = true;


Messageheader header;
MessageHeader header;
header.message_proto_ = protos; header.message_proto_ = protos;
header.message_meta_length_ = SizeToUint(meta->ByteSizeLong()); header.message_meta_length_ = SizeToUint(meta->ByteSizeLong());
header.message_length_ = size + header.message_meta_length_; header.message_length_ = size + header.message_meta_length_;


+ 1
- 1
mindspore/ccsrc/ps/core/tcp_server.cc View File

@@ -73,7 +73,7 @@ bool TcpConnection::SendMessage(std::shared_ptr<MessageMeta> meta, const Protos
MS_EXCEPTION_IF_NULL(data); MS_EXCEPTION_IF_NULL(data);
bufferevent_lock(buffer_event_); bufferevent_lock(buffer_event_);
bool res = true; bool res = true;
Messageheader header;
MessageHeader header;
header.message_proto_ = protos; header.message_proto_ = protos;
header.message_meta_length_ = SizeToUint(meta->ByteSizeLong()); header.message_meta_length_ = SizeToUint(meta->ByteSizeLong());
header.message_length_ = size + header.message_meta_length_; header.message_length_ = size + header.message_meta_length_;


Loading…
Cancel
Save