GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-01-29 00:51:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

simplified AioNotifier()

Browse Source
This commit is contained in:
Maxim Devaev
2022-08-07 19:04:32 +03:00
parent aa630988cc
commit ec9785b4be
18 changed files with 43 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
kvmd/aiotools.py
View File

@@ -232,10 +232,7 @@ class AioNotifier:
def __init__(self) -> None:
self.__queue: "asyncio.Queue[None]" = asyncio.Queue()
async def notify(self) -> None:
await self.__queue.put(None)
def notify_sync(self) -> None:
def notify(self) -> None:
self.__queue.put_nowait(None)
async def wait(self, timeout: Optional[float]=None) -> None:
@@ -299,7 +296,7 @@ class AioExclusiveRegion:
self.__busy = True
try:
if self.__notifier:
await self.__notifier.notify()
self.__notifier.notify()
except: # noqa: E722
self.__busy = False
raise
@@ -309,7 +306,7 @@ class AioExclusiveRegion:
async def exit(self) -> None:
self.__busy = False
if self.__notifier:
await self.__notifier.notify()
self.__notifier.notify()
async def __aenter__(self) -> None:
await self.enter()