package system import ( "errors" "sync" "github.com/casbin/casbin/v2" gormadapter "github.com/casbin/gorm-adapter/v3" "github.com/flipped-aurora/gin-vue-admin/server/global" "github.com/flipped-aurora/gin-vue-admin/server/model/system" "github.com/flipped-aurora/gin-vue-admin/server/model/system/request" _ "github.com/go-sql-driver/mysql" ) //@author: [piexlmax](https://github.com/piexlmax) //@function: UpdateCasbin //@description: 更新casbin权限 //@param: authorityId string, casbinInfos []request.CasbinInfo //@return: error type CasbinService struct { } var CasbinServiceApp = new(CasbinService) func (casbinService *CasbinService) UpdateCasbin(authorityId string, casbinInfos []request.CasbinInfo) error { casbinService.ClearCasbin(0, authorityId) rules := [][]string{} for _, v := range casbinInfos { cm := system.CasbinModel{ Ptype: "p", AuthorityId: authorityId, Path: v.Path, Method: v.Method, } rules = append(rules, []string{cm.AuthorityId, cm.Path, cm.Method}) } e := casbinService.Casbin() success, _ := e.AddPolicies(rules) if !success { return errors.New("存在相同api,添加失败,请联系管理员") } return nil } //@author: [piexlmax](https://github.com/piexlmax) //@function: UpdateCasbinApi //@description: API更新随动 //@param: oldPath string, newPath string, oldMethod string, newMethod string //@return: error func (casbinService *CasbinService) UpdateCasbinApi(oldPath string, newPath string, oldMethod string, newMethod string) error { err := global.GVA_DB.Table("casbin_rule").Model(&system.CasbinModel{}).Where("v1 = ? AND v2 = ?", oldPath, oldMethod).Updates(map[string]interface{}{ "v1": newPath, "v2": newMethod, }).Error return err } //@author: [piexlmax](https://github.com/piexlmax) //@function: GetPolicyPathByAuthorityId //@description: 获取权限列表 //@param: authorityId string //@return: pathMaps []request.CasbinInfo func (casbinService *CasbinService) GetPolicyPathByAuthorityId(authorityId string) (pathMaps []request.CasbinInfo) { e := casbinService.Casbin() list := e.GetFilteredPolicy(0, authorityId) for _, v := range list { pathMaps = append(pathMaps, request.CasbinInfo{ Path: v[1], Method: v[2], }) } return pathMaps } //@author: [piexlmax](https://github.com/piexlmax) //@function: ClearCasbin //@description: 清除匹配的权限 //@param: v int, p ...string //@return: bool func (casbinService *CasbinService) ClearCasbin(v int, p ...string) bool { e := casbinService.Casbin() success, _ := e.RemoveFilteredPolicy(v, p...) return success } //@author: [piexlmax](https://github.com/piexlmax) //@function: Casbin //@description: 持久化到数据库 引入自定义规则 //@return: *casbin.Enforcer var ( syncedEnforcer *casbin.SyncedEnforcer once sync.Once ) func (casbinService *CasbinService) Casbin() *casbin.SyncedEnforcer { once.Do(func() { a, _ := gormadapter.NewAdapterByDB(global.GVA_DB) syncedEnforcer, _ = casbin.NewSyncedEnforcer(global.GVA_CONFIG.Casbin.ModelPath, a) }) _ = syncedEnforcer.LoadPolicy() return syncedEnforcer }