mirror of
https://github.com/umap-project/umap.git
synced 2025-05-06 06:21:49 +02:00
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import asyncio
|
|
import websockets
|
|
from websockets.server import serve
|
|
import json
|
|
|
|
# Just relay all messages to other connected peers for now
|
|
|
|
CONNECTIONS = set()
|
|
|
|
async def join_and_listen(websocket):
|
|
print(f"Someone joined: {id(websocket)}")
|
|
CONNECTIONS.add(websocket)
|
|
try:
|
|
async for message in websocket:
|
|
# recompute the peers-list at the time of message-sending.
|
|
# doing so beforehand would miss new connections
|
|
peers = CONNECTIONS - {websocket}
|
|
print("message", message)
|
|
print("peers", peers)
|
|
websockets.broadcast(peers, message)
|
|
finally:
|
|
CONNECTIONS.remove(websocket)
|
|
|
|
|
|
async def handler(websocket):
|
|
message = await websocket.recv()
|
|
event = json.loads(message)
|
|
|
|
# The first event should always be 'join'
|
|
assert event["kind"] == "join"
|
|
await join_and_listen(websocket)
|
|
|
|
async def main():
|
|
print("WS server started, will relay messages")
|
|
async with serve(handler, "localhost", 8001):
|
|
await asyncio.Future() # run forever
|
|
|
|
asyncio.run(main())
|