欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

程序员文章站 2022-07-12 08:34:56
本文实例讲述了yii2框架中使用rbac对模块,控制器,方法的权限控制及规则的使用。分享给大家供大家参考,具体如下:在使用yii2中自带的rbac时,需要先配置config/web.php:retur...

本文实例讲述了yii2框架中使用rbac对模块,控制器,方法的权限控制及规则的使用。分享给大家供大家参考,具体如下:

在使用yii2中自带的rbac时,需要先配置config/web.php:

return [
  // ...
  'components' => [
    'authmanager' => [
      'class' => 'yii\rbac\dbmanager',
    ],
    // ...
  ],
];

如果你需要运行yii migrate来创建表,那么config/console.php也需要同上面一样配置一下。

cmd进入项目目录,运行如下命令:

yii migrate --migrationpath=@yii/rbac/migrations

你会发现在数据库中创建了四张表

auth_assignment 角色与用户的关联表
auth_item 存放角色与权限,通过type字段区分
auth_item_child 存放角色与权限的上下级关系
auth_rule 规则表,用于扩展权限功能

为了演示,我们在控制器下分别写四个方法,分别用来创建权限,创建角色,指派角色,使用规则。

indexcontroller.php代码如下:

<?php

namespace app\controllers;

use yii;
use app\models\myuserlogin;
use app\rbac\userupdselfrule;
use app\controllers\basecontroller;

class indexcontroller extends basecontroller
{

  //首页
  public function actionindex()
  {
    $this->renderpartial('index');
  }

  //登陆
  public function actionlogin()
  {
    if (yii::$app->request->ispost) {
      $user = new myuserlogin();
      $user->load(yii::$app->request->post(), '');

      if ($user->login()) {
        echo '登陆成功';
      } else {
        echo '登陆失败';
      }

    } else {
      return $this->renderpartial('login');
    }
  }

  //为了演示,这里我们添加几条权限
  public function actionper()
  {
    $auth = yii::$app->authmanager;
    //创建用户删除权限
    $per = $auth->createpermission('user/del');
    $per->description = '删除用户';
    $auth->add($per);
    //创建用户更新权限
    $per = $auth->createpermission('user/upd');
    $per->description = '更新用户';
    $auth->add($per);
    //创建用户添加权限
    $per = $auth->createpermission('user/add');
    $per->description = '添加用户';
    $auth->add($per);
    //创建用户查看权限
    $per = $auth->createpermission('user/list');
    $per->description = '查看用户列表';
    $auth->add($per);
  }

  //添加角色
  public function actionrole()
  {
    $auth = yii::$app->authmanager;

    //添加管理员角色
    $admin = $auth->createrole('admin');
    $admin->description = '管理员';
    $auth->add($admin);
    //给管理员赋予权限
    $auth->addchild($admin, $auth->getpermission('user/del'));
    $auth->addchild($admin, $auth->getpermission('user/upd'));
    $auth->addchild($admin, $auth->getpermission('user/add'));
    $auth->addchild($admin, $auth->getpermission('user/list'));

    //添加普通员工角色
    $employee = $auth->createrole('employee');
    $employee->description = '普通员工';
    $auth->add($employee);
    $auth->addchild($employee, $auth->getpermission('user/list'));
    $auth->addchild($employee, $auth->getpermission('user/add'));
  }

  //给用户指派角色
  public function actionassign()
  {
    $auth = yii::$app->authmanager;

    //注意这里的2是用户的id,即你用户表user里的id
    //也可通过yii::$app->user->id获取
    $auth->assign($auth->getrole('admin'), 1);

    $auth->assign($auth->getrole('employee'), 2);
  }

  //添加规则
  public function actionrule()
  {
    $auth = yii::$app->authmanager;
    $rule = new userupdselfrule();
    $auth->add($rule);

    //创建权限,与规则关联
    $per = $auth->createpermission('user/upd/updself');
    $per->description = '用户只能修改自已';
    $per->rulename = $rule->name;
    $auth->add($per);

    //这里,要注意,要把user/upd/updself权限设为user/upd的父级
    //要不然,普通员工访问user/upd这个方法会被拦住
    $auth->addchild($per, $auth->getpermission('user/upd'));
    //给普通员工赋予user/upd/updself权限,注意我们这里并没有给员工赋予user/upd权限
    $auth->addchild($auth->getrole('employee'), $per);
  }
}

我们在项目目录下创建rbac目录,并创建userupdselfrule.php,来实现用户只能修改自已信息的规则。

<?php

//注意命名空间要跟你的目录对应
namespace app\rbac;

use yii\rbac\rule;

//必须继承自yii\rbac\rule
class userupdselfrule extends rule
{
  public $name = 'userupdself';

  //必须要实现execute方法
  //$user表示用户id
  //$item规则相关的角色或者权限
  //$params传递过来的参数
  public function execute($user, $item, $params)
  {
    //如果没有设置参数id,直接返回true
    if (!isset($params['id'])) {
      return true;
    }
    //判断id是否是当前用户id
    return ($params['id'] == $user) ? true : false;
  }
}

我们访问index/per查看数据表中的变化。

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

访问index/role结果如下:

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

访问index/assign结果如下:

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

访问index/rule结果如下:

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

为了能够对我们的模块,控制器,方法进行权限控制,我们需要创建一个基类来统一处理,上面的控制器就是继承自基类。

basecontroller.php代码如下:

<?php

namespace app\controllers;

use yii;
use yii\web\controller;

class basecontroller extends controller
{
  //不需要验证的
  protected $nocheckaccess = [
    'index/index',
    'index/per',
    'index/role',
    'index/assign',
    'index/rule',
    'index/login',
  ];

  //不需要登陆的
  protected $nologin = [
    'index/login',
  ];

  //验证权限
  //注意,不要把获取模块名,控制器名,方法名写到init()函数里,那样是获取不到的
  //这个坑我已经踩了,大家就不用再去踩了
  public function beforeaction($action)
  {
    $mid = !empty($this->module->id) ? $this->module->id : '';
    $cid = !empty($this->id) ? $this->id : '';
    $aid = !empty($action->id) ? $action->id : '';

    //如果模块为basic,我们只验证控制器/方法
    if ($mid == 'basic') {
      $per = "{$cid}/{$aid}";
    } else {
      $per = "{$mid}/{$cid}/{$aid}";
    }

    if (!in_array($per, $this->nologin)) {
      if (!$this->checkonline()) {
        $this->redirect('index/login');
      }
    }

    if (!in_array($per, $this->nocheckaccess)) {
      if (!yii::$app->user->can($per)) {
        die('你没有权限');
      }
    }

    return parent::beforeaction($action);
  }

  //检查是否在线
  public function checkonline()
  {
    return !empty(yii::$app->user->id) ? true : false;
  }
}

为了演示,我们创建一个usercontroller.php,代码如下:

<?php

namespace app\controllers;

use yii;
use app\controllers\basecontroller;

class usercontroller extends basecontroller
{
  public function actionupd()
  {
    $id = yii::$app->request->get('id', 0);

    echo 'user id : ', yii::$app->user->id, '<br>';

    //先判断用户有没有只能修改自已的权限
    if (yii::$app->user->can('user/upd/updself')) {
      //然后再判断修改id是否与自已的id一样,在userupdselfrule里进行判断
      if (yii::$app->user->can('user/upd/updself', ['id' => $id])) {
        echo '有权修改自已';
      } else {
        echo '不能修改除自已以外的';
      }
    } else {
      echo '修改所有';
    }
  }

  public function actiondel()
  {
    echo 'user id : ', yii::$app->user->id, '<br>';
    echo 'user del';
  }

  public function actionlist()
  {
    echo 'user id : ', yii::$app->user->id, '<br>';
    echo 'user list';
  }

  public function actionadd()
  {
    echo 'user id : ', yii::$app->user->id, '<br>';
    echo 'user add';
  }
}

我的用户表里有两个用户

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

分别登陆这两个用户,然后让他们访问user/add,user/del,user/list,user/upd,结果如下:

admin用户状态如下:

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

test用户状态如下:

YII2框架中使用RBAC对模块,控制器,方法的权限控制及规则的使用示例

test之所以能够访问user/upd是因为我们把user/upd/updself设为了user/upd的父级,如果没有设置,这里是会被拦住的。