Function sync_token_stats_to_db

app/utils/token_pool.py:638–685

将内存中的 Token 统计同步到数据库 应在服务关闭或定期调用，确保统计数据不丢失

()

async def sync_token_stats_to_db():
    """
    将内存中的 Token 统计同步到数据库

    应在服务关闭或定期调用，确保统计数据不丢失
    """
    pool = get_token_pool()
    if not pool:
        return

    from app.services.token_dao import get_token_dao

    dao = get_token_dao()

    pending_updates = []
    with pool._lock:
        for token, status in pool.token_statuses.items():
            pending_success = max(
                0,
                status.successful_requests - status.db_synced_successful_requests,
            )
            pending_failure = max(
                0,
                status.failed_requests - status.db_synced_failed_requests,
            )
            if pending_success > 0 or pending_failure > 0:
                pending_updates.append(
                    (
                        token,
                        status.token_id,
                        pending_success,
                        pending_failure,
                    )
                )

    for token, token_id, pending_success, pending_failure in pending_updates:
        for _ in range(pending_success):
            await dao.record_success(token_id)
        for _ in range(pending_failure):
            await dao.record_failure(token_id)

        with pool._lock:
            if token in pool.token_statuses:
                status = pool.token_statuses[token]
                status.db_synced_successful_requests += pending_success
                status.db_synced_failed_requests += pending_failure

    logger.info("✅ Token 统计已同步到数据库")