def _run_grpc_server(self, address): self._grpc_server = grpc.server( futures.ThreadPoolExecutor( max_workers=8, thread_name_prefix="TrainerMasterServerThreadPoolExecutor")) tm_grpc.add_TrainerMasterServiceServicer_to_server( self, self._grpc_server) self._grpc_server.add_insecure_port(address) self._grpc_server.start() fl_logging.info('Trainer Master Server start on address: %s', address)
def run(self, listen_port): self._server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) tm_grpc.add_TrainerMasterServiceServicer_to_server( TrainerMasterServer(self._data_block_response, self._get_checkpoint_fn, self._restore_checkpoint_fn), self._server) self._server.add_insecure_port('[::]:%d' % listen_port) self._server.start() logging.info('Trainer Master Server start on port[%d].', listen_port) self._server.wait_for_termination()
def run(self, listen_port): # TrainerMaster need to load data_block from store or disk at first. # if TrainerMaster shutdown, need to recovery from previous checkpoint. self._load_data() self._server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) tm_grpc.add_TrainerMasterServiceServicer_to_server( TrainerMasterServer(self._data_block_response), self._server) self._server.add_insecure_port('[::]:%d' % listen_port) self._server.start() logging.info('Trainer Master Server start on port[%d].', listen_port) self._server.wait_for_termination()