role.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746
  1. package service
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "log"
  7. "strings"
  8. "sync"
  9. "time"
  10. "github.com/mhaya/game/game_cluster/internal/mdb"
  11. "github.com/mhaya/game/game_cluster/internal/mdb/models"
  12. "github.com/mhaya/game/game_cluster/nodes/webadmin/entity"
  13. "github.com/mhaya/game/game_cluster/nodes/webadmin/model"
  14. "go.mongodb.org/mongo-driver/bson"
  15. "go.mongodb.org/mongo-driver/bson/primitive"
  16. "go.mongodb.org/mongo-driver/mongo"
  17. "go.mongodb.org/mongo-driver/mongo/options"
  18. )
  19. type Role struct {
  20. }
  21. func NewRole() *Role {
  22. return &Role{}
  23. }
  24. // List 角色列表
  25. func (r *Role) List(ctx context.Context, req entity.RoleListReq) ([]entity.RoleResp, error) {
  26. roles := models.Roles{}
  27. rolesCollection := mdb.MDB.Collection(roles.TableName())
  28. // 构建过滤器
  29. filter := bson.M{}
  30. if req.Name != "" {
  31. filter["name"] = req.Name
  32. }
  33. if req.Status > 0 {
  34. filter["status"] = req.Status
  35. }
  36. // 数据验证
  37. if req.Page <= 0 || req.Size <= 0 {
  38. return nil, fmt.Errorf("invalid page or size")
  39. }
  40. // 设置分页选项
  41. findOptions := options.Find().SetSkip(int64((req.Page - 1) * req.Size)).SetLimit(int64(req.Size))
  42. // 防御性编程
  43. tableName := roles.TableName()
  44. if tableName == "" {
  45. return nil, fmt.Errorf("invalid table name")
  46. }
  47. cursor, err := rolesCollection.Find(ctx, filter, findOptions)
  48. if err != nil {
  49. log.Printf("Failed to execute query: %v", err)
  50. return nil, err
  51. }
  52. defer func() {
  53. if err := cursor.Close(ctx); err != nil {
  54. log.Printf("Failed to close cursor: %v", err)
  55. }
  56. }()
  57. var result []entity.RoleResp
  58. for cursor.Next(ctx) {
  59. var role entity.RoleResp
  60. if err := cursor.Decode(&role); err != nil {
  61. log.Printf("Failed to decode document: %v", err)
  62. return nil, err
  63. }
  64. result = append(result, role)
  65. }
  66. if err := cursor.Err(); err != nil {
  67. log.Printf("Cursor error: %v", err)
  68. return nil, err
  69. }
  70. return result, nil
  71. }
  72. // Add 新增角色
  73. func (r *Role) Add(ctx context.Context, req entity.RoleAddReq) error {
  74. // 验证角色名称是否已存在
  75. if r.checkRoleNameExist(req.Name) == false {
  76. return fmt.Errorf("角色名称已存在")
  77. }
  78. // 检查上下文是否有效
  79. if ctx.Err() != nil {
  80. return ctx.Err()
  81. }
  82. // 插入新角色记录
  83. roles := models.Roles{}
  84. insertData := bson.M{}
  85. insertData["name"] = req.Name
  86. insertData["desc"] = req.Desc
  87. insertData["status"] = req.Status
  88. insertData["created_at"] = time.Now().Unix()
  89. // 确保 Collection 方法不会返回错误
  90. collection := mdb.MDB.Collection(roles.TableName())
  91. _, insertErr := collection.InsertOne(ctx, req)
  92. if insertErr != nil {
  93. log.Printf("Failed to insert role: %s", insertErr)
  94. return insertErr
  95. }
  96. return nil
  97. }
  98. // Update 修改角色
  99. func (r *Role) Update(ctx context.Context, req entity.RoleUpdateReq) error {
  100. // 更新条件
  101. objID, err := primitive.ObjectIDFromHex(req.Id)
  102. if err != nil {
  103. return fmt.Errorf("invalid ObjectID: %v", err)
  104. }
  105. updateCondition := bson.M{"_id": objID}
  106. // 更新内容
  107. updateContent := bson.M{
  108. "$set": bson.M{
  109. "name": req.Name,
  110. "desc": req.Desc,
  111. "status": req.Status,
  112. },
  113. }
  114. // 设置更新选项
  115. roles := models.Roles{}
  116. collection := mdb.MDB.Collection(roles.TableName())
  117. updateOptions := options.Update().SetUpsert(true) // 设置 upsert 选项
  118. // 执行更新操作
  119. _, err = collection.UpdateOne(context.TODO(), updateCondition, updateContent, updateOptions)
  120. if err != nil {
  121. return err
  122. }
  123. return nil
  124. }
  125. // checkRoleNameExist 检查角色名称是否已存在
  126. func (r *Role) checkRoleNameExist(name string) bool {
  127. // 创建带超时的上下文
  128. ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  129. defer cancel()
  130. roles := models.Roles{}
  131. collection := mdb.MDB.Collection(roles.TableName())
  132. // 构建过滤器
  133. filter := bson.M{"name": name}
  134. // 执行查询
  135. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  136. if errors.Is(err, mongo.ErrNoDocuments) {
  137. log.Printf("No document found with role name: %s", name)
  138. return true
  139. }
  140. return false
  141. }
  142. return false
  143. }
  144. // Del 删除角色
  145. func (r *Role) Del(ctx context.Context, req entity.RoleDelReq) error {
  146. // 创建带超时的上下文
  147. ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
  148. defer cancel()
  149. roles := models.Roles{}
  150. collection := mdb.MDB.Collection(roles.TableName())
  151. id, _ := primitive.ObjectIDFromHex(req.Id)
  152. _, err := collection.DeleteOne(ctx, bson.M{"id": id})
  153. return err
  154. }
  155. // AddRoleAccess 添加角色权限
  156. func (r *Role) AddRoleAccess(ctx context.Context, req entity.RoleAccessAddReq) error {
  157. // 检查上下文是否有效
  158. if ctx.Err() != nil {
  159. return ctx.Err()
  160. }
  161. // 检查角色是否存在
  162. roles := models.Roles{}
  163. collection := mdb.MDB.Collection(roles.TableName())
  164. roleIDobj, _ := primitive.ObjectIDFromHex(req.RoleId)
  165. if err := collection.FindOne(ctx, bson.M{"_id": roleIDobj}).Err(); err != nil {
  166. if errors.Is(err, mongo.ErrNoDocuments) {
  167. return fmt.Errorf("角色不存在")
  168. }
  169. }
  170. // 检查权限是否存在 ->具体的权限规则表-存放路由、菜单等
  171. access := models.Access{}
  172. collection = mdb.MDB.Collection(access.TableName())
  173. var accessIDS []primitive.ObjectID
  174. for _, v := range req.AccessId {
  175. accessIDobj, _ := primitive.ObjectIDFromHex(v)
  176. accessIDS = append(accessIDS, accessIDobj)
  177. }
  178. filter := bson.M{"_id": bson.M{"$in": accessIDS}} // 数组查询
  179. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  180. if errors.Is(err, mongo.ErrNoDocuments) {
  181. return fmt.Errorf("权限不存在")
  182. }
  183. }
  184. // 插入新角色权限记录
  185. roleAccess := models.RoleAccess{}
  186. // 确保 Collection 方法不会返回错误
  187. collection = mdb.MDB.Collection(roleAccess.TableName())
  188. _, insertErr := collection.UpdateOne(ctx, bson.M{"role_id": req.RoleId}, bson.M{"$addToSet": bson.M{"access_id": bson.M{"$each": req.AccessId}}}, options.Update().SetUpsert(true))
  189. if insertErr != nil {
  190. log.Printf("Failed to insert role: %s", insertErr)
  191. return insertErr
  192. }
  193. return nil
  194. }
  195. // UpdateRoleAccess 修改角色权限
  196. func (r *Role) UpdateRoleAccess(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  197. // 验证请求数据的有效性
  198. if err := validateConcurrently(ctx, req); err != nil {
  199. return err
  200. }
  201. // 更新角色权限
  202. if err := r.updateAccessInDatabase(ctx, req); err != nil {
  203. return err
  204. }
  205. // 返回成功
  206. return nil
  207. }
  208. // DelRoleAccess 根据角色ID删除角色权限
  209. func (r *Role) DelRoleAccess(ctx context.Context, req entity.RoleAccessDelReq) error {
  210. roleAccess := models.RoleAccess{}
  211. collection := mdb.MDB.Collection(roleAccess.TableName())
  212. filter := bson.M{"role_id": req.RoleId}
  213. _, err := collection.DeleteOne(ctx, filter)
  214. if err != nil {
  215. return err
  216. }
  217. return nil
  218. }
  219. // GetRoleAccessList 获取角色权限列表根据角色ID
  220. func (r *Role) GetRoleAccessList(ctx context.Context, req entity.RoleAccessListReq) ([]*entity.AccessResp, error) {
  221. // 查询角色权限列表
  222. roleAccess := models.RoleAccess{}
  223. collection := mdb.MDB.Collection(roleAccess.TableName())
  224. filter := bson.M{"role_id": req.RoleId} // 数组查询
  225. cursor, err := collection.Find(ctx, filter)
  226. defer cursor.Close(ctx)
  227. if err != nil {
  228. return nil, err
  229. }
  230. var accessIDS []string
  231. for cursor.Next(ctx) {
  232. var roleAccess models.RoleAccess
  233. if err := cursor.Decode(&roleAccess); err != nil {
  234. log.Printf("Failed to decode document: %v", err)
  235. return nil, err
  236. }
  237. accessIDS = append(accessIDS, roleAccess.AccessID...)
  238. }
  239. // 判断accessIDS是否为空
  240. if len(accessIDS) == 0 {
  241. return nil, nil
  242. }
  243. var objIDS []primitive.ObjectID
  244. for _, v := range accessIDS {
  245. objID, _ := primitive.ObjectIDFromHex(v)
  246. objIDS = append(objIDS, objID)
  247. }
  248. // 根据accessIDS查询权限列表 并且按照下级关系组合
  249. access := models.Access{}
  250. collection = mdb.MDB.Collection(access.TableName())
  251. accessFilter := bson.M{"_id": bson.M{"$in": objIDS}}
  252. cursor, err = collection.Find(ctx, accessFilter)
  253. defer cursor.Close(ctx)
  254. if err != nil {
  255. return nil, err
  256. }
  257. var accessList []*entity.AccessResp
  258. for cursor.Next(ctx) {
  259. var accesss *models.Access
  260. if err := cursor.Decode(&accesss); err != nil {
  261. log.Printf("Failed to decode document: %v", err)
  262. return nil, err
  263. }
  264. accessList = append(accessList, &entity.AccessResp{
  265. ID: accesss.ID,
  266. ActionName: accesss.ActionName,
  267. ModuleName: accesss.ModuleName,
  268. Description: accesss.Description,
  269. URL: accesss.URL,
  270. ParentId: accesss.ParentId,
  271. Sort: accesss.Sort,
  272. Type: accesss.Type,
  273. Status: accesss.Status,
  274. })
  275. }
  276. // 格式化数据并且按照下级关系组合
  277. return formatAccessData(accessList), nil
  278. }
  279. // formatAccessData formats and organizes access data into a hierarchical structure
  280. func formatAccessData(accessData []*entity.AccessResp) []*entity.AccessResp {
  281. nodeMap := make(map[interface{}]*entity.AccessResp)
  282. var rootNodes []*entity.AccessResp
  283. for i := range accessData {
  284. node := accessData[i]
  285. nodeMap[node.ID] = node
  286. }
  287. for i := range accessData {
  288. node := accessData[i]
  289. if node.ParentId == "0" {
  290. rootNodes = append(rootNodes, node)
  291. } else {
  292. if parentNode, exists := nodeMap[node.ParentId]; exists {
  293. parentNode.AccessItem = append(parentNode.AccessItem, node)
  294. }
  295. }
  296. }
  297. return rootNodes
  298. }
  299. // AddAccess 添加权限路由
  300. func (r *Role) AddAccess(ctx context.Context, req entity.AccessAddReq) error {
  301. // 检查上下文是否有效
  302. if ctx.Err() != nil {
  303. return ctx.Err()
  304. }
  305. access := models.Access{}
  306. collection := mdb.MDB.Collection(access.TableName())
  307. // 判断是否有相同的数据
  308. filter := bson.M{"path": req.URL}
  309. if err := collection.FindOne(ctx, filter).Err(); err == nil {
  310. return fmt.Errorf("权限已存在")
  311. }
  312. // 插入新角色权限记录
  313. _, err := collection.InsertOne(ctx, req)
  314. return err
  315. }
  316. // DelAccess 删除权限路由
  317. func (r *Role) DelAccess(ctx context.Context, req entity.AccessDelReq) error {
  318. access := models.Access{}
  319. collection := mdb.MDB.Collection(access.TableName())
  320. // 判断是否有角色使用了该权限路由
  321. roleAccess := models.RoleAccess{}
  322. collection = mdb.MDB.Collection(roleAccess.TableName())
  323. filter := bson.M{"access_id": bson.M{"$in": req.Id}} // 数组查询
  324. if err := collection.FindOne(ctx, filter).Err(); err == nil {
  325. return fmt.Errorf("权限已被角色使用,无法删除")
  326. }
  327. objID := primitive.ObjectID{}
  328. objID, _ = primitive.ObjectIDFromHex(req.Id)
  329. _, err := collection.DeleteOne(ctx, bson.M{"_id": objID})
  330. return err
  331. }
  332. // UpdateAccess 修改权限路由
  333. func (r *Role) UpdateAccess(ctx context.Context, req entity.AccessUpdateReq) error {
  334. access := models.Access{}
  335. collection := mdb.MDB.Collection(access.TableName())
  336. // update
  337. var updateFields = bson.M{
  338. "module_name": req.ModuleName,
  339. "action_name": req.ActionName,
  340. "url": req.URL,
  341. "type": req.Type,
  342. "description": req.Description,
  343. "sort": req.Sort,
  344. "status": req.Status}
  345. // 去掉为空的字段
  346. for k, v := range updateFields {
  347. switch v.(type) {
  348. case string:
  349. if v == "" {
  350. delete(updateFields, k)
  351. }
  352. case int:
  353. if v == 0 {
  354. delete(updateFields, k)
  355. }
  356. default:
  357. continue
  358. }
  359. }
  360. // 检查是否有需要更新的字段
  361. if len(updateFields) == 0 {
  362. return errors.New("no fields to update")
  363. }
  364. // 确保 req.Id 是一个有效的 ObjectID
  365. objID, err := primitive.ObjectIDFromHex(req.Id)
  366. if err != nil {
  367. return fmt.Errorf("invalid ObjectID: %v", err)
  368. }
  369. _, err = collection.UpdateByID(ctx, objID, bson.M{"$set": updateFields})
  370. if err != nil {
  371. return fmt.Errorf("update failed: %v", err)
  372. }
  373. return err
  374. }
  375. // ListAccess listAccessa
  376. func (r *Role) ListAccess(ctx context.Context, req entity.AccessListReq) ([]*entity.AccessResp, error) {
  377. access := models.Access{}
  378. collection := mdb.MDB.Collection(access.TableName())
  379. filter := bson.M{}
  380. if req.ActionName != "" {
  381. filter["action_name"] = req.ActionName
  382. }
  383. if req.ModuleName != "" {
  384. filter["module_name"] = req.ModuleName
  385. }
  386. if req.Status != 0 {
  387. filter["status"] = req.Status
  388. }
  389. if req.Type != 0 {
  390. filter["type"] = req.Type
  391. }
  392. if req.ParentId != "" {
  393. filter["parent_id"] = req.ParentId
  394. }
  395. if req.URL != "" {
  396. filter["url"] = req.URL
  397. }
  398. // 数据验证
  399. if req.Page <= 0 || req.Size <= 0 {
  400. return nil, fmt.Errorf("invalid page or size")
  401. }
  402. // 设置分页选项
  403. findOptions := options.Find().SetSkip(int64((req.Page - 1) * req.Size)).SetLimit(int64(req.Size))
  404. cursor, err := collection.Find(ctx, filter, findOptions)
  405. defer cursor.Close(ctx)
  406. if err != nil {
  407. return nil, err
  408. }
  409. var accessList []*entity.AccessResp
  410. for cursor.Next(ctx) {
  411. var accesss *models.Access
  412. if err := cursor.Decode(&accesss); err != nil {
  413. log.Printf("Failed to decode document: %v", err)
  414. return nil, err
  415. }
  416. accessList = append(accessList, &entity.AccessResp{
  417. ID: accesss.ID,
  418. ActionName: accesss.ActionName,
  419. ModuleName: accesss.ModuleName,
  420. Description: accesss.Description,
  421. URL: accesss.URL,
  422. ParentId: accesss.ParentId,
  423. Sort: accesss.Sort,
  424. Type: accesss.Type,
  425. Status: accesss.Status,
  426. })
  427. }
  428. // 格式化数据并且按照下级关系组合
  429. return formatAccessData(accessList), nil
  430. }
  431. // updateAccessInDatabase 在数据库中更新角色权限
  432. func (r *Role) updateAccessInDatabase(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  433. // 例如更新角色权限表中的记录
  434. roleAccess := models.RoleAccess{}
  435. collection := mdb.MDB.Collection(roleAccess.TableName())
  436. filter := bson.M{"role_id": req.RoleId}
  437. update := bson.M{"$set": bson.M{"access_id": req.AccessId}}
  438. _, err := collection.UpdateOne(ctx, filter, update)
  439. if err != nil {
  440. return err
  441. }
  442. return nil // 假设更新成功,实际应根据业务逻辑处理
  443. }
  444. // AdminBindRole 绑定角色
  445. func (r *Role) AdminBindRole(ctx context.Context, req *entity.AdminBindRoleReq) error {
  446. // 例如更新角色权限表中的记录
  447. role := models.Roles{}
  448. collection := mdb.MDB.Collection(role.TableName())
  449. roleId, _ := primitive.ObjectIDFromHex(req.RoleId)
  450. filter := bson.M{"_id": roleId, "status": 1}
  451. // 判断你是否存在
  452. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  453. return fmt.Errorf("角色不存在,或者已经被禁用")
  454. }
  455. // 判断管理员是否存在
  456. admin := model.Admin{}
  457. collection = mdb.MDB.Collection(admin.TableName())
  458. objID, _ := primitive.ObjectIDFromHex(req.AdminId)
  459. filter = bson.M{"_id": objID, "status": 1}
  460. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  461. return fmt.Errorf("管理员不存在 或者 已经被禁用")
  462. }
  463. // 更新管理员数据
  464. _, err := collection.UpdateByID(ctx, objID, bson.M{"$set": bson.M{"role_id": req.RoleId}})
  465. if err != nil {
  466. return fmt.Errorf("更新管理员失败")
  467. }
  468. return nil // 假设更新成功,实际应根据业务逻辑处理
  469. }
  470. // AdminUnBindRole 取消绑定角色
  471. func (r *Role) AdminUnBindRole(ctx context.Context, req *entity.AdminBindRoleReq) error {
  472. // 例如更新角色权限表中的记录
  473. admin := model.Admin{}
  474. collection := mdb.MDB.Collection(admin.TableName())
  475. objID, _ := primitive.ObjectIDFromHex(req.AdminId)
  476. filter := bson.M{"_id": objID}
  477. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  478. return fmt.Errorf("管理员不存在")
  479. }
  480. // 更新管理员数据
  481. _, err := collection.UpdateByID(ctx, objID, bson.M{"$set": bson.M{"role_id": ""}})
  482. if err != nil {
  483. return fmt.Errorf("更新管理员失败")
  484. }
  485. return nil
  486. }
  487. // GetAdminRole GetAdminBindRole 根据角色id 获取Access
  488. func getAdmin(ctx context.Context, id string) (*model.Admin, error) {
  489. objID, err := primitive.ObjectIDFromHex(id)
  490. if err != nil {
  491. return nil, fmt.Errorf("解析管理员ID失败: %v", err)
  492. }
  493. admin := model.Admin{}
  494. adminCollection := mdb.MDB.Collection(admin.TableName())
  495. adminFilter := bson.M{"_id": objID}
  496. err = adminCollection.FindOne(ctx, adminFilter).Decode(&admin)
  497. if err != nil {
  498. if errors.Is(err, mongo.ErrNoDocuments) {
  499. return nil, fmt.Errorf("找不到管理员")
  500. }
  501. return nil, fmt.Errorf("查询管理员失败: %v", err)
  502. }
  503. return &admin, nil
  504. }
  505. func getRole(ctx context.Context, id string) (*models.Roles, error) {
  506. objID, err := primitive.ObjectIDFromHex(id)
  507. if err != nil {
  508. return nil, fmt.Errorf("解析角色ID失败: %v", err)
  509. }
  510. role := models.Roles{}
  511. roleCollection := mdb.MDB.Collection(role.TableName())
  512. roleFilter := bson.M{"_id": objID}
  513. err = roleCollection.FindOne(ctx, roleFilter).Decode(&role)
  514. if err != nil {
  515. if errors.Is(err, mongo.ErrNoDocuments) {
  516. return nil, fmt.Errorf("找不到角色")
  517. }
  518. return nil, fmt.Errorf("查询角色失败: %v", err)
  519. }
  520. return &role, nil
  521. }
  522. func getRoleAccess(ctx context.Context, roleId string) (*models.RoleAccess, error) {
  523. roleAccess := models.RoleAccess{}
  524. roleAccessCollection := mdb.MDB.Collection(roleAccess.TableName())
  525. roleAccessFilter := bson.M{"role_id": roleId}
  526. err := roleAccessCollection.FindOne(ctx, roleAccessFilter).Decode(&roleAccess)
  527. if err != nil {
  528. if errors.Is(err, mongo.ErrNoDocuments) {
  529. return nil, fmt.Errorf("找不到角色权限")
  530. }
  531. return nil, fmt.Errorf("查询角色权限失败: %v", err)
  532. }
  533. return &roleAccess, nil
  534. }
  535. func (r *Role) GetAdminRole(ctx context.Context, req *entity.AdminBindRoleReq) (*entity.AdminBindRoleResp, error) {
  536. admin, err := getAdmin(ctx, req.AdminId)
  537. if err != nil {
  538. return nil, err
  539. }
  540. role, err := getRole(ctx, req.RoleId)
  541. if err != nil {
  542. return nil, err
  543. }
  544. roleAccess, err := getRoleAccess(ctx, req.RoleId)
  545. if err != nil {
  546. return nil, err
  547. }
  548. var AccessIds []primitive.ObjectID
  549. var invalidAccessIds []string
  550. for _, v := range roleAccess.AccessID {
  551. objId, err := primitive.ObjectIDFromHex(v)
  552. if err != nil {
  553. invalidAccessIds = append(invalidAccessIds, v)
  554. continue
  555. }
  556. AccessIds = append(AccessIds, objId)
  557. }
  558. if len(AccessIds) == 0 {
  559. if len(invalidAccessIds) > 0 {
  560. return nil, fmt.Errorf("无效的权限ID: %v", strings.Join(invalidAccessIds, ", "))
  561. }
  562. return nil, fmt.Errorf("没有权限")
  563. }
  564. access := models.Access{}
  565. collection := mdb.MDB.Collection(access.TableName())
  566. filter := bson.M{}
  567. filter["_id"] = bson.M{"$in": AccessIds}
  568. cursor, err := collection.Find(ctx, filter)
  569. defer cursor.Close(ctx)
  570. if err != nil {
  571. return nil, fmt.Errorf("查询权限失败: %v", err)
  572. }
  573. var accessList []*entity.AccessResp
  574. for cursor.Next(ctx) {
  575. var accesss *models.Access
  576. if err := cursor.Decode(&accesss); err != nil {
  577. log.Printf("Failed to decode document: %v", err)
  578. return nil, fmt.Errorf("解码权限失败: %v", err)
  579. }
  580. accessList = append(accessList, &entity.AccessResp{
  581. ID: accesss.ID,
  582. ActionName: accesss.ActionName,
  583. ModuleName: accesss.ModuleName,
  584. Description: accesss.Description,
  585. URL: accesss.URL,
  586. ParentId: accesss.ParentId,
  587. Sort: accesss.Sort,
  588. Type: accesss.Type,
  589. Status: accesss.Status,
  590. })
  591. }
  592. return &entity.AdminBindRoleResp{
  593. AdminId: req.AdminId,
  594. AdminName: admin.Username,
  595. RoleName: role.Name,
  596. RoleId: req.RoleId,
  597. AccessList: formatAccessData(accessList)},
  598. nil
  599. }
  600. // 使用协程和通道并发执行验证操作
  601. func validateConcurrently(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  602. // 创建通道
  603. ch := make(chan error, 3)
  604. // 并发执行验证操作
  605. var wg sync.WaitGroup
  606. wg.Add(3)
  607. go func() {
  608. defer wg.Done()
  609. ch <- validateRoleExistence(ctx, req)
  610. }()
  611. go func() {
  612. defer wg.Done()
  613. ch <- validateAccessExistence(ctx, req)
  614. }()
  615. go func() {
  616. defer wg.Done()
  617. ch <- validateRoleAccessExistence(ctx, req)
  618. }()
  619. // 收集所有验证结果
  620. var e []error
  621. go func() {
  622. defer wg.Wait()
  623. for i := 0; i < 3; i++ {
  624. if err := <-ch; err != nil {
  625. e = append(e, err)
  626. }
  627. }
  628. if len(e) > 0 {
  629. ch <- fmt.Errorf("验证失败: %v", e)
  630. } else {
  631. ch <- nil
  632. }
  633. }()
  634. // 等待所有并发任务完成
  635. select {
  636. case err := <-ch:
  637. return err
  638. case <-ctx.Done():
  639. return ctx.Err()
  640. }
  641. }
  642. // 验证角色是否存在
  643. func validateRoleExistence(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  644. roles := models.Roles{}
  645. collection := mdb.MDB.Collection(roles.TableName())
  646. objID, err := primitive.ObjectIDFromHex(req.RoleId)
  647. if err != nil {
  648. return fmt.Errorf("解析角色ID失败: %v", err)
  649. }
  650. if err := collection.FindOne(ctx, bson.M{"_id": objID}).Err(); err != nil {
  651. if errors.Is(err, mongo.ErrNoDocuments) {
  652. return fmt.Errorf("角色不存在")
  653. }
  654. return err
  655. }
  656. return nil
  657. }
  658. // 验证权限是否存在
  659. func validateAccessExistence(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  660. access := models.Access{}
  661. collection := mdb.MDB.Collection(access.TableName())
  662. objIDS := make([]primitive.ObjectID, len(req.AccessId))
  663. for i, id := range req.AccessId {
  664. objID, err := primitive.ObjectIDFromHex(id)
  665. if err != nil {
  666. return fmt.Errorf("解析权限ID失败: %v", err)
  667. }
  668. objIDS[i] = objID
  669. }
  670. filter := bson.M{"_id": bson.M{"$in": objIDS}} // 数组查询
  671. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  672. if errors.Is(err, mongo.ErrNoDocuments) {
  673. return fmt.Errorf("权限不存在")
  674. }
  675. return err
  676. }
  677. return nil
  678. }
  679. // 验证角色权限是否已存在
  680. func validateRoleAccessExistence(ctx context.Context, req entity.RoleAccessUpdateReq) error {
  681. roleAccess := models.RoleAccess{}
  682. collection := mdb.MDB.Collection(roleAccess.TableName())
  683. filter := bson.M{"role_id": req.RoleId}
  684. if err := collection.FindOne(ctx, filter).Err(); err != nil {
  685. if errors.Is(err, mongo.ErrNoDocuments) {
  686. return fmt.Errorf("角色权限不存在 不能更新")
  687. }
  688. return err
  689. }
  690. return nil
  691. }