Browse Source

Serving, grpc agent 0310

tags/v1.2.0
xuyongfei 5 years ago
parent
commit
6e50133edb
2 changed files with 3 additions and 4 deletions
  1. +1
    -1
      mindspore_serving/ccsrc/worker/distributed_worker/notify_distributed/notify_worker.cc
  2. +2
    -3
      mindspore_serving/ccsrc/worker/distributed_worker/worker_agent.cc

+ 1
- 1
mindspore_serving/ccsrc/worker/distributed_worker/notify_distributed/notify_worker.cc View File

@@ -141,7 +141,7 @@ Status GrpcNotifyDistributeWorker::GetAgentsConfigsFromWorker(const std::string
if (ExitSignalHandle::Instance().HasStopped()) {
return INFER_STATUS_LOG_WARNING(FAILED) << "Agent exit, stop get Agents configs from Worker";
}
return INFER_STATUS_LOG_ERROR(SYSTEM_ERROR) << "Failed to get Agents configs from Worker";
return INFER_STATUS_LOG_ERROR(SYSTEM_ERROR) << "Failed to get Agents configs from Worker, worker is not available.";
}
Status GrpcNotifyDistributeWorker::ParseAgentConfigAcquireReply(const proto::AgentConfigAcquireReply &reply,
DistributedServableConfig *config) {


+ 2
- 3
mindspore_serving/ccsrc/worker/distributed_worker/worker_agent.cc View File

@@ -84,9 +84,8 @@ Status WorkerAgent::StartAgent(const AgentStartUpConfig &config) {

Status WorkerAgent::StartGrpcServer() {
std::string server_address = config_.agent_ip + ":" + std::to_string(config_.agent_port);
grpc_server_.Start(std::make_shared<MSAgentImpl>(server_address), config_.agent_ip, config_.agent_port,
gRpcMaxMBMsgSize, "Agent");
return SUCCESS;
return grpc_server_.Start(std::make_shared<MSAgentImpl>(server_address), config_.agent_ip, config_.agent_port,
gRpcMaxMBMsgSize, "Agent");
}

Status WorkerAgent::RegisterAgent() {


Loading…
Cancel
Save