PYTHON-5846 Decompose Hello response in _HeartbeatTelemetry.succeeded

Author: blink1073

pymongo/_telemetry.py

@@ -403,22 +403,23 @@ def emit_started_log(self, conn_id: int, server_conn_id: Optional[int]) -> None:
     def succeeded(
         self,
         round_trip_time: float,
-        response: Any,
+        reply: _DocumentOut,
+        awaited: bool,
         conn_id: int,
         server_conn_id: Optional[int],
     ) -> None:
         """Emit the SUCCEEDED log entry and APM event."""
         if self._should_publish:
             assert self._listeners is not None
             self._listeners.publish_server_heartbeat_succeeded(
-                self._address, round_trip_time, response, response.awaitable
+                self._address, round_trip_time, reply, awaited
             )
         self._emit_log(
             _SDAMStatusMessage.HEARTBEAT_SUCCESS,
             driverConnectionId=conn_id,
             serverConnectionId=server_conn_id,
             durationMS=round_trip_time * 1000,
-            reply=response.document,
+            reply=reply,
         )
 
     def failed(self, error: Exception, conn_id: Optional[int]) -> None:

pymongo/asynchronous/monitor.py

@@ -304,7 +304,13 @@ async def _check_once(self) -> ServerDescription:
 
             avg_rtt, min_rtt = await self._rtt_monitor.get()
             sd = ServerDescription(address, response, avg_rtt, min_round_trip_time=min_rtt)
-            hb.succeeded(round_trip_time, response, conn.id, conn.server_connection_id)
+            hb.succeeded(
+                round_trip_time,
+                response.document,
+                response.awaitable,
+                conn.id,
+                conn.server_connection_id,
+            )
             return sd
 
     async def _check_with_socket(self, conn: AsyncConnection) -> tuple[Hello, float]:  # type: ignore[type-arg]

pymongo/synchronous/monitor.py

@@ -302,7 +302,13 @@ def _check_once(self) -> ServerDescription:
 
             avg_rtt, min_rtt = self._rtt_monitor.get()
             sd = ServerDescription(address, response, avg_rtt, min_round_trip_time=min_rtt)
-            hb.succeeded(round_trip_time, response, conn.id, conn.server_connection_id)
+            hb.succeeded(
+                round_trip_time,
+                response.document,
+                response.awaitable,
+                conn.id,
+                conn.server_connection_id,
+            )
             return sd
 
     def _check_with_socket(self, conn: Connection) -> tuple[Hello, float]:  # type: ignore[type-arg]