1 Star 3 Fork 21

go-course / keyauth-g7

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
interface.go 2.05 KB
一键复制 编辑 原始数据 按行查看 历史
Mr.Yu 提交于 2022-06-18 17:12 . add allow all
package role
import (
"context"
"time"
"github.com/go-playground/validator/v10"
"github.com/infraboard/mcube/http/request"
"github.com/rs/xid"
)
const (
AppName = "role"
)
var (
validate = validator.New()
)
type Service interface {
CreateRole(context.Context, *CreateRoleRequest) (*Role, error)
RPCServer
}
func NewQueryRoleRequestWithName(names []string) *QueryRoleRequest {
return &QueryRoleRequest{
Page: request.NewDefaultPageRequest(),
RoleNames: names,
}
}
func (s *RoleSet) HasPermission(req *PermissionRequest) (bool, *Role) {
for i := range s.Items {
if s.Items[i].HasPermission(req) {
return true, s.Items[i]
}
}
return false, nil
}
// 单个角色如何判断有没有权限
func (r *Role) HasPermission(req *PermissionRequest) bool {
for i := range r.Spec.Permissions {
if r.Spec.Permissions[i].HasPermission(req) {
return true
}
}
return false
}
//
func (p *Permission) HasPermission(req *PermissionRequest) bool {
// 确认是不是同一个服务
if p.Service != req.Service {
return false
}
// 放行所有功能
if p.AllowAll {
return true
}
// 查询所有的功能,确认是否允许
for i := range p.Featrues {
f := p.Featrues[i]
if f.Resource == req.Resource && f.Action == req.Action {
return true
}
}
return false
}
func NewRoleSet() *RoleSet {
return &RoleSet{
Items: []*Role{},
}
}
func (s *RoleSet) Add(item *Role) {
s.Items = append(s.Items, item)
}
func NewPermissionRequest(service, resource, action string) *PermissionRequest {
return &PermissionRequest{
Service: service,
Resource: resource,
Action: action,
}
}
func (req *CreateRoleRequest) Validate() error {
return validate.Struct(req)
}
func NewRole(req *CreateRoleRequest) *Role {
return &Role{
Id: xid.New().String(),
CreateAt: time.Now().UnixMilli(),
Spec: req,
}
}
func NewCreateRoleRequest() *CreateRoleRequest {
return &CreateRoleRequest{
Permissions: []*Permission{},
Meta: map[string]string{},
}
}
func NewDefaultRole() *Role {
return &Role{
Spec: NewCreateRoleRequest(),
}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/go-course/keyauth-g7.git
git@gitee.com:go-course/keyauth-g7.git
go-course
keyauth-g7
keyauth-g7
v0.0.7

搜索帮助

344bd9b3 5694891 D2dac590 5694891