break reference cycles

home-assistant · Jun 19, 2023 · 7a9e023 · 7a9e023
1 parent 09839c9
commit 7a9e023
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/homeassistant/components/websocket_api/http.py b/homeassistant/components/websocket_api/http.py
@@ -392,6 +392,8 @@ def handle_hass_stop(event: Event) -> None:
         finally:
             unsub_stop()
 
+            self._cancel_peak_checker()
+
             if connection is not None:
                 connection.async_handle_close()
 
@@ -424,4 +426,14 @@ def handle_hass_stop(event: Event) -> None:
 
                     async_dispatcher_send(self.hass, SIGNAL_WEBSOCKET_DISCONNECTED)
 
+                    # Break reference cycles to make sure GC can happen sooner
+                    self.wsock = None  # type: ignore[assignment]
+                    self.request = None  # type: ignore[assignment]
+                    self.hass = None  # type: ignore[assignment]
+                    self._logger = None  # type: ignore[assignment]
+                    self._message_queue = None  # type: ignore[assignment]
+                    self._handle_task = None
+                    self._writer_task = None
+                    self._ready_future = None
+
         return wsock