Merge pull request finos#2806 from s22chan/master

fix aiohttp disconnect error on websockets
tomjakubowski · Oct 29, 2024 · 09204c5 · 09204c5
2 parents fb11915 + e1c1a03
commit 09204c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rust/perspective-python/perspective/handlers/aiohttp.py b/rust/perspective-python/perspective/handlers/aiohttp.py
@@ -39,7 +39,7 @@ def __init__(self, **kwargs):
         self._request = kwargs.pop("request")
         super().__init__(**kwargs)
 
-    async def run(self) -> None:
+    async def run(self) -> web.WebSocketResponse:
         def inner(msg):
             asyncio.get_running_loop().create_task(self._ws.send_bytes(msg))
 
@@ -53,3 +53,4 @@ def inner(msg):
 
         finally:
             self.session.close()
+        return self._ws