Add `open_pikerd()` and `spawn_emsd()`

The direct open is needed for running `pikerd` cmd and
the ems spawn point is the first step toward detaching UI based order
entry from the engine itself.
supervise
Tyler Goodlet 2021-03-21 21:52:53 -04:00
parent 8812aff3b8
commit 79d37646a2
1 changed files with 69 additions and 22 deletions

View File

@ -1,7 +1,7 @@
""" """
pikerd daemon lifecylcle & rpc pikerd daemon lifecylcle & rpc
""" """
from typing import Optional from typing import Optional, Dict, Callable
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
import tractor import tractor
@ -14,14 +14,41 @@ 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.exchange._ems', 'piker.clearing._ems',
'piker.exchange._client', 'piker.clearing._client',
] ]
@asynccontextmanager
async def open_pikerd(
loglevel: Optional[str] = None
) -> Optional[tractor._portal.Portal]:
global _root_nursery
async with tractor.open_root_actor(
name=_root_dname,
loglevel=loglevel,
# TODO: eventually we should be able to avoid
# having the root have more then permissions to
# spawn other specialized daemons I think?
# enable_modules=[__name__],
enable_modules=_root_modules
):
# init root nursery
try:
async with tractor.open_nursery() as nursery:
_root_nursery = nursery
yield nursery
finally:
# client code may block indefinitely so cancel when
# teardown is invoked
await nursery.cancel()
@asynccontextmanager @asynccontextmanager
async def maybe_open_pikerd( async def maybe_open_pikerd(
loglevel: Optional[str] = None loglevel: Optional[str] = None
@ -41,20 +68,12 @@ async def maybe_open_pikerd(
yield portal yield portal
else: # assume role else: # assume role
async with tractor.open_root_actor( async with open_pikerd(loglevel) as nursery:
name=_root_dname, # in the case where we're starting up the
loglevel=loglevel, # tractor-piker runtime stack in **this** process
enable_modules=_root_modules # we want to hand off a nursery for starting (as a sub)
): # whatever actor is requesting pikerd.
# init root nursery
try:
async with tractor.open_nursery() as nursery:
_root_nursery = nursery
yield None yield None
finally:
# client code may block indefinitely so cancel when
# teardown is invoked
await nursery.cancel()
# brokerd enabled modules # brokerd enabled modules
@ -72,10 +91,12 @@ async def spawn_brokerd(
**tractor_kwargs **tractor_kwargs
) -> tractor._portal.Portal: ) -> tractor._portal.Portal:
log.info(f'Spawning {brokername} broker daemon')
brokermod = get_brokermod(brokername) brokermod = get_brokermod(brokername)
dname = f'brokerd.{brokername}' dname = f'brokerd.{brokername}'
log.info(f'Spawning {brokername} broker daemon')
tractor_kwargs = getattr(brokermod, '_spawnkwargs', {}) extra_tractor_kwargs = getattr(brokermod, '_spawnkwargs', {})
# TODO: raise exception when _root_nursery == None? # TODO: raise exception when _root_nursery == None?
global _root_nursery global _root_nursery
@ -83,6 +104,32 @@ async def spawn_brokerd(
dname, dname,
enable_modules=_data_mods + [brokermod.__name__], enable_modules=_data_mods + [brokermod.__name__],
loglevel=loglevel, loglevel=loglevel,
**tractor_kwargs **extra_tractor_kwargs
) )
return portal return dname
async def spawn_emsd(
brokername,
loglevel: Optional[str] = None,
**extra_tractor_kwargs
) -> tractor._portal.Portal:
from .clearing import _client
log.info('Spawning emsd')
# TODO: raise exception when _root_nursery == None?
global _root_nursery
assert _root_nursery
portal = await _root_nursery.start_actor(
'emsd',
enable_modules=[
'piker.clearing._ems',
'piker.clearing._client',
],
loglevel=loglevel,
**extra_tractor_kwargs
)
return 'emsd'