jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 14:47:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add sqlalchemy LRUs to the profiler lru service (#90428)

Browse Source
This commit is contained in:
J. Nick Koston 2023-03-28 12:02:35 -10:00 committed by GitHub
parent 9ae0236208
commit 9dc936f8b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/profiler/__init__.py
View File

@ -35,6 +35,7 @@ SERVICE_LOG_THREAD_FRAMES = "log_thread_frames"
SERVICE_LOG_EVENT_LOOP_SCHEDULED = "log_event_loop_scheduled" SERVICE_LOG_EVENT_LOOP_SCHEDULED = "log_event_loop_scheduled"
_LRU_CACHE_WRAPPER_OBJECT = _lru_cache_wrapper.__name__ _LRU_CACHE_WRAPPER_OBJECT = _lru_cache_wrapper.__name__
_SQLALCHEMY_LRU_OBJECT = "LRUCache"
_KNOWN_LRU_CLASSES = ( _KNOWN_LRU_CLASSES = (
"EventDataManager", "EventDataManager",
@ -67,7 +68,9 @@ LOG_INTERVAL_SUB = "log_interval_subscription"
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_setup_entry( # noqa: C901
hass: HomeAssistant, entry: ConfigEntry
) -> bool:
"""Set up Profiler from a config entry.""" """Set up Profiler from a config entry."""
lock = asyncio.Lock() lock = asyncio.Lock()
domain_data = hass.data[DOMAIN] = {} domain_data = hass.data[DOMAIN] = {}
@ -176,6 +179,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
maybe_lru.get_stats(), maybe_lru.get_stats(),
) )
for lru in objgraph.by_type(_SQLALCHEMY_LRU_OBJECT):
if (data := getattr(lru, "_data", None)) and isinstance(data, dict):
for key, value in dict(data).items():
_LOGGER.critical(
"Cache data for sqlalchemy LRUCache %s: %s: %s", lru, key, value
)
persistent_notification.create( persistent_notification.create(
hass, hass,
( (

10
tests/components/profiler/test_init.py
View File

@ -11,6 +11,7 @@ import pytest
from homeassistant.components.profiler import ( from homeassistant.components.profiler import (
_LRU_CACHE_WRAPPER_OBJECT, _LRU_CACHE_WRAPPER_OBJECT,
_SQLALCHEMY_LRU_OBJECT,
CONF_SECONDS, CONF_SECONDS,
SERVICE_DUMP_LOG_OBJECTS, SERVICE_DUMP_LOG_OBJECTS,
SERVICE_LOG_EVENT_LOOP_SCHEDULED, SERVICE_LOG_EVENT_LOOP_SCHEDULED,
@ -254,9 +255,17 @@ async def test_lru_stats(hass: HomeAssistant, caplog: pytest.LogCaptureFixture)
domain_data = DomainData() domain_data = DomainData()
assert hass.services.has_service(DOMAIN, SERVICE_LRU_STATS) assert hass.services.has_service(DOMAIN, SERVICE_LRU_STATS)
class LRUCache:
def __init__(self):
self._data = {"sqlalchemy_test": 1}
sqlalchemy_lru_cache = LRUCache()
def _mock_by_type(type_): def _mock_by_type(type_):
if type_ == _LRU_CACHE_WRAPPER_OBJECT: if type_ == _LRU_CACHE_WRAPPER_OBJECT:
return [_dummy_test_lru_stats] return [_dummy_test_lru_stats]
if type_ == _SQLALCHEMY_LRU_OBJECT:
return [sqlalchemy_lru_cache]
return [domain_data] return [domain_data]
with patch("objgraph.by_type", side_effect=_mock_by_type): with patch("objgraph.by_type", side_effect=_mock_by_type):
@ -266,3 +275,4 @@ async def test_lru_stats(hass: HomeAssistant, caplog: pytest.LogCaptureFixture)
assert "(0, 0)" in caplog.text assert "(0, 0)" in caplog.text
assert "_dummy_test_lru_stats" in caplog.text assert "_dummy_test_lru_stats" in caplog.text
assert "CacheInfo" in caplog.text assert "CacheInfo" in caplog.text
assert "sqlalchemy_test" in caplog.text