import asyncio
import json
import websockets
from logger_config import logger
from data_processing import messages


async def ws_inited(ws, path):
    logger.info("客户端已连接上，ws://localhost:6789初始化完毕")
    while True:
        message = await asyncio.get_running_loop().run_in_executor(None, messages.get)

        # 检查连接是否仍然开放
        if ws.open:
            message_data = json.dumps(message)  # 将字典序列化为JSON字符串
            await ws.send(message_data)
        else:
            logger.info("WebSocket 连接已关闭")
            break


async def start_websocket_server():
    server = await websockets.serve(ws_inited, "localhost", 6789)
    logger.info("WebSocket 服务器启动在 ws://localhost:6789")
    await server.wait_closed()