如何在Yii2中全局声明behaviors方法，实现全控制器限流

嘿，要让所有控制器都用上这个限流功能，咱们不用在每个控制器里重复写behaviors方法，最优雅的方式是通过基础控制器类来全局复用。下面一步步来操作：

1. 创建全局基础控制器 #

在你的项目controllers目录下新建一个BaseController.php文件，这个类会作为所有业务控制器的父类。代码如下：

<?php

namespace app\controllers;

use yii\web\Controller;
use yii\filters\RateLimiter;

class BaseController extends Controller
{
    public function behaviors()
    {
        $behaviors = parent::behaviors();
        
        // 全局添加限流Behavior
        $behaviors['rateLimiter'] = [
            'class' => RateLimiter::className(),
            'rateLimit' => 5,
            'timePeriod' => 10,
            'separateRates' => false,
            'enableRateLimitHeaders' => false,
        ];
        
        return $behaviors;
    }
}

2. 让所有业务控制器继承基础控制器 #

比如你的SiteController、UserController等，都改成继承BaseController而不是原来的Controller：

<?php

namespace app\controllers;

class SiteController extends BaseController // 这里改成继承BaseController
{
    // 你的控制器逻辑...
}

3. （可选）针对单个控制器自定义限流规则 #

如果某个控制器需要单独调整限流参数，只需要在该控制器里重写behaviors方法，覆盖全局配置即可：

public function behaviors()
{
    $behaviors = parent::behaviors();
    
    // 自定义当前控制器的限流规则，比如把请求次数改成10次/10秒
    $behaviors['rateLimiter']['rateLimit'] = 10;
    
    return $behaviors;
}

注意事项 #

别忘了确保你的用户模型（实现IdentityInterface的类）已经实现了RateLimiter需要的三个方法：getRateLimit()、loadAllowance()、saveAllowance()，否则限流功能不会正常工作。示例参考：

use yii\web\IdentityInterface;

class User extends \yii\db\ActiveRecord implements IdentityInterface
{
    // ...其他已实现的方法
    
    public function getRateLimit($request, $action)
    {
        return [$this->rateLimit, 60]; // 这里可以和全局配置对应，或者单独设置
    }

    public function loadAllowance($request, $action)
    {
        return [$this->allowance, $this->allowance_updated_at];
    }

    public function saveAllowance($request, $action, $allowance, $timestamp)
    {
        $this->allowance = $allowance;
        $this->allowance_updated_at = $timestamp;
        $this->save(false);
    }
}

内容的提问来源于stack exchange，提问作者Vijay Makwana