genericWebsocket: add reconnect attempt on disconnect

This commit is contained in:
Jacob Plaster
2019-01-18 10:34:45 +00:00
committed by Jacob Plaster
parent 6e4012f3c1
commit 3c8c3fb71e
5 changed files with 43 additions and 17 deletions

View File

@@ -369,7 +369,7 @@ class BfxWebsocket(GenericWebsocket):
jdata = generate_auth_payload(self.API_KEY, self.API_SECRET)
if self.dead_man_switch:
jdata['dms'] = 4
await self.ws.send(json.dumps(jdata))
await self.get_ws().send(json.dumps(jdata))
async def on_open(self):
self.logger.info("Websocket opened.")
@@ -380,17 +380,19 @@ class BfxWebsocket(GenericWebsocket):
# enable order book checksums
if self.manageOrderBooks:
await self.enable_flag(Flags.CHECKSUM)
# resubscribe to any channels
await self.subscriptionManager.resubscribe_all()
async def _send_auth_command(self, channel_name, data):
payload = [0, channel_name, None, data]
await self.ws.send(json.dumps(payload))
await self.get_ws().send(json.dumps(payload))
async def enable_flag(self, flag):
payload = {
"event": 'conf',
"flags": flag
}
await self.ws.send(json.dumps(payload))
await self.get_ws().send(json.dumps(payload))
def get_orderbook(self, symbol):
return self.orderBooks.get(symbol, None)