59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
|
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()
|