fix(backend): properly handle async run state for generic

Kitware · Oct 5, 2023 · 247b046 · 247b046
1 parent 25e0703
commit 247b046
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/src/wslink/backends/generic/core.py b/python/src/wslink/backends/generic/core.py
@@ -74,6 +74,7 @@ class GenericServer(AbstractWebApp):
     def __init__(self, server_config):
         AbstractWebApp.__init__(self, server_config)
         self._websockets = {}
+        self._stop_event = asyncio.Event()
 
         if "ws" in server_config:
             for route, server_protocol in server_config["ws"].items():
@@ -111,7 +112,11 @@ async def start(self, port_callback):
         if port_callback is not None:
             port_callback(self.get_port())
 
-        await asyncio.Event().wait()
+        self._stop_event.clear()
+        await self._stop_event.wait()
+
+    async def stop(self):
+        self._stop_event.set()
 
 
 def startWebServer(*args, **kwargs):