aljaz/bitfinex-api-py
1
0
Fork 0
mirror of https://github.com/aljazceru/bitfinex-api-py.git synced 2025-12-19 06:44:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add automatic deletion for buckets that reach zero subscriptions (e.g. after a call to BfxWebSocketClient::unsubscribe).

Browse Source
This commit is contained in:
Davide Casale
2023-10-16 04:45:47 +02:00
parent e5ec94b757
commit 133db74a72
3 changed files with 27 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
bfxapi/websocket/_client/bfx_websocket_bucket.py
View File

@@ -44,6 +44,11 @@ class BfxWebSocketBucket(Connection):
return self.count == \
BfxWebSocketBucket.__MAXIMUM_SUBSCRIPTIONS_AMOUNT
@property
def ids(self) -> List[str]:
return [ pending["subId"] for pending in self.__pendings ] + \
[ subscription["sub_id"] for subscription in self.__subscriptions.values() ]
async def start(self) -> None:
async with websockets.client.connect(self._host) as websocket:
self._websocket = websocket
@@ -59,11 +64,6 @@ class BfxWebSocketBucket(Connection):
if isinstance(message, dict):
if message["event"] == "subscribed":
self.__on_subscribed(message)
elif message["event"] == "unsubscribed":
if message["status"] == "OK":
chan_id = cast(int, message["chan_id"])
del self.__subscriptions[chan_id]
if isinstance(message, list):
if (chan_id := cast(int, message[0])) and \
@@ -117,12 +117,14 @@ class BfxWebSocketBucket(Connection):
@Connection.require_websocket_connection
async def unsubscribe(self, sub_id: str) -> None:
for chan_id, subscription in self.__subscriptions.items():
for chan_id, subscription in list(self.__subscriptions.items()):
if subscription["sub_id"] == sub_id:
unsubscription = {
"event": "unsubscribe",
"chanId": chan_id }
del self.__subscriptions[chan_id]
await self._websocket.send(message = \
json.dumps(unsubscription))