43 lines
797 B
Go
43 lines
797 B
Go
package common
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/eryajf/go-ldap-admin/config"
|
||
|
||
"github.com/casbin/casbin/v2"
|
||
gormadapter "github.com/casbin/gorm-adapter/v3"
|
||
)
|
||
|
||
// 全局CasbinEnforcer
|
||
var CasbinEnforcer *casbin.Enforcer
|
||
|
||
// 初始化casbin策略管理器
|
||
func InitCasbinEnforcer() {
|
||
e, err := mysqlCasbin()
|
||
if err != nil {
|
||
Log.Panicf("初始化Casbin失败：%v", err)
|
||
panic(fmt.Sprintf("初始化Casbin失败：%v", err))
|
||
}
|
||
|
||
CasbinEnforcer = e
|
||
Log.Info("初始化Casbin完成!")
|
||
}
|
||
|
||
func mysqlCasbin() (*casbin.Enforcer, error) {
|
||
a, err := gormadapter.NewAdapterByDB(DB)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
e, err := casbin.NewEnforcer(config.Conf.Casbin.ModelPath, a)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
err = e.LoadPolicy()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return e, nil
|
||
}
|