Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow insecure gRPC connections for federated learning (#8181)

Browse Source 
* Allow insecure gRPC connections for federated learning

* format
This commit is contained in:
Rong Ou
2022-08-18 21:16:14 -07:00
committed by GitHub
parent 53d2a733b0
commit ad3bc0edee
7 changed files with 75 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
plugin/federated/federated_server.cc
View File

@@ -231,5 +231,20 @@ void RunServer(int port, int world_size, char const* server_key_file, char const
server->Wait();
}
void RunInsecureServer(int port, int world_size) {
std::string const server_address = "0.0.0.0:" + std::to_string(port);
FederatedService service{world_size};
grpc::ServerBuilder builder;
builder.SetMaxReceiveMessageSize(std::numeric_limits<int>::max());
builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
builder.RegisterService(&service);
std::unique_ptr<grpc::Server> server(builder.BuildAndStart());
LOG(CONSOLE) << "Insecure federated server listening on " << server_address << ", world size "
<< world_size;
server->Wait();
}
} // namespace federated
} // namespace xgboost