diff options
Diffstat (limited to 'resources/external/websocket_test.py')
| -rw-r--r-- | resources/external/websocket_test.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/resources/external/websocket_test.py b/resources/external/websocket_test.py index 9c29fcf..25ebcff 100644 --- a/resources/external/websocket_test.py +++ b/resources/external/websocket_test.py @@ -3,16 +3,25 @@ """Echo server using the asyncio API.""" import asyncio +import sys from websockets.asyncio.server import serve DEFAULT_PORT = 8181 DEFAULT_IP = "127.0.0.1" +async def delayed_func(): + await asyncio.sleep(5) + print("after delay") async def echo(websocket): print("client connected") - async for message in websocket: - await websocket.send(message) + try: + async for message in websocket: + if message == "Hello world!": + asyncio.run(delayed_func()) + await websocket.send(message) + finally: + print("client DC") async def main(ip, port): @@ -35,6 +44,6 @@ if __name__ == "__main__": PORT = int(args[arg_i]) except: print("Port must be an integer.") - return + quit() asyncio.run(main(IP, PORT))
\ No newline at end of file |