From 5503379a3b0597a0523dc83c1ad0a96fe0183263 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Thu, 27 Jun 2024 01:50:41 -0500
Subject: [PATCH] Fix performance regression in integration from state_reported
 (#120621)

* Fix performance regression in integration from state_reported

Because the callbacks were no longer indexed by entity id, users
saw upwards of 1M calls/min

https://github.com/home-assistant/core/pull/113869/files#r1655580523

* Update homeassistant/helpers/event.py

* coverage

---------

Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
---
 .../components/integration/sensor.py          | 24 +++++++++----------
 1 file changed, 11 insertions(+), 13 deletions(-)

diff --git a/homeassistant/components/integration/sensor.py b/homeassistant/components/integration/sensor.py
index 4fca92e9b40..8cc5341f081 100644
--- a/homeassistant/components/integration/sensor.py
+++ b/homeassistant/components/integration/sensor.py
@@ -27,8 +27,6 @@ from homeassistant.const import (
     CONF_METHOD,
     CONF_NAME,
     CONF_UNIQUE_ID,
-    EVENT_STATE_CHANGED,
-    EVENT_STATE_REPORTED,
     STATE_UNAVAILABLE,
     UnitOfTime,
 )
@@ -45,7 +43,11 @@ from homeassistant.helpers import config_validation as cv, entity_registry as er
 from homeassistant.helpers.device import async_device_info_to_link_from_entity
 from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.event import async_call_later
+from homeassistant.helpers.event import (
+    async_call_later,
+    async_track_state_change_event,
+    async_track_state_reported_event,
+)
 from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
 from .const import (
@@ -440,21 +442,17 @@ class IntegrationSensor(RestoreSensor):
             self._derive_and_set_attributes_from_state(state)
 
         self.async_on_remove(
-            self.hass.bus.async_listen(
-                EVENT_STATE_CHANGED,
+            async_track_state_change_event(
+                self.hass,
+                self._sensor_source_id,
                 handle_state_change,
-                event_filter=callback(
-                    lambda event_data: event_data["entity_id"] == self._sensor_source_id
-                ),
             )
         )
         self.async_on_remove(
-            self.hass.bus.async_listen(
-                EVENT_STATE_REPORTED,
+            async_track_state_reported_event(
+                self.hass,
+                self._sensor_source_id,
                 handle_state_report,
-                event_filter=callback(
-                    lambda event_data: event_data["entity_id"] == self._sensor_source_id
-                ),
             )
         )