edb_classify.go 57 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683
  1. package data
  2. import (
  3. "errors"
  4. "eta/eta_api/models"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/data_manage/cross_variety"
  7. "eta/eta_api/models/data_manage/excel"
  8. "eta/eta_api/models/system"
  9. "eta/eta_api/services/data/data_manage_permission"
  10. "eta/eta_api/services/data_stat"
  11. "eta/eta_api/utils"
  12. "fmt"
  13. "strconv"
  14. "time"
  15. )
  16. func EdbClassifyListMakeTree(allNode []*data_manage.EdbClassifyItems, node *data_manage.EdbClassifyItems) {
  17. childs, _ := edbClassifyHaveChild(allNode, node) //判断节点是否有子节点并返回
  18. allEdbInfo, _ := GetEdbClassifyEdbInfo()
  19. if len(childs) > 0 {
  20. node.Children = append(node.Children, childs[0:]...) //添加子节点
  21. for _, v := range childs { //查询子节点的子节点,并添加到子节点
  22. _, has := edbClassifyHaveChild(allNode, v)
  23. if has {
  24. EdbClassifyListMakeTree(allNode, v) //递归添加节点
  25. } else {
  26. edbInfoList := allEdbInfo[v.ClassifyId]
  27. v.Children = edbInfoList
  28. if v.Children == nil {
  29. childrenArr := make([]*data_manage.EdbClassifyItems, 0)
  30. v.Children = childrenArr
  31. }
  32. }
  33. }
  34. } else {
  35. childrenArr := make([]*data_manage.EdbClassifyItems, 0)
  36. node.Children = childrenArr
  37. }
  38. }
  39. func edbClassifyHaveChild(allNode []*data_manage.EdbClassifyItems, node *data_manage.EdbClassifyItems) (childs []*data_manage.EdbClassifyItems, yes bool) {
  40. for _, v := range allNode {
  41. if v.ParentId == node.ClassifyId {
  42. childs = append(childs, v)
  43. }
  44. }
  45. if len(childs) > 0 {
  46. yes = true
  47. }
  48. return
  49. }
  50. // GetClassifyTreeRecursive 递归获取分类树形结构
  51. func GetClassifyTreeRecursive(list []*data_manage.EdbClassifyItems, parentId, level int) []*data_manage.EdbClassifyItems {
  52. res := make([]*data_manage.EdbClassifyItems, 0)
  53. for _, v := range list {
  54. if v.Level == level {
  55. v.SelectDisable = false
  56. }
  57. if v.ParentId == parentId {
  58. v.Children = GetClassifyTreeRecursive(list, v.ClassifyId, level)
  59. res = append(res, v)
  60. }
  61. }
  62. return res
  63. }
  64. func GetFullClassifyByClassifyId(targetClassifyId int) (targetList []*data_manage.EdbClassifyIdItems, err error, errMsg string) {
  65. //判断是否是挂在顶级目录下
  66. targetClassify, err := data_manage.GetEdbClassifyById(targetClassifyId)
  67. if err != nil {
  68. if err.Error() == utils.ErrNoRow() {
  69. errMsg = "当前分类不存在"
  70. err = errors.New(errMsg)
  71. return
  72. }
  73. errMsg = "获取失败"
  74. err = errors.New("获取分类信息失败,Err:" + err.Error())
  75. return
  76. }
  77. if targetClassify.ParentId == 0 {
  78. targetItem := new(data_manage.EdbClassifyIdItems)
  79. targetItem.ClassifyId = targetClassify.ClassifyId
  80. targetItem.ParentId = targetClassify.ParentId
  81. targetItem.RootId = targetClassify.RootId
  82. targetItem.UniqueCode = targetClassify.UniqueCode
  83. targetItem.Level = targetClassify.Level
  84. targetItem.ClassifyName = targetClassify.ClassifyName
  85. targetItem.ClassifyNameEn = targetClassify.ClassifyNameEn
  86. targetItem.IsJoinPermission = targetClassify.IsJoinPermission
  87. targetList = append(targetList, targetItem)
  88. return
  89. }
  90. tmpList, err := data_manage.GetEdbClassifyByRootIdLevel(targetClassify.RootId, targetClassify.ClassifyType, "")
  91. if err != nil && err.Error() != utils.ErrNoRow() {
  92. errMsg = "获取失败"
  93. err = errors.New("获取数据失败,Err:" + err.Error())
  94. return
  95. }
  96. idMap := make(map[int]struct{})
  97. if len(tmpList) > 0 {
  98. for _, v := range tmpList {
  99. if v.ClassifyId == targetClassify.ClassifyId {
  100. idMap[v.ClassifyId] = struct{}{}
  101. idMap[v.ParentId] = struct{}{}
  102. }
  103. }
  104. for _, v := range tmpList {
  105. if _, ok := idMap[v.ClassifyId]; ok {
  106. idMap[v.ParentId] = struct{}{}
  107. }
  108. }
  109. for _, v := range tmpList {
  110. if _, ok := idMap[v.ClassifyId]; ok {
  111. targetItem := new(data_manage.EdbClassifyIdItems)
  112. targetItem.ClassifyId = v.ClassifyId
  113. targetItem.ParentId = v.ParentId
  114. targetItem.RootId = v.RootId
  115. targetItem.UniqueCode = v.UniqueCode
  116. targetItem.Level = v.Level
  117. targetItem.ClassifyName = v.ClassifyName
  118. targetItem.ClassifyNameEn = v.ClassifyNameEn
  119. targetItem.IsJoinPermission = v.IsJoinPermission
  120. targetList = append(targetList, targetItem)
  121. }
  122. }
  123. }
  124. return
  125. }
  126. // GetFullClassifyByRootId 查询指标列表里的分类信息
  127. func GetFullClassifyByRootId(targetClassify *data_manage.EdbClassify, tmpList []*data_manage.EdbClassifyItems) (targetList []*data_manage.EdbClassifyIdItems, err error, errMsg string) {
  128. if targetClassify.ParentId == 0 {
  129. targetItem := new(data_manage.EdbClassifyIdItems)
  130. targetItem.ClassifyId = targetClassify.ClassifyId
  131. targetItem.ParentId = targetClassify.ParentId
  132. targetItem.RootId = targetClassify.RootId
  133. targetItem.UniqueCode = targetClassify.UniqueCode
  134. targetItem.Level = targetClassify.Level
  135. targetItem.ClassifyName = targetClassify.ClassifyName
  136. targetItem.ClassifyName = targetClassify.ClassifyName
  137. targetItem.IsJoinPermission = targetClassify.IsJoinPermission
  138. targetList = append(targetList, targetItem)
  139. return
  140. }
  141. idMap := make(map[int]struct{})
  142. if len(tmpList) > 0 {
  143. for _, v := range tmpList {
  144. if v.ClassifyId == targetClassify.ClassifyId {
  145. idMap[v.ClassifyId] = struct{}{}
  146. idMap[v.ParentId] = struct{}{}
  147. }
  148. }
  149. for _, v := range tmpList {
  150. if _, ok := idMap[v.ClassifyId]; ok {
  151. idMap[v.ParentId] = struct{}{}
  152. }
  153. }
  154. for _, v := range tmpList {
  155. if _, ok := idMap[v.ClassifyId]; ok {
  156. targetItem := new(data_manage.EdbClassifyIdItems)
  157. targetItem.ClassifyId = v.ClassifyId
  158. targetItem.ParentId = v.ParentId
  159. targetItem.RootId = v.RootId
  160. targetItem.UniqueCode = v.UniqueCode
  161. targetItem.Level = v.Level
  162. targetItem.ClassifyName = v.ClassifyName
  163. targetItem.IsJoinPermission = v.IsJoinPermission
  164. targetList = append(targetList, targetItem)
  165. }
  166. }
  167. }
  168. return
  169. }
  170. func GetChildClassifyByClassifyId(targetClassifyId int) (targetList []*data_manage.EdbClassifyIdItems, err error, errMsg string) {
  171. //判断是否是挂在顶级目录下
  172. targetClassify, err := data_manage.GetEdbClassifyById(targetClassifyId)
  173. if err != nil {
  174. if err.Error() == utils.ErrNoRow() {
  175. errMsg = "当前分类不存在"
  176. err = errors.New(errMsg)
  177. return
  178. }
  179. errMsg = "获取失败"
  180. err = errors.New("获取分类信息失败,Err:" + err.Error())
  181. return
  182. }
  183. orderStr := ` order by level asc, sort asc, classify_id asc`
  184. tmpList, err := data_manage.GetEdbClassifyByRootIdLevel(targetClassify.RootId, targetClassify.ClassifyType, orderStr)
  185. if err != nil && err.Error() != utils.ErrNoRow() {
  186. errMsg = "获取失败"
  187. err = errors.New("获取数据失败,Err:" + err.Error())
  188. return
  189. }
  190. idMap := make(map[int]struct{})
  191. if len(tmpList) > 0 {
  192. for _, v := range tmpList {
  193. if v.ClassifyId == targetClassify.ClassifyId {
  194. idMap[v.ClassifyId] = struct{}{}
  195. }
  196. }
  197. for _, v := range tmpList {
  198. if _, ok := idMap[v.ParentId]; ok {
  199. idMap[v.ClassifyId] = struct{}{}
  200. }
  201. }
  202. for _, v := range tmpList {
  203. if _, ok := idMap[v.ClassifyId]; ok {
  204. targetItem := new(data_manage.EdbClassifyIdItems)
  205. targetItem.ClassifyId = v.ClassifyId
  206. targetItem.ParentId = v.ParentId
  207. targetItem.RootId = v.RootId
  208. targetItem.UniqueCode = v.UniqueCode
  209. targetItem.Level = v.Level
  210. targetItem.ClassifyName = v.ClassifyName
  211. targetItem.IsJoinPermission = v.IsJoinPermission
  212. targetList = append(targetList, targetItem)
  213. }
  214. }
  215. }
  216. return
  217. }
  218. func GetEdbClassifyEdbInfo() (result map[int][]*data_manage.EdbClassifyItems, err error) {
  219. cMap := make(map[int][]*data_manage.EdbClassifyItems)
  220. items, err := data_manage.GetEdbInfoAll(0)
  221. if err != nil {
  222. return
  223. }
  224. for _, v := range items {
  225. childrenArr := make([]*data_manage.EdbClassifyItems, 0)
  226. v.Children = childrenArr
  227. if items, ok := cMap[v.ClassifyId]; ok {
  228. items = append(items, v)
  229. cMap[v.ClassifyId] = items
  230. } else {
  231. infoList := make([]*data_manage.EdbClassifyItems, 0)
  232. infoList = append(infoList, v)
  233. cMap[v.ClassifyId] = infoList
  234. }
  235. }
  236. result = cMap
  237. return
  238. }
  239. func EdbClassifyItemsMakeTree(allNode []*data_manage.EdbClassifyItems, node *data_manage.EdbClassifyItems) {
  240. childs, _ := edbClassifyHaveChild(allNode, node) //判断节点是否有子节点并返回
  241. if len(childs) > 0 {
  242. node.Children = append(node.Children, childs[0:]...) //添加子节点
  243. for _, v := range childs { //查询子节点的子节点,并添加到子节点
  244. _, has := edbClassifyHaveChild(allNode, v)
  245. if has {
  246. EdbClassifyItemsMakeTree(allNode, v) //递归添加节点
  247. } else {
  248. childrenArr := make([]*data_manage.EdbClassifyItems, 0)
  249. v.Children = childrenArr
  250. }
  251. }
  252. } else {
  253. childrenArr := make([]*data_manage.EdbClassifyItems, 0)
  254. node.Children = childrenArr
  255. }
  256. }
  257. // GetEdbClassifyListByAdminId 根据账户类型获取分类id集合
  258. func GetEdbClassifyListByAdminId(adminId int64) (classifyIdList []string, err error) {
  259. list, err := models.GetEdbdataClassify(adminId)
  260. if err != nil {
  261. return
  262. }
  263. for _, parent := range list {
  264. if parent.Child != nil {
  265. for _, classify := range parent.Child {
  266. classifyIdList = append(classifyIdList, fmt.Sprint(classify.ClassifyId))
  267. }
  268. }
  269. }
  270. return
  271. }
  272. // GetEdbClassifyNameListByAdminId 根据账户类型获取分类名称集合
  273. func GetEdbClassifyNameListByAdminId(adminId int64) (classifyNameList []string, edbDataClassifyMap map[string]*models.EdbdataClassify, err error) {
  274. list, err := models.GetEdbdataClassify(adminId)
  275. edbDataClassifyMap = make(map[string]*models.EdbdataClassify)
  276. if err != nil {
  277. return
  278. }
  279. for _, parent := range list {
  280. if parent.Child != nil {
  281. for _, classify := range parent.Child {
  282. classifyNameList = append(classifyNameList, classify.ClassifyName)
  283. edbDataClassifyMap[classify.ClassifyName] = classify
  284. }
  285. }
  286. }
  287. return
  288. }
  289. // AddEdbClassify 添加指标分类
  290. func AddEdbClassify(classifyName string, parentId, level int, classifyType uint8, sysUserId int, sysUserName, lang string) (classifyInfo *data_manage.EdbClassify, err error, errMsg string) {
  291. // 校验分类名称相同的数量
  292. {
  293. var count int
  294. switch lang {
  295. case utils.EnLangVersion:
  296. count, err = data_manage.GetEdbClassifyEnCount(classifyName, parentId, classifyType)
  297. default:
  298. count, err = data_manage.GetEdbClassifyCount(classifyName, parentId, classifyType)
  299. }
  300. if err != nil {
  301. errMsg = `判断名称是否已存在失败`
  302. return
  303. }
  304. if count > 0 {
  305. errMsg = `分类名称已存在,请重新输入`
  306. err = errors.New(errMsg)
  307. return
  308. }
  309. }
  310. // 层级校验
  311. if level > 6 {
  312. errMsg = `最高只支持添加6级分类`
  313. return
  314. }
  315. //获取该层级下最大的排序数
  316. maxSort, err := GetEdbClassifyMaxSort(parentId, classifyType)
  317. if err != nil {
  318. errMsg = "获取失败"
  319. err = errors.New("查询排序信息失败,Err:" + err.Error())
  320. return
  321. }
  322. //查询顶级rootId
  323. rootId := 0
  324. if parentId > 0 {
  325. parentClassify, tErr := data_manage.GetEdbClassifyById(parentId)
  326. if tErr != nil {
  327. if tErr.Error() == utils.ErrNoRow() {
  328. errMsg = "父级分类不存在"
  329. err = errors.New(errMsg)
  330. return
  331. }
  332. errMsg = "获取失败"
  333. err = errors.New("获取分类信息失败,Err:" + tErr.Error())
  334. return
  335. }
  336. rootId = parentClassify.RootId
  337. }
  338. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  339. classifyInfo = &data_manage.EdbClassify{
  340. //ClassifyId: 0,
  341. ClassifyType: classifyType,
  342. ClassifyName: classifyName,
  343. ClassifyNameEn: classifyName,
  344. ParentId: parentId,
  345. RootId: rootId,
  346. HasData: 0,
  347. CreateTime: time.Now(),
  348. ModifyTime: time.Now(),
  349. SysUserId: sysUserId,
  350. SysUserRealName: sysUserName,
  351. Level: level + 1,
  352. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  353. Sort: maxSort + 1,
  354. }
  355. newId, err := data_manage.AddEdbClassify(classifyInfo)
  356. if err != nil {
  357. errMsg = "保存分类失败"
  358. return
  359. }
  360. // 继承分类权限
  361. {
  362. source := 3
  363. if classifyType == 1 {
  364. source = 4
  365. }
  366. go data_manage_permission.InheritParentClassify(source, int(classifyType), int(newId), classifyInfo.ParentId, classifyInfo.ClassifyName)
  367. }
  368. if parentId == 0 { //一级目录的rootId等于自己本身
  369. classifyInfo.ClassifyId = int(newId)
  370. classifyInfo.RootId = int(newId)
  371. err = classifyInfo.Update([]string{"RootId"})
  372. if err != nil {
  373. errMsg = "更新分类失败"
  374. return
  375. }
  376. }
  377. return
  378. }
  379. // EditEdbClassify 编辑指标分类
  380. func EditEdbClassify(classifyId, parentId int, classifyName, lang string, sysUser *system.Admin) (err error, errMsg string) {
  381. item, err := data_manage.GetEdbClassifyById(classifyId)
  382. if err != nil {
  383. errMsg = `保存失败`
  384. return
  385. }
  386. // 权限校验
  387. {
  388. // 已授权分类id
  389. permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbClassifyPermissionList(sysUser.AdminId, classifyId)
  390. if tmpErr != nil {
  391. errMsg = "保存失败"
  392. err = errors.New("获取已授权分类id数据失败,Err:" + tmpErr.Error())
  393. return
  394. }
  395. // 数据权限
  396. haveOperaAuth := data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(item.IsJoinPermission, item.ClassifyId, permissionClassifyIdList)
  397. if item.ClassifyType == 0 { // 普通指标
  398. button := GetEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  399. if !button.OpButton {
  400. errMsg = "无操作权限"
  401. err = errors.New(errMsg)
  402. return
  403. }
  404. } else if item.ClassifyType == 1 { // 预测指标
  405. button := GetPredictEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  406. if !button.OpButton {
  407. errMsg = "无操作权限"
  408. err = errors.New(errMsg)
  409. return
  410. }
  411. }
  412. }
  413. if parentId != item.ParentId {
  414. parentClassifyInfo, e := data_manage.GetEdbClassifyById(parentId)
  415. if e != nil {
  416. err = e
  417. return
  418. }
  419. if item.Level != parentClassifyInfo.Level - 1 {
  420. err = errors.New("父级分类层级异常")
  421. }
  422. }
  423. // 需要变更的字段
  424. updateCols := make([]string, 0)
  425. if parentId != item.ParentId {
  426. item.ParentId = parentId
  427. item.ModifyTime = time.Now()
  428. updateCols = append(updateCols, "ParentId", "ModifyTime")
  429. }
  430. switch lang {
  431. case utils.EnLangVersion:
  432. // 名字相同,那么就直接返回
  433. if item.ClassifyNameEn == classifyName {
  434. return
  435. }
  436. // 判断名称是否已存在
  437. count, tmpErr := data_manage.GetEdbClassifyEnCount(classifyName, item.ParentId, item.ClassifyType)
  438. if tmpErr != nil {
  439. err = tmpErr
  440. errMsg = "判断名称是否已存在失败"
  441. return
  442. }
  443. if count > 0 {
  444. errMsg = "分类名称已存在,请重新输入"
  445. err = errors.New(errMsg)
  446. return
  447. }
  448. item.ClassifyNameEn = classifyName
  449. item.LastModifyUserId = sysUser.AdminId
  450. item.LastModifyUserRealName = sysUser.RealName
  451. updateCols = append(updateCols, "ClassifyNameEn", "LastModifyUserId", "LastModifyUserRealName")
  452. default:
  453. // 名字相同,那么就直接返回
  454. if item.ClassifyName == classifyName {
  455. return
  456. }
  457. // 判断名称是否已存在
  458. count, tmpErr := data_manage.GetEdbClassifyCount(classifyName, item.ParentId, item.ClassifyType)
  459. if tmpErr != nil {
  460. err = tmpErr
  461. errMsg = "判断名称是否已存在失败"
  462. return
  463. }
  464. if count > 0 {
  465. errMsg = "分类名称已存在,请重新输入"
  466. err = errors.New(errMsg)
  467. return
  468. }
  469. item.ClassifyName = classifyName
  470. item.LastModifyUserId = sysUser.AdminId
  471. item.LastModifyUserRealName = sysUser.RealName
  472. updateCols = append(updateCols, "ClassifyName", "LastModifyUserId", "LastModifyUserRealName")
  473. }
  474. // 修改数据
  475. if len(updateCols) > 0 {
  476. err = item.Update(updateCols)
  477. if err != nil {
  478. errMsg = "保存失败"
  479. }
  480. }
  481. return
  482. }
  483. // DeleteCheck 删除检测
  484. func DeleteCheck(classifyId, edbInfoId int, sysUser *system.Admin) (deleteStatus int, tipsMsg string, tableList []*data_manage.ExcelBaseInfo, err error, errMsg string) {
  485. //删除分类
  486. if classifyId > 0 && edbInfoId == 0 {
  487. // 查找分类
  488. item, tmpErr := data_manage.GetEdbClassifyById(classifyId)
  489. if tmpErr != nil {
  490. errMsg = `查找分类失败`
  491. err = tmpErr
  492. return
  493. }
  494. // 已授权分类id
  495. permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbClassifyPermissionList(sysUser.AdminId, classifyId)
  496. if tmpErr != nil {
  497. errMsg = "删除检测失败"
  498. err = errors.New("获取已授权分类id数据失败,Err:" + tmpErr.Error())
  499. return
  500. }
  501. // 权限校验
  502. {
  503. // 数据权限
  504. haveOperaAuth := data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(item.IsJoinPermission, item.ClassifyId, permissionClassifyIdList)
  505. if item.ClassifyType == 0 { // 普通指标
  506. button := GetEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  507. if !button.DeleteButton {
  508. errMsg = "无操作权限"
  509. err = errors.New(errMsg)
  510. return
  511. }
  512. } else if item.ClassifyType == 1 { // 预测指标
  513. button := GetPredictEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  514. if !button.DeleteButton {
  515. errMsg = "无操作权限"
  516. err = errors.New(errMsg)
  517. return
  518. }
  519. }
  520. }
  521. //判断分类下,是否含有指标
  522. count, tmpErr := data_manage.GetEdbInfoCountByClassifyId(classifyId)
  523. if tmpErr != nil {
  524. errMsg = "删除失败"
  525. err = errors.New("分类下是否含有指标失败,Err:" + tmpErr.Error())
  526. return
  527. }
  528. if count > 0 {
  529. deleteStatus = 1
  530. tipsMsg = "若目录关联指标不可删除"
  531. }
  532. }
  533. if deleteStatus != 1 && edbInfoId == 0 {
  534. classifyCount, tmpErr := data_manage.GetClassifyCountByClassifyId(classifyId)
  535. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  536. errMsg = "删除失败"
  537. err = errors.New("分类下是否含有指标失败,Err:" + tmpErr.Error())
  538. return
  539. }
  540. if classifyCount > 0 {
  541. deleteStatus = 2
  542. tipsMsg = "确认删除当前目录及包含的子目录吗"
  543. }
  544. }
  545. //删除指标
  546. if edbInfoId > 0 {
  547. edbInfo, tmpErr := data_manage.GetEdbInfoById(edbInfoId)
  548. if tmpErr != nil {
  549. if tmpErr.Error() == utils.ErrNoRow() {
  550. errMsg = "指标已删除,请刷新页面"
  551. err = errors.New("指标不存在,Err:" + tmpErr.Error())
  552. return
  553. } else {
  554. errMsg = "删除失败"
  555. err = errors.New("删除失败,获取指标信息失败,Err:" + tmpErr.Error())
  556. return
  557. }
  558. }
  559. if edbInfo == nil {
  560. errMsg = "指标已删除,请刷新页面"
  561. return
  562. }
  563. //判断指标是否用于作图,如果用于作图,则不可删除
  564. chartCount, tmpErr := data_manage.GetChartEdbMappingCount(edbInfoId)
  565. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  566. errMsg = "删除失败"
  567. err = errors.New("判断指标是否被用于作图失败,Err:" + tmpErr.Error())
  568. return
  569. }
  570. if chartCount > 0 {
  571. deleteStatus = 3
  572. tipsMsg = "当前指标已用作画图,不可删除"
  573. return
  574. }
  575. // 查询是否用于相关性图表
  576. correlationChart := new(data_manage.ChartInfoCorrelation)
  577. correlationChartCount, tmpErr := correlationChart.GetChartEdbMappingCount(edbInfoId)
  578. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  579. errMsg = "删除失败"
  580. err = errors.New("判断指标是否被用于相关性图表失败,Err:" + tmpErr.Error())
  581. return
  582. }
  583. if correlationChartCount > 0 {
  584. deleteStatus = 3
  585. tipsMsg = "当前指标已用作画图,不可删除"
  586. return
  587. }
  588. //判断指标是否用于计算
  589. {
  590. calculateCount, tmpErr := data_manage.GetEdbInfoCalculateMappingCount(edbInfoId)
  591. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  592. errMsg = "删除失败"
  593. err = errors.New("判断指标是否被用于计算失败,GetEdbInfoCalculateCount Err:" + tmpErr.Error())
  594. return
  595. }
  596. if calculateCount > 0 {
  597. deleteStatus = 4
  598. tipsMsg = "当前指标已用作指标运算,不可删除"
  599. return
  600. }
  601. }
  602. //如果是普通指标,那么还需要判断是否被预测指标作为源指标
  603. if edbInfo.EdbInfoType == 0 {
  604. predictEdbInfoCount, tmpErr := data_manage.GetPredictEdbConfCount(edbInfoId)
  605. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  606. errMsg = "删除失败"
  607. err = errors.New("判断指标是否被用于预测指标失败,Err:" + tmpErr.Error())
  608. return
  609. }
  610. if predictEdbInfoCount > 0 {
  611. deleteStatus = 5
  612. tipsMsg = "当前指标已用作预测指标,不可删除"
  613. return
  614. }
  615. }
  616. // 判断指标是否用作表格引用
  617. {
  618. tableItems, tmpErr := excel.GetNoCustomAnalysisExcelEdbMapping(edbInfoId)
  619. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  620. errMsg = "删除失败"
  621. err = errors.New("判断指标是否用作表格引用,GetNoCustomAnalysisExcelEdbMappingCount Err:" + tmpErr.Error())
  622. return
  623. }
  624. //英文翻译:
  625. //1、当前指标已用作画图,不可删除:The current metric is in use for charting and cannot be deleted
  626. //2、当前指标已被表格引用,不可删除:The current metric is referenced by a table and cannot be deleted
  627. //3、删除失败:Deletion failed
  628. //4、知道了:Understood
  629. if len(tableItems) > 0 {
  630. deleteStatus = 6
  631. tipsMsg = "当前指标已被表格引用,不可删除"
  632. var excelIds []int
  633. for _, tableItem := range tableItems {
  634. if tableItem.ParentId > 0 {
  635. excelIds = append(excelIds, tableItem.ParentId)
  636. } else {
  637. excelIds = append(excelIds, tableItem.ExcelInfoId)
  638. }
  639. }
  640. tableList, tmpErr = excel.GetExcelBaseInfoByExcelInfoIdList(excelIds)
  641. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  642. errMsg = "删除失败"
  643. err = errors.New("判断指标是否用作表格引用,GetExcelBaseInfoByExcelInfoIdList Err:" + tmpErr.Error())
  644. return
  645. }
  646. return
  647. }
  648. }
  649. // 判断指标是否用作跨品种图表使用
  650. {
  651. // todo 如果绑定的标签未画图,则允许删除绑定
  652. // 查询跨品种的图表
  653. tagXList, tmpErr := cross_variety.GetChartInfoCrossVarietyByXEdbInfoId(edbInfoId)
  654. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  655. errMsg = "删除失败"
  656. err = errors.New("判断指标是否用作跨品种图表使用,GetChartInfoCrossVarietyByXEdbInfoId Err:" + tmpErr.Error())
  657. return
  658. }
  659. if len(tagXList) > 0 {
  660. deleteStatus = 7
  661. tipsMsg = "当前指标已添加到跨品种分析,不可删除"
  662. return
  663. }
  664. tagYList, tmpErr := cross_variety.GetChartInfoCrossVarietyByYEdbInfoId(edbInfoId)
  665. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  666. errMsg = "删除失败"
  667. err = errors.New("判断指标是否用作跨品种图表使用,GetChartInfoCrossVarietyByYEdbInfoId Err:" + tmpErr.Error())
  668. return
  669. }
  670. if len(tagYList) > 0 {
  671. deleteStatus = 7
  672. tipsMsg = "当前指标已添加到跨品种分析,不可删除"
  673. return
  674. }
  675. /*calculateCount, tmpErr := cross_variety.GetCountByEdbInfoId(edbInfoId)
  676. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  677. errMsg = "删除失败"
  678. err = errors.New("判断指标是否用作跨品种图表使用,GetCountByEdbInfoId Err:" + tmpErr.Error())
  679. return
  680. }
  681. if calculateCount > 0 {
  682. deleteStatus = 3
  683. tipsMsg = "当前指标已添加到跨品种分析,不可删除"
  684. return
  685. }*/
  686. }
  687. // 判断指标是否用于因子指标系列
  688. {
  689. ob := new(data_manage.FactorEdbSeriesMapping)
  690. cond := fmt.Sprintf(" AND %s = ?", ob.Cols().EdbInfoId)
  691. pars := make([]interface{}, 0)
  692. pars = append(pars, edbInfoId)
  693. count, e := ob.GetCountByCondition(cond, pars)
  694. if e != nil {
  695. errMsg = "删除失败"
  696. err = fmt.Errorf("获取指标是否用于系列失败, err: %v", e)
  697. return
  698. }
  699. if count > 0 {
  700. deleteStatus = 3
  701. tipsMsg = "当前指标已用于因子指标系列, 不可删除"
  702. return
  703. }
  704. }
  705. }
  706. return
  707. }
  708. // Delete 删除分类/指标
  709. func Delete(classifyId, edbInfoId int, sysUser *system.Admin, requestBody, requestUrl string) (nextItem *data_manage.EdbInfo, tableList []*data_manage.ExcelBaseInfo, err error, errMsg string) {
  710. //删除分类
  711. if classifyId > 0 && edbInfoId == 0 {
  712. // 查找分类
  713. item, tmpErr := data_manage.GetEdbClassifyById(classifyId)
  714. if tmpErr != nil {
  715. errMsg = `查找分类失败`
  716. err = tmpErr
  717. return
  718. }
  719. // 已授权分类id
  720. permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbClassifyPermissionList(sysUser.AdminId, classifyId)
  721. if tmpErr != nil {
  722. errMsg = "删除失败"
  723. err = errors.New("获取已授权分类id数据失败,Err:" + tmpErr.Error())
  724. return
  725. }
  726. // 权限校验
  727. {
  728. // 数据权限
  729. haveOperaAuth := data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(item.IsJoinPermission, item.ClassifyId, permissionClassifyIdList)
  730. if item.ClassifyType == 0 { // 普通指标
  731. button := GetEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  732. if !button.DeleteButton {
  733. errMsg = "无操作权限"
  734. err = errors.New(errMsg)
  735. return
  736. }
  737. } else if item.ClassifyType == 1 { // 预测指标
  738. button := GetPredictEdbClassifyOpButton(sysUser, item.SysUserId, haveOperaAuth)
  739. if !button.DeleteButton {
  740. errMsg = "无操作权限"
  741. err = errors.New(errMsg)
  742. return
  743. }
  744. }
  745. }
  746. //判断是否含有指标
  747. count, tmpErr := data_manage.GetEdbInfoCountByClassifyId(classifyId)
  748. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  749. errMsg = "删除失败"
  750. err = errors.New("分类下是否含有指标失败,Err:" + tmpErr.Error())
  751. return
  752. }
  753. if count > 0 {
  754. errMsg = "该目录下存在关联指标,不可删除"
  755. return
  756. }
  757. tmpErr = data_manage.DeleteEdbClassify(classifyId)
  758. if tmpErr != nil {
  759. errMsg = "删除失败"
  760. err = errors.New("删除失败,Err:" + tmpErr.Error())
  761. return
  762. }
  763. }
  764. //删除指标
  765. if edbInfoId > 0 {
  766. edbInfo, tmpErr := data_manage.GetEdbInfoById(edbInfoId)
  767. if tmpErr != nil {
  768. if tmpErr.Error() == utils.ErrNoRow() {
  769. errMsg = "指标已删除,请刷新页面"
  770. err = errors.New("指标不存在,Err:" + tmpErr.Error())
  771. return
  772. } else {
  773. errMsg = "删除失败"
  774. err = errors.New("删除失败,获取指标信息失败,Err:" + tmpErr.Error())
  775. return
  776. }
  777. }
  778. if edbInfo == nil {
  779. errMsg = "指标已删除,请刷新页面"
  780. return
  781. }
  782. //判断指标是否用于作图,如果用于作图,则不可删除
  783. chartCount, tmpErr := data_manage.GetChartEdbMappingCount(edbInfoId)
  784. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  785. errMsg = "删除失败"
  786. err = errors.New("判断指标是否被用于作图失败,Err:" + tmpErr.Error())
  787. return
  788. }
  789. if chartCount > 0 {
  790. errMsg = "当前指标已用作画图,不可删除"
  791. return
  792. }
  793. // 查询是否用于相关性图表
  794. correlationChart := new(data_manage.ChartInfoCorrelation)
  795. correlationChartCount, tmpErr := correlationChart.GetChartEdbMappingCount(edbInfoId)
  796. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  797. errMsg = "删除失败"
  798. err = errors.New("判断指标是否被用于相关性图表失败,Err:" + tmpErr.Error())
  799. return
  800. }
  801. if correlationChartCount > 0 {
  802. errMsg = "当前指标已用作画图,不可删除"
  803. return
  804. }
  805. //如果是普通指标,那么还需要判断是否被预测指标作为源指标
  806. if edbInfo.EdbInfoType == 0 {
  807. predictEdbInfoCount, tmpErr := data_manage.GetPredictEdbConfCount(edbInfoId)
  808. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  809. errMsg = "删除失败"
  810. err = errors.New("判断指标是否被用于预测指标失败,Err:" + tmpErr.Error())
  811. return
  812. }
  813. if predictEdbInfoCount > 0 {
  814. errMsg = "当前指标已用作预测指标,不可删除"
  815. return
  816. }
  817. }
  818. //判断指标是否用作其他指标的计算
  819. {
  820. calculateCount, tmpErr := data_manage.GetEdbInfoCalculateMappingCount(edbInfoId)
  821. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  822. errMsg = "删除失败"
  823. err = errors.New("判断指标是否被用于计算失败,GetEdbInfoCalculateCount Err:" + tmpErr.Error())
  824. return
  825. }
  826. if calculateCount > 0 {
  827. errMsg = "当前指标已用作,指标运算,不可删除"
  828. return
  829. }
  830. }
  831. // 判断指标是否用作表格引用
  832. {
  833. tableItems, tmpErr := excel.GetNoCustomAnalysisExcelEdbMapping(edbInfoId)
  834. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  835. errMsg = "删除失败"
  836. err = errors.New("判断指标是否用作表格引用,GetNoCustomAnalysisExcelEdbMappingCount Err:" + tmpErr.Error())
  837. return
  838. }
  839. //英文翻译:
  840. //1、当前指标已用作画图,不可删除:The current metric is in use for charting and cannot be deleted
  841. //2、当前指标已被表格引用,不可删除:The current metric is referenced by a table and cannot be deleted
  842. //3、删除失败:Deletion failed
  843. //4、知道了:Understood
  844. if len(tableItems) > 0 {
  845. errMsg = "当前指标已被表格引用,不可删除"
  846. return
  847. }
  848. }
  849. // 判断指标是否用作跨品种图表使用
  850. {
  851. // todo 如果绑定的标签未画图,则允许删除绑定
  852. // 查询跨品种的图表
  853. tagXList, tmpErr := cross_variety.GetChartInfoCrossVarietyByXEdbInfoId(edbInfoId)
  854. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  855. errMsg = "删除失败"
  856. err = errors.New("判断指标是否用作跨品种图表使用,GetChartInfoCrossVarietyByXEdbInfoId Err:" + tmpErr.Error())
  857. return
  858. }
  859. if len(tagXList) > 0 {
  860. errMsg = "当前指标已添加到跨品种分析,不可删除"
  861. return
  862. }
  863. tagYList, tmpErr := cross_variety.GetChartInfoCrossVarietyByYEdbInfoId(edbInfoId)
  864. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  865. errMsg = "删除失败"
  866. err = errors.New("判断指标是否用作跨品种图表使用,GetChartInfoCrossVarietyByYEdbInfoId Err:" + tmpErr.Error())
  867. return
  868. }
  869. if len(tagYList) > 0 {
  870. errMsg = "当前指标已添加到跨品种分析,不可删除"
  871. return
  872. }
  873. /*calculateCount, tmpErr := cross_variety.GetCountByEdbInfoId(edbInfoId)
  874. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  875. errMsg = "删除失败"
  876. err = errors.New("判断指标是否用作跨品种图表使用,GetCountByEdbInfoId Err:" + tmpErr.Error())
  877. return
  878. }
  879. if calculateCount > 0 {
  880. errMsg = "当前指标已添加到跨品种分析,不可删除"
  881. return
  882. }*/
  883. }
  884. //真实删除
  885. tmpErr = data_manage.DeleteEdbInfoAndData(edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource)
  886. if tmpErr != nil {
  887. errMsg = "删除失败"
  888. err = errors.New("删除失败,Err:" + tmpErr.Error())
  889. return
  890. }
  891. // 如果是同花顺高频数据或类似数据, 还需要删除base_from_edb_mapping对应关系
  892. baseMappingOb := new(data_manage.BaseFromEdbMapping)
  893. {
  894. cond := fmt.Sprintf(" %s = ? AND %s = ? AND %s = ?", baseMappingOb.Cols().EdbCode, baseMappingOb.Cols().Source, baseMappingOb.Cols().SubSource)
  895. pars := make([]interface{}, 0)
  896. pars = append(pars, edbInfo.EdbCode, edbInfo.Source, edbInfo.SubSource)
  897. if e := baseMappingOb.RemoveByCondition(cond, pars); e != nil {
  898. errMsg = "删除失败"
  899. err = fmt.Errorf("删除源指标映射失败, %v", e)
  900. return
  901. }
  902. }
  903. go data_stat.AddEdbDeleteLog(edbInfo, sysUser)
  904. // 删除指标后的操作
  905. go handleByDelEdbInfo(edbInfo)
  906. // 返回下一个表格的信息
  907. {
  908. var condition string
  909. var pars []interface{}
  910. condition += " AND edb_info_type=? AND classify_id=? "
  911. pars = append(pars, edbInfo.EdbInfoType, edbInfo.ClassifyId)
  912. condition += " AND (sort>? OR (sort=? AND edb_info_id<?) ) "
  913. pars = append(pars, edbInfo.Sort, edbInfo.Sort, edbInfo.EdbInfoId)
  914. nextItem, tmpErr = data_manage.GetNextEdbInfoByCondition(condition, pars)
  915. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  916. errMsg = "删除失败"
  917. err = errors.New("获取下一级预测指标信息失败,Err:" + tmpErr.Error())
  918. return
  919. }
  920. // 如果没找到,那么查找下一个分类的第一个表格
  921. if nextItem == nil {
  922. currClassifyInfo, tmpErr := data_manage.GetEdbClassifyById(edbInfo.ClassifyId)
  923. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  924. errMsg = "删除失败"
  925. err = errors.New("获取当前预测指标分类信息失败,Err:" + tmpErr.Error())
  926. return
  927. }
  928. nextItem, err = data_manage.GetNextEdbInfo(edbInfo.ClassifyId, currClassifyInfo.Sort, int(currClassifyInfo.ClassifyType), edbInfo.EdbInfoType)
  929. if err != nil && err.Error() != utils.ErrNoRow() {
  930. errMsg = "删除失败"
  931. err = errors.New("获取下一级预测指标信息失败,Err:" + tmpErr.Error())
  932. return
  933. }
  934. }
  935. }
  936. //新增操作日志
  937. {
  938. edbLog := new(data_manage.EdbInfoLog)
  939. edbLog.EdbInfoId = edbInfo.EdbInfoId
  940. edbLog.SourceName = edbInfo.SourceName
  941. edbLog.Source = edbInfo.Source
  942. edbLog.EdbCode = edbInfo.EdbCode
  943. edbLog.EdbName = edbInfo.EdbName
  944. edbLog.ClassifyId = edbInfo.ClassifyId
  945. edbLog.SysUserId = sysUser.AdminId
  946. edbLog.SysUserRealName = sysUser.RealName
  947. edbLog.CreateTime = time.Now()
  948. edbLog.Content = requestBody
  949. edbLog.Status = "删除指标"
  950. edbLog.Method = requestUrl
  951. go data_manage.AddEdbInfoLog(edbLog)
  952. }
  953. // 删除es中的数据
  954. DeleteEdbInfoToEs(edbInfoId)
  955. }
  956. return
  957. }
  958. // MoveEdbClassify 移动指标分类
  959. func MoveEdbClassify(req data_manage.MoveEdbClassifyReq, sysUser *system.Admin, classifyType uint8) (err error, errMsg string) {
  960. // req.ClassifyId, req.ParentClassifyId, req.PrevClassifyId, req.NextClassifyId
  961. classifyId := req.ClassifyId
  962. parentClassifyId := req.ParentClassifyId
  963. prevClassifyId := req.PrevClassifyId
  964. nextClassifyId := req.NextClassifyId
  965. edbInfoId := req.EdbInfoId
  966. prevEdbInfoId := req.PrevEdbInfoId
  967. nextEdbInfoId := req.NextEdbInfoId
  968. //首先确定移动的对象是分类还是指标
  969. //判断上一个节点是分类还是指标
  970. //判断下一个节点是分类还是指标
  971. //同时更新分类目录下的分类sort和指标sort
  972. //更新当前移动的分类或者指标sort
  973. var parentEdbClassifyInfo *data_manage.EdbClassify
  974. if parentClassifyId > 0 {
  975. parentEdbClassifyInfo, err = data_manage.GetEdbClassifyById(parentClassifyId)
  976. if err != nil {
  977. errMsg = "移动失败"
  978. err = errors.New("获取上级分类信息失败,Err:" + err.Error())
  979. return
  980. }
  981. }
  982. //如果有传入 上一个兄弟节点分类id
  983. var (
  984. edbClassifyInfo *data_manage.EdbClassify
  985. prevClassify *data_manage.EdbClassify
  986. nextClassify *data_manage.EdbClassify
  987. edbInfo *data_manage.EdbInfo
  988. prevEdbInfo *data_manage.EdbInfo
  989. nextEdbInfo *data_manage.EdbInfo
  990. prevSort int
  991. nextSort int
  992. )
  993. // 移动对象为分类, 判断权限
  994. if edbInfoId == 0 {
  995. edbClassifyInfo, err = data_manage.GetEdbClassifyById(classifyId)
  996. if err != nil {
  997. if err.Error() == utils.ErrNoRow() {
  998. errMsg = "当前分类不存在"
  999. err = errors.New("获取分类信息失败,Err:" + err.Error())
  1000. return
  1001. }
  1002. errMsg = "移动失败"
  1003. err = errors.New("获取分类信息失败,Err:" + err.Error())
  1004. return
  1005. }
  1006. if parentClassifyId > 0 && parentEdbClassifyInfo.Level == 6 {
  1007. errMsg = "最高只支持添加6级分类"
  1008. err = errors.New(errMsg)
  1009. return
  1010. }
  1011. // 如果是移动目录, 那么校验一下父级目录下是否有重名目录
  1012. exists, e := data_manage.GetEdbClassifyByParentIdAndName(parentClassifyId, edbClassifyInfo.ClassifyName, classifyId)
  1013. if e != nil && e.Error() != utils.ErrNoRow() {
  1014. errMsg = "移动失败"
  1015. err = fmt.Errorf("获取父级分类下的同名分类失败, Err: %s", e.Error())
  1016. return
  1017. }
  1018. if exists != nil {
  1019. errMsg = "移动失败,分类名称已存在"
  1020. return
  1021. }
  1022. // 权限校验
  1023. {
  1024. // 已授权分类id
  1025. permissionClassifyIdList, tmpErr := data_manage_permission.GetUserEdbClassifyPermissionList(sysUser.AdminId, classifyId)
  1026. if tmpErr != nil {
  1027. errMsg = "移动失败"
  1028. err = errors.New("获取已授权分类id数据失败,Err:" + tmpErr.Error())
  1029. return
  1030. }
  1031. // 数据权限
  1032. haveOperaAuth := data_manage_permission.CheckEdbClassifyPermissionByPermissionIdList(edbClassifyInfo.IsJoinPermission, edbClassifyInfo.ClassifyId, permissionClassifyIdList)
  1033. if edbClassifyInfo.ClassifyType == 0 { // 普通指标
  1034. button := GetEdbClassifyOpButton(sysUser, edbClassifyInfo.SysUserId, haveOperaAuth)
  1035. if !button.MoveButton {
  1036. errMsg = "无操作权限"
  1037. err = errors.New(errMsg)
  1038. return
  1039. }
  1040. } else if edbClassifyInfo.ClassifyType == 1 { // 预测指标
  1041. button := GetPredictEdbClassifyOpButton(sysUser, edbClassifyInfo.SysUserId, haveOperaAuth)
  1042. if !button.MoveButton {
  1043. errMsg = "无操作权限"
  1044. err = errors.New(errMsg)
  1045. return
  1046. }
  1047. }
  1048. }
  1049. } else {
  1050. edbInfo, err = data_manage.GetEdbInfoById(req.EdbInfoId)
  1051. if err != nil {
  1052. if err.Error() == utils.ErrNoRow() {
  1053. errMsg = "当前指标不存在"
  1054. err = errors.New("获取分类信息失败,Err:" + err.Error())
  1055. return
  1056. }
  1057. errMsg = "移动失败"
  1058. err = errors.New("获取分类信息失败,Err:" + err.Error())
  1059. return
  1060. }
  1061. if parentClassifyId == 0 {
  1062. errMsg = "移动失败,指标必须挂在分类下"
  1063. err = errors.New(errMsg)
  1064. return
  1065. }
  1066. var haveOperaAuth bool
  1067. // 权限校验
  1068. {
  1069. haveOperaAuth, err = data_manage_permission.CheckEdbPermissionByEdbInfoId(edbInfo.EdbInfoId, edbInfo.ClassifyId, edbInfo.IsJoinPermission, sysUser.AdminId)
  1070. if err != nil {
  1071. errMsg = "移动失败"
  1072. err = errors.New("获取指标权限信息失败,Err:" + err.Error())
  1073. return
  1074. }
  1075. }
  1076. // 移动权限校验
  1077. button := GetEdbOpButton(sysUser, edbInfo.SysUserId, edbInfo.EdbType, edbInfo.EdbInfoType, haveOperaAuth)
  1078. if !button.MoveButton {
  1079. errMsg = "无操作权限"
  1080. err = errors.New(errMsg)
  1081. return
  1082. }
  1083. }
  1084. if prevClassifyId > 0 {
  1085. prevClassify, err = data_manage.GetEdbClassifyById(prevClassifyId)
  1086. if err != nil {
  1087. errMsg = "移动失败"
  1088. err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
  1089. return
  1090. }
  1091. prevSort = prevClassify.Sort
  1092. } else if prevEdbInfoId > 0 {
  1093. prevEdbInfo, err = data_manage.GetEdbInfoById(prevEdbInfoId)
  1094. if err != nil {
  1095. errMsg = "移动失败"
  1096. err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
  1097. return
  1098. }
  1099. prevSort = prevEdbInfo.Sort
  1100. }
  1101. if nextClassifyId > 0 {
  1102. //下一个兄弟节点
  1103. nextClassify, err = data_manage.GetEdbClassifyById(nextClassifyId)
  1104. if err != nil {
  1105. errMsg = "移动失败"
  1106. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
  1107. return
  1108. }
  1109. nextSort = nextClassify.Sort
  1110. } else if nextEdbInfoId > 0 {
  1111. //下一个兄弟节点
  1112. nextEdbInfo, err = data_manage.GetEdbInfoById(nextEdbInfoId)
  1113. if err != nil {
  1114. errMsg = "移动失败"
  1115. err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
  1116. return
  1117. }
  1118. nextSort = nextEdbInfo.Sort
  1119. }
  1120. err, errMsg = moveEdbClassify(parentEdbClassifyInfo, edbClassifyInfo, prevClassify, nextClassify, edbInfo, prevEdbInfo, nextEdbInfo, parentClassifyId, prevSort, nextSort, classifyType)
  1121. return
  1122. }
  1123. // moveEdbClassify 移动指标分类
  1124. func moveEdbClassify(parentEdbClassifyInfo, edbClassifyInfo, prevClassify, nextClassify *data_manage.EdbClassify, edbInfo, prevEdbInfo, nextEdbInfo *data_manage.EdbInfo, parentClassifyId int, prevSort, nextSort int, classifyType uint8) (err error, errMsg string) {
  1125. updateCol := make([]string, 0)
  1126. // 移动对象为分类, 判断分类是否存在
  1127. if edbClassifyInfo != nil {
  1128. oldParentId := edbClassifyInfo.ParentId
  1129. oldLevel := edbClassifyInfo.Level
  1130. var classifyIds []int
  1131. if oldParentId != parentClassifyId {
  1132. //更新子分类对应的level
  1133. childList, e, m := GetChildClassifyByClassifyId(edbClassifyInfo.ClassifyId)
  1134. if e != nil {
  1135. errMsg = "移动失败"
  1136. err = errors.New("查询子分类失败,Err:" + e.Error() + m)
  1137. return
  1138. }
  1139. if len(childList) > 0 {
  1140. for _, v := range childList {
  1141. if v.ClassifyId == edbClassifyInfo.ClassifyId {
  1142. continue
  1143. }
  1144. classifyIds = append(classifyIds, v.ClassifyId)
  1145. }
  1146. }
  1147. }
  1148. //判断上级id是否一致,如果不一致的话,那么需要移动该分类层级
  1149. if edbClassifyInfo.ParentId != parentClassifyId && parentClassifyId != 0 {
  1150. if edbClassifyInfo.Level != parentEdbClassifyInfo.Level+1 { //禁止层级调整
  1151. errMsg = "移动失败"
  1152. err = errors.New("不支持目录层级变更")
  1153. return
  1154. }
  1155. edbClassifyInfo.ParentId = parentEdbClassifyInfo.ClassifyId
  1156. edbClassifyInfo.RootId = parentEdbClassifyInfo.RootId
  1157. edbClassifyInfo.Level = parentEdbClassifyInfo.Level + 1
  1158. edbClassifyInfo.ModifyTime = time.Now()
  1159. updateCol = append(updateCol, "ParentId", "RootId", "Level", "ModifyTime")
  1160. } else if edbClassifyInfo.ParentId != parentClassifyId && parentClassifyId == 0 {
  1161. errMsg = "移动失败"
  1162. err = errors.New("不支持目录层级变更")
  1163. return
  1164. }
  1165. if prevSort > 0 {
  1166. //如果是移动在两个兄弟节点之间
  1167. if nextSort > 0 {
  1168. //下一个兄弟节点
  1169. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  1170. if prevSort == nextSort || prevSort == edbClassifyInfo.Sort {
  1171. //变更兄弟节点的排序
  1172. updateSortStr := `sort + 2`
  1173. //变更分类
  1174. if prevClassify != nil {
  1175. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, prevClassify.ClassifyId, prevClassify.Sort, updateSortStr, classifyType)
  1176. } else {
  1177. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, prevSort, updateSortStr, classifyType)
  1178. }
  1179. //变更指标
  1180. if prevEdbInfo != nil {
  1181. //变更兄弟节点的排序
  1182. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, prevEdbInfo.EdbInfoId, updateSortStr)
  1183. } else {
  1184. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, 0, updateSortStr)
  1185. }
  1186. } else {
  1187. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  1188. if nextSort-prevSort == 1 {
  1189. //变更兄弟节点的排序
  1190. updateSortStr := `sort + 1`
  1191. //变更分类
  1192. if prevClassify != nil {
  1193. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, prevClassify.ClassifyId, prevSort, updateSortStr, classifyType)
  1194. } else {
  1195. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, prevSort, updateSortStr, classifyType)
  1196. }
  1197. //变更指标
  1198. if prevEdbInfo != nil {
  1199. //变更兄弟节点的排序
  1200. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, prevEdbInfo.EdbInfoId, updateSortStr)
  1201. } else {
  1202. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, 0, updateSortStr)
  1203. }
  1204. }
  1205. }
  1206. }
  1207. edbClassifyInfo.Sort = prevSort + 1
  1208. edbClassifyInfo.ModifyTime = time.Now()
  1209. updateCol = append(updateCol, "Sort", "ModifyTime")
  1210. } else if prevClassify == nil && nextClassify == nil && prevEdbInfo == nil && nextEdbInfo == nil && parentClassifyId > 0 {
  1211. //处理只拖动到目录里,默认放到目录底部的情况
  1212. var maxSort int
  1213. maxSort, err = GetEdbClassifyMaxSort(parentClassifyId, classifyType)
  1214. if err != nil {
  1215. errMsg = "移动失败"
  1216. err = errors.New("查询组内排序信息失败,Err:" + err.Error())
  1217. return
  1218. }
  1219. edbClassifyInfo.Sort = maxSort + 1 //那就是排在组内最后一位
  1220. edbClassifyInfo.ModifyTime = time.Now()
  1221. updateCol = append(updateCol, "Sort", "ModifyTime")
  1222. } else {
  1223. // 拖动到父级分类的第一位
  1224. firstClassify, tmpErr := data_manage.GetFirstEdbClassifyByParentId(parentClassifyId)
  1225. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  1226. errMsg = "移动失败"
  1227. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tmpErr.Error())
  1228. return
  1229. }
  1230. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  1231. if firstClassify != nil && firstClassify.Sort == 0 {
  1232. updateSortStr := ` sort + 1 `
  1233. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, firstClassify.ClassifyId-1, 0, updateSortStr, classifyType)
  1234. //该分类下的所有指标也需要+1
  1235. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, 0, 0, updateSortStr)
  1236. } else {
  1237. //如果该分类下存在指标,且第一个指标的排序等于0,那么需要调整排序
  1238. firstEdb, tErr := data_manage.GetFirstEdbInfoByClassifyId(parentClassifyId)
  1239. if tErr != nil && tErr.Error() != utils.ErrNoRow() {
  1240. errMsg = "移动失败"
  1241. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tErr.Error())
  1242. return
  1243. }
  1244. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  1245. if firstEdb != nil && firstEdb.Sort == 0 {
  1246. updateSortStr := ` sort + 1 `
  1247. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, 0, firstEdb.EdbInfoId-1, updateSortStr)
  1248. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, 0, updateSortStr, classifyType)
  1249. }
  1250. }
  1251. edbClassifyInfo.Sort = 0 //那就是排在第一位
  1252. edbClassifyInfo.ModifyTime = time.Now()
  1253. updateCol = append(updateCol, "Sort", "ModifyTime")
  1254. }
  1255. //更新
  1256. if len(updateCol) > 0 {
  1257. err = edbClassifyInfo.Update(updateCol)
  1258. if err != nil {
  1259. errMsg = "移动失败"
  1260. err = errors.New("修改失败,Err:" + err.Error())
  1261. return
  1262. }
  1263. //更新对应分类的root_id和层级
  1264. if oldParentId != parentClassifyId {
  1265. if len(classifyIds) > 0 {
  1266. levelStep := edbClassifyInfo.Level - oldLevel
  1267. err = data_manage.UpdateEdbClassifyChildByParentClassifyId(classifyIds, edbClassifyInfo.RootId, levelStep)
  1268. if err != nil {
  1269. errMsg = "移动失败"
  1270. err = errors.New("更新子分类失败,Err:" + err.Error())
  1271. return
  1272. }
  1273. }
  1274. }
  1275. }
  1276. } else {
  1277. if edbInfo == nil {
  1278. errMsg = "当前指标不存在"
  1279. err = errors.New(errMsg)
  1280. return
  1281. }
  1282. //如果改变了分类,那么移动该指标数据
  1283. if edbInfo.ClassifyId != parentClassifyId {
  1284. edbInfo.ClassifyId = parentClassifyId
  1285. edbInfo.ModifyTime = time.Now()
  1286. updateCol = append(updateCol, "ClassifyId", "ModifyTime")
  1287. }
  1288. if prevSort > 0 {
  1289. //如果是移动在两个兄弟节点之间
  1290. if nextSort > 0 {
  1291. //下一个兄弟节点
  1292. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  1293. if prevSort == nextSort || prevSort == edbInfo.Sort {
  1294. //变更兄弟节点的排序
  1295. updateSortStr := `sort + 2`
  1296. //变更分类
  1297. if prevClassify != nil {
  1298. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, prevClassify.ClassifyId, prevClassify.Sort, updateSortStr, classifyType)
  1299. } else {
  1300. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, prevSort, updateSortStr, classifyType)
  1301. }
  1302. //变更指标
  1303. if prevEdbInfo != nil {
  1304. //变更兄弟节点的排序
  1305. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, prevEdbInfo.EdbInfoId, updateSortStr)
  1306. } else {
  1307. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, 0, updateSortStr)
  1308. }
  1309. } else {
  1310. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  1311. if nextSort-prevSort == 1 {
  1312. //变更兄弟节点的排序
  1313. updateSortStr := `sort + 1`
  1314. //变更分类
  1315. if prevClassify != nil {
  1316. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, prevClassify.ClassifyId, prevSort, updateSortStr, classifyType)
  1317. } else {
  1318. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, prevSort, updateSortStr, classifyType)
  1319. }
  1320. //变更指标
  1321. if prevEdbInfo != nil {
  1322. //变更兄弟节点的排序
  1323. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, prevEdbInfo.EdbInfoId, updateSortStr)
  1324. } else {
  1325. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, prevSort, 0, updateSortStr)
  1326. }
  1327. }
  1328. }
  1329. }
  1330. edbInfo.Sort = prevSort + 1
  1331. edbInfo.ModifyTime = time.Now()
  1332. updateCol = append(updateCol, "Sort", "ModifyTime")
  1333. } else if prevClassify == nil && nextClassify == nil && prevEdbInfo == nil && nextEdbInfo == nil && parentClassifyId > 0 {
  1334. //处理只拖动到目录里,默认放到目录底部的情况
  1335. var maxSort int
  1336. maxSort, err = GetEdbClassifyMaxSort(parentClassifyId, classifyType)
  1337. if err != nil {
  1338. errMsg = "移动失败"
  1339. err = errors.New("查询组内排序信息失败,Err:" + err.Error())
  1340. return
  1341. }
  1342. edbInfo.Sort = maxSort + 1 //那就是排在组内最后一位
  1343. edbInfo.ModifyTime = time.Now()
  1344. updateCol = append(updateCol, "Sort", "ModifyTime")
  1345. } else {
  1346. // 拖动到父级分类的第一位
  1347. firstClassify, tmpErr := data_manage.GetFirstEdbClassifyByParentId(parentClassifyId)
  1348. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  1349. errMsg = "移动失败"
  1350. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tmpErr.Error())
  1351. return
  1352. }
  1353. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  1354. if firstClassify != nil && firstClassify.Sort == 0 {
  1355. updateSortStr := ` sort + 1 `
  1356. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, firstClassify.ClassifyId-1, 0, updateSortStr, classifyType)
  1357. //该分类下的所有指标也需要+1
  1358. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, 0, 0, updateSortStr)
  1359. } else {
  1360. //如果该分类下存在指标,且第一个指标的排序等于0,那么需要调整排序
  1361. firstEdb, tErr := data_manage.GetFirstEdbInfoByClassifyId(parentClassifyId)
  1362. if tErr != nil && tErr.Error() != utils.ErrNoRow() {
  1363. errMsg = "移动失败"
  1364. err = errors.New("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tErr.Error())
  1365. return
  1366. }
  1367. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  1368. if firstEdb != nil && firstEdb.Sort == 0 {
  1369. updateSortStr := ` sort + 1 `
  1370. _ = data_manage.UpdateEdbInfoSortByClassifyId(parentClassifyId, 0, firstEdb.EdbInfoId-1, updateSortStr)
  1371. _ = data_manage.UpdateEdbClassifySortByParentId(parentClassifyId, 0, 0, updateSortStr, classifyType)
  1372. }
  1373. }
  1374. edbInfo.Sort = 0 //那就是排在第一位
  1375. edbInfo.ModifyTime = time.Now()
  1376. updateCol = append(updateCol, "Sort", "ModifyTime")
  1377. }
  1378. //更新
  1379. if len(updateCol) > 0 {
  1380. err = edbInfo.Update(updateCol)
  1381. if err != nil {
  1382. errMsg = "移动失败"
  1383. err = errors.New("修改失败,Err:" + err.Error())
  1384. return
  1385. }
  1386. }
  1387. }
  1388. return
  1389. }
  1390. // GetEdbOpButton 获取ETA指标的操作权限
  1391. func GetEdbOpButton(sysUser *system.Admin, belongUserId, edbType, edbInfoType int, haveOperaAuth bool) (button data_manage.EdbClassifyItemsButton) {
  1392. // 没有数据权限就直接返回
  1393. if !haveOperaAuth {
  1394. return
  1395. }
  1396. //2、用户对于自己添加的分类,有权限编辑、移动和删除该分类;
  1397. //3、ficc管理员和超管对所有分类有编辑、移动和删除权限;
  1398. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId || sysUser.EdbPermission == 1 {
  1399. button.AddButton = true
  1400. button.OpButton = true
  1401. button.DeleteButton = true
  1402. button.MoveButton = true
  1403. }
  1404. // ETA指标的基础指标才允许插入值
  1405. if edbInfoType == 0 && edbType == 1 {
  1406. button.InsertNewDataButton = true
  1407. }
  1408. button.ShowEdbRelation = true
  1409. button.ShowChartRelation = true
  1410. return
  1411. }
  1412. // GetEdbClassifyOpButton 获取ETA指标分类的操作权限
  1413. func GetEdbClassifyOpButton(sysUser *system.Admin, belongUserId int, haveOperaAuth bool) (button data_manage.EdbClassifyItemsButton) {
  1414. // 没有数据权限就直接返回
  1415. if !haveOperaAuth {
  1416. return
  1417. }
  1418. //ficc管理员和超管和ficc研究员有权限创建和管理分类,可以编辑分类名称(分类名称不允许重复),可以拖动分类,改变分类顺序,可以拖动分类下模型,改变顺序,可以删除分类,若分类下有预测指标,则不允许删除;
  1419. //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) {
  1420. button.AddButton = true
  1421. button.OpButton = true
  1422. button.DeleteButton = true
  1423. button.MoveButton = true
  1424. //}
  1425. return
  1426. }
  1427. // GetPredictEdbOpButton 获取ETA预测指标的操作权限
  1428. func GetPredictEdbOpButton(sysUser *system.Admin, belongUserId int, haveOperaAuth bool) (button data_manage.EdbClassifyItemsButton) {
  1429. // 没有数据权限就直接返回
  1430. if !haveOperaAuth {
  1431. return
  1432. }
  1433. //预测指标的添加人对该预测指标有全部操作权限,ficc管理员、超管对所有预测指标有全部操作权限;
  1434. if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN || sysUser.AdminId == belongUserId || sysUser.PredictEdbPermission == 1 {
  1435. button.AddButton = true
  1436. button.OpButton = true
  1437. button.DeleteButton = true
  1438. button.MoveButton = true
  1439. }
  1440. button.ShowEdbRelation = true
  1441. button.ShowChartRelation = true
  1442. return
  1443. }
  1444. // GetPredictEdbClassifyOpButton 获取ETA预测指标分类的操作权限
  1445. func GetPredictEdbClassifyOpButton(sysUser *system.Admin, belongUserId int, haveOperaAuth bool) (button data_manage.EdbClassifyItemsButton) {
  1446. // 没有数据权限就直接返回
  1447. if !haveOperaAuth {
  1448. return
  1449. }
  1450. //ficc管理员和超管和ficc研究员有权限创建和管理分类,可以编辑分类名称(分类名称不允许重复),可以拖动分类,改变分类顺序,可以拖动分类下模型,改变顺序,可以删除分类,若分类下有预测指标,则不允许删除;
  1451. //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) {
  1452. button.AddButton = true
  1453. button.OpButton = true
  1454. button.DeleteButton = true
  1455. button.MoveButton = true
  1456. //}
  1457. return
  1458. }
  1459. func GetEdbClassifyMaxSort(parentId int, classifyType uint8) (maxSort int, err error) {
  1460. //获取该层级下最大的排序数
  1461. classifyMaxSort, err := data_manage.GetEdbClassifyMaxSort(parentId, classifyType)
  1462. if err != nil {
  1463. return
  1464. }
  1465. maxSort = classifyMaxSort
  1466. edbMaxSort, err := data_manage.GetEdbInfoMaxSortByClassifyId(parentId)
  1467. if err != nil {
  1468. return
  1469. }
  1470. if maxSort < edbMaxSort {
  1471. maxSort = edbMaxSort
  1472. }
  1473. return
  1474. }
  1475. func GetEdbClassifyByIsMe(adminId, parentId, edbInfoType int, classifyList []*data_manage.EdbClassifyItems) (list []*data_manage.EdbClassifyItems, err error) {
  1476. edbInfoList, err := data_manage.GetEdbInfoListByUserId([]int{adminId}, edbInfoType)
  1477. if err != nil {
  1478. return
  1479. }
  1480. classifyIdList := make([]int, 0)
  1481. for _, edbInfo := range edbInfoList {
  1482. if edbInfo.ClassifyId > 0 {
  1483. classifyIdList = append(classifyIdList, edbInfo.ClassifyId)
  1484. }
  1485. }
  1486. if parentId > 0 {
  1487. for _, v := range classifyList {
  1488. if v.ClassifyId > 0 && v.EdbInfoId == 0 {
  1489. classifyItems, er, _ := GetChildClassifyByClassifyId(v.ClassifyId)
  1490. if er != nil {
  1491. err = er
  1492. return
  1493. }
  1494. existClassifyMap := make(map[int]struct{})
  1495. for _, classify := range classifyItems {
  1496. existClassifyMap[classify.ClassifyId] = struct{}{}
  1497. }
  1498. for _, edb := range edbInfoList {
  1499. if _, ok := existClassifyMap[edb.ClassifyId]; ok {
  1500. list = append(list, v)
  1501. break
  1502. }
  1503. }
  1504. } else {
  1505. list = append(list, v)
  1506. }
  1507. }
  1508. } else {
  1509. edbInfoClassifyList, er := data_manage.GetEdbClassifyByIdList(classifyIdList)
  1510. if er != nil {
  1511. err = er
  1512. return
  1513. }
  1514. existClassifyIdMap := make(map[int]struct{})
  1515. for _, classify := range edbInfoClassifyList {
  1516. existClassifyIdMap[classify.RootId] = struct{}{}
  1517. }
  1518. for _, classify := range classifyList {
  1519. if _, ok := existClassifyIdMap[classify.ClassifyId]; ok {
  1520. list = append(list, classify)
  1521. }
  1522. }
  1523. }
  1524. return
  1525. }
  1526. // GetEdbClassifyChildrenRecursive 根据父目录递归子级目录
  1527. func GetEdbClassifyChildrenRecursive(list []*data_manage.EdbClassifyItems, parentId int) []*data_manage.EdbClassifyItems {
  1528. var res []*data_manage.EdbClassifyItems
  1529. for _, v := range list {
  1530. if v.ParentId == parentId {
  1531. // 递归调用以获取更深层次的子级
  1532. children := GetEdbClassifyChildrenRecursive(list, v.ClassifyId)
  1533. // 将当前节点和所有子节点添加到结果中
  1534. res = append(res, v)
  1535. res = append(res, children...)
  1536. } else if v.ClassifyId == parentId {
  1537. // 将当前节点添加到结果中
  1538. res = append(res, v)
  1539. }
  1540. }
  1541. return res
  1542. }
  1543. // GetEdbClassifyChildrenRecursiveByParentIds 根据父目录递归子级目录
  1544. func GetEdbClassifyChildrenRecursiveByParentIds(list []*data_manage.EdbClassifyItems, parentIds []string) []*data_manage.EdbClassifyItems{
  1545. var res []*data_manage.EdbClassifyItems
  1546. for _, v := range list {
  1547. for _, id := range parentIds {
  1548. parentId, _ := strconv.Atoi(id)
  1549. if v.ParentId == parentId || v.ClassifyId == parentId {
  1550. // 递归调用以获取更深层次的子级
  1551. children := GetEdbClassifyChildrenRecursive(list, v.ClassifyId)
  1552. // 将当前节点和所有子节点添加到结果中
  1553. res = append(res, v)
  1554. res = append(res, children...)
  1555. }
  1556. }
  1557. }
  1558. return res
  1559. }