casbin_util.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package utils
  2. import (
  3. "sync"
  4. "github.com/casbin/casbin/v2"
  5. "github.com/casbin/casbin/v2/model"
  6. gormadapter "github.com/casbin/gorm-adapter/v3"
  7. "github.com/flipped-aurora/gin-vue-admin/server/global"
  8. "go.uber.org/zap"
  9. )
  10. var (
  11. syncedCachedEnforcer *casbin.SyncedCachedEnforcer
  12. once sync.Once
  13. )
  14. // GetCasbin 获取casbin实例
  15. func GetCasbin() *casbin.SyncedCachedEnforcer {
  16. once.Do(func() {
  17. a, err := gormadapter.NewAdapterByDB(global.GVA_DB)
  18. if err != nil {
  19. zap.L().Error("适配数据库失败请检查casbin表是否为InnoDB引擎!", zap.Error(err))
  20. return
  21. }
  22. text := `
  23. [request_definition]
  24. r = sub, obj, act
  25. [policy_definition]
  26. p = sub, obj, act
  27. [role_definition]
  28. g = _, _
  29. [policy_effect]
  30. e = some(where (p.eft == allow))
  31. [matchers]
  32. m = r.sub == p.sub && keyMatch2(r.obj,p.obj) && r.act == p.act
  33. `
  34. m, err := model.NewModelFromString(text)
  35. if err != nil {
  36. zap.L().Error("字符串加载模型失败!", zap.Error(err))
  37. return
  38. }
  39. syncedCachedEnforcer, _ = casbin.NewSyncedCachedEnforcer(m, a)
  40. syncedCachedEnforcer.SetExpireTime(60 * 60)
  41. _ = syncedCachedEnforcer.LoadPolicy()
  42. })
  43. return syncedCachedEnforcer
  44. }