From b5f2bff41d0d4ed248752faacbaf8ef20f9bb3df Mon Sep 17 00:00:00 2001 From: Andrei Zbikowski Date: Mon, 22 Jan 2018 06:27:24 -0800 Subject: [PATCH] Close w/ non-1000 code, make variable private --- disco/gateway/client.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/disco/gateway/client.py b/disco/gateway/client.py index b8ceecc..a7b5d36 100644 --- a/disco/gateway/client.py +++ b/disco/gateway/client.py @@ -68,7 +68,7 @@ class GatewayClient(LoggingClass): # Heartbeat self._heartbeat_task = None - self.heartbeat_acknowledged = True + self._heartbeat_acknowledged = True def send(self, op, data): self.limiter.check() @@ -84,14 +84,14 @@ class GatewayClient(LoggingClass): def heartbeat_task(self, interval): while True: - if not self.heartbeat_acknowledged: + if not self._heartbeat_acknowledged: self.log.warning('Received HEARTBEAT without HEARTBEAT_ACK, forcing a fresh reconnect') - self.heartbeat_acknowledged = True - self.ws.close() + self._heartbeat_acknowledged = True + self.ws.close(status=4000) return self._send(OPCode.HEARTBEAT, self.seq) - self.heartbeat_acknowledged = False + self._heartbeat_acknowledged = False gevent.sleep(interval / 1000) def handle_dispatch(self, packet): @@ -106,7 +106,7 @@ class GatewayClient(LoggingClass): def handle_heartbeat_acknowledge(self, _): self.log.debug('Received HEARTBEAT_ACK') - self.heartbeat_acknowledged = True + self._heartbeat_acknowledged = True def handle_reconnect(self, _): self.log.warning('Received RECONNECT request, forcing a fresh reconnect')