|
|
|
@ -247,9 +247,6 @@ void AsyncGRPCServer::StartServer() {
|
|
|
|
|
std::bind(&AsyncGRPCServer::TryToRegisterNewOne, this,
|
|
|
|
|
std::placeholders::_1, std::placeholders::_2);
|
|
|
|
|
|
|
|
|
|
LOG(INFO) << "Server StartServer on "
|
|
|
|
|
<< "TryToRegisterNewOne bind finished";
|
|
|
|
|
|
|
|
|
|
for (auto& t : rpc_call_map_) {
|
|
|
|
|
auto& rpc_name = t.first;
|
|
|
|
|
auto& cq = rpc_cq_[rpc_name];
|
|
|
|
@ -258,7 +255,7 @@ void AsyncGRPCServer::StartServer() {
|
|
|
|
|
|
|
|
|
|
reqs.reserve(kRequestBufSize);
|
|
|
|
|
|
|
|
|
|
LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name << "I: " << i;
|
|
|
|
|
LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name << " I: " << i;
|
|
|
|
|
|
|
|
|
|
for (int i = 0; i < kRequestBufSize; i++) {
|
|
|
|
|
TryToRegisterNewOne(rpc_name, i);
|
|
|
|
@ -313,8 +310,11 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name,
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
VLOG(4) << "register send rpc_name:" << rpc_name
|
|
|
|
|
<< ", handler:" << rpc_call_map_[kRequestSend];
|
|
|
|
|
LOG(INFO) << "TryToRegisterNewOne on RPC NAME: " << rpc_name
|
|
|
|
|
<< " REQ ID: " << req_id;
|
|
|
|
|
|
|
|
|
|
// VLOG(4) << "register send rpc_name:" << rpc_name
|
|
|
|
|
// << ", handler:" << rpc_call_map_[kRequestSend];
|
|
|
|
|
|
|
|
|
|
auto& reqs = rpc_reqs_[rpc_name];
|
|
|
|
|
auto& handler = rpc_call_map_[rpc_name];
|
|
|
|
@ -328,6 +328,7 @@ void AsyncGRPCServer::TryToRegisterNewOne(const std::string& rpc_name,
|
|
|
|
|
} else if (rpc_name == kRequestPrefetch) {
|
|
|
|
|
b = new RequestPrefetch(&service_, cq.get(), handler, req_id);
|
|
|
|
|
} else if (rpc_name == kRequestCheckpoint) {
|
|
|
|
|
LOG(INFO) << "TryToRegisterNewOne on RPC kRequestCheckpoint";
|
|
|
|
|
b = new RequestCheckpointNotify(&service_, cq.get(), handler, req_id);
|
|
|
|
|
} else {
|
|
|
|
|
PADDLE_ENFORCE(false, "not supported rpc");
|
|
|
|
|