signals handling

1 files changed, 27 insertions, 30 deletions

diff --git a/kvmd/apps/janus/runner.py b/kvmd/apps/janus/runner.py
index 28956e0a..b62e47bd 100644
--- a/kvmd/apps/janus/runner.py
+++ b/kvmd/apps/janus/runner.py
@@ -62,43 +62,38 @@ class JanusRunner:  # pylint: disable=too-many-instance-attributes
         logger = get_logger(0)
         logger.info("Starting Janus Runner ...")
         try:
-            asyncio.run(self.__run())
+            aiotools.run(self.__run())
         except (SystemExit, KeyboardInterrupt):
-            pass
+            aiotools.run(self.__stop_janus())
         logger.info("Bye-bye")
 
     # =====
 
     async def __run(self) -> None:
         logger = get_logger(0)
-        try:
-            prev_netcfg: Optional[_Netcfg] = None
-            while True:
-                retry = 0
-                netcfg = _Netcfg()
-                for retry in range(self.__check_retries):
-                    netcfg = await self.__get_netcfg()
-                    if netcfg.ext_ip:
-                        break
-                    await asyncio.sleep(self.__check_retries_delay)
-                if retry != 0 and netcfg.ext_ip:
-                    logger.info("I'm fine, continue working ...")
-
-                if netcfg != prev_netcfg:
-                    logger.info("Got new %s", netcfg)
-                    if netcfg.src_ip and netcfg.ext_ip:
-                        logger.info("Okay, restarting Janus ...")
-                        await self.__stop_janus()
-                        await self.__start_janus(netcfg)
-                    else:
-                        logger.error("Empty src_ip or ext_ip; stopping Janus ...")
-                        await self.__stop_janus()
-                    prev_netcfg = netcfg
-
-                await asyncio.sleep(self.__check_interval)
-        except:  # noqa: E722
-            await self.__stop_janus()
-            raise
+        prev_netcfg: Optional[_Netcfg] = None
+        while True:
+            retry = 0
+            netcfg = _Netcfg()
+            for retry in range(self.__check_retries):
+                netcfg = await self.__get_netcfg()
+                if netcfg.ext_ip:
+                    break
+                await asyncio.sleep(self.__check_retries_delay)
+            if retry != 0 and netcfg.ext_ip:
+                logger.info("I'm fine, continue working ...")
+
+            if netcfg != prev_netcfg:
+                logger.info("Got new %s", netcfg)
+                if netcfg.src_ip and netcfg.ext_ip:
+                    await self.__stop_janus()
+                    await self.__start_janus(netcfg)
+                else:
+                    logger.error("Empty src_ip or ext_ip; stopping Janus ...")
+                    await self.__stop_janus()
+                prev_netcfg = netcfg
+
+            await asyncio.sleep(self.__check_interval)
 
     async def __get_netcfg(self) -> _Netcfg:
         src_ip = (self.__get_default_ip() or "0.0.0.0")
@@ -136,12 +131,14 @@ class JanusRunner:  # pylint: disable=too-many-instance-attributes
 
     @aiotools.atomic
     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
     async def __stop_janus(self) -> None:
         if self.__janus_task:
+            get_logger(0).info("Stopping Janus ...")
             self.__janus_task.cancel()
             await asyncio.gather(self.__janus_task, return_exceptions=True)
         await self.__kill_janus_proc()