use AutoReconnect

Author: blink1073

pymongo/asynchronous/pool.py

@@ -1102,11 +1102,12 @@ async def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> A
 
             await conn.authenticate()
         # Catch KeyboardInterrupt, CancelledError, etc. and cleanup.
-        except BaseException:
+        except BaseException as e:
             async with self.lock:
                 self.active_contexts.discard(conn.cancel_context)
             # Enter backoff mode and reconnect on establishment failure.
-            if conn.conn_closed():
+            if type(e) == AutoReconnect:
+                await conn.close_conn(ConnectionClosedReason.ERROR)
                 self._backoff += 1
                 # TODO: emit a message about backoff.
                 print("backing off", self._backoff)  # noqa: T201

pymongo/synchronous/pool.py

@@ -1098,11 +1098,12 @@ def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> Connect
 
             conn.authenticate()
         # Catch KeyboardInterrupt, CancelledError, etc. and cleanup.
-        except BaseException:
+        except BaseException as e:
             with self.lock:
                 self.active_contexts.discard(conn.cancel_context)
             # Enter backoff mode and reconnect on establishment failure.
-            if conn.conn_closed():
+            if type(e) == AutoReconnect:
+                conn.close_conn(ConnectionClosedReason.ERROR)
                 self._backoff += 1
                 # TODO: emit a message about backoff.
                 print("backing off", self._backoff)  # noqa: T201