Factor msg loop into new func: `handle_order_updates()`
parent
9723d2737a
commit
4d19c0f910
|
@ -329,10 +329,41 @@ async def trades_dialogue(
|
|||
ids,
|
||||
)
|
||||
|
||||
# process and relay trades events to ems
|
||||
# https://docs.kraken.com/websockets/#message-ownTrades
|
||||
# enter relay loop
|
||||
await handle_order_updates(
|
||||
ws,
|
||||
ems_stream,
|
||||
emsflow,
|
||||
ids,
|
||||
trans,
|
||||
acctid,
|
||||
acc_name,
|
||||
token,
|
||||
)
|
||||
|
||||
|
||||
async def handle_order_updates(
|
||||
ws: NoBsWs,
|
||||
ems_stream: tractor.MsgStream,
|
||||
emsflow: dict[str, list[MsgUnion]],
|
||||
ids: bidict[str, int],
|
||||
trans: list[pp.Transaction],
|
||||
acctid: str,
|
||||
acc_name: str,
|
||||
token: str,
|
||||
|
||||
) -> None:
|
||||
'''
|
||||
Main msg handling loop for all things order management.
|
||||
|
||||
This code is broken out to make the context explicit and state variables
|
||||
defined in the signature clear to the reader.
|
||||
|
||||
'''
|
||||
async for msg in stream_messages(ws):
|
||||
match msg:
|
||||
# process and relay clearing trade events to ems
|
||||
# https://docs.kraken.com/websockets/#message-ownTrades
|
||||
case [
|
||||
trades_msgs,
|
||||
'ownTrades',
|
||||
|
@ -405,7 +436,7 @@ async def trades_dialogue(
|
|||
t.bsuid for t in trans),
|
||||
)
|
||||
|
||||
# emit pp msgs
|
||||
# emit any new pp msgs to ems
|
||||
for pos in filter(
|
||||
bool,
|
||||
chain(active.values(), closed.values()),
|
||||
|
@ -432,19 +463,16 @@ async def trades_dialogue(
|
|||
)
|
||||
await ems_stream.send(pp_msg.dict())
|
||||
|
||||
# process and relay order state change events
|
||||
# https://docs.kraken.com/websockets/#message-openOrders
|
||||
case [
|
||||
order_msgs,
|
||||
'openOrders',
|
||||
{'sequence': seq},
|
||||
]:
|
||||
# TODO: async order update handling which we
|
||||
# should remove from `handle_order_requests()`
|
||||
# above:
|
||||
# https://github.com/pikers/piker/issues/293
|
||||
# https://github.com/pikers/piker/issues/310
|
||||
for order_msg in order_msgs:
|
||||
log.info(
|
||||
'Order msg update_{seq}:\n'
|
||||
f'Order msg update_{seq}:\n'
|
||||
f'{pformat(order_msg)}'
|
||||
)
|
||||
txid, update_msg = list(order_msg.items())[0]
|
||||
|
|
Loading…
Reference in New Issue