Yii2 最简单的实现管理员才能登录后台

场景

有时候一个小项目，想实管理员才能登录后台，其实没必要引入复杂的 RBAC，可以利用 Yii User 表里面的 role 字段来实现这个效果。

实战：后台只有管理员才能登录

第一步：修改 common\models\User 文件，添加如下静态常量：

const ROLE_USER = 10;const ROLE_ADMIN = 20;

第二步：修改 common\models\User 文件，修改 rules 规则：

['role', 'default', 'value' => 10],['role', 'in', 'range' => [self::ROLE_USER, self::ROLE_ADMIN]],

第三步：修改 common\models\User 文件，添加判断是否是超级管理员的 isUserAdmin 静态方法：

/** * @param $username * @return bool */public static function isUserAdmin($username){    if (static::findOne(['username' => $username, 'role' => self::ROLE_ADMIN])) {        return true;    } else {        return false;    }}

第四步：修改 common\models\LoginForm 文件，添加 loginAdmin 方法：

/** * @return bool */public function loginAdmin(){   if ($this->validate()) {        if (User::isUserAdmin($this->username)) {            return Yii::$app->user->login($this->getUser(), $this->rememberMe ? 3600 * 24 * 30 : 0);        }        $this->addError('username', 'You don\'t have permission to login.');    } else {        $this->addError('password', Yii::t('app', 'Incorrect username or password.'));    }    return false;}

第五步：修改 backend\controllers\SiteController.php, 修改 actionLogin 方法：

把 $model->login() 改成 $model->loginAdmin()。

public function actionLogin(){    if (!\Yii::$app->user->isGuest) {        return $this->goHome();    }    $model = new LoginForm();    if ($model->load(Yii::$app->request->post()) && $model->loginAdmin()) {        return $this->goBack();    } else {        return $this->render('login', [            'model' => $model,        ]);    }}

第六步：未登录用户只能访问登录页面

实现方式一：新建 backend\controllers\Controller.php 文件，代码如下：

除了 SiteController其他控制器都要继承此控制器，方可实现我们最终效果，切换其他控制器要使用 behaviors 的时候记得要ArrayHelper::merge 父类 behaviors。

<?php/** * author     : forecho <[e-mail]> * createTime : 2016/3/10 14:39 * description: */namespace backend\controllers;use yii\filters\AccessControl;class Controller extends \yii\web\Controller{    public function behaviors()    {        return [            // 后台必须登录才能使用            'access' => [                'class' => AccessControl::className(),                'rules' => [                    [                        'allow' => true,                        'roles' => ['@'],                    ],                ],            ],        ];    }}

实现方式二：新建 common\components\AccessControl.php 文件，代码如下：

<?php/** * author     : forecho <[e-mail]> * createTime : 16/8/14 下午10:38 * description: */namespace common\components;use Yii;use yii\web\User;use yii\di\Instance;class AccessControl extends \yii\base\ActionFilter{    /**     * @var User User for check access.     */    private $_user = 'user';    /**     * Get user     * @return User     */    public function getUser()    {        if (!$this->_user instanceof User) {            $this->_user = Instance::ensure($this->_user, User::className());        }        return $this->_user;    }    /**     * Set user     * @param User|string $user     */    public function setUser($user)    {        $this->_user = $user;    }    /**     * @inheritdoc     */    public function beforeAction($action)    {        $user = $this->getUser();        return $this->denyAccess($user);    }    /**     * @param User|string $user     * @return bool     */    protected function denyAccess($user)    {        if ($user->getIsGuest()) {            $user->loginRequired();        }        return true;    }    /**     * @param \yii\base\Action $action     * @return bool     */    protected function isActive($action)    {        $uniqueId = $action->getUniqueId();        if ($uniqueId === Yii::$app->getErrorHandler()->errorAction) {            return false;        }        $user = $this->getUser();        if ($user->getIsGuest() && is_array($user->loginUrl) && isset($user->loginUrl[0]) && $uniqueId === trim($user->loginUrl[0], '/')) {            return false;        }        return true;    }}

修改 `` 配置文件，添加代码：

'as access' => [    'class' => 'common\components\AccessControl',],'components' => [    // ···],

本文由 forecho 创作，出自https://getyii.com/topic/585。