llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-05-24 08:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: optimize getEnforcer, only load filtered polices (#1669)

Browse Source
This commit is contained in:
June 2023-03-19 19:13:48 +07:00 committed by GitHub
parent 89d29c2519
commit 4c4ad8320d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
object/permission_enforcer.go
View File

@ -20,6 +20,7 @@ import (
"github.com/casbin/casbin/v2" "github.com/casbin/casbin/v2"
"github.com/casbin/casbin/v2/config" "github.com/casbin/casbin/v2/config"
"github.com/casbin/casbin/v2/log"
"github.com/casbin/casbin/v2/model" "github.com/casbin/casbin/v2/model"
"github.com/casdoor/casdoor/conf" "github.com/casdoor/casdoor/conf"
xormadapter "github.com/casdoor/xorm-adapter/v3" xormadapter "github.com/casdoor/xorm-adapter/v3"
@ -50,27 +51,30 @@ func getEnforcer(permission *Permission) *casbin.Enforcer {
panic(err) panic(err)
} }
policyFilter := xormadapter.Filter{} // Init an enforcer instance without specifying a model or adapter.
// If you specify an adapter, it will load all policies, which is a
if !HasRoleDefinition(m) { // heavy process that can slow down the application.
policyFilter.Ptype = []string{"p"} enforcer, err := casbin.NewEnforcer(&log.DefaultLogger{}, false)
err = adapter.LoadFilteredPolicy(m, policyFilter)
if err != nil {
panic(err)
}
}
enforcer, err := casbin.NewEnforcer(m, adapter)
if err != nil { if err != nil {
panic(err) panic(err)
} }
// load Policy with a specific Permission enforcer.InitWithModelAndAdapter(m, nil)
policyFilter.V5 = []string{permission.GetId()} enforcer.SetAdapter(adapter)
policyFilter := xormadapter.Filter{
V5: []string{permission.GetId()},
}
if !HasRoleDefinition(m) {
policyFilter.Ptype = []string{"p"}
}
err = enforcer.LoadFilteredPolicy(policyFilter) err = enforcer.LoadFilteredPolicy(policyFilter)
if err != nil { if err != nil {
panic(err) panic(err)
} }
return enforcer return enforcer
} }