GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-12 09:10:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

improved streamer killing

Browse Source
This commit is contained in:
Devaev Maxim 2018-06-29 23:59:43 +03:00
parent 93e28c62e5
commit 5589ecbac4
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
kvmd/kvmd/streamer.py
View File

@ -57,8 +57,8 @@ class Streamer: # pylint: disable=too-many-instance-attributes
async def __process(self) -> None:
logger = get_logger(0)
proc: Optional[asyncio.subprocess.Process] = None # pylint: disable=no-member
while True:
while True: # pylint: disable=too-many-nested-blocks
proc: Optional[asyncio.subprocess.Process] = None # pylint: disable=no-member
try:
proc = await asyncio.create_subprocess_shell(
self.__cmd,
@ -78,27 +78,36 @@ class Streamer: # pylint: disable=too-many-instance-attributes
if empty == 100: # asyncio bug
break
await self.__kill(proc)
raise RuntimeError("WTF")
except asyncio.CancelledError:
break
except Exception as err:
if proc:
logger.exception("Unexpected finished streamer pid=%d with retcode=%d", proc.pid, proc.returncode)
logger.exception("Unexpected streamer error: pid=%d", proc.pid)
else:
logger.exception("Can't start streamer: %s", err)
await asyncio.sleep(1)
if proc:
await self.__kill(proc)
finally:
if proc and proc.returncode is None:
await self.__kill(proc)
async def __kill(self, proc: asyncio.subprocess.Process) -> None: # pylint: disable=no-member
try:
proc.terminate()
await asyncio.sleep(1)
if proc.returncode is None:
proc.kill()
try:
proc.kill()
except Exception:
if proc.returncode is not None:
raise
await proc.wait()
get_logger().info("Streamer killed: pid=%d; retcode=%d")
except Exception:
pass
if proc.returncode is None:
get_logger().exception("Can't kill streamer pid=%d", proc.pid)
else:
get_logger().info("Streamer killed: pid=%d; retcode=%d", proc.pid, proc.returncode)