From 97bad7f4fbc209b026739348deb03812eef16851 Mon Sep 17 00:00:00 2001 From: Davide Casale Date: Thu, 14 Dec 2023 18:35:00 +0100 Subject: [PATCH] Fix low latency bug in public method BfxWebSocketBucket::start. --- bfxapi/websocket/_client/bfx_websocket_bucket.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/bfxapi/websocket/_client/bfx_websocket_bucket.py b/bfxapi/websocket/_client/bfx_websocket_bucket.py index e30ec18..fe98f3e 100644 --- a/bfxapi/websocket/_client/bfx_websocket_bucket.py +++ b/bfxapi/websocket/_client/bfx_websocket_bucket.py @@ -67,8 +67,9 @@ class BfxWebSocketBucket(Connection): if isinstance(message, list): if (chan_id := cast(int, message[0])) and \ - (message[1] != Connection._HEARTBEAT): - self.__handler.handle(self.__subscriptions[chan_id], message[1:]) + (subscription := self.__subscriptions.get(chan_id)) and \ + (message[1] != Connection._HEARTBEAT): + self.__handler.handle(subscription, message[1:]) def __on_subscribed(self, message: Dict[str, Any]) -> None: chan_id = cast(int, message["chan_id"])