Handle ping/pong events in ASGI apps

frankie567 · Feb 21, 2024 · 4958baf · 4958baf
1 parent cd0a3dc
commit 4958baf
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/httpx_ws/transport.py b/httpx_ws/transport.py
@@ -104,6 +104,11 @@ async def write(
                 await self.send({"type": "websocket.receive", "text": event.data})
             elif isinstance(event, wsproto.events.BytesMessage):
                 await self.send({"type": "websocket.receive", "bytes": event.data})
+            elif isinstance(event, wsproto.events.Ping):
+                data = self.connection.send(event.response())
+                await self.write(data)
+            elif isinstance(event, wsproto.events.Pong):
+                pass
             else:
                 raise UnhandledWebSocketEvent(event)