Casbin之自定函数

最新推荐文章于 2023-12-02 14:56:10 发布

~庞贝

最新推荐文章于 2023-12-02 14:56:10 发布

阅读量199

点赞数

分类专栏： Casbin 文章标签： golang

本文链接：https://blog.csdn.net/qq_53267860/article/details/126759979

版权

Casbin 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

Casbin之自定函数

1.讲解

2.model.conf文件

[request_definition]
r = sub,obj,act


[policy_definition]
p = sub,obj,act


[policy_effect]
e = some(where (p.eft == allow))

[matchers]
m = r.sub == p.sub && my_func(r.obj, p.obj) && r.act == p.act

3.Policy.csv文件

p, dajun, data/*, read

4.main

package main

import (
	"fmt"
	"github.com/casbin/casbin/v2"
	"log"
	"strings"
)

//我们当然也可以定义自己的函数。先定义一个函数，返回 bool：
func KeyMatch(key1, key2 string) bool {
	i := strings.Index(key2, "*")
	if i == -1 {
		return key1 == key2
	}

	if len(key1) > i {
		return key1[:i] == key2[:i]
	}

	return key1 == key2[:i]
}

//然后将这个函数用interface{}类型包装一层：
func KeyMatchFunc(args ...interface{}) (interface{}, error) {
	name1 := args[0].(string)
	name2 := args[1].(string)

	return (bool)(KeyMatch(name1, name2)), nil
}
func check(e *casbin.Enforcer, sub, obj, act string) {
	ok, _ := e.Enforce(sub, obj, act)
	if ok {
		fmt.Printf("%s CAN %s %s\n", sub, act, obj)
	} else {
		fmt.Printf("%s CANNOT %s %s\n", sub, act, obj)
	}
}
func main() {
	e, err := casbin.NewEnforcer("./casbin/model.conf", "./casbin/policy.csv")
	//然后添加到权限认证器中：
	e.AddFunction("my_func", KeyMatchFunc)
	if err != nil {
		log.Fatalf("NewEnforecer failed:%v\n", err)
	}

	check(e, "dajun", "data/1", "read")
	check(e, "dajun", "data/2", "read")
	check(e, "dajun", "data/1", "write")
	check(e, "dajun", "mydata", "read")
}