def Foo(self, request, context):
print('Accept new request')
time.sleep(5)
print('Done with request')
return service_pb2.Empty()
def main():
channel = grpc.insecure_channel('localhost:50001')
stub = service_pb2_grpc.ServerStub(channel)
print('Send request')
stub.Foo(service_pb2.Empty())
print('Request successful')
async def main():
with grpc.insecure_channel('localhost:50001') as channel:
stub = service_pb2_grpc.ServerStub(channel)
print('Send request')
await _submit(stub.Foo, service_pb2.Empty())
print('Request successful')
def Aggregate(self, request, _):
LOGGER.info("Aggregate: request [%s]", request)
self.request_validate(request)
threading.Thread(target=self.send_response,
daemon=True,
args=(request.aggregatedModel.path, )).start()
return service_pb2.Empty()
def LocalTrain(self, request, _):
LOGGER.info("LocalTrain: request [%s]", request)
self.request_validate(request)
threading.Thread(target=self.send_response,
daemon=True,
args=(request.localModel.path, )).start()
return service_pb2.Empty()
def Aggregate(self, request, context):
logging.info("received Aggregate message [%s]", request)
threading.Thread(
target=aggregate,
args=(request.localModels, request.aggregatedModel),
daemon=True
).start()
response = service_pb2.Empty()
return response
def LocalTrain(self, request, context):
logging.info("LocalTrain")
threading.Thread(target=train,
args=(request.baseModel, request.localModel.path,
request.EpR),
daemon=True).start()
resp = service_pb2.Empty()
logging.info("Sending response: {}".format(resp))
return resp
def main():
host = 'localhost'
port = 1337
with open('server.crt') as f:
trusted_certs = f.read()
credentials = grpc.ssl_channel_credentials(root_certificates=trusted_certs)
channel = grpc.secure_channel('{}:{}'.format(host, port), credentials)
stub = service_pb2.ServerStub(channel)
stub.Foo(service_pb2.Empty())
def AggregateFinish(self, request, _):
print(f"Aggregate finish: [{request}]")
result = torch.load("/repos/output/weights.tar", 'cpu')
print(f"Aggregate result: [{result}]")
if operator.eq(result, EXPECTED):
os.kill(os.getpid(), signal.SIGUSR1)
else:
print(f"Expect [{EXPECTED}], Got [{result}]")
os.kill(os.getpid(), signal.SIGUSR2)
sys.stdout.flush()
return service_pb2.Empty()
def main():
host = 'localhost'
port = 1337
with open('server.crt', 'rb') as f:
trusted_certs = f.read()
credentials = grpc.ssl_channel_credentials(root_certificates=trusted_certs)
channel = grpc.secure_channel('{}:{}'.format(host, port), credentials)
stub = service_pb2_grpc.ServerStub(channel)
stub.Foo(service_pb2.Empty())
response = stub.SayHello(service_pb2.HelloRequest(name='you'))
print("Greeter client received: " + response.message)
response = stub.SayHelloAgain(service_pb2.HelloRequest(name='you'))
print("Greeter client received: " + response.message)
async def Foo(self, request, context):
print('Accept new request')
await asyncio.sleep(1)
print('Done with request')
return service_pb2.Empty()
def Foo(self, request, context):
return service_pb2.Empty(message='Hello! Current time is ' + time.ctime())
def Foo(self, request, context):
return service_pb2.Empty()
def TrainFinish(self, _request, _context):
logging.info("TrainFinish")
STOP_EVENT.set()
return service_pb2.Empty()
def TrainInterrupt(self, _request, _context):
# Not Implemented
return service_pb2.Empty()
def TrainInit(self, _0, _1):
LOGGER.info("Initialize train")
return service_pb2.Empty()
def TrainInit(self, request, context):
logging.info("TrainInit")
resp = service_pb2.Empty()
logging.info(f"Sending response: {resp}")
return resp
def Unsubscribe(self, request, unused_context):
DramaInfo.Unsubscribe(request.user_id, request.drama_id)
return service_pb2.Empty()
def TrainFinish(self, _0, _1):
LOGGER.info("Train Finish")
STOP_EVENT.set()
return service_pb2.Empty()
