Use new |-union style type annots in sampling routines

agg_feedz
Tyler Goodlet 2022-11-09 13:05:51 -05:00
parent 32b36aa042
commit 25bfe6f035
1 changed files with 8 additions and 3 deletions

View File

@ -22,7 +22,10 @@ financial data flows.
from __future__ import annotations from __future__ import annotations
from collections import Counter from collections import Counter
import time import time
from typing import TYPE_CHECKING, Optional, Union from typing import (
TYPE_CHECKING,
Union,
)
import tractor import tractor
import trio import trio
@ -147,7 +150,7 @@ async def increment_ohlc_buffer(
async def broadcast( async def broadcast(
delay_s: int, delay_s: int,
shm: Optional[ShmArray] = None, shm: ShmArray | None = None,
) -> None: ) -> None:
''' '''
@ -241,6 +244,8 @@ async def sample_and_broadcast(
# iterate stream delivered by broker # iterate stream delivered by broker
async for quotes in quote_stream: async for quotes in quote_stream:
# print(quotes)
# TODO: ``numba`` this! # TODO: ``numba`` this!
for broker_symbol, quote in quotes.items(): for broker_symbol, quote in quotes.items():
# TODO: in theory you can send the IPC msg *before* writing # TODO: in theory you can send the IPC msg *before* writing
@ -314,7 +319,7 @@ async def sample_and_broadcast(
tuple[ tuple[
Union[tractor.MsgStream, trio.MemorySendChannel], Union[tractor.MsgStream, trio.MemorySendChannel],
tractor.Context, tractor.Context,
Optional[float], # tick throttle in Hz float | None, # tick throttle in Hz
] ]
] = bus._subscribers[broker_symbol.lower()] ] = bus._subscribers[broker_symbol.lower()]