最近想用beego框架想开发一个定时任务管理的系统,发现用户权限这里需要一个rabc的功能。我之前一直是PHP开发,用yii已经4年,所以基于yii框架的rabc功能,用go也写了一个beego框架的插件。目前刚写完,欢迎大家使用吐槽,后期会在2.0的版本增加菜单和权限缓存的功能。。话不多说,来看看目前的功能~~
GRABC插件包括路由、权限、角色、用户角色绑定的功能。路由功能会自动扫描controller里面的方法,然后根据需要将路由分配给权限,权限分配给角色,角色分配给用户。grabc功能简单实用,在beego框架中安装只需要不到20行的代码,即可实现rabc的功能。
安装:
go get github.com/codyi/grabc
配置:
第一步:在你项目中的数据库中导入rabc.sql,生成对应数据表,用于保存路由,权限,角色的数据
第二步:在项目中引入grabc库(可以在项目中的main.go或router.go中引入)
在router.go或main.go中配置如下:
//引入grabc库
import "github.com/codyi/grabc"
func init() {
//将路由注册到grabc,用于反射出对应的网址
grabc.RegisterController(&controllers.SiteController{})
grabc.RegisterController(&controllers.UserController{})
//注册用户系统模型到grabc,用于用户ID和grabc插件绑定
//注意:注册的这个用户模型,需要实现IUserModel中的方法
grabc.RegisterUserModel(&models.User{})
//增加忽律权限检查的页面
grabc.AppendIgnoreRoute("site", "login")
//403页面地址注册到grabc中,用于grabc插件禁止权限的页面跳转
grabc.Http_403("/site/nopermission")
//设置模板,为了让grabc更具有通用性,可以设置模板
//目前设置模板只支持传入模板的内容,也可以使用默认的模板
grabc.SetLayout(libs.Grabc_layout, nil)
}
添加好上面的配置之后,剩下就是在controller中增加权限判了,个人建议做一个BaseController,然后每个controller都继承这个base,然后在BaseController中的Prepare方法中增加grabc的权限检查~~
BaseController.go中Prepare方法中配置如下
//注册当前登录的用户,注意:user需要继承IUserIdentify接口
grabc.RegisterIdentify(user)
if !grabc.CheckAccess(this.controllerName, this.actionName) {
this.redirect(this.URLFor("SiteController.NoPermission"))
}
到此grabc的功能都加完了,是不是很简单~~~
注意:
增加完权限判断之后,会发现很多页面都不能访问了,那么就在忽律权限中增加如下代码
grabc.AppendIgnoreRoute("*", "*")
以上配置将会忽律所有的权限检查,这时候需要去/route/index中增加路由,然后添加权限,角色和用户分配,都配置好之后,就可以将grabc.AppendIgnoreRoute("*", "*")代码删掉,然后重启项目~~权限起作用了
接口说明:
IUserModel接口
//用于定义用户model
type IUserModel interface {
//用户列表返回可用用户的id和姓名
//参数:pageIndex 分页的页数
//参数:pageCount 每页显示的用户数量
//返回值:userList [用户ID]用户姓名,用户列表展示
//返回值:totalNum 全部的用户数目,用于计算分页的数量
//返回值:err
UserList(pageIndex, pageCount int) (userList map[int]string, totalNum int, err error)
//根据用户ID获取用户姓名
FindNameById(id int) string
}
IUserIdentify接口
type IUserIdentify interface {
GetId() int //返回当前登录用户的ID
}
路由扫描不全的问题:
grabc对注册的控制器会进行反射,然后获取每个controller的名称和controller内的公共方法,由于每个controller都继承了beego.Controller,在获取controller下的方法名称时,会将beego.Controller继承的方法也会全部获取到,所以目前还不能区分出方法名到底是beego和用户自己定义的,所以grabc将beego继承的方法都进行了忽律,如果在route扫描中,没有找到自定义的方法,可以在controller中增加如下方法,进行方法返回~~
func (this *SiteController) RABCMethods() []string {
return []string{"Get", "Post"}
}
项目地址:
https://github.com/codyi/grabc
GRABC示例地址:
https://github.com/codyi/grabc_example
有疑问加站长微信联系(非本文作者)