<?php
declare(strict_types=1);

namespace app\mcp\service;

use think\facade\Cache;
use think\facade\Log;

/**
 * 基于系统缓存的固定窗口限流（缓存驱动为 redis 时计数更准确；文件缓存下为近似值）。
 * 计数只是保护措施：缓存读写出错（如文件缓存在并发调用时读到写了一半的文件）时放行并记日志，
 * 不能让记账失败把正常查询变成“内部错误”。
 */
class RateLimiter
{
    /** 记一次并判断是否仍在限额内 */
    public static function hit(string $key, int $limit, int $windowSeconds): bool
    {
        try {
            $bucket = 'ai_mcp_rl_' . $key . '_' . intdiv(time(), $windowSeconds);
            $count = (int) Cache::get($bucket, 0) + 1;
            Cache::set($bucket, $count, $windowSeconds * 2);
            return $count <= $limit;
        } catch (\Throwable $e) {
            Log::warning('[ai_mcp] 限流计数失败，本次放行: ' . $e->getMessage());
            return true;
        }
    }

    public static function rowsToday(int $adminId): int
    {
        try {
            return (int) Cache::get(self::rowsKey($adminId), 0);
        } catch (\Throwable $e) {
            Log::warning('[ai_mcp] 读取今日行数失败: ' . $e->getMessage());
            return 0;
        }
    }

    public static function addRows(int $adminId, int $rows): void
    {
        if ($rows <= 0) {
            return;
        }
        try {
            Cache::set(self::rowsKey($adminId), self::rowsToday($adminId) + $rows, 90000);
        } catch (\Throwable $e) {
            Log::warning('[ai_mcp] 记录今日行数失败: ' . $e->getMessage());
        }
    }

    private static function rowsKey(int $adminId): string
    {
        return 'ai_mcp_rows_' . $adminId . '_' . date('Ymd');
    }
}