Add config to enable checksums in BfxWebSocketBucket.

This commit is contained in:
Davide Casale
2023-07-28 14:57:35 +02:00
parent 3038027f35
commit d9267de009

View File

@@ -14,6 +14,8 @@ if TYPE_CHECKING:
from websockets.client import WebSocketClientProtocol
from pyee import EventEmitter
_CHECKSUM_FLAG_VALUE = 131_072
class BfxWebSocketBucket(Connection):
VERSION = 2
@@ -45,6 +47,8 @@ class BfxWebSocketBucket(Connection):
await self.__recover_state()
await self.__set_conf(flags=_CHECKSUM_FLAG_VALUE)
async with self.__condition:
self.__condition.notify(1)
@@ -84,6 +88,10 @@ class BfxWebSocketBucket(Connection):
self.__subscriptions.clear()
async def __set_conf(self, flags: int) -> None:
await self._websocket.send(json.dumps( \
{ "event": "conf", "flags": flags }))
@Connection.require_websocket_connection
async def subscribe(self,
channel: str,