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
parent
8812aff3b8
commit
79d37646a2
|
@ -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
|
yield None
|
||||||
try:
|
|
||||||
async with tractor.open_nursery() as nursery:
|
|
||||||
_root_nursery = nursery
|
|
||||||
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'
|
||||||
|
|
Loading…
Reference in New Issue