From ec85c25342101d91345e6134c286f85dc02060ad Mon Sep 17 00:00:00 2001 From: Andrei Date: Thu, 15 Dec 2016 12:38:59 -0600 Subject: [PATCH] bugfix - reset reconnection attempts after a successful resume --- disco/gateway/client.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/disco/gateway/client.py b/disco/gateway/client.py index a744890..9544af5 100644 --- a/disco/gateway/client.py +++ b/disco/gateway/client.py @@ -42,6 +42,7 @@ class GatewayClient(LoggingClass): # Bind to ready payload self.events.on('Ready', self.on_ready) + self.events.on('Resumed', self.on_resumed) # Websocket connection self.ws = None @@ -103,6 +104,10 @@ class GatewayClient(LoggingClass): self.session_id = ready.session_id self.reconnects = 0 + def on_resumed(self, _): + self.log.info('Recieved RESUMED') + self.reconnects = 0 + def connect_and_run(self, gateway_url=None): if not gateway_url: if not self._cached_gateway_url: @@ -188,7 +193,7 @@ class GatewayClient(LoggingClass): self.log.info('WS Closed: [%s] %s (%s)', code, reason, self.reconnects) if self.MAX_RECONNECTS and self.reconnects > self.MAX_RECONNECTS: - raise Exception('Failed to reconect after {} attempts, giving up'.format(self.MAX_RECONNECTS)) + raise Exception('Failed to reconnect after {} attempts, giving up'.format(self.MAX_RECONNECTS)) # Don't resume for these error codes if code and 4000 <= code <= 4010: