33 lines
744 B
Go
33 lines
744 B
Go
|
package mycasbin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"ferry/global/orm"
|
||
|
|
"ferry/tools/config"
|
||
|
|
|
||
|
|
"github.com/casbin/casbin/v2"
|
||
|
|
gormadapter "github.com/casbin/gorm-adapter/v2"
|
||
|
|
"github.com/go-kit/kit/endpoint"
|
||
|
|
_ "github.com/go-sql-driver/mysql"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
var Em endpoint.Middleware
|
||
|
|
|
||
|
|
func Casbin() (*casbin.Enforcer, error) {
|
||
|
|
conn := orm.MysqlConn
|
||
|
|
Apter, err := gormadapter.NewAdapter(config.DatabaseConfig.Dbtype, conn, true)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
e, err := casbin.NewEnforcer("config/rbac_model.conf", Apter)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if err := e.LoadPolicy(); err == nil {
|
||
|
|
return e, err
|
||
|
|
} else {
|
||
|
|
log.Printf("casbin rbac_model or policy init error, message: %v \r\n", err.Error())
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|