report_v2.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. package services
  2. import (
  3. "errors"
  4. "eta/eta_api/models"
  5. "eta/eta_api/models/report"
  6. "eta/eta_api/models/system"
  7. "eta/eta_api/utils"
  8. "strconv"
  9. "time"
  10. )
  11. // AddReportAndChapter
  12. // @Description: 新增报告(包含章节)
  13. // @author: Roc
  14. // @datetime 2024-06-04 11:23:20
  15. // @param reportInfo *models.Report
  16. // @param inheritReportId int
  17. // @return err error
  18. // @return errMsg string
  19. func AddReportAndChapter(reportInfo *models.Report, inheritReportId int, grantAdminIdList []int) (err error, errMsg string) {
  20. // 获取最小分类id
  21. minClassifyId := reportInfo.ClassifyIdThird
  22. if minClassifyId <= 0 {
  23. minClassifyId = reportInfo.ClassifyIdSecond
  24. }
  25. if minClassifyId <= 0 {
  26. minClassifyId = reportInfo.ClassifyIdFirst
  27. }
  28. if minClassifyId <= 0 {
  29. errMsg = "分类异常"
  30. err = errors.New(errMsg)
  31. return
  32. }
  33. errMsg = "生成报告失败"
  34. // 报告继承
  35. if inheritReportId > 0 {
  36. inheritReport, tmpErr := models.GetReportByReportId(inheritReportId)
  37. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  38. errMsg = "获取待继承的报告失败"
  39. err = tmpErr
  40. return
  41. }
  42. if inheritReport != nil {
  43. // 判断当前的报告分类与继承的报告分类是否一致
  44. if inheritReport.ClassifyIdFirst != reportInfo.ClassifyIdFirst || inheritReport.ClassifyIdSecond != reportInfo.ClassifyIdSecond || inheritReport.ClassifyIdThird != reportInfo.ClassifyIdThird {
  45. errMsg = "分类异常,与继承的报告分类不一致"
  46. err = tmpErr
  47. return
  48. }
  49. reportInfo.ChapterType = inheritReport.ChapterType
  50. reportInfo.Content = inheritReport.Content
  51. reportInfo.ContentSub = inheritReport.ContentSub
  52. reportInfo.ContentStruct = inheritReport.ContentStruct
  53. reportInfo.HeadImg = inheritReport.HeadImg
  54. reportInfo.EndImg = inheritReport.EndImg
  55. reportInfo.CanvasColor = inheritReport.CanvasColor
  56. reportInfo.NeedSplice = inheritReport.NeedSplice
  57. reportInfo.HeadResourceId = inheritReport.HeadResourceId
  58. reportInfo.EndResourceId = inheritReport.EndResourceId
  59. reportInfo.InheritReportId = inheritReport.Id
  60. }
  61. }
  62. // 获取待生成的报告章节
  63. addChapterList, allGrantUserList, err, errMsg := getAddChapter(reportInfo, minClassifyId, inheritReportId, grantAdminIdList)
  64. // 新增报告及章节
  65. var reportId int64
  66. reportId, err = models.AddReportAndChapter(reportInfo, allGrantUserList, addChapterList)
  67. if err != nil {
  68. err = errors.New("新增报告及章节失败, Err: " + err.Error())
  69. return
  70. }
  71. reportCode := utils.MD5(strconv.Itoa(reportInfo.Id))
  72. reportInfo.ReportCode = reportCode
  73. // 修改唯一编码
  74. {
  75. go models.ModifyReportCode(reportId, reportCode)
  76. }
  77. // TODO 报告权限处理
  78. //处理权限
  79. //go func() {
  80. // permissionItems, e := models.GetPermission(req.ClassifyIdSecond)
  81. // if e != nil {
  82. // alarm_msg.SendAlarmMsg("获取权限失败,Err:"+e.Error(), 3)
  83. // return
  84. // }
  85. // for _, v := range permissionItems {
  86. // e = models.AddChartPermissionChapterMapping(v.ChartPermissionId, int64(item.Id))
  87. // if e != nil {
  88. // alarm_msg.SendAlarmMsg("新增权限失败,Err:"+e.Error(), 3)
  89. // return
  90. // }
  91. // }
  92. // // 同步crm权限
  93. // _ = EditReportPermissionSync(int64(item.Id), req.ClassifyIdSecond)
  94. //}()
  95. return
  96. }
  97. // EditReport
  98. // @Description: 修改报告的基础信息、授权用户权限
  99. // @author: Roc
  100. // @datetime 2024-06-06 17:16:58
  101. // @param reportInfo *models.Report
  102. // @param req models.EditReq
  103. // @param sysUser *system.Admin
  104. // @return err error
  105. // @return errMsg string
  106. func EditReport(reportInfo *models.Report, req models.EditReq, sysUser *system.Admin) (err error, errMsg string) {
  107. errMsg = `保存失败`
  108. var stage int
  109. if reportInfo.ClassifyNameFirst != req.ClassifyNameFirst || reportInfo.ClassifyNameSecond != req.ClassifyNameSecond {
  110. // 报告期数
  111. maxStage, _ := models.GetReportStageEdit(req.ClassifyIdFirst, req.ClassifyIdSecond, req.ClassifyIdThird, int(req.ReportId))
  112. maxStage = maxStage + 1
  113. stage = maxStage
  114. } else {
  115. stage = reportInfo.Stage
  116. }
  117. //if req.State != reportInfo.State {
  118. // recordItem := &models.ReportStateRecord{
  119. // ReportId: int(req.ReportId),
  120. // ReportType: 1,
  121. // State: req.State,
  122. // AdminId: this.SysUser.AdminId,
  123. // AdminName: this.SysUser.AdminName,
  124. // CreateTime: time.Now(),
  125. // }
  126. // go func() {
  127. // _, _ = models.AddReportStateRecord(recordItem)
  128. // }()
  129. //}
  130. //item := new(models.Report)
  131. reportInfo.ClassifyIdFirst = req.ClassifyIdFirst
  132. reportInfo.ClassifyNameFirst = req.ClassifyNameFirst
  133. reportInfo.ClassifyIdSecond = req.ClassifyIdSecond
  134. reportInfo.ClassifyNameSecond = req.ClassifyNameSecond
  135. reportInfo.ClassifyIdThird = req.ClassifyIdThird
  136. reportInfo.ClassifyNameThird = req.ClassifyNameThird
  137. reportInfo.Title = req.Title
  138. reportInfo.Abstract = req.Abstract
  139. reportInfo.Author = req.Author
  140. reportInfo.Frequency = req.Frequency
  141. //reportInfo.State = reportInfo.State // 编辑不变更状态
  142. reportInfo.Stage = stage // 编辑不变更期数
  143. //reportInfo.Content = html.EscapeString(req.Content) // 编辑不变更具体内容
  144. //reportInfo.ContentSub = html.EscapeString(contentSub) // 编辑不变更具体内容
  145. reportInfo.CreateTime = req.CreateTime
  146. //reportInfo.CollaborateType = req.CollaborateType
  147. //reportInfo.ReportLayout = req.ReportLayout
  148. if req.IsPublicPublish <= 0 {
  149. req.IsPublicPublish = 1
  150. }
  151. reportInfo.IsPublicPublish = req.IsPublicPublish
  152. reportInfo.LastModifyAdminId = sysUser.AdminId
  153. reportInfo.LastModifyAdminName = sysUser.RealName
  154. reportInfo.ModifyTime = time.Now()
  155. updateCols := []string{"ClassifyIdFirst", "ClassifyNameFirst", "ClassifyIdSecond", "ClassifyNameSecond", "ClassifyIdThird", "ClassifyNameThird", "Title", "Abstract", "Author", "Frequency", "Stage", "CreateTime", "IsPublicPublish", "LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  156. // 需要添加的报告授权数据
  157. addReportAdminList := make([]*report.ReportGrant, 0)
  158. // 待移除的报告授权数据id
  159. delReportGrantIdList := make([]int, 0)
  160. // 处理当前报告需要新增/移除的授权信息
  161. {
  162. reportGrantObj := report.ReportGrant{}
  163. // 获取当前报告已经授权的用户信息
  164. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  165. if tmpErr != nil {
  166. err = tmpErr
  167. return
  168. }
  169. // 当前报告已经授权的用户信息
  170. currReportAdminMap := make(map[int]*report.ReportGrant)
  171. // 需要删除的报告授权数据
  172. delReportAdminMap := make(map[int]*report.ReportGrant)
  173. for _, v := range reportGrantList {
  174. currReportAdminMap[v.AdminId] = v
  175. delReportAdminMap[v.AdminId] = v
  176. }
  177. // 先看需要新增哪些用户
  178. for _, tmpAdminId := range req.GrantAdminIdList {
  179. _, ok := currReportAdminMap[tmpAdminId]
  180. // 如果章节中需要新增的用户 已经在 报告授权用户里面,那么就忽略,可以不用新增了
  181. if ok {
  182. delete(delReportAdminMap, tmpAdminId)
  183. continue
  184. }
  185. // 如果不存在,那么就新增授权
  186. addReportAdminList = append(addReportAdminList, &report.ReportGrant{
  187. //GrantId: 0,
  188. ReportId: reportInfo.Id,
  189. AdminId: tmpAdminId,
  190. CreateTime: time.Now(),
  191. })
  192. }
  193. // 查出需要移除的授权id
  194. for _, v := range delReportAdminMap {
  195. delReportGrantIdList = append(delReportGrantIdList, v.GrantId)
  196. }
  197. }
  198. // 修改报告的基本信息,以及报告的授权用户
  199. err = models.EditReportAndPermission(reportInfo, updateCols, addReportAdminList, delReportGrantIdList)
  200. if err != nil {
  201. return
  202. }
  203. // TODO 权限处理
  204. //处理权限
  205. //go func() {
  206. // e := models.RemoveChartPermissionChapterMapping(req.ReportId)
  207. // if e != nil {
  208. // alarm_msg.SendAlarmMsg("修改删除报告权限失败,Err:"+e.Error(), 3)
  209. // return
  210. // }
  211. // permissionItems, e := models.GetPermission(req.ClassifyIdSecond)
  212. // if e != nil {
  213. // alarm_msg.SendAlarmMsg("获取权限失败,Err:"+e.Error(), 3)
  214. // return
  215. // }
  216. // for _, v := range permissionItems {
  217. // e = models.AddChartPermissionChapterMapping(v.ChartPermissionId, req.ReportId)
  218. // if e != nil {
  219. // alarm_msg.SendAlarmMsg("新增权限失败,Err:"+e.Error(), 3)
  220. // return
  221. // }
  222. // }
  223. // // 同步crm权限
  224. // _ = services.EditReportPermissionSync(req.ReportId, req.ClassifyIdSecond)
  225. //}()
  226. return
  227. }
  228. // getAddChapter
  229. // @Description: 获取待新增的报告章节列表
  230. // @author: Roc
  231. // @datetime 2024-06-04 13:10:58
  232. // @param reportInfo *models.Report
  233. // @param minClassifyId int
  234. // @param inheritReportId int
  235. // @param grantAdminIdList []int
  236. // @return chapterList []*models.ReportChapter
  237. // @return err error
  238. // @return errMsg string
  239. func getAddChapter(reportInfo *models.Report, minClassifyId, inheritReportId int, grantAdminIdList []int) (chapterList []models.AddReportChapter, allGrantUserList []*report.ReportGrant, err error, errMsg string) {
  240. // 待生成的报告章节内容
  241. chapterList = make([]models.AddReportChapter, 0)
  242. // 所有的授权用户
  243. allGrantUserList = make([]*report.ReportGrant, 0)
  244. // 报告授权的用户ID
  245. needAdminIdMap := make(map[int]bool)
  246. for _, adminId := range grantAdminIdList {
  247. needAdminIdMap[adminId] = true
  248. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  249. //GrantId: 0,
  250. //ReportId: 0,
  251. AdminId: adminId,
  252. CreateTime: time.Now(),
  253. })
  254. }
  255. if reportInfo.HasChapter != 1 {
  256. return
  257. }
  258. // TODO 弘则得单独处理启用禁用的情况
  259. //if utils.BusinessCode != utils.BusinessCodeRelease && utils.BusinessCode != utils.BusinessCodeSandbox && utils.BusinessCode != utils.BusinessCodeDebug {
  260. // br.Ret = 200
  261. // br.Success = true
  262. // br.Msg = "操作成功"
  263. // return
  264. //}
  265. errMsg = "生成报告章节失败"
  266. // 章节类型列表
  267. typeList, err := models.GetReportChapterTypeListByClassifyId(minClassifyId)
  268. if err != nil {
  269. err = errors.New(errMsg)
  270. return
  271. }
  272. // 分类章节的授权用户
  273. typeGrantListMap := make(map[int][]*report.ReportChapterGrant)
  274. // 自定义章节列表
  275. customAddChapterList := make([]models.AddReportChapter, 0)
  276. // 报告继承
  277. inheritChapterMap := make(map[int]*models.ReportChapter)
  278. if inheritReportId > 0 {
  279. // 继承待继承的报告章节内容
  280. inheritReportChapters, tmpErr := models.GetChapterListByReportId(inheritReportId)
  281. if tmpErr != nil {
  282. errMsg = "获取待继承的报告章节失败"
  283. err = tmpErr
  284. return
  285. }
  286. reportChaptersIdList := make([]int, 0)
  287. for _, v := range inheritReportChapters {
  288. reportChaptersIdList = append(reportChaptersIdList, v.ReportChapterId)
  289. }
  290. // 继承的报告章节用户map
  291. grantListMap := make(map[int][]*report.ReportChapterGrant)
  292. // 授权数据列表
  293. if len(reportChaptersIdList) > 0 {
  294. obj := report.ReportChapterGrant{}
  295. grantList, tmpErr := obj.GetGrantListByIdList(reportChaptersIdList)
  296. if tmpErr != nil {
  297. errMsg = "获取待继承的报告章节的授权用户列表失败"
  298. err = tmpErr
  299. return
  300. }
  301. for _, v := range grantList {
  302. // 如果不在报告授权的用户ID里面,那么该章节就不继承该授权用户
  303. if _, ok := needAdminIdMap[v.AdminId]; !ok {
  304. continue
  305. }
  306. currReportChapterId := v.ReportChapterId
  307. tmpGrantList, ok := grantListMap[currReportChapterId]
  308. if !ok {
  309. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  310. }
  311. v.ReportChapterId = 0
  312. v.GrantId = 0
  313. tmpGrantList = append(tmpGrantList, v)
  314. grantListMap[currReportChapterId] = tmpGrantList
  315. }
  316. }
  317. // 继承的报告章节内容
  318. for i := 0; i < len(inheritReportChapters); i++ {
  319. customChapter := inheritReportChapters[i]
  320. // 授权用户列表
  321. tmpGrantList, ok := grantListMap[customChapter.ReportChapterId]
  322. if !ok {
  323. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  324. }
  325. typeGrantListMap[customChapter.TypeId] = tmpGrantList
  326. // 判断该章节是否是系统章节,如果是的话,那就是需要额外创建的
  327. if customChapter.TypeId > 0 {
  328. inheritChapterMap[customChapter.TypeId] = customChapter
  329. continue
  330. }
  331. // 自定义的报告内容
  332. customChapter.ReportId = reportInfo.Id
  333. customChapter.ReportChapterId = 0
  334. customChapter.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  335. customChapter.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  336. customChapter.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  337. customChapter.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  338. customChapter.ClassifyIdThird = reportInfo.ClassifyIdThird
  339. customChapter.ClassifyNameThird = reportInfo.ClassifyNameThird
  340. customChapter.Stage = reportInfo.Stage
  341. customChapter.PublishState = 1
  342. customChapter.CreateTime = reportInfo.CreateTime
  343. customChapter.ModifyTime = time.Now()
  344. customChapter.LastModifyAdminId = reportInfo.LastModifyAdminId
  345. customChapter.LastModifyAdminName = reportInfo.LastModifyAdminName
  346. customChapter.ContentModifyTime = time.Now()
  347. customAddChapter := models.AddReportChapter{
  348. ReportChapter: customChapter,
  349. GrantList: tmpGrantList,
  350. }
  351. customAddChapterList = append(customAddChapterList, customAddChapter)
  352. }
  353. }
  354. // 最大排序
  355. var maxSort int
  356. for _, typeItem := range typeList {
  357. v := inheritChapterMap[typeItem.ReportChapterTypeId]
  358. chapterItem := new(models.ReportChapter)
  359. if v != nil {
  360. chapterItem.AddType = 2
  361. chapterItem.Title = v.Title
  362. chapterItem.ReportType = v.ReportType
  363. chapterItem.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  364. chapterItem.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  365. chapterItem.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  366. chapterItem.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  367. chapterItem.ClassifyIdThird = reportInfo.ClassifyIdThird
  368. chapterItem.ClassifyNameThird = reportInfo.ClassifyNameThird
  369. chapterItem.TypeId = typeItem.ReportChapterTypeId
  370. chapterItem.TypeName = typeItem.ReportChapterTypeName
  371. chapterItem.Content = v.Content
  372. chapterItem.ContentSub = v.ContentSub
  373. chapterItem.Stage = reportInfo.Stage
  374. chapterItem.PublishState = 1
  375. chapterItem.Sort = typeItem.Sort
  376. chapterItem.CreateTime = reportInfo.CreateTime
  377. chapterItem.ModifyTime = time.Now()
  378. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  379. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  380. chapterItem.ContentModifyTime = time.Now()
  381. chapterItem.ContentStruct = v.ContentStruct
  382. chapterItem.CanvasColor = v.CanvasColor
  383. chapterItem.HeadResourceId = v.HeadResourceId
  384. chapterItem.EndResourceId = v.EndResourceId
  385. chapterItem.CollaborateType = v.CollaborateType
  386. chapterItem.ReportLayout = v.ReportLayout
  387. chapterItem.ReportCreateTime = time.Now()
  388. } else {
  389. chapterItem.AddType = 1
  390. chapterItem.Title = typeItem.ReportChapterTypeName
  391. //chapterItem.ReportType = reportType
  392. chapterItem.ClassifyIdFirst = reportInfo.ClassifyIdFirst
  393. chapterItem.ClassifyNameFirst = reportInfo.ClassifyNameFirst
  394. chapterItem.ClassifyIdSecond = reportInfo.ClassifyIdSecond
  395. chapterItem.ClassifyNameSecond = reportInfo.ClassifyNameSecond
  396. chapterItem.ClassifyIdThird = reportInfo.ClassifyIdThird
  397. chapterItem.ClassifyNameThird = reportInfo.ClassifyNameThird
  398. chapterItem.TypeId = typeItem.ReportChapterTypeId
  399. chapterItem.TypeName = typeItem.ReportChapterTypeName
  400. chapterItem.Stage = reportInfo.Stage
  401. chapterItem.PublishState = 1
  402. chapterItem.Sort = typeItem.Sort
  403. chapterItem.CreateTime = reportInfo.CreateTime
  404. chapterItem.ModifyTime = time.Now()
  405. chapterItem.LastModifyAdminId = reportInfo.LastModifyAdminId
  406. chapterItem.LastModifyAdminName = reportInfo.LastModifyAdminName
  407. chapterItem.ContentModifyTime = time.Now()
  408. //chapterItem.ContentStruct = v.ContentStruct
  409. //chapterItem.HeadImg = v.HeadImg
  410. //chapterItem.EndImg = v.EndImg
  411. //chapterItem.CanvasColor = v.CanvasColor
  412. //chapterItem.HeadResourceId = v.HeadResourceId
  413. //chapterItem.EndResourceId = v.EndResourceId
  414. chapterItem.CollaborateType = reportInfo.CollaborateType
  415. chapterItem.ReportLayout = reportInfo.ReportLayout
  416. chapterItem.ReportCreateTime = time.Now()
  417. }
  418. if typeItem.Sort > maxSort {
  419. maxSort = typeItem.Sort
  420. }
  421. // 授权用户列表
  422. tmpGrantList, ok := typeGrantListMap[chapterItem.TypeId]
  423. if !ok {
  424. tmpGrantList = make([]*report.ReportChapterGrant, 0)
  425. }
  426. addChapter := models.AddReportChapter{
  427. ReportChapter: chapterItem,
  428. GrantList: tmpGrantList,
  429. }
  430. chapterList = append(chapterList, addChapter)
  431. }
  432. // 将自定义的章节内容添加到待生成的章节内容中
  433. for _, addChapterItem := range customAddChapterList {
  434. maxSort++
  435. addChapterItem.ReportChapter.Sort = maxSort
  436. chapterList = append(chapterList, addChapterItem)
  437. }
  438. hasGrantUserMap := make(map[int]bool)
  439. for _, grantList := range typeGrantListMap {
  440. for _, grant := range grantList {
  441. if _, ok := hasGrantUserMap[grant.AdminId]; !ok {
  442. allGrantUserList = append(allGrantUserList, &report.ReportGrant{
  443. //GrantId: 0,
  444. //ReportId: 0,
  445. AdminId: grant.AdminId,
  446. CreateTime: time.Now(),
  447. })
  448. }
  449. }
  450. }
  451. return
  452. }
  453. // AddChapterBaseInfoAndPermission
  454. // @Description: 添加章节基本信息及权限
  455. // @author: Roc
  456. // @datetime 2024-06-11 15:35:23
  457. // @param reportInfo *models.Report
  458. // @param reportChapterInfo *models.ReportChapter
  459. // @param permissionIdList []int
  460. // @param adminIdList []int
  461. // @return err error
  462. // @return errMsg string
  463. func AddChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  464. errMsg = "新增失败"
  465. if reportInfo.State == 2 {
  466. errMsg = "该报告已发布,不允许编辑"
  467. err = errors.New(errMsg)
  468. return
  469. }
  470. // 需要添加的报告章节授权数据
  471. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  472. for _, adminId := range adminIdList {
  473. //新增授权
  474. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  475. //GrantId: 0,
  476. //ReportChapterId: reportChapterInfo.ReportChapterId,
  477. AdminId: adminId,
  478. CreateTime: time.Now(),
  479. })
  480. }
  481. // 需要添加的报告章节品种权限数据
  482. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  483. // 品种权限
  484. for _, permissionId := range permissionIdList {
  485. // 如果不存在,那么就新增品种权限配置
  486. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  487. //ReportChapterPermissionMappingId: 0,
  488. //ReportChapterId: reportChapterInfo.ReportChapterId,
  489. ChartPermissionId: permissionId,
  490. CreateTime: time.Now(),
  491. })
  492. }
  493. err = models.AddChapterBaseInfoAndPermission(reportChapterInfo, addChapterAdminList, addChapterPermissionList)
  494. return
  495. }
  496. // EditChapterBaseInfoAndPermission
  497. // @Description: 修改报告章节的基础信息、授权用户权限、品种权限
  498. // @author: Roc
  499. // @datetime 2024-06-05 11:49:11
  500. // @param reportInfo *models.Report
  501. // @param reportChapterInfo *models.ReportChapter
  502. // @param title string
  503. // @param permissionIdList []int
  504. // @param adminIdList []int
  505. // @return err error
  506. // @return errMsg string
  507. func EditChapterBaseInfoAndPermission(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, title string, permissionIdList []int, adminIdList []int) (err error, errMsg string) {
  508. errMsg = "修改失败"
  509. if reportInfo.State == 2 {
  510. errMsg = "该报告已发布,不允许编辑"
  511. err = errors.New(errMsg)
  512. return
  513. }
  514. updateCols := make([]string, 0)
  515. // 如果标题内容,那么就修改
  516. if title != `` {
  517. reportChapterInfo.Title = title
  518. reportChapterInfo.ModifyTime = time.Now()
  519. updateCols = append(updateCols, "Title", "ModifyTime")
  520. reportChapterInfo.UpdateChapter(updateCols)
  521. }
  522. reportGrantObj := report.ReportGrant{}
  523. chapterGrantObj := report.ReportChapterGrant{}
  524. chapterPermissionObj := report.ReportChapterPermissionMapping{}
  525. // 报告授权的用户map
  526. reportGrantAdminIdMap := make(map[int]bool)
  527. // 获取报告授权的用户信息
  528. {
  529. // 获取当前报告已经授权的用户信息
  530. reportGrantList, tmpErr := reportGrantObj.GetGrantListById(reportInfo.Id)
  531. if tmpErr != nil {
  532. err = tmpErr
  533. return
  534. }
  535. for _, v := range reportGrantList {
  536. reportGrantAdminIdMap[v.AdminId] = true
  537. }
  538. }
  539. // 需要添加的报告章节授权数据
  540. addChapterAdminList := make([]*report.ReportChapterGrant, 0)
  541. // 待移除的报告章节授权数据id
  542. delReportChapterGrantIdList := make([]int, 0)
  543. // 处理当前报告章节需要新增/移除的授权信息
  544. {
  545. // 获取当前章节已经授权的用户信息
  546. chapterGrantList, tmpErr := chapterGrantObj.GetGrantListById(reportChapterInfo.ReportChapterId)
  547. if tmpErr != nil {
  548. err = tmpErr
  549. return
  550. }
  551. // 当前章节已经授权的用户信息
  552. currChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  553. // 需要删除的报告章节授权数据
  554. delChapterAdminMap := make(map[int]*report.ReportChapterGrant)
  555. for _, v := range chapterGrantList {
  556. currChapterAdminMap[v.AdminId] = v
  557. delChapterAdminMap[v.AdminId] = v
  558. }
  559. for _, adminId := range adminIdList {
  560. // 如果该用户 不在 报告授权的用户map 里面,说明这个用户是要移除的
  561. if _, ok := reportGrantAdminIdMap[adminId]; !ok {
  562. continue
  563. }
  564. _, ok := currChapterAdminMap[adminId]
  565. // 如果存在,那么从 “需要删除的报告章节授权数据” 的map中移除
  566. if ok {
  567. delete(delChapterAdminMap, adminId)
  568. continue
  569. }
  570. // 如果不存在,那么就新增授权
  571. addChapterAdminList = append(addChapterAdminList, &report.ReportChapterGrant{
  572. //GrantId: 0,
  573. ReportChapterId: reportChapterInfo.ReportChapterId,
  574. AdminId: adminId,
  575. CreateTime: time.Now(),
  576. })
  577. }
  578. // 查出需要移除的授权id
  579. for _, v := range delChapterAdminMap {
  580. delReportChapterGrantIdList = append(delReportChapterGrantIdList, v.GrantId)
  581. }
  582. }
  583. // 其他章节授权的用户
  584. otherChapterAdminMap := make(map[int]bool)
  585. {
  586. // 获取报告所有的章节id
  587. reportChapterIdList, tmpErr := models.GetReportChapterIdList(reportInfo.Id)
  588. if tmpErr != nil {
  589. err = tmpErr
  590. return
  591. }
  592. if len(reportChapterIdList) > 0 {
  593. list, tmpErr := chapterGrantObj.GetGrantListByIdList(reportChapterIdList)
  594. if tmpErr != nil {
  595. err = tmpErr
  596. return
  597. }
  598. for _, v := range list {
  599. // 如果是当前章节,因为涉及到重新授权,所以得过滤
  600. if v.ReportChapterId == reportChapterInfo.ReportChapterId {
  601. continue
  602. }
  603. otherChapterAdminMap[v.AdminId] = true
  604. }
  605. }
  606. }
  607. // 需要添加的报告章节品种权限数据
  608. addChapterPermissionList := make([]*report.ReportChapterPermissionMapping, 0)
  609. // 待移除的报告章节品种权限数据id
  610. delChapterPermissionMappingIdList := make([]int, 0)
  611. // 处理当前报告章节需要新增/移除的品种权限信息
  612. {
  613. // 获取当前章节已经配置的品种权限信息
  614. chapterPermissionList, tmpErr := chapterPermissionObj.GetPermissionListById(reportChapterInfo.ReportChapterId)
  615. if tmpErr != nil {
  616. err = tmpErr
  617. return
  618. }
  619. // 当前章节已经配置的品种权限信息
  620. currChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  621. // 需要删除的报告章节品种权限配置
  622. delChapterPermissionMap := make(map[int]*report.ReportChapterPermissionMapping)
  623. for _, v := range chapterPermissionList {
  624. currChapterPermissionMap[v.ChartPermissionId] = v
  625. delChapterPermissionMap[v.ChartPermissionId] = v
  626. }
  627. for _, permissionId := range permissionIdList {
  628. _, ok := currChapterPermissionMap[permissionId]
  629. // 如果存在,那么从 “需要删除的报告章节品种权限配置” 的map中移除
  630. if ok {
  631. delete(delChapterPermissionMap, permissionId)
  632. continue
  633. }
  634. // 如果不存在,那么就新增品种权限配置
  635. addChapterPermissionList = append(addChapterPermissionList, &report.ReportChapterPermissionMapping{
  636. //ReportChapterPermissionMappingId: 0,
  637. ReportChapterId: reportChapterInfo.ReportChapterId,
  638. ChartPermissionId: permissionId,
  639. CreateTime: time.Now(),
  640. })
  641. }
  642. // 查出需要移除的品种权限配置
  643. for _, v := range delChapterPermissionMap {
  644. delChapterPermissionMappingIdList = append(delChapterPermissionMappingIdList, v.ReportChapterPermissionMappingId)
  645. }
  646. }
  647. err = models.EditChapterBaseInfoAndPermission(reportChapterInfo, updateCols, addChapterAdminList, addChapterPermissionList, delReportChapterGrantIdList, delChapterPermissionMappingIdList)
  648. return
  649. }
  650. // DelChapter
  651. // @Description: 删除报告章节、授权用户权限、品种权限
  652. // @author: Roc
  653. // @datetime 2024-06-06 17:28:37
  654. // @param reportInfo *models.Report
  655. // @param reportChapterInfo *models.ReportChapter
  656. // @param sysUser *system.Admin
  657. // @return err error
  658. // @return errMsg string
  659. func DelChapter(reportInfo *models.Report, reportChapterInfo *models.ReportChapter, sysUser *system.Admin) (err error, errMsg string) {
  660. errMsg = "删除失败"
  661. if reportInfo.State == 2 {
  662. errMsg = "该报告已发布,不允许删除"
  663. err = errors.New(errMsg)
  664. return
  665. }
  666. reportInfo.LastModifyAdminId = sysUser.AdminId
  667. reportInfo.LastModifyAdminName = sysUser.RealName
  668. reportInfo.ModifyTime = time.Now()
  669. updateReportCols := []string{"LastModifyAdminId", "LastModifyAdminName", "ModifyTime"}
  670. err = models.DelChapterAndPermission(reportInfo, updateReportCols, reportChapterInfo)
  671. return
  672. }