52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
import websockets
|
|
from websockets import WebSocketServerProtocol, broadcast
|
|
|
|
from settings import settings
|
|
|
|
connected_clients = set()
|
|
|
|
|
|
def setup_logging() -> None:
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format="%(asctime)s %(levelname)s - %(message)s",
|
|
)
|
|
|
|
|
|
async def ws_handler(websocket: WebSocketServerProtocol):
|
|
connected_clients.add(websocket)
|
|
logging.info(f"Add client: {websocket}")
|
|
|
|
try:
|
|
async for message in websocket:
|
|
print(">>>>>>>>>>", message)
|
|
broadcast(connected_clients, message)
|
|
finally:
|
|
connected_clients.remove(websocket)
|
|
logging.info(f"Remove client: {websocket}")
|
|
|
|
|
|
async def main():
|
|
setup_logging()
|
|
|
|
logging.info(
|
|
f"Starting FairHopper Websockets Server on {settings.ws_server.HOST}:{settings.ws_server.PORT}"
|
|
)
|
|
|
|
async with websockets.serve(
|
|
ws_handler=ws_handler,
|
|
host=settings.ws_server.HOST,
|
|
port=settings.ws_server.PORT,
|
|
):
|
|
await asyncio.Future() # run forever
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except (KeyboardInterrupt, SystemExit):
|
|
logging.info(f"FairHopper Websockets Server terminated")
|