#!/usr/bin/env python3
"""
Caffeine缓存管理模块演示脚本

展示Caffeine缓存的核心功能。
"""

import time
import threading
from core.caffeine_cache import CaffeineCache, cache_manager


def demo_basic_operations():
    """演示基本操作"""
    print("\n" + "="*60)
    print("演示1: 缓存基本操作")
    print("="*60)
    
    cache = CaffeineCache()
    
    # Put操作
    cache.put("user:1", {"name": "张三", "age": 25})
    cache.put("user:2", {"name": "李四", "age": 30})
    print("✅ 添加2个缓存项")
    
    # Get操作
    user1 = cache.get("user:1")
    print(f"✅ 获取user:1 = {user1}")
    
    # Exists操作
    exists = cache.exists("user:1")
    print(f"✅ user:1存在: {exists}")
    
    # Delete操作
    cache.delete("user:1")
    user1_after_delete = cache.get("user:1")
    print(f"✅ 删除后user:1 = {user1_after_delete}")


def demo_expiration():
    """演示过期时间"""
    print("\n" + "="*60)
    print("演示2: 缓存过期时间")
    print("="*60)
    
    cache = CaffeineCache()
    
    # 设置1秒过期时间
    cache.put("temp_data", "临时数据", expire_seconds=1)
    print("✅ 设置1秒过期时间的缓存")
    
    # 立即获取
    value = cache.get("temp_data")
    print(f"✅ 立即获取: {value}")
    
    # 等待1.5秒后获取
    print("⏱️ 等待1.5秒...")
    time.sleep(1.5)
    value_after_expire = cache.get("temp_data")
    print(f"✅ 过期后获取: {value_after_expire}")


def demo_capacity_limit():
    """演示容量限制"""
    print("\n" + "="*60)
    print("演示3: 缓存容量限制(LRU淘汰)")
    print("="*60)
    
    # 创建容量为3的缓存
    cache = CaffeineCache(max_size=3)
    
    # 添加3个缓存项
    cache.put("key1", "value1")
    cache.put("key2", "value2")
    cache.put("key3", "value3")
    print("✅ 添加3个缓存项")
    
    # 访问key1使其最近使用
    cache.get("key1")
    print("✅ 访问key1使其最近使用")
    
    # 添加第4个缓存项
    cache.put("key4", "value4")
    print("✅ 添加第4个缓存项key4")
    
    # 验证key2被移除
    value2 = cache.get("key2")
    value1 = cache.get("key1")
    print(f"✅ key2值: {value2} (应该为None，被LRU淘汰)")
    print(f"✅ key1值: {value1} (应该保留，最近使用)")


def demo_statistics():
    """演示统计信息"""
    print("\n" + "="*60)
    print("演示4: 缓存统计信息")
    print("="*60)
    
    cache = CaffeineCache(record_stats=True)
    
    # 执行缓存操作
    cache.put("key1", "value1")
    cache.get("key1")  # 命中
    cache.get("key1")  # 命中
    cache.get("key2")  # 未命中
    
    # 获取统计信息
    stats = cache.get_stats()
    print(f"✅ 命中次数: {stats['hit_count']}")
    print(f"✅ 未命中次数: {stats['miss_count']}")
    print(f"✅ 命中率: {stats['hit_rate']:.2%}")
    print(f"✅ 缓存大小: {stats['size']}")
    print(f"✅ 最大容量: {stats['max_size']}")


def demo_batch_operations():
    """演示批量操作"""
    print("\n" + "="*60)
    print("演示5: 批量操作")
    print("="*60)
    
    cache = CaffeineCache()
    
    # 批量添加
    data = {
        "batch_key1": "batch_value1",
        "batch_key2": "batch_value2",
        "batch_key3": "batch_value3",
    }
    cache.put_all(data)
    print(f"✅ 批量添加: {list(data.keys())}")
    
    # 批量获取
    keys = ["batch_key1", "batch_key2", "batch_key3"]
    values = cache.get_all(keys)
    print(f"✅ 批量获取: {values}")
    
    # 批量删除
    deleted_count = cache.delete_all(keys)
    print(f"✅ 批量删除: {deleted_count}个")
    
    # 验证删除
    values_after_delete = cache.get_all(keys)
    print(f"✅ 删除后: {values_after_delete}")


def demo_thread_safety():
    """演示线程安全"""
    print("\n" + "="*60)
    print("演示6: 线程安全")
    print("="*60)
    
    cache = CaffeineCache()
    errors = []
    
    def write_data(thread_id: int):
        try:
            for i in range(10):
                cache.put(f"thread_{thread_id}_key_{i}", f"value_{i}")
        except Exception as e:
            errors.append(str(e))
    
    # 创建5个线程并发写入
    threads = []
    for i in range(5):
        t = threading.Thread(target=write_data, args=(i,))
        threads.append(t)
        t.start()
    
    for t in threads:
        t.join()
    
    print(f"✅ 5个线程并发写入完成")
    print(f"✅ 错误数: {len(errors)}")
    
    # 验证数据完整性
    total_keys = 5 * 10
    count = 0
    for thread_id in range(5):
        for i in range(10):
            if cache.get(f"thread_{thread_id}_key_{i}") is not None:
                count += 1
    
    print(f"✅ 成功写入: {count}/{total_keys} 条")


def demo_cache_manager():
    """演示缓存管理器"""
    print("\n" + "="*60)
    print("演示7: 缓存管理器")
    print("="*60)
    
    # 获取命名缓存
    user_cache = cache_manager.get_cache("users", max_size=100, record_stats=True)
    product_cache = cache_manager.get_cache("products", max_size=50, record_stats=True)
    
    # 添加数据
    user_cache.put("user:1", {"name": "张三"})
    product_cache.put("product:1", {"name": "iPhone"})
    
    print("✅ 创建2个命名缓存: users, products")
    print(f"✅ user:1 = {user_cache.get('user:1')}")
    print(f"✅ product:1 = {product_cache.get('product:1')}")
    
    # 获取所有统计信息
    all_stats = cache_manager.get_all_stats()
    print(f"✅ 所有缓存统计: {list(all_stats.keys())}")


def main():
    """主函数"""
    print("\n" + "="*60)
    print("Caffeine缓存管理模块演示")
    print("="*60)
    
    demo_basic_operations()
    demo_expiration()
    demo_capacity_limit()
    demo_statistics()
    demo_batch_operations()
    demo_thread_safety()
    demo_cache_manager()
    
    print("\n" + "="*60)
    print("✅ 所有演示完成!")
    print("="*60)


if __name__ == "__main__":
    main()