amazing-data-service/backend/app/api/v1/cache.py

"""
缓存管理路由
"""
from typing import List
from fastapi import APIRouter, Depends, Query
from sqlalchemy.orm import Session

from app.db.session import get_db
from app.schemas.base import ResponseModel, PaginatedResponse
from app.schemas.cache import (
    DetectMissingRequest, DetectMissingResponse,
    BatchCacheRequest, CacheTaskResponse, CacheStatusResponse
)
from app.services.cache_service import CacheService
from app.core.security import get_current_user
from app.models.user import User
from app.utils.date_utils import parse_date

router = APIRouter()


@router.post("/detect-missing", response_model=ResponseModel)
async def detect_missing_data(
    request: DetectMissingRequest,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """检测缺失数据"""
    service = CacheService(db)
    start = parse_date(request.start_date)
    end = parse_date(request.end_date)
    
    task = service.detect_missing_data(
        request.security_type,
        request.period_type,
        start,
        end,
        request.code_list
    )
    
    # 获取缺失详情
    details = service.get_task_details(task.id)
    missing_codes = [d for d in details if d.is_missing]
    
    missing_info = []
    for code in request.code_list:
        code_details = [d for d in details if d.code == code and d.is_missing]
        if code_details:
            missing_info.append({
                "code": code,
                "missing_dates": [{
                    "date": format_date(d.trade_date),
                    "expected": d.expected_count,
                    "actual": d.actual_count,
                    "missing_ratio": (d.expected_count - d.actual_count) / d.expected_count if d.expected_count > 0 else 0
                } for d in code_details]
            })
    
    return ResponseModel(data={
        "task_id": task.id,
        "total_codes": len(request.code_list),
        "missing_codes": missing_info
    })


@router.post("/batch-cache", response_model=ResponseModel[CacheTaskResponse])
async def batch_cache_data(
    request: BatchCacheRequest,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """批量缓存数据"""
    service = CacheService(db)
    start = parse_date(request.start_date)
    end = parse_date(request.end_date)
    
    task = service.batch_cache_data(
        request.security_type,
        request.period_type,
        start,
        end,
        request.code_list
    )
    
    return ResponseModel(data=CacheTaskResponse.model_validate(task))


@router.get("/tasks", response_model=ResponseModel)
async def get_cache_tasks(
    page: int = Query(1, ge=1),
    page_size: int = Query(20, ge=1, le=100),
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """获取缓存任务列表"""
    service = CacheService(db)
    result = service.get_tasks(page, page_size)
    
    return ResponseModel(data={
        "items": [CacheTaskResponse.model_validate(t) for t in result["items"]],
        "total": result["total"],
        "page": result["page"],
        "page_size": result["page_size"],
        "total_pages": result["total_pages"]
    })


@router.get("/tasks/{task_id}", response_model=ResponseModel)
async def get_cache_task(
    task_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """获取缓存任务详情"""
    service = CacheService(db)
    task = service.get_task(task_id)
    
    if not task:
        return ResponseModel(code=404, message="任务不存在")
    
    details = service.get_task_details(task_id)
    
    return ResponseModel(data={
        "task": CacheTaskResponse.model_validate(task),
        "details": [{
            "id": d.id,
            "code": d.code,
            "trade_date": d.trade_date.isoformat() if d.trade_date else None,
            "expected_count": d.expected_count,
            "actual_count": d.actual_count,
            "is_missing": bool(d.is_missing),
            "status": d.status,
            "error_message": d.error_message
        } for d in details]
    })


@router.delete("/tasks/{task_id}", response_model=ResponseModel)
async def cancel_cache_task(
    task_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """取消缓存任务"""
    service = CacheService(db)
    success = service.cancel_task(task_id)
    
    if success:
        return ResponseModel(message="任务已取消")
    else:
        return ResponseModel(code=400, message="任务不存在或已完成")


@router.get("/status/{code}", response_model=ResponseModel)
async def get_cache_status(
    code: str,
    security_type: str = Query("stock", description="证券类型: stock, future"),
    period_type: str = Query("daily", description="周期类型: daily, min1, min5, etc."),
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """获取代码缓存状态"""
    service = CacheService(db)
    status = service.get_cache_status(code, security_type, period_type)
    
    return ResponseModel(data=status)


from app.utils.date_utils import format_date