chart_classify.go 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060
  1. package data
  2. import (
  3. "errors"
  4. "eta/eta_api/models/data_manage"
  5. "eta/eta_api/models/system"
  6. "eta/eta_api/services/data/data_manage_permission"
  7. "eta/eta_api/utils"
  8. "fmt"
  9. "strconv"
  10. "time"
  11. )
  12. func chartClassifyHaveChild(allNode []*data_manage.ChartClassifyItems, node *data_manage.ChartClassifyItems) (childs []*data_manage.ChartClassifyItems, yes bool) {
  13. for _, v := range allNode {
  14. if v.ParentId == node.ChartClassifyId {
  15. childs = append(childs, v)
  16. }
  17. }
  18. if len(childs) > 0 {
  19. yes = true
  20. }
  21. return
  22. }
  23. func ChartClassifyItemsMakeTree(sysUser *system.Admin, allNode []*data_manage.ChartClassifyItems, node *data_manage.ChartClassifyItems, permissionClassifyIdList []int, level int) {
  24. node.HaveOperaAuth = data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(node.IsJoinPermission, node.ChartClassifyId, permissionClassifyIdList)
  25. button := GetChartClassifyOpButton(sysUser, node.SysUserId, node.HaveOperaAuth)
  26. node.Button = button
  27. node.Disable = true
  28. if node.Level == level {
  29. node.Disable = false
  30. }
  31. childs, _ := chartClassifyHaveChild(allNode, node) //判断节点是否有子节点并返回
  32. if len(childs) > 0 {
  33. for _, child := range childs {
  34. child.HaveOperaAuth = data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(child.IsJoinPermission, child.ChartClassifyId, permissionClassifyIdList)
  35. childButton := GetChartClassifyOpButton(sysUser, child.SysUserId, child.HaveOperaAuth)
  36. if child.Level == 6 {
  37. childButton.AddButton = false //第六级的话,默认图表都是没有添加按钮的
  38. }
  39. child.Disable = true
  40. //if child.Level == level {
  41. // child.Disable = false
  42. //}
  43. child.Button = childButton
  44. //node.Children = append(node.Children, child)
  45. }
  46. node.Children = append(node.Children, childs[0:]...) //添加子节点
  47. for _, v := range childs {
  48. v.HaveOperaAuth = data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(v.IsJoinPermission, v.ChartClassifyId, permissionClassifyIdList)
  49. //查询子节点的子节点,并添加到子节点
  50. _, has := chartClassifyHaveChild(allNode, v)
  51. if has {
  52. ChartClassifyItemsMakeTree(sysUser, allNode, v, permissionClassifyIdList, level) //递归添加节点
  53. } else {
  54. //childrenArr := make([]*data_manage.ChartClassifyItems, 0)
  55. v.Children = nil
  56. }
  57. }
  58. } else {
  59. //childrenArr := make([]*data_manage.ChartClassifyItems, 0)
  60. node.Children = nil
  61. }
  62. }
  63. func FixChartClassify() {
  64. //新增三级分类
  65. list, err := data_manage.GetChartClassifyByLevel(2, 1)
  66. if err != nil {
  67. return
  68. }
  69. for _, v := range list {
  70. classify := new(data_manage.ChartClassify)
  71. classify.ParentId = v.ChartClassifyId
  72. classify.ChartClassifyName = v.ChartClassifyName
  73. classify.HasData = 0
  74. classify.CreateTime = time.Now()
  75. classify.ModifyTime = time.Now()
  76. if utils.RunMode == "debug" {
  77. classify.SysUserId = 11
  78. } else {
  79. classify.SysUserId = 163
  80. }
  81. classify.SysUserRealName = "admin"
  82. classify.Level = 3
  83. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  84. classify.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  85. classify.Sort = v.Sort
  86. classify.Source = utils.CHART_SOURCE_DEFAULT
  87. id, err := data_manage.AddChartClassify(classify)
  88. if err != nil {
  89. return
  90. }
  91. err = data_manage.FIXChartClassifyId(int(id), v.ChartClassifyId)
  92. if err != nil {
  93. return
  94. }
  95. }
  96. }
  97. func FixChartClassifySysUserId() {
  98. //获取所有父级分类信息
  99. list, err := data_manage.GetChartClassifyByLevel(2, 1)
  100. if err != nil {
  101. return
  102. }
  103. for _, v := range list {
  104. err := data_manage.EditChartClassifySysUser(v.ChartClassifyId, v.SysUserId, v.SysUserRealName)
  105. if err != nil {
  106. return
  107. }
  108. fmt.Println("ChartClassifyId:", v.ChartClassifyId)
  109. }
  110. fmt.Println("FixChartClassifySysUserId end")
  111. }
  112. // GetChartClassifyOpButton 获取ETA图库分类的操作权限
  113. func GetChartClassifyOpButton(sysUser *system.Admin, belongUserId int, haveOperaAuth bool) (button data_manage.ChartClassifyItemsButton) {
  114. // 没有数据权限就直接返回
  115. if !haveOperaAuth {
  116. return
  117. }
  118. //ficc管理员和超管和ficc研究员有权限创建和管理分类,可以编辑分类名称(分类名称不允许重复),可以拖动分类,改变分类顺序,可以拖动分类下模型,改变顺序,可以删除分类,若分类下有预测指标,则不允许删除;
  119. //if utils.InArrayByStr([]string{utils.ROLE_TYPE_CODE_ADMIN, utils.ROLE_TYPE_CODE_FICC_ADMIN, utils.ROLE_TYPE_CODE_RESEARCHR, utils.ROLE_TYPE_CODE_FICC_RESEARCHR}, sysUser.RoleTypeCode) {
  120. button.AddButton = true
  121. button.OpButton = true
  122. button.DeleteButton = true
  123. button.MoveButton = true
  124. //}
  125. return
  126. }
  127. // GetCanOpChartClassify 获取是否允许添加图表分类
  128. func GetCanOpChartClassify(roleTypeCode string) (canOpClassify bool) {
  129. // 如果不是 超管 或者 ficc管理员 或者 ficc研究员,那么就没有权限
  130. if utils.InArrayByStr([]string{utils.ROLE_TYPE_CODE_ADMIN, utils.ROLE_TYPE_CODE_FICC_ADMIN, utils.ROLE_TYPE_CODE_RESEARCHR, utils.ROLE_TYPE_CODE_FICC_RESEARCHR}, roleTypeCode) {
  131. canOpClassify = true
  132. }
  133. return
  134. }
  135. // GetChartOpButton 获取ETA图库的操作权限
  136. func GetChartOpButton(sysUser *system.Admin, belongUserId int, haveOperaAuth bool) (button data_manage.ChartClassifyItemsButton) {
  137. // 数据权限判断
  138. if !haveOperaAuth {
  139. return
  140. }
  141. //预测指标的添加人对该预测指标有全部操作权限,ficc管理员、超管对所有预测指标有全部操作权限;
  142. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId || sysUser.PredictEdbPermission == 1 {
  143. button.AddButton = true
  144. button.OpButton = true
  145. button.DeleteButton = true
  146. button.MoveButton = true
  147. }
  148. return
  149. }
  150. // HandleNoPermissionChart 图表列表返回,将没有权限的图表移除
  151. func HandleNoPermissionChart(allNodes []*data_manage.ChartClassifyItems, noPermissionChartIdMap map[int]bool, sysUserId int) (newAllNodes []*data_manage.ChartClassifyItems) {
  152. // 涉及分类id
  153. classifyIdList := make([]int, 0)
  154. // 移除没有权限的图表
  155. newAllNodes = make([]*data_manage.ChartClassifyItems, 0)
  156. for _, node := range allNodes {
  157. // 二级分类
  158. tmpNodeInfo := *node
  159. tmpNodeList := make([]*data_manage.ChartClassifyItems, 0)
  160. if node.Children != nil {
  161. for _, chartInfo := range node.Children {
  162. // 如果指标不可见,那么就不返回该指标
  163. if _, ok := noPermissionChartIdMap[chartInfo.ChartInfoId]; ok {
  164. continue
  165. }
  166. tmpNodeList = append(tmpNodeList, chartInfo)
  167. classifyIdList = append(classifyIdList, chartInfo.ChartClassifyId)
  168. }
  169. }
  170. tmpNodeInfo.Children = tmpNodeList
  171. newAllNodes = append(newAllNodes, &tmpNodeInfo)
  172. classifyIdList = append(classifyIdList, node.ChartClassifyId)
  173. }
  174. // 数据权限
  175. if len(classifyIdList) > 0 {
  176. // 图表分类
  177. chartClassifyList, err := data_manage.GetChartClassifyByIdList(classifyIdList)
  178. if err != nil {
  179. return
  180. }
  181. chartClassifyMap := make(map[int]*data_manage.ChartClassify)
  182. for _, v := range chartClassifyList {
  183. chartClassifyMap[v.ChartClassifyId] = v
  184. }
  185. // 已授权分类id
  186. permissionChartIdList, permissionClassifyIdList, err := data_manage_permission.GetUserChartAndClassifyPermissionList(sysUserId, 0, 0)
  187. if err != nil {
  188. return
  189. }
  190. for _, node := range newAllNodes {
  191. if node.Children != nil {
  192. for _, chartInfo := range node.Children {
  193. if currClassify, ok := chartClassifyMap[chartInfo.ChartClassifyId]; ok {
  194. chartInfo.HaveOperaAuth = data_manage_permission.CheckChartPermissionByPermissionIdList(chartInfo.IsJoinPermission, currClassify.IsJoinPermission, chartInfo.ChartInfoId, chartInfo.ChartClassifyId, permissionChartIdList, permissionClassifyIdList)
  195. }
  196. }
  197. }
  198. if currClassify, ok := chartClassifyMap[node.ChartClassifyId]; ok {
  199. node.HaveOperaAuth = data_manage_permission.CheckChartPermissionByPermissionIdList(node.IsJoinPermission, currClassify.IsJoinPermission, node.ChartInfoId, node.ChartClassifyId, permissionChartIdList, permissionClassifyIdList)
  200. }
  201. }
  202. }
  203. return
  204. }
  205. // AddChartClassify
  206. // @Description: 添加图表分类
  207. // @author: Roc
  208. // @datetime 2024-04-17 14:48:23
  209. // @param chartClassifyName string
  210. // @param parentId int
  211. // @param level int
  212. // @param source int
  213. // @param lang string
  214. // @param sysUser *system.Admin
  215. // @return classifyInfo *data_manage.ChartClassify
  216. // @return err error
  217. // @return errMsg string
  218. // @return isSendEmail bool
  219. func AddChartClassify(chartClassifyName string, parentId, level, source int, lang string, sysUser *system.Admin) (classifyInfo *data_manage.ChartClassify, err error, errMsg string, isSendEmail bool) {
  220. isSendEmail = true
  221. errMsg = "保存分类失败"
  222. // 校验分类名称相同的数量
  223. {
  224. var count int
  225. switch lang {
  226. case utils.EnLangVersion:
  227. count, err = data_manage.GetChartClassifyEnCount(chartClassifyName, parentId, source)
  228. default:
  229. count, err = data_manage.GetChartClassifyCount(chartClassifyName, parentId, source)
  230. }
  231. if err != nil {
  232. errMsg = "判断名称是否已存在失败"
  233. return
  234. }
  235. if count > 0 {
  236. errMsg = "分类名称已存在,请重新输入"
  237. err = errors.New(errMsg)
  238. isSendEmail = false
  239. return
  240. }
  241. }
  242. if level > 6 {
  243. errMsg = `最高只支持添加6级分类`
  244. return
  245. }
  246. //获取该层级下最大的排序数
  247. maxSort, err := data_manage.GetChartClassifyMaxSort(parentId, source)
  248. if err != nil {
  249. errMsg = "获取失败"
  250. err = errors.New("查询排序信息失败,Err:" + err.Error())
  251. return
  252. }
  253. //查询顶级rootId
  254. rootId := 0
  255. if parentId > 0 {
  256. parentClassify, tErr := data_manage.GetChartClassifyById(parentId)
  257. if tErr != nil {
  258. if tErr.Error() == utils.ErrNoRow() {
  259. errMsg = "父级分类不存在"
  260. err = errors.New(errMsg)
  261. return
  262. }
  263. errMsg = "获取失败"
  264. err = errors.New("获取分类信息失败,Err:" + tErr.Error())
  265. return
  266. }
  267. rootId = parentClassify.RootId
  268. }
  269. classifyInfo = new(data_manage.ChartClassify)
  270. classifyInfo.ParentId = parentId
  271. classifyInfo.ChartClassifyName = chartClassifyName
  272. classifyInfo.ChartClassifyNameEn = chartClassifyName
  273. classifyInfo.HasData = 0
  274. classifyInfo.CreateTime = time.Now()
  275. classifyInfo.ModifyTime = time.Now()
  276. classifyInfo.SysUserId = sysUser.AdminId
  277. classifyInfo.SysUserRealName = sysUser.RealName
  278. classifyInfo.Level = level + 1
  279. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  280. classifyInfo.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  281. classifyInfo.Sort = maxSort + 1
  282. classifyInfo.Source = source
  283. classifyInfo.RootId = rootId
  284. newId, err := data_manage.AddChartClassify(classifyInfo)
  285. if err != nil {
  286. return
  287. }
  288. if parentId == 0 { //一级目录的rootId等于自己本身
  289. classifyInfo.RootId = int(newId)
  290. err = classifyInfo.Update([]string{"RootId"})
  291. if err != nil {
  292. errMsg = "更新分类失败"
  293. return
  294. }
  295. }
  296. // 目前只有ETA图库需要继承分类权限
  297. if classifyInfo.Source == utils.CHART_SOURCE_DEFAULT {
  298. go data_manage_permission.InheritParentClassify(5, classifyInfo.Source, classifyInfo.ChartClassifyId, classifyInfo.ParentId, classifyInfo.ChartClassifyName)
  299. }
  300. return
  301. }
  302. // EditChartClassify
  303. // @Description: 修改图表分类
  304. // @author: Roc
  305. // @datetime 2024-04-17 14:59:37
  306. // @param chartClassifyId int
  307. // @param source int
  308. // @param chartClassifyName string
  309. // @param lang string
  310. // @param sysUser *system.Admin
  311. // @return classifyInfo *data_manage.ChartClassify
  312. // @return err error
  313. // @return errMsg string
  314. // @return isSendEmail bool
  315. func EditChartClassify(chartClassifyId, source int, chartClassifyName, lang string, sysUser *system.Admin) (classifyInfo *data_manage.ChartClassify, err error, errMsg string, isSendEmail bool) {
  316. isSendEmail = true
  317. errMsg = "保存失败"
  318. // 获取分类信息
  319. classifyInfo, err = data_manage.GetChartClassifyById(chartClassifyId)
  320. if err != nil {
  321. return
  322. }
  323. // 分类来源校验
  324. if classifyInfo.Source != source {
  325. errMsg = "图表分类异常"
  326. err = errors.New(fmt.Sprintf("图表分类来源异常,修改来源:%d,分类来源:%d", source, classifyInfo.Source))
  327. isSendEmail = false
  328. return
  329. }
  330. // 需要变更的字段
  331. updateCols := make([]string, 0)
  332. // 语言版本校验
  333. switch lang {
  334. case utils.EnLangVersion:
  335. if classifyInfo.ChartClassifyNameEn != chartClassifyName {
  336. count, tmpErr := data_manage.GetChartClassifyEnCount(chartClassifyName, classifyInfo.ParentId, source)
  337. if tmpErr != nil {
  338. errMsg = "判断名称是否已存在失败"
  339. err = errors.New("判断名称是否已存在失败,Err:" + tmpErr.Error())
  340. return
  341. }
  342. if count > 0 {
  343. errMsg = "分类名称已存在,请重新输入"
  344. err = errors.New(errMsg)
  345. isSendEmail = false
  346. return
  347. }
  348. classifyInfo.ChartClassifyNameEn = chartClassifyName
  349. classifyInfo.ModifyTime = time.Now()
  350. updateCols = append(updateCols, "ChartClassifyNameEn", "ModifyTime")
  351. }
  352. default:
  353. if classifyInfo.ChartClassifyName != chartClassifyName {
  354. count, tmpErr := data_manage.GetChartClassifyCount(chartClassifyName, classifyInfo.ParentId, source)
  355. if tmpErr != nil {
  356. errMsg = "判断名称是否已存在失败"
  357. err = errors.New("判断名称是否已存在失败,Err:" + tmpErr.Error())
  358. return
  359. }
  360. if count > 0 {
  361. errMsg = "分类名称已存在,请重新输入"
  362. err = errors.New(errMsg)
  363. isSendEmail = false
  364. return
  365. }
  366. classifyInfo.ChartClassifyName = chartClassifyName
  367. classifyInfo.ModifyTime = time.Now()
  368. updateCols = append(updateCols, "ChartClassifyName", "ModifyTime")
  369. }
  370. }
  371. if len(updateCols) > 0 {
  372. err = classifyInfo.Update(updateCols)
  373. }
  374. return
  375. }
  376. // MoveChartClassify 移动图表分类
  377. func MoveChartClassify(req data_manage.MoveChartClassifyReq, sysUser *system.Admin, source int) (err error, errMsg string) {
  378. classifyId := req.ClassifyId
  379. parentClassifyId := req.ParentClassifyId
  380. prevClassifyId := req.PrevClassifyId
  381. nextClassifyId := req.NextClassifyId
  382. chartInfoId := req.ChartInfoId
  383. prevChartInfoId := req.PrevChartInfoId
  384. nextChartInfoId := req.NextChartInfoId
  385. //首先确定移动的对象是分类还是指标
  386. //判断上一个节点是分类还是指标
  387. //判断下一个节点是分类还是指标
  388. //同时更新分类目录下的分类sort和指标sort
  389. //更新当前移动的分类或者指标sort
  390. var parentChartClassifyInfo *data_manage.ChartClassify
  391. if parentClassifyId > 0 {
  392. parentChartClassifyInfo, err = data_manage.GetChartClassifyById(parentClassifyId)
  393. if err != nil {
  394. errMsg = "移动失败"
  395. err = errors.New("获取上级分类信息失败,Err:" + err.Error())
  396. return
  397. }
  398. }
  399. //如果有传入 上一个兄弟节点分类id
  400. var (
  401. chartClassifyInfo *data_manage.ChartClassify
  402. prevClassify *data_manage.ChartClassify
  403. nextClassify *data_manage.ChartClassify
  404. chartInfo *data_manage.ChartInfo
  405. prevChartInfo *data_manage.ChartInfo
  406. nextChartInfo *data_manage.ChartInfo
  407. prevSort int
  408. nextSort int
  409. )
  410. // 移动对象为分类, 判断权限
  411. if chartInfoId == 0 {
  412. chartClassifyInfo, err = data_manage.GetChartClassifyById(classifyId)
  413. if err != nil {
  414. if err.Error() == utils.ErrNoRow() {
  415. errMsg = "当前分类不存在"
  416. err = errors.New("获取分类信息失败,Err:" + err.Error())
  417. return
  418. }
  419. errMsg = "移动失败"
  420. err = errors.New("获取分类信息失败,Err:" + err.Error())
  421. return
  422. }
  423. if parentClassifyId > 0 && parentChartClassifyInfo.Level == 6 {
  424. errMsg = "最高只支持添加6级分类"
  425. err = errors.New(errMsg)
  426. return
  427. }
  428. // 如果是移动目录, 那么校验一下父级目录下是否有重名目录
  429. exists, e := data_manage.GetChartClassifyByParentIdAndName(parentClassifyId, chartClassifyInfo.ChartClassifyName, classifyId)
  430. if e != nil && e.Error() != utils.ErrNoRow() {
  431. errMsg = "移动失败"
  432. err = fmt.Errorf("获取父级分类下的同名分类失败, Err: %s", e.Error())
  433. return
  434. }
  435. if exists != nil {
  436. errMsg = "移动失败,分类名称已存在"
  437. return
  438. }
  439. // 权限校验
  440. {
  441. // 已授权分类id
  442. permissionClassifyIdList, tmpErr := data_manage_permission.GetUserChartClassifyPermissionList(sysUser.AdminId, classifyId)
  443. if tmpErr != nil {
  444. errMsg = "移动失败"
  445. err = errors.New("获取已授权分类id数据失败,Err:" + tmpErr.Error())
  446. return
  447. }
  448. // 数据权限
  449. haveOperaAuth := data_manage_permission.CheckChartClassifyPermissionByPermissionIdList(chartClassifyInfo.IsJoinPermission, chartClassifyInfo.ChartClassifyId, permissionClassifyIdList)
  450. button := GetChartClassifyOpButton(sysUser, chartClassifyInfo.SysUserId, haveOperaAuth)
  451. if !button.MoveButton {
  452. errMsg = "无操作权限"
  453. err = errors.New(errMsg)
  454. return
  455. }
  456. }
  457. } else {
  458. chartInfo, err = data_manage.GetChartInfoById(req.ChartInfoId)
  459. if err != nil {
  460. if err.Error() == utils.ErrNoRow() {
  461. errMsg = "当前图表不存在"
  462. err = errors.New("获取分类信息失败,Err:" + err.Error())
  463. return
  464. }
  465. errMsg = "移动失败"
  466. err = errors.New("获取分类信息失败,Err:" + err.Error())
  467. return
  468. }
  469. if parentClassifyId == 0 {
  470. errMsg = "移动失败,图表必须挂在分类下"
  471. err = errors.New(errMsg)
  472. return
  473. }
  474. var haveOperaAuth bool
  475. // 权限校验
  476. {
  477. haveOperaAuth, err = data_manage_permission.CheckChartPermissionByChartInfoId(chartInfo.ChartInfoId, chartInfo.ChartClassifyId, chartInfo.IsJoinPermission, sysUser.AdminId)
  478. if err != nil {
  479. errMsg = "移动失败"
  480. err = errors.New("获取图表权限信息失败,Err:" + err.Error())
  481. return
  482. }
  483. }
  484. // 移动权限校验
  485. button := GetChartOpButton(sysUser, chartInfo.SysUserId, haveOperaAuth)
  486. if !button.MoveButton {
  487. errMsg = "无操作权限"
  488. err = errors.New(errMsg)
  489. return
  490. }
  491. }
  492. if prevClassifyId > 0 {
  493. prevClassify, err = data_manage.GetChartClassifyById(prevClassifyId)
  494. if err != nil {
  495. errMsg = "移动失败"
  496. err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
  497. return
  498. }
  499. prevSort = prevClassify.Sort
  500. } else if prevChartInfoId > 0 {
  501. prevChartInfo, err = data_manage.GetChartInfoById(prevChartInfoId)
  502. if err != nil {
  503. errMsg = "移动失败"
  504. err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
  505. return
  506. }
  507. prevSort = prevChartInfo.Sort
  508. }
  509. if nextClassifyId > 0 {
  510. //下一个兄弟节点
  511. nextClassify, err = data_manage.GetChartClassifyById(nextClassifyId)
  512. if err != nil {
  513. errMsg = "移动失败"
  514. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
  515. return
  516. }
  517. nextSort = nextClassify.Sort
  518. } else if nextChartInfoId > 0 {
  519. //下一个兄弟节点
  520. nextChartInfo, err = data_manage.GetChartInfoById(nextChartInfoId)
  521. if err != nil {
  522. errMsg = "移动失败"
  523. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
  524. return
  525. }
  526. nextSort = nextChartInfo.Sort
  527. }
  528. err, errMsg = moveChartClassify(parentChartClassifyInfo, chartClassifyInfo, prevClassify, nextClassify, chartInfo, prevChartInfo, nextChartInfo, parentClassifyId, prevSort, nextSort, source)
  529. return
  530. }
  531. // moveChartClassify 移动图表分类
  532. func moveChartClassify(parentChartClassifyInfo, chartClassifyInfo, prevClassify, nextClassify *data_manage.ChartClassify, chartInfo, prevChartInfo, nextChartInfo *data_manage.ChartInfo, parentClassifyId, prevSort, nextSort, source int) (err error, errMsg string) {
  533. updateCol := make([]string, 0)
  534. // 移动对象为分类, 判断分类是否存在
  535. if chartClassifyInfo != nil {
  536. oldParentId := chartClassifyInfo.ParentId
  537. oldLevel := chartClassifyInfo.Level
  538. var classifyIds []int
  539. if oldParentId != parentClassifyId {
  540. //更新子分类对应的level
  541. childList, e, m := GetChildChartClassifyByClassifyId(chartClassifyInfo.ChartClassifyId)
  542. if e != nil {
  543. errMsg = "移动失败"
  544. err = errors.New("查询子分类失败,Err:" + e.Error() + m)
  545. return
  546. }
  547. if len(childList) > 0 {
  548. for _, v := range childList {
  549. if v.ChartClassifyId == chartClassifyInfo.ChartClassifyId {
  550. continue
  551. }
  552. classifyIds = append(classifyIds, v.ChartClassifyId)
  553. }
  554. }
  555. }
  556. //判断上级id是否一致,如果不一致的话,那么需要移动该分类层级
  557. if chartClassifyInfo.ParentId != parentClassifyId && parentClassifyId != 0 {
  558. if chartClassifyInfo.Level != parentChartClassifyInfo.Level+1 { //禁止层级调整
  559. errMsg = "移动失败"
  560. err = errors.New("不支持目录层级变更")
  561. return
  562. }
  563. chartClassifyInfo.ParentId = parentChartClassifyInfo.ChartClassifyId
  564. chartClassifyInfo.RootId = parentChartClassifyInfo.RootId
  565. chartClassifyInfo.Level = parentChartClassifyInfo.Level + 1
  566. chartClassifyInfo.ModifyTime = time.Now()
  567. updateCol = append(updateCol, "ParentId", "RootId", "Level", "ModifyTime")
  568. } else if chartClassifyInfo.ParentId != parentClassifyId && parentClassifyId == 0 {
  569. errMsg = "移动失败"
  570. err = errors.New("不支持目录层级变更")
  571. return
  572. }
  573. if prevSort > 0 {
  574. //如果是移动在两个兄弟节点之间
  575. if nextSort > 0 {
  576. //下一个兄弟节点
  577. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  578. if prevSort == nextSort || prevSort == chartClassifyInfo.Sort {
  579. //变更兄弟节点的排序
  580. updateSortStr := `sort + 2`
  581. //变更分类
  582. if prevClassify != nil {
  583. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, prevClassify.ChartClassifyId, prevClassify.Sort, updateSortStr, source)
  584. } else {
  585. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, prevSort, updateSortStr, source)
  586. }
  587. //变更图表
  588. if prevChartInfo != nil {
  589. //变更兄弟节点的排序
  590. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, prevChartInfo.ChartInfoId, updateSortStr)
  591. } else {
  592. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, 0, updateSortStr)
  593. }
  594. } else {
  595. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  596. if nextSort-prevSort == 1 {
  597. //变更兄弟节点的排序
  598. updateSortStr := `sort + 1`
  599. //变更分类
  600. if prevClassify != nil {
  601. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, prevClassify.ChartClassifyId, prevSort, updateSortStr, source)
  602. } else {
  603. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, prevSort, updateSortStr, source)
  604. }
  605. //变更图表
  606. if prevChartInfo != nil {
  607. //变更兄弟节点的排序
  608. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, prevChartInfo.ChartInfoId, updateSortStr)
  609. } else {
  610. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, 0, updateSortStr)
  611. }
  612. }
  613. }
  614. }
  615. chartClassifyInfo.Sort = prevSort + 1
  616. chartClassifyInfo.ModifyTime = time.Now()
  617. updateCol = append(updateCol, "Sort", "ModifyTime")
  618. } else if prevClassify == nil && nextClassify == nil && prevChartInfo == nil && nextChartInfo == nil && parentClassifyId > 0 {
  619. //处理只拖动到目录里,默认放到目录底部的情况
  620. var maxSort int
  621. maxSort, err = GetChartClassifyMaxSort(parentClassifyId, source)
  622. if err != nil {
  623. errMsg = "移动失败"
  624. err = errors.New("查询组内排序信息失败,Err:" + err.Error())
  625. return
  626. }
  627. chartClassifyInfo.Sort = maxSort + 1 //那就是排在组内最后一位
  628. chartClassifyInfo.ModifyTime = time.Now()
  629. updateCol = append(updateCol, "Sort", "ModifyTime")
  630. } else {
  631. // 拖动到父级分类的第一位
  632. firstClassify, tmpErr := data_manage.GetFirstChartClassifyByParentId(parentClassifyId)
  633. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  634. errMsg = "移动失败"
  635. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tmpErr.Error())
  636. return
  637. }
  638. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  639. if firstClassify != nil && firstClassify.Sort == 0 {
  640. updateSortStr := ` sort + 1 `
  641. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, firstClassify.ChartClassifyId-1, 0, updateSortStr, source)
  642. //该分类下的所有图表也需要+1
  643. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, 0, 0, updateSortStr)
  644. } else {
  645. //如果该分类下存在图表,且第一个图表的排序等于0,那么需要调整排序
  646. firstEdb, tErr := data_manage.GetFirstChartInfoByClassifyId(parentClassifyId)
  647. if tErr != nil && tErr.Error() != utils.ErrNoRow() {
  648. errMsg = "移动失败"
  649. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tErr.Error())
  650. return
  651. }
  652. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  653. if firstEdb != nil && firstEdb.Sort == 0 {
  654. updateSortStr := ` sort + 1 `
  655. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, 0, firstEdb.ChartInfoId-1, updateSortStr)
  656. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, 0, updateSortStr, source)
  657. }
  658. }
  659. chartClassifyInfo.Sort = 0 //那就是排在第一位
  660. chartClassifyInfo.ModifyTime = time.Now()
  661. updateCol = append(updateCol, "Sort", "ModifyTime")
  662. }
  663. //更新
  664. if len(updateCol) > 0 {
  665. err = chartClassifyInfo.Update(updateCol)
  666. if err != nil {
  667. errMsg = "移动失败"
  668. err = errors.New("修改失败,Err:" + err.Error())
  669. return
  670. }
  671. //更新对应分类的root_id和层级
  672. if oldParentId != parentClassifyId {
  673. if len(classifyIds) > 0 {
  674. levelStep := chartClassifyInfo.Level - oldLevel
  675. err = data_manage.UpdateEdbClassifyChildByParentClassifyId(classifyIds, chartClassifyInfo.RootId, levelStep)
  676. if err != nil {
  677. errMsg = "移动失败"
  678. err = errors.New("更新子分类失败,Err:" + err.Error())
  679. return
  680. }
  681. }
  682. }
  683. }
  684. } else {
  685. if chartInfo == nil {
  686. errMsg = "当前图表不存在"
  687. err = errors.New(errMsg)
  688. return
  689. }
  690. //如果改变了分类,那么移动该图表数据
  691. if chartInfo.ChartClassifyId != parentClassifyId {
  692. chartInfo.ChartClassifyId = parentClassifyId
  693. chartInfo.ModifyTime = time.Now()
  694. updateCol = append(updateCol, "ChartClassifyId", "ModifyTime")
  695. }
  696. if prevSort > 0 {
  697. //如果是移动在两个兄弟节点之间
  698. if nextSort > 0 {
  699. //下一个兄弟节点
  700. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  701. if prevSort == nextSort || prevSort == chartInfo.Sort {
  702. //变更兄弟节点的排序
  703. updateSortStr := `sort + 2`
  704. //变更分类
  705. if prevClassify != nil {
  706. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, prevClassify.ChartClassifyId, prevClassify.Sort, updateSortStr, source)
  707. } else {
  708. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, prevSort, updateSortStr, source)
  709. }
  710. //变更图表
  711. if prevChartInfo != nil {
  712. //变更兄弟节点的排序
  713. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, prevChartInfo.ChartInfoId, updateSortStr)
  714. } else {
  715. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, 0, updateSortStr)
  716. }
  717. } else {
  718. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  719. if nextSort-prevSort == 1 {
  720. //变更兄弟节点的排序
  721. updateSortStr := `sort + 1`
  722. //变更分类
  723. if prevClassify != nil {
  724. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, prevClassify.ChartClassifyId, prevSort, updateSortStr, source)
  725. } else {
  726. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, prevSort, updateSortStr, source)
  727. }
  728. //变更图表
  729. if prevChartInfo != nil {
  730. //变更兄弟节点的排序
  731. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, prevChartInfo.ChartInfoId, updateSortStr)
  732. } else {
  733. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, prevSort, 0, updateSortStr)
  734. }
  735. }
  736. }
  737. }
  738. chartInfo.Sort = prevSort + 1
  739. chartInfo.ModifyTime = time.Now()
  740. updateCol = append(updateCol, "Sort", "ModifyTime")
  741. } else if prevClassify == nil && nextClassify == nil && prevChartInfo == nil && nextChartInfo == nil && parentClassifyId > 0 {
  742. //处理只拖动到目录里,默认放到目录底部的情况
  743. var maxSort int
  744. maxSort, err = GetChartClassifyMaxSort(parentClassifyId, source)
  745. if err != nil {
  746. errMsg = "移动失败"
  747. err = errors.New("查询组内排序信息失败,Err:" + err.Error())
  748. return
  749. }
  750. chartInfo.Sort = maxSort + 1 //那就是排在组内最后一位
  751. chartInfo.ModifyTime = time.Now()
  752. updateCol = append(updateCol, "Sort", "ModifyTime")
  753. } else {
  754. // 拖动到父级分类的第一位
  755. firstClassify, tmpErr := data_manage.GetFirstChartClassifyByParentId(parentClassifyId)
  756. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  757. errMsg = "移动失败"
  758. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tmpErr.Error())
  759. return
  760. }
  761. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  762. if firstClassify != nil && firstClassify.Sort == 0 {
  763. updateSortStr := ` sort + 1 `
  764. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, firstClassify.ChartClassifyId-1, 0, updateSortStr, source)
  765. //该分类下的所有图表也需要+1
  766. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, 0, 0, updateSortStr)
  767. } else {
  768. //如果该分类下存在图表,且第一个图表的排序等于0,那么需要调整排序
  769. firstEdb, tErr := data_manage.GetFirstChartInfoByClassifyId(parentClassifyId)
  770. if tErr != nil && tErr.Error() != utils.ErrNoRow() {
  771. errMsg = "移动失败"
  772. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tErr.Error())
  773. return
  774. }
  775. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  776. if firstEdb != nil && firstEdb.Sort == 0 {
  777. updateSortStr := ` sort + 1 `
  778. _ = data_manage.UpdateChartInfoSortByClassifyIdV2(parentClassifyId, 0, firstEdb.ChartInfoId-1, updateSortStr)
  779. _ = data_manage.UpdateChartClassifySortByParentIdAndSource(parentClassifyId, 0, 0, updateSortStr, source)
  780. }
  781. }
  782. chartInfo.Sort = 0 //那就是排在第一位
  783. chartInfo.ModifyTime = time.Now()
  784. updateCol = append(updateCol, "Sort", "ModifyTime")
  785. }
  786. //更新
  787. if len(updateCol) > 0 {
  788. err = chartInfo.Update(updateCol)
  789. if err != nil {
  790. errMsg = "移动失败"
  791. err = errors.New("修改失败,Err:" + err.Error())
  792. return
  793. }
  794. }
  795. }
  796. return
  797. }
  798. func GetChildChartClassifyByClassifyId(targetClassifyId int) (targetList []*data_manage.ChartClassifyIdItems, err error, errMsg string) {
  799. //判断是否是挂在顶级目录下
  800. targetClassify, err := data_manage.GetChartClassifyById(targetClassifyId)
  801. if err != nil {
  802. if err.Error() == utils.ErrNoRow() {
  803. errMsg = "当前分类不存在"
  804. err = errors.New(errMsg)
  805. return
  806. }
  807. errMsg = "获取失败"
  808. err = errors.New("获取分类信息失败,Err:" + err.Error())
  809. return
  810. }
  811. orderStr := ` order by level asc, sort asc, chart_classify_id asc`
  812. tmpList, err := data_manage.GetChartClassifyByRootIdLevel(targetClassify.RootId, targetClassify.Source, orderStr)
  813. if err != nil && err.Error() != utils.ErrNoRow() {
  814. errMsg = "获取失败"
  815. err = errors.New("获取数据失败,Err:" + err.Error())
  816. return
  817. }
  818. idMap := make(map[int]struct{})
  819. if len(tmpList) > 0 {
  820. for _, v := range tmpList {
  821. if v.ChartClassifyId == targetClassify.ChartClassifyId {
  822. idMap[v.ChartClassifyId] = struct{}{}
  823. }
  824. }
  825. for _, v := range tmpList {
  826. if _, ok := idMap[v.ParentId]; ok {
  827. idMap[v.ChartClassifyId] = struct{}{}
  828. }
  829. }
  830. for _, v := range tmpList {
  831. if _, ok := idMap[v.ChartClassifyId]; ok {
  832. targetItem := new(data_manage.ChartClassifyIdItems)
  833. targetItem.ChartClassifyId = v.ChartClassifyId
  834. targetItem.ParentId = v.ParentId
  835. targetItem.RootId = v.RootId
  836. targetItem.UniqueCode = v.UniqueCode
  837. targetItem.Level = v.Level
  838. targetItem.ChartClassifyName = v.ChartClassifyName
  839. targetItem.IsJoinPermission = v.IsJoinPermission
  840. targetList = append(targetList, targetItem)
  841. }
  842. }
  843. }
  844. return
  845. }
  846. func GetChartClassifyMaxSort(parentId, source int) (maxSort int, err error) {
  847. //获取该层级下最大的排序数
  848. classifyMaxSort, err := data_manage.GetChartClassifyMaxSort(parentId, source)
  849. if err != nil {
  850. return
  851. }
  852. maxSort = classifyMaxSort
  853. edbMaxSort, err := data_manage.GetChartInfoMaxSortByClassifyId(parentId)
  854. if err != nil {
  855. return
  856. }
  857. if maxSort < edbMaxSort {
  858. maxSort = edbMaxSort
  859. }
  860. return
  861. }
  862. // GetChartClassifyTreeRecursive 递归获取分类树形结构
  863. func GetChartClassifyTreeRecursive(list []*data_manage.ChartClassifyItems, parentId int) []*data_manage.ChartClassifyItems {
  864. res := make([]*data_manage.ChartClassifyItems, 0)
  865. for _, v := range list {
  866. if v.ParentId == parentId {
  867. v.Children = GetChartClassifyTreeRecursive(list, v.ChartClassifyId)
  868. res = append(res, v)
  869. }
  870. }
  871. return res
  872. }
  873. // GetChartClassifyParentRecursive 根据子目录递归父级目录
  874. func GetChartClassifyParentRecursive(list []*data_manage.ChartClassifyItems, classifyId int) []*data_manage.ChartClassifyItems {
  875. res := make([]*data_manage.ChartClassifyItems, 0)
  876. for _, v := range list {
  877. if v.ChartClassifyId == classifyId {
  878. if v.ParentId > 0 {
  879. res = GetChartClassifyParentRecursive(list, v.ParentId)
  880. }
  881. res = append(res, v)
  882. }
  883. }
  884. return res
  885. }
  886. // 修改图表分类,可以修改父级
  887. func EditChartClassifyV2(chartClassifyId, praentId, source int, chartClassifyName, lang string) (classifyInfo *data_manage.ChartClassify, err error, errMsg string, isSendEmail bool) {
  888. isSendEmail = true
  889. errMsg = "保存失败"
  890. // 获取分类信息
  891. classifyInfo, err = data_manage.GetChartClassifyById(chartClassifyId)
  892. if err != nil {
  893. return
  894. }
  895. if praentId != classifyInfo.ParentId {
  896. parentClassifyInfo, e := data_manage.GetChartClassifyById(praentId)
  897. if e != nil {
  898. err = e
  899. return
  900. }
  901. if classifyInfo.Level != parentClassifyInfo.Level - 1 {
  902. err = errors.New("父级分类层级异常")
  903. }
  904. }
  905. // 分类来源校验
  906. if classifyInfo.Source != source {
  907. errMsg = "图表分类异常"
  908. err = errors.New(fmt.Sprintf("图表分类来源异常,修改来源:%d,分类来源:%d", source, classifyInfo.Source))
  909. isSendEmail = false
  910. return
  911. }
  912. // 需要变更的字段
  913. updateCols := make([]string, 0)
  914. if praentId != classifyInfo.ParentId {
  915. classifyInfo.ParentId = praentId
  916. classifyInfo.ModifyTime = time.Now()
  917. updateCols = append(updateCols, "ParentId")
  918. }
  919. // 语言版本校验
  920. switch lang {
  921. case utils.EnLangVersion:
  922. if classifyInfo.ChartClassifyNameEn != chartClassifyName {
  923. count, tmpErr := data_manage.GetChartClassifyEnCount(chartClassifyName, classifyInfo.ParentId, source)
  924. if tmpErr != nil {
  925. errMsg = "判断名称是否已存在失败"
  926. err = errors.New("判断名称是否已存在失败,Err:" + tmpErr.Error())
  927. return
  928. }
  929. if count > 0 {
  930. errMsg = "分类名称已存在,请重新输入"
  931. err = errors.New(errMsg)
  932. isSendEmail = false
  933. return
  934. }
  935. classifyInfo.ChartClassifyNameEn = chartClassifyName
  936. classifyInfo.ModifyTime = time.Now()
  937. updateCols = append(updateCols, "ChartClassifyNameEn", "ModifyTime")
  938. }
  939. default:
  940. if classifyInfo.ChartClassifyName != chartClassifyName {
  941. count, tmpErr := data_manage.GetChartClassifyCount(chartClassifyName, classifyInfo.ParentId, source)
  942. if tmpErr != nil {
  943. errMsg = "判断名称是否已存在失败"
  944. err = errors.New("判断名称是否已存在失败,Err:" + tmpErr.Error())
  945. return
  946. }
  947. if count > 0 {
  948. errMsg = "分类名称已存在,请重新输入"
  949. err = errors.New(errMsg)
  950. isSendEmail = false
  951. return
  952. }
  953. classifyInfo.ChartClassifyName = chartClassifyName
  954. classifyInfo.ModifyTime = time.Now()
  955. updateCols = append(updateCols, "ChartClassifyName", "ModifyTime")
  956. }
  957. }
  958. if len(updateCols) > 0 {
  959. err = classifyInfo.Update(updateCols)
  960. }
  961. return
  962. }