mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-09 23:39:50 +00:00
Code Issues Projects Releases Wiki Activity

Handle HEARTBEAT_ACK

Browse Source
This commit is contained in:
Rapptz 2017-02-08 07:23:09 -05:00
parent dae3c66a54
commit 9341be6be8
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
discord/gateway.py
View File

@ -25,6 +25,7 @@ DEALINGS IN THE SOFTWARE.
""" """
import sys import sys
import time
import websockets import websockets
import asyncio import asyncio
import aiohttp import aiohttp
@ -64,9 +65,23 @@ class KeepAliveHandler(threading.Thread):
self.daemon = True self.daemon = True
self.msg = 'Keeping websocket alive with sequence {0[d]}' self.msg = 'Keeping websocket alive with sequence {0[d]}'
self._stop_ev = threading.Event() self._stop_ev = threading.Event()
self._last_ack = time.time()
def run(self): def run(self):
while not self._stop_ev.wait(self.interval): while not self._stop_ev.wait(self.interval):
if self._last_ack + 2 * self.interval < time.time():
log.warn("We have stopped responding to the gateway.")
coro = self.ws.close(1006)
f = compat.run_coroutine_threadsafe(coro, loop=self.ws.loop)
try:
f.result()
except:
pass
finally:
self.stop()
return
data = self.get_payload() data = self.get_payload()
log.debug(self.msg.format(data)) log.debug(self.msg.format(data))
coro = self.ws.send_as_json(data) coro = self.ws.send_as_json(data)
@ -86,12 +101,16 @@ class KeepAliveHandler(threading.Thread):
def stop(self): def stop(self):
self._stop_ev.set() self._stop_ev.set()
def ack(self):
self._last_ack = time.time()
class VoiceKeepAliveHandler(KeepAliveHandler): class VoiceKeepAliveHandler(KeepAliveHandler):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.msg = 'Keeping voice websocket alive with timestamp {0[d]}' self.msg = 'Keeping voice websocket alive with timestamp {0[d]}'
def get_payload(self): def get_payload(self):
self.ack()
return { return {
'op': self.ws.HEARTBEAT, 'op': self.ws.HEARTBEAT,
'd': int(time.time() * 1000) 'd': int(time.time() * 1000)
@ -302,7 +321,8 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
raise ReconnectWebSocket() raise ReconnectWebSocket()
if op == self.HEARTBEAT_ACK: if op == self.HEARTBEAT_ACK:
return # disable noisy logging for now self._keep_alive.ack()
return
if op == self.HEARTBEAT: if op == self.HEARTBEAT:
beat = self._keep_alive.get_payload() beat = self._keep_alive.get_payload()