From 8aa333ba895e7b498e867f7b430335852a4a9f84 Mon Sep 17 00:00:00 2001 From: Devaev Maxim Date: Wed, 5 Jun 2019 06:30:21 +0300 Subject: better atomic ops --- kvmd/aiotools.py | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 kvmd/aiotools.py (limited to 'kvmd/aiotools.py') diff --git a/kvmd/aiotools.py b/kvmd/aiotools.py new file mode 100644 index 00000000..d0c12537 --- /dev/null +++ b/kvmd/aiotools.py @@ -0,0 +1,48 @@ +# ========================================================================== # +# # +# KVMD - The main Pi-KVM daemon. # +# # +# Copyright (C) 2018 Maxim Devaev # +# # +# This program is free software: you can redistribute it and/or modify # +# it under the terms of the GNU General Public License as published by # +# the Free Software Foundation, either version 3 of the License, or # +# (at your option) any later version. # +# # +# This program is distributed in the hope that it will be useful, # +# but WITHOUT ANY WARRANTY; without even the implied warranty of # +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # +# GNU General Public License for more details. # +# # +# You should have received a copy of the GNU General Public License # +# along with this program. If not, see . # +# # +# ========================================================================== # + + +import asyncio +import functools + +import typing + +from typing import Callable +from typing import TypeVar +from typing import Any + + +# ===== +_AtomicF = TypeVar("_AtomicF", bound=Callable[..., Any]) + + +def atomic(method: _AtomicF) -> _AtomicF: + @functools.wraps(method) + async def wrapper(*args: object, **kwargs: object) -> Any: + return (await asyncio.shield(method(*args, **kwargs))) + return typing.cast(_AtomicF, wrapper) + + +def task(method: Callable[..., Any]) -> Callable[..., asyncio.Task]: + @functools.wraps(method) + async def wrapper(*args: object, **kwargs: object) -> asyncio.Task: + return asyncio.create_task(method(*args, **kwargs)) + return typing.cast(Callable[..., asyncio.Task], wrapper) -- cgit v1.2.3