pikers
/
piker
4
1
Fork 0
Code Issues 10 Pull Requests 14 Packages Projects Releases Wiki Activity

Support no spawning `brokerd` with no real-time quotes

mkts_backup
Tyler Goodlet 2022-03-09 15:48:33 -05:00
parent 02ba7b6b96
commit e33d0aac15
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
piker/data/feed.py
View File

@ -318,6 +318,7 @@ async def allocate_persistent_feed(
brokername: str, brokername: str,
symbol: str, symbol: str,
loglevel: str, loglevel: str,
start_stream: bool = True,
task_status: TaskStatus[trio.CancelScope] = trio.TASK_STATUS_IGNORED, task_status: TaskStatus[trio.CancelScope] = trio.TASK_STATUS_IGNORED,
@ -404,6 +405,9 @@ async def allocate_persistent_feed(
bus.feeds[symbol.lower()] = (init_msg, first_quotes) bus.feeds[symbol.lower()] = (init_msg, first_quotes)
task_status.started((init_msg, first_quotes)) task_status.started((init_msg, first_quotes))
if not start_stream:
await trio.sleep_forever()
# backend will indicate when real-time quotes have begun. # backend will indicate when real-time quotes have begun.
await feed_is_live.wait() await feed_is_live.wait()
@ -462,12 +466,6 @@ async def open_feed_bus(
# brokerd yet, start persistent stream and shm writer task in # brokerd yet, start persistent stream and shm writer task in
# service nursery # service nursery
if entry is None: if entry is None:
if not start_stream:
raise RuntimeError(
f'No stream feed exists for {fqsn}?\n'
f'You may need a `brokerd` started first.'
)
# allocate a new actor-local stream bus which will persist for # allocate a new actor-local stream bus which will persist for
# this `brokerd`. # this `brokerd`.
async with bus.task_lock: async with bus.task_lock:
@ -477,13 +475,12 @@ async def open_feed_bus(
bus=bus, bus=bus,
brokername=brokername, brokername=brokername,
# here we pass through the selected symbol in native # here we pass through the selected symbol in native
# "format" (i.e. upper vs. lowercase depending on # "format" (i.e. upper vs. lowercase depending on
# provider). # provider).
symbol=symbol, symbol=symbol,
loglevel=loglevel, loglevel=loglevel,
start_stream=start_stream,
) )
) )
# TODO: we can remove this? # TODO: we can remove this?