pikers
/
piker
3
0
Fork 0
Code Issues 5 Pull Requests 7 Packages Projects Releases Wiki Activity

Set global registry addr after first entry point spawns `pikerd`

agg_feedz
Tyler Goodlet 2023-01-09 17:27:27 -05:00
parent f6b7057b0d
commit 81585d9e6e
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
piker/_daemon.py
View File

@ -37,10 +37,15 @@ _root_dname = 'pikerd'
_registry_host: str = '127.0.0.1' _registry_host: str = '127.0.0.1'
_registry_port: int = 6116 _registry_port: int = 6116
_registry_addr = ( _default_reg_addr: tuple[str, int] = (
_registry_host, _registry_host,
_registry_port, _registry_port,
) )
# NOTE: this value is set as an actor-global once the first endpoint
# who is capable, spawns a `pikerd` service tree.
_registry_addr: tuple[str, int] | None = None
_tractor_kwargs: dict[str, Any] = { _tractor_kwargs: dict[str, Any] = {
# use a different registry addr then tractor's default # use a different registry addr then tractor's default
'arbiter_addr': _registry_addr 'arbiter_addr': _registry_addr
@ -152,13 +157,17 @@ async def open_pikerd(
''' '''
global _services global _services
global _registry_addr
if _registry_addr is None:
_registry_addr = registry_addr or _default_reg_addr
# XXX: this may open a root actor as well # XXX: this may open a root actor as well
async with ( async with (
tractor.open_root_actor( tractor.open_root_actor(
# passed through to ``open_root_actor`` # passed through to ``open_root_actor``
arbiter_addr=registry_addr or _registry_addr, arbiter_addr=_registry_addr,
name=_root_dname, name=_root_dname,
loglevel=loglevel, loglevel=loglevel,
debug_mode=debug_mode, debug_mode=debug_mode,
@ -197,7 +206,7 @@ async def open_piker_runtime(
# XXX: you should pretty much never want debug mode # XXX: you should pretty much never want debug mode
# for data daemons when running in production. # for data daemons when running in production.
debug_mode: bool = False, debug_mode: bool = False,
registry_addr: None | tuple[str, int] = _registry_addr, registry_addr: None | tuple[str, int] = None,
) -> tractor.Actor: ) -> tractor.Actor:
''' '''
@ -206,13 +215,17 @@ async def open_piker_runtime(
''' '''
global _services global _services
global _registry_addr
if _registry_addr is None:
_registry_addr = registry_addr or _default_reg_addr
# XXX: this may open a root actor as well # XXX: this may open a root actor as well
async with ( async with (
tractor.open_root_actor( tractor.open_root_actor(
# passed through to ``open_root_actor`` # passed through to ``open_root_actor``
arbiter_addr=registry_addr, arbiter_addr=_registry_addr,
name=name, name=name,
loglevel=loglevel, loglevel=loglevel,
debug_mode=debug_mode, debug_mode=debug_mode,