chart_classify.go 34 KB

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