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

Switch recorder to use async_timeout instead of asyncio.wait_for (#78607)

Browse Source
This commit is contained in:
J. Nick Koston 2022-09-17 12:40:10 -05:00 committed by GitHub
parent 51b01fcd80
commit 6e4a0b9fdc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/recorder/core.py
View File

@ -13,6 +13,7 @@ import threading
import time import time
from typing import Any, TypeVar, cast from typing import Any, TypeVar, cast
import async_timeout
from awesomeversion import AwesomeVersion from awesomeversion import AwesomeVersion
from lru import LRU # pylint: disable=no-name-in-module from lru import LRU # pylint: disable=no-name-in-module
from sqlalchemy import create_engine, event as sqlalchemy_event, exc, func, select from sqlalchemy import create_engine, event as sqlalchemy_event, exc, func, select
@ -1030,7 +1031,8 @@ class Recorder(threading.Thread):
task = DatabaseLockTask(database_locked, threading.Event(), False) task = DatabaseLockTask(database_locked, threading.Event(), False)
self.queue_task(task) self.queue_task(task)
try: try:
await asyncio.wait_for(database_locked.wait(), timeout=DB_LOCK_TIMEOUT) async with async_timeout.timeout(DB_LOCK_TIMEOUT):
await database_locked.wait()
except asyncio.TimeoutError as err: except asyncio.TimeoutError as err:
task.database_unlock.set() task.database_unlock.set()
raise TimeoutError( raise TimeoutError(