minimal fixup for daemon signal handling regression (#13570)

This commit is contained in:
Kyle Altendorf 2022-09-30 18:49:42 -04:00 committed by GitHub
parent c837711bd0
commit 0b3201635d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,4 +1,5 @@
import asyncio
import functools
import json
import logging
import os
@ -186,8 +187,14 @@ class WebSocketServer:
async def setup_process_global_state(self) -> None:
try:
asyncio.get_running_loop().add_signal_handler(signal.SIGINT, self._accept_signal)
asyncio.get_running_loop().add_signal_handler(signal.SIGTERM, self._accept_signal)
asyncio.get_running_loop().add_signal_handler(
signal.SIGINT,
functools.partial(self._accept_signal, signal_number=signal.SIGINT),
)
asyncio.get_running_loop().add_signal_handler(
signal.SIGTERM,
functools.partial(self._accept_signal, signal_number=signal.SIGTERM),
)
except NotImplementedError:
self.log.info("Not implemented")