initial rework separating server and client impl

scripts/test_ws_client.py

@@ -0,0 +1,58 @@
+import asyncio
+import logging
+import socket
+
+import websockets
+
+
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+
+async def _client():
+    try:
+        while True:
+            try:
+                url = "ws://localhost:8080/api/v1/message/ws?token=testtoken"
+                async with websockets.connect(url) as ws:
+                    logger.info("Connection successful")
+                    while True:
+                        try:
+                            data = await asyncio.wait_for(
+                                ws.recv(),
+                                timeout=5
+                            )
+                            logger.info(f"Data received - {data}")
+                        except (asyncio.TimeoutError, websockets.exceptions.ConnectionClosed):
+                            # We haven't received data yet. Check the connection and continue.
+                            try:
+                                # ping
+                                ping = await ws.ping()
+                                await asyncio.wait_for(ping, timeout=2)
+                                logger.debug(f"Connection to {url} still alive...")
+                                continue
+                            except Exception:
+                                logger.info(
+                                    f"Ping error {url} - retrying in 5s")
+                                asyncio.sleep(2)
+                                break
+
+            except (socket.gaierror, ConnectionRefusedError):
+                logger.info("Connection Refused - retrying connection in 5s")
+                await asyncio.sleep(2)
+                continue
+            except websockets.exceptions.InvalidStatusCode as e:
+                logger.error(f"Connection Refused - {e}")
+                await asyncio.sleep(2)
+                continue
+
+    except (asyncio.CancelledError, KeyboardInterrupt):
+        pass
+
+
+def main():
+    asyncio.run(_client())
+
+
+if __name__ == "__main__":
+    main()