wait short tasks

5 files changed, 12 insertions, 7 deletions

diff --git a/kvmd/apps/kvmd/atx.py b/kvmd/apps/kvmd/atx.py
index 861c3dc2..6f39e01e 100644
--- a/kvmd/apps/kvmd/atx.py
+++ b/kvmd/apps/kvmd/atx.py
@@ -175,7 +175,7 @@ class Atx:  # pylint: disable=too-many-instance-attributes
 
     # =====
 
-    @aiotools.task
+    @aiotools.tasked
     @aiotools.atomic
     async def __click(self, pin: int, delay: float) -> None:
         with self.__region:
diff --git a/kvmd/apps/kvmd/hid.py b/kvmd/apps/kvmd/hid.py
index da1c7d43..5b51b850 100644
--- a/kvmd/apps/kvmd/hid.py
+++ b/kvmd/apps/kvmd/hid.py
@@ -165,7 +165,7 @@ class Hid(multiprocessing.Process):  # pylint: disable=too-many-instance-attribu
             yield self.get_state()
             await asyncio.sleep(self.__state_poll)
 
-    @aiotools.task
+    @aiotools.tasked
     @aiotools.atomic
     async def reset(self) -> None:
         async with self.__lock:
diff --git a/kvmd/apps/kvmd/msd.py b/kvmd/apps/kvmd/msd.py
index 6033b25a..49c16651 100644
--- a/kvmd/apps/kvmd/msd.py
+++ b/kvmd/apps/kvmd/msd.py
@@ -315,7 +315,7 @@ class MassStorageDevice:  # pylint: disable=too-many-instance-attributes
             return state
 
     @_msd_working
-    @aiotools.task
+    @aiotools.tasked
     @aiotools.atomic
     async def reset(self) -> None:
         with self.__region:
diff --git a/kvmd/apps/kvmd/server.py b/kvmd/apps/kvmd/server.py
index d4c4c13a..18e8b65f 100644
--- a/kvmd/apps/kvmd/server.py
+++ b/kvmd/apps/kvmd/server.py
@@ -40,6 +40,8 @@ import aiohttp
 import aiohttp.web
 import setproctitle
 
+from ... import aiotools
+
 from ...logging import get_logger
 
 from ...aioregion import RegionIsBusyError
@@ -144,9 +146,9 @@ async def _get_multipart_field(reader: aiohttp.MultipartReader, name: str) -> ai
     return field
 
 
-_ATTR_EXPOSED = "exposed"
-_ATTR_EXPOSED_METHOD = "exposed_method"
-_ATTR_EXPOSED_PATH = "exposed_path"
+_ATTR_EXPOSED = "_server_exposed"
+_ATTR_EXPOSED_METHOD = "_server_exposed_method"
+_ATTR_EXPOSED_PATH = "_server_exposed_path"
 _ATTR_SYSTEM_TASK = "system_task"
 
 _HEADER_AUTH_USER = "X-KVMD-User"
@@ -549,6 +551,9 @@ class Server:  # pylint: disable=too-many-instance-attributes
     async def __on_shutdown(self, _: aiohttp.web.Application) -> None:
         logger = get_logger(0)
 
+        logger.info("Waiting short tasks ...")
+        await aiotools.gather_short_tasks()
+
         logger.info("Cancelling system tasks ...")
         for task in self.__system_tasks:
             task.cancel()
diff --git a/kvmd/apps/kvmd/streamer.py b/kvmd/apps/kvmd/streamer.py
index c8c89c50..4afc018c 100644
--- a/kvmd/apps/kvmd/streamer.py
+++ b/kvmd/apps/kvmd/streamer.py
@@ -172,7 +172,7 @@ class Streamer:  # pylint: disable=too-many-instance-attributes
     async def __inner_start(self) -> None:
         assert not self.__streamer_task
         await self.__set_hw_enabled(True)
-        self.__streamer_task = asyncio.get_running_loop().create_task(self.__run_streamer())
+        self.__streamer_task = asyncio.create_task(self.__run_streamer())
 
     async def __inner_stop(self) -> None:
         assert self.__streamer_task