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

Port daemon(s) to new clearing system apis

supervise
Tyler Goodlet 2021-03-20 13:33:32 -04:00
parent 189c56c012
commit 2b9ac8d8ec
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
piker/_daemon.py
View File

@ -13,10 +13,12 @@ from .brokers import get_brokermod
log = get_logger(__name__) log = get_logger(__name__)
_root_nursery: Optional[tractor._trionics.ActorNursery] = None _root_nursery: Optional[tractor._trionics.ActorNursery] = None
root_dname = 'pikerd' _root_dname = 'pikerd'
root_modules = [
_root_modules = [
__name__, __name__,
'piker._ems' 'piker.exchange._ems',
'piker.exchange._client',
] ]
@ -33,16 +35,16 @@ async def maybe_open_pikerd(
if loglevel: if loglevel:
get_console_log(loglevel) get_console_log(loglevel)
async with tractor.find_actor(root_dname) as portal: async with tractor.find_actor(_root_dname) as portal:
if portal is not None: # pikerd exists if portal is not None: # pikerd exists
yield portal yield portal
else: # assume role else: # assume role
async with tractor.open_root_actor( async with tractor.open_root_actor(
name=root_dname, name=_root_dname,
loglevel=loglevel, loglevel=loglevel,
enable_modules=root_modules enable_modules=_root_modules
): ):
# init root nursery # init root nursery
try: try:
@ -55,7 +57,7 @@ async def maybe_open_pikerd(
await nursery.cancel() await nursery.cancel()
# brokerd enable modules # brokerd enabled modules
_data_mods = [ _data_mods = [
'piker.brokers.core', 'piker.brokers.core',
'piker.brokers.data', 'piker.brokers.data',
@ -63,11 +65,12 @@ _data_mods = [
'piker.data._buffer' 'piker.data._buffer'
] ]
async def spawn_brokerd( async def spawn_brokerd(
brokername, brokername,
loglevel: Optional[str] = None, loglevel: Optional[str] = None,
**tractor_kwargs **tractor_kwargs
): ) -> tractor._portal.Portal:
brokermod = get_brokermod(brokername) brokermod = get_brokermod(brokername)
dname = f'brokerd.{brokername}' dname = f'brokerd.{brokername}'
@ -76,10 +79,10 @@ async def spawn_brokerd(
# TODO: raise exception when _root_nursery == None? # TODO: raise exception when _root_nursery == None?
global _root_nursery global _root_nursery
await _root_nursery.start_actor( portal = await _root_nursery.start_actor(
dname, dname,
enable_modules=_data_mods + [brokermod.__name__], enable_modules=_data_mods + [brokermod.__name__],
loglevel=loglevel, loglevel=loglevel,
**tractor_kwargs **tractor_kwargs
) )
return portal