🐛 gateway no longer starts multiple instances

pincer/core/gateway.py

@@ -110,7 +110,8 @@ def __init__(
         }
 
         # 4000 and 4009 are not included. The client will reconnect when receiving
-        # either.
+        # either. Code 4000 is also used for internal disconnects that will lead to a
+        # reconnect.
         self.__close_codes: Dict[int, GatewayError] = {
             4001: GatewayError("Invalid opcode was sent"),
             4002: GatewayError("Invalid payload was sent."),
@@ -140,7 +141,7 @@ def __init__(
 
         # The gateway can be disconnected from Discord. This variable stores if the
         # gateway should send a hello or reconnect.
-        self.__should_reconnect: bool = False
+        self.__should_resume: bool = False
 
         # The sequence number for the last received payload. This is used reconnecting.
         self.__sequence_number: int = 0
@@ -258,8 +259,8 @@ async def event_loop(self):
             "%s Disconnected from Gateway due without any errors. Reconnecting.",
             self.shard_key
         )
-        self.__should_reconnect = True
-        await self.start_loop()
+        # ensure_future prevents a stack overflow
+        ensure_future(self.start_loop())
 
     async def handle_data(self, data: Dict[Any]):
         """|coro|
@@ -310,19 +311,23 @@ async def handle_reconnect(self, payload: GatewayDispatch):
             self.shard_key
         )
 
+        self.__should_resume = True
         await self.__socket.close()
-        self.__should_reconnect = True
-        await self.start_loop()
 
     async def handle_invalid_session(self, payload: GatewayDispatch):
         """|coro|
         Opcode 9 - Invalid connection
         Attempt to relog. This is probably because the session was already invalidated
         when we tried to reconnect.
         """
-        _log.debug("%s Invalid session, attempting to relog...", self.shard_key)
-        self.__should_reconnect = False
-        await self.start_loop()
+
+        _log.debug(
+            "%s Invalid session, attempting to %s...",
+            self.shard_key,
+            "reconnect" if payload.data else "relog"
+        )
+
+        self.__should_resume = payload.data
 
     async def identify_and_handle_hello(self, payload: GatewayDispatch):
         """|coro|
@@ -334,7 +339,7 @@ async def identify_and_handle_hello(self, payload: GatewayDispatch):
 
         Successful reconnects are handled in the `resumed` middleware.
         """
-        if self.__should_reconnect:
+        if self.__should_resume:
             _log.debug("%s Resuming connection with Discord", self.shard_key)
 
             await self.send(str(GatewayDispatch(
@@ -363,8 +368,6 @@ async def identify_and_handle_hello(self, payload: GatewayDispatch):
         ))
         self.__heartbeat_interval = payload.data["heartbeat_interval"]
 
-        # This process should already be forked to the background so there is no need to
-        # `ensure_future()` here.
         self.start_heartbeat()
 
     async def handle_heartbeat(self, payload: GatewayDispatch):
@@ -452,14 +455,13 @@ async def __heartbeat_loop(self):
                 # Close code is specified to be anything that is not 1000 in the docs.
                 _log.debug(
                     "%s %s ack not received. Attempting to reconnect."
-                    " Closing socket with close code 1001. \U0001f480",
+                    " Closing socket with close code 4000. \U0001f480",
                     datetime.now(),
                     self.shard_key
                 )
-                await self.__socket.close(code=1001)
-                self.__should_reconnect = True
+                await self.__socket.close(code=4000)
+                self.__should_resume = True
                 # A new loop is started in the background while this one is stopped.
-                ensure_future(self.start_loop())
                 self.stop_heartbeat()
                 return