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

using shield_fg() in atomic (now atomic_fg)

Browse Source
This commit is contained in:
Maxim Devaev
2022-08-07 19:35:08 +03:00
parent fdc3edfa79
commit e37a7254d5
10 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kvmd/aiotools.py
View File

@@ -125,10 +125,10 @@ def shield_fg(aw: Awaitable): # type: ignore
return outer
def atomic(func: _FunctionT) -> _FunctionT:
def atomic_fg(func: _FunctionT) -> _FunctionT:
@functools.wraps(func)
async def wrapper(*args: Any, **kwargs: Any) -> Any:
return (await asyncio.shield(func(*args, **kwargs)))
return (await shield_fg(func(*args, **kwargs)))
return typing.cast(_FunctionT, wrapper)