76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package casbin
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strconv"
|
|
"sync"
|
|
|
|
"github.com/casbin/casbin/v2"
|
|
gormadapter "github.com/casbin/gorm-adapter/v3"
|
|
"pet-house.com/core/g"
|
|
"pet-house.com/core/helper/dir"
|
|
"pet-house.com/core/server/database"
|
|
"pet-house.com/core/server/zap_server"
|
|
)
|
|
|
|
var (
|
|
once sync.Once
|
|
enforcer *casbin.Enforcer
|
|
)
|
|
|
|
// Instance casbin instance
|
|
func Instance() *casbin.Enforcer {
|
|
once.Do(func() {
|
|
enforcer = getEnforcer()
|
|
})
|
|
return enforcer
|
|
}
|
|
|
|
// getEnforcer get casbin.Enforcer
|
|
func getEnforcer() *casbin.Enforcer {
|
|
if database.Instance() == nil {
|
|
zap_server.ZAPLOG.Error(database.ErrDatabaseInit.Error())
|
|
return nil
|
|
}
|
|
c, err := gormadapter.NewAdapterByDBUseTableName(database.Instance(), "", "casbin_rule") // Your driver and data source.
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
enforcer, err := casbin.NewEnforcer(filepath.Join(dir.GetCurrentAbPath(), g.CasbinFileName), c)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
if enforcer == nil {
|
|
zap_server.ZAPLOG.Error("Casbin init")
|
|
return nil
|
|
}
|
|
|
|
err = enforcer.LoadPolicy()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
return enforcer
|
|
}
|
|
|
|
// GetRolesForUser get user's roles
|
|
func GetRolesForUser(uid uint) []string {
|
|
uids, err := Instance().GetRolesForUser(strconv.FormatUint(uint64(uid), 10))
|
|
if err != nil {
|
|
return []string{}
|
|
}
|
|
|
|
return uids
|
|
}
|
|
|
|
// ClearCasbin clean rules
|
|
func ClearCasbin(v int, p ...string) error {
|
|
_, err := Instance().RemoveFilteredPolicy(v, p...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|