role.go 26 KB

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