def __init__(self, args: Union['argparse.Namespace', Dict]):
super().__init__()
self.args = args
self.name = self.__class__.__name__ #: this is the process name
self.is_ready_event = _get_event(self)
self.is_shutdown = _get_event(self)
self.ready_or_shutdown = _make_or_event(self, self.is_ready_event,
self.is_shutdown)
self.is_shutdown.clear()
if 'daemon' in args:
self.daemon = args.daemon
if 'name' in self.args and self.args.name:
self.name = f'runtime-{self.args.name}'
if 'role' in self.args and self.args.role == PeaRoleType.PARALLEL:
self.name = f'runtime-{self.args.name}-{self.args.pea_id}'
if 'role' in self.args and self.args.role == PeaRoleType.HEAD:
self.name = f'runtime-{self.args.name}-head'
if 'role' in self.args and self.args.role == PeaRoleType.TAIL:
self.name = f'runtime-{self.args.name}-tail'
if 'host' in self.args and 'port_ctrl' in self.args and 'ctrl_with_ipc' in self.args:
self.ctrl_addr, self.ctrl_with_ipc = Zmqlet.get_ctrl_address(
self.args.host, self.args.port_ctrl, self.args.ctrl_with_ipc)
if 'log_id' in self.args and 'log_config' in self.args:
self.logger = JinaLogger(self.name,
log_id=self.args.log_id,
log_config=self.args.log_config)
else:
self.logger = JinaLogger(self.name)
def test_get_ctrl_addr(host):
p = set_pea_parser().parse_args(['--host', '[email protected]', '--port-ctrl', '56789'])
assert Zmqlet.get_ctrl_address(p)[0] == 'tcp://192.0.0.1:56789'
def test_get_ctrl_addr_ipc(host):
assert Zmqlet.get_ctrl_address(host, 56789, True)[0].startswith('ipc')
def test_get_ctrl_addr(host):
assert Zmqlet.get_ctrl_address(host, 56789,
False)[0] == 'tcp://192.0.0.1:56789'
def test_get_ctrl_addr_ipc(host):
assert 'ipc' == Zmqlet.get_ctrl_address(host, 56789, True)[0][0:3]