jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-28 03:07:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ensure storage task loads are always garbage collected (#42917)

Browse Source
This commit is contained in:
J. Nick Koston 2020-11-07 06:54:56 -10:00 committed by GitHub
parent 3ec7258c21
commit b47a5449e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/helpers/storage.py
View File

@ -105,6 +105,13 @@ class Store:
return await self._load_task return await self._load_task
async def _async_load(self): async def _async_load(self):
"""Load the data and ensure the task is removed."""
try:
return await self._async_load_data()
finally:
self._load_task = None
async def _async_load_data(self):
"""Load the data.""" """Load the data."""
# Check if we have a pending write # Check if we have a pending write
if self._data is not None: if self._data is not None:
@ -131,7 +138,6 @@ class Store:
) )
stored = await self._async_migrate_func(data["version"], data["data"]) stored = await self._async_migrate_func(data["version"], data["data"])
self._load_task = None
return stored return stored
async def async_save(self, data: Union[Dict, List]) -> None: async def async_save(self, data: Union[Dict, List]) -> None: