#!/usr/bin/env python3
"""
测试股票名称获取的缓存优化
"""

import sys
from pathlib import Path
import time

# 添加src目录到路径
current_dir = Path(__file__).parent
src_dir = current_dir / "src"
sys.path.insert(0, str(src_dir))

from loguru import logger
from src.data.data_fetcher import ADataFetcher

def test_stock_name_cache():
    """测试股票名称缓存机制"""
    logger.info("🧪 开始测试股票名称缓存优化")

    # 初始化数据获取器
    data_fetcher = ADataFetcher()

    # 测试股票列表
    test_stocks = ['000001.SZ', '000002.SZ', '600000.SH', '600036.SH', '000858.SZ']

    # 第一次获取股票名称（会触发缓存构建）
    logger.info("📊 第一次获取股票名称（构建缓存）...")
    start_time = time.time()

    names_first = {}
    for stock_code in test_stocks:
        name = data_fetcher.get_stock_name(stock_code)
        names_first[stock_code] = name
        logger.info(f"   {stock_code}: {name}")

    first_duration = time.time() - start_time
    logger.info(f"⏱️ 第一次获取耗时: {first_duration:.2f}秒")

    # 等待一秒
    time.sleep(1)

    # 第二次获取股票名称（应该从缓存读取）
    logger.info("📊 第二次获取股票名称（从缓存读取）...")
    start_time = time.time()

    names_second = {}
    for stock_code in test_stocks:
        name = data_fetcher.get_stock_name(stock_code)
        names_second[stock_code] = name
        logger.info(f"   {stock_code}: {name}")

    second_duration = time.time() - start_time
    logger.info(f"⏱️ 第二次获取耗时: {second_duration:.2f}秒")

    # 比较结果
    logger.info("📈 性能对比:")
    if second_duration < first_duration:
        speedup = first_duration / second_duration
        logger.info(f"✅ 缓存优化成功! 第二次比第一次快 {speedup:.1f}x")
    else:
        logger.warning("❌ 缓存优化效果不明显")

    # 验证数据一致性
    consistent = names_first == names_second
    logger.info(f"🔍 数据一致性: {'✅ 一致' if consistent else '❌ 不一致'}")

    # 显示缓存状态
    logger.info(f"📦 当前缓存中的股票数量: {len(data_fetcher._stock_name_cache)}")

    return first_duration, second_duration, consistent

if __name__ == "__main__":
    # 设置日志
    logger.remove()
    logger.add(sys.stdout, level="INFO", format="{time:HH:mm:ss} | {level} | {message}")

    print("=" * 60)
    print("🧪 股票名称缓存优化测试")
    print("=" * 60)

    try:
        first_time, second_time, is_consistent = test_stock_name_cache()

        print("\n" + "=" * 60)
        print("📊 测试结果总结:")
        print(f"   第一次获取耗时: {first_time:.2f}秒")
        print(f"   第二次获取耗时: {second_time:.2f}秒")
        print(f"   性能提升倍数: {first_time/second_time:.1f}x")
        print(f"   数据一致性: {'✅ 通过' if is_consistent else '❌ 失败'}")
        print("=" * 60)

    except Exception as e:
        logger.error(f"测试过程中发生错误: {e}")
        import traceback
        traceback.print_exc()