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

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/apps/janus/runner.py
View File

@@ -126,13 +126,13 @@ class JanusRunner: # pylint: disable=too-many-instance-attributes
# =====
@aiotools.atomic
@aiotools.atomic_fg
async def __start_janus(self, netcfg: _Netcfg) -> None:
get_logger(0).info("Starting Janus ...")
assert not self.__janus_task
self.__janus_task = asyncio.create_task(self.__janus_task_loop(netcfg))
@aiotools.atomic
@aiotools.atomic_fg
async def __stop_janus(self) -> None:
if self.__janus_task:
get_logger(0).info("Stopping Janus ...")

2
kvmd/apps/kvmd/auth.py
View File

@@ -113,7 +113,7 @@ class AuthManager:
assert self.__enabled
return self.__tokens.get(token)
@aiotools.atomic
@aiotools.atomic_fg
async def cleanup(self) -> None:
if self.__enabled:
assert self.__internal_service

10
kvmd/apps/kvmd/streamer.py
View File

@@ -221,7 +221,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes
# =====
@aiotools.atomic
@aiotools.atomic_fg
async def ensure_start(self, reset: bool) -> None:
if not self.__streamer_task or self.__stop_task:
logger = get_logger(0)
@@ -241,7 +241,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes
logger.info("Starting streamer ...")
await self.__inner_start()
@aiotools.atomic
@aiotools.atomic_fg
async def ensure_stop(self, immediately: bool) -> None:
if self.__streamer_task:
logger = get_logger(0)
@@ -384,7 +384,7 @@ class Streamer: # pylint: disable=too-many-instance-attributes
# =====
@aiotools.atomic
@aiotools.atomic_fg
async def cleanup(self) -> None:
await self.ensure_stop(immediately=True)
if self.__http_session:
@@ -409,12 +409,12 @@ class Streamer: # pylint: disable=too-many-instance-attributes
# =====
@aiotools.atomic
@aiotools.atomic_fg
async def __inner_start(self) -> None:
assert not self.__streamer_task
self.__streamer_task = asyncio.create_task(self.__streamer_task_loop())
@aiotools.atomic
@aiotools.atomic_fg
async def __inner_stop(self) -> None:
assert self.__streamer_task
self.__streamer_task.cancel()

10
kvmd/apps/kvmd/ugpio.py
View File

@@ -172,7 +172,7 @@ class _GpioOutput: # pylint: disable=too-many-instance-attributes
raise GpioSwitchNotSupported()
await self.__run_action(wait, "switch", self.__inner_switch, state)
@aiotools.atomic
@aiotools.atomic_fg
async def pulse(self, delay: float, wait: bool) -> None:
if not self.__pulse_delay:
raise GpioPulseNotSupported()
@@ -181,7 +181,7 @@ class _GpioOutput: # pylint: disable=too-many-instance-attributes
# =====
@aiotools.atomic
@aiotools.atomic_fg
async def __run_action(self, wait: bool, name: str, func: Callable, *args: Any) -> None:
if wait:
async with self.__region:
@@ -192,20 +192,20 @@ class _GpioOutput: # pylint: disable=too-many-instance-attributes
self.__region, self.__action_task_wrapper, name, func, *args,
)
@aiotools.atomic
@aiotools.atomic_fg
async def __action_task_wrapper(self, name: str, func: Callable, *args: Any) -> None:
try:
return (await func(*args))
except GpioDriverOfflineError:
get_logger(0).error("Can't perform %s of %s or operation was not completed: driver offline", name, self)
@aiotools.atomic
@aiotools.atomic_fg
async def __inner_switch(self, state: bool) -> None:
await self.__write(state)
get_logger(0).info("Ensured switch %s to state=%d", self, state)
await asyncio.sleep(self.__busy_delay)
@aiotools.atomic
@aiotools.atomic_fg
async def __inner_pulse(self, delay: float) -> None:
try:
await self.__write(True)