PHP限制函数调用：装饰器与Redis实现

class FunctionCallLimiter 
{
    private $redis;
    private $callback;
    private $key;
    private $limit;
    private $ttl;
<pre class="brush:php;toolbar:false;">public function __construct($redis, callable $callback, string $key, int $limit, int $ttl = 3600)
{
    $this->redis = $redis;
    $this->callback = $callback;
    $this->key = $key;
    $this->limit = $limit;
    $this->ttl = $ttl;
}

public function __invoke(...$args)
{
    $count = $this->redis->get($this->key);
    if ($count >= $this->limit) {
        error_log("Function call limit exceeded for key: {$this->key}");
        throw new RuntimeException("调用次数超限");
    }

    // 记录调用日志
    $this->redis->incr($this->key);
    $this->redis->expire($this->key, $this->ttl);

    // 写入调用日志（可选：记录时间、参数等）
    $logKey = "log:" . $this->key;
    $this->redis->rPush($logKey, json_encode([
        'time' => date('Y-m-d H:i:s'),
        'args' => $args
    ]));
    $this->redis->expire($logKey, $this->ttl * 2); // 日志保留稍久

    return call_user_func_array($this->callback, $args);
}

}

$redis = new Redis();
$redis->connect('127.0.0.1', 6379);
<p>// 被限制的函数
$sendEmail = function($to, $subject) {
echo "发送邮件至: $to, 主题: $subject\n";
// 实际发送逻辑...
};</p><p>// 用户唯一标识（如用户ID + 函数名）
$userId = 12345;
$key = "user:{$userId}:send_email";
$limit = 5;
$ttl = 3600; // 1小时</p><p>// 使用装饰器包装函数
$limitedSendEmail = new FunctionCallLimiter($redis, $sendEmail, $key, $limit, $ttl);</p><p>// 调用（模拟多次）
try {
$limitedSendEmail('test@example.com', '欢迎信');
} catch (RuntimeException $e) {
echo "错误: " . $e->getMessage() . "\n";
}</p>

$logKey = "log:user:12345:send_email";
$logs = $redis->lRange($logKey, 0, -1);
foreach ($logs as $log) {
    print_r(json_decode($log, true));
}