Merge pull request #1559 from Dorukyum/resume_gateway_url

Implement `resume_gateway_url`

Author: BobDotCom

discord/gateway.py

@@ -292,6 +292,7 @@ def __init__(self, socket, *, loop):
         # ws related stuff
         self.session_id = None
         self.sequence = None
+        self.resume_gateway_url = None
         self._zlib = zlib.decompressobj()
         self._buffer = bytearray()
         self._close_code = None
@@ -511,6 +512,7 @@ async def received_message(self, msg, /):
             self._trace = trace = data.get("_trace", [])
             self.sequence = msg["s"]
             self.session_id = data["session_id"]
+            self.resume_gateway_url = data["resume_gateway_url"]
             # pass back shard ID to ready handler
             data["__shard_id__"] = self.shard_id
             _log.info(

discord/shard.py

@@ -221,7 +221,11 @@ async def reidentify(self, exc: ReconnectWebSocket) -> None:
     async def reconnect(self) -> None:
         self._cancel_task()
         try:
-            coro = DiscordWebSocket.from_client(self._client, shard_id=self.id)
+            coro = DiscordWebSocket.from_client(
+                self._client,
+                gateway=self.ws.resume_gateway_url,
+                shard_id=self.id,
+            )
             self.ws = await asyncio.wait_for(coro, timeout=60.0)
         except self._handled_exceptions as e:
             await self._handle_disconnect(e)