#!/usr/bin/env python3
"""
测试 WebSocket 价格监控服务
"""
import sys
import os
import asyncio
import threading
import time

# 确保路径正确
script_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(script_dir)
backend_dir = os.path.join(project_root, 'backend')
sys.path.insert(0, backend_dir)

from app.services.websocket_monitor import get_ws_price_monitor
from app.utils.logger import logger


# 用于接收价格更新的队列
price_updates = []


def on_price_update(symbol: str, price: float):
    """价格更新回调"""
    price_updates.append((symbol, price))
    print(f"📊 [回调] {symbol}: ${price:,.2f}")


async def main():
    print("=" * 60)
    print("🔌 测试 WebSocket 价格监控服务")
    print("=" * 60)

    ws_monitor = get_ws_price_monitor()

    # 订阅几个交易对
    symbols = ['BTCUSDT', 'ETHUSDT']

    print(f"\n订阅交易对: {', '.join(symbols)}")
    for symbol in symbols:
        print(f"  订阅 {symbol}...")
        ws_monitor.subscribe_symbol(symbol)

    # 注册回调
    print("注册价格回调...")
    ws_monitor.add_price_callback(on_price_update)

    print("\n等待价格推送（30秒）...")
    print("提示: WebSocket 连接可能需要几秒钟建立...")
    print("-" * 60)

    # 给 WebSocket 线程一些时间启动
    await asyncio.sleep(2)

    # 检查状态
    print(f"\n📡 WebSocket 运行状态: {ws_monitor.is_running()}")
    print(f"📡 已订阅交易对: {ws_monitor.get_subscribed_symbols()}")
    print(f"📡 WebSocket 线程存活: {ws_monitor._thread.is_alive() if ws_monitor._thread else 'None'}")
    print(f"📡 事件循环: {'已创建' if ws_monitor._loop else '未创建'}")
    print("-" * 60)

    # 等待价格更新
    start_time = time.time()
    last_price_count = 0

    for i in range(30):
        await asyncio.sleep(1)

        # 每5秒打印一次状态
        if (i + 1) % 5 == 0:
            elapsed = time.time() - start_time
            current_count = len(price_updates)
            new_updates = current_count - last_price_count
            last_price_count = current_count

            print(f"\n⏱️  已运行 {i+1} 秒 | 收到 {current_count} 次价格更新 (最近5秒: +{new_updates})")
            print(f"   WebSocket 状态: {'🟢 运行中' if ws_monitor.is_running() else '🔴 未运行'}")

            # 显示当前缓存的价格
            for symbol in symbols:
                price = ws_monitor.get_latest_price(symbol)
                if price:
                    print(f"   📊 {symbol}: ${price:,.2f}")

    # 显示获取到的价格
    print("\n" + "=" * 60)
    print("📊 最终结果:")
    print("=" * 60)
    print(f"   共收到 {len(price_updates)} 次价格更新")
    for symbol in symbols:
        price = ws_monitor.get_latest_price(symbol)
        if price:
            print(f"   {symbol}: ${price:,.2f}")
        else:
            print(f"   {symbol}: 未获取到价格")

    # 停止服务
    print("\n停止服务...")
    ws_monitor.stop()

    # 等待线程结束
    if ws_monitor._thread:
        ws_monitor._thread.join(timeout=3)
        print(f"线程已结束: {not ws_monitor._thread.is_alive()}")

    print("=" * 60)
    print("✅ 测试完成")
    print("=" * 60)


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n\n⚠️  测试中断")
        sys.exit(0)