chart_classify.go 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980
  1. package data_manage
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "hongze/hz_eta_api/controllers"
  6. "hongze/hz_eta_api/models"
  7. "hongze/hz_eta_api/models/data_manage"
  8. "hongze/hz_eta_api/models/system"
  9. "hongze/hz_eta_api/services/data"
  10. "hongze/hz_eta_api/utils"
  11. "strconv"
  12. "time"
  13. )
  14. // 数据管理-分类模块
  15. type ChartClassifyController struct {
  16. controllers.BaseAuthController
  17. }
  18. //
  19. //// @Title 图表分类列表
  20. //// @Description 图表分类列表接口
  21. //// @Success 200 {object} data_manage.ChartClassifyListResp
  22. //// @router /chart_classify/list [get]
  23. //func (this *ChartClassifyController) ChartClassifyList() {
  24. // br := new(models.BaseResponse).Init()
  25. // defer func() {
  26. // this.Data["json"] = br
  27. // this.ServeJSON()
  28. // }()
  29. // rootList, err := data_manage.GetChartClassifyByParentId(0)
  30. // if err != nil {
  31. // br.Msg = "获取失败"
  32. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  33. // return
  34. // }
  35. //
  36. // classifyAll, err := data_manage.GetChartClassifyAll()
  37. // if err != nil {
  38. // br.Msg = "获取失败"
  39. // br.ErrMsg = "获取数据失败,Err:" + err.Error()
  40. // return
  41. // }
  42. //
  43. // nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  44. // for k := range rootList {
  45. // rootNode := rootList[k]
  46. // data.ChartClassifyListMakeTree(classifyAll, rootNode)
  47. // nodeAll = append(nodeAll, rootNode)
  48. // }
  49. // resp := new(data_manage.ChartClassifyListResp)
  50. // resp.AllNodes = nodeAll
  51. // br.Ret = 200
  52. // br.Success = true
  53. // br.Msg = "获取成功"
  54. // br.Data = resp
  55. //}
  56. // ChartClassifyListV2
  57. // @Title 图表分类列表
  58. // @Description 图表分类列表接口
  59. // @Param IsShowMe query bool true "是否只看我的,true、false"
  60. // @Success 200 {object} data_manage.ChartClassifyListResp
  61. // @router /chart_classify/list [get]
  62. func (this *ChartClassifyController) ChartClassifyListV2() {
  63. br := new(models.BaseResponse).Init()
  64. defer func() {
  65. this.Data["json"] = br
  66. this.ServeJSON()
  67. }()
  68. resp := new(data_manage.ChartClassifyListResp)
  69. // 获取当前账号的不可见指标
  70. noPermissionChartIdMap := make(map[int]bool)
  71. {
  72. obj := data_manage.EdbInfoNoPermissionAdmin{}
  73. confList, err := obj.GetAllChartListByAdminId(this.SysUser.AdminId)
  74. if err != nil && err.Error() != utils.ErrNoRow() {
  75. br.Msg = "获取失败"
  76. br.ErrMsg = "获取不可见指标配置数据失败,Err:" + err.Error()
  77. return
  78. }
  79. for _, v := range confList {
  80. noPermissionChartIdMap[v.ChartInfoId] = true
  81. }
  82. }
  83. isShowMe, _ := this.GetBool("IsShowMe")
  84. if isShowMe {
  85. errMsg, err := getChartClassifyListForMe(*this.SysUser, resp)
  86. if err != nil {
  87. br.Msg = errMsg
  88. br.ErrMsg = err.Error()
  89. return
  90. }
  91. // 移除没有权限的图表
  92. allNodes := handleNoPermissionChart(resp.AllNodes, noPermissionChartIdMap)
  93. resp.AllNodes = allNodes
  94. br.Ret = 200
  95. br.Success = true
  96. br.Msg = "获取成功"
  97. br.Data = resp
  98. fmt.Println("source my classify")
  99. return
  100. }
  101. //判断是否存在缓存,如果存在缓存,那么直接从缓存中获取
  102. key := utils.CACHE_CHART_CLASSIFY
  103. if utils.Re == nil {
  104. if utils.Re == nil && utils.Rc.IsExist(key) {
  105. if data, err1 := utils.Rc.RedisBytes(key); err1 == nil {
  106. err := json.Unmarshal(data, &resp)
  107. if err == nil && resp != nil {
  108. // 移除没有权限的图表
  109. allNodes := handleNoPermissionChart(resp.AllNodes, noPermissionChartIdMap)
  110. resp.AllNodes = allNodes
  111. br.Ret = 200
  112. br.Success = true
  113. br.Msg = "获取成功"
  114. br.Data = resp
  115. fmt.Println("source redis")
  116. return
  117. }
  118. }
  119. }
  120. }
  121. rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_DEFAULT)
  122. if err != nil && err.Error() != utils.ErrNoRow() {
  123. br.Msg = "获取失败"
  124. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  125. return
  126. }
  127. classifyAll, err := data_manage.GetChartClassifyAll(utils.CHART_SOURCE_DEFAULT)
  128. if err != nil && err.Error() != utils.ErrNoRow() {
  129. br.Msg = "获取失败"
  130. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  131. return
  132. }
  133. allChartInfo, err := data_manage.GetChartInfoAll([]int{utils.CHART_SOURCE_DEFAULT})
  134. if err != nil && err.Error() != utils.ErrNoRow() {
  135. br.Msg = "获取失败"
  136. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  137. return
  138. }
  139. chartInfoMap := make(map[int][]*data_manage.ChartClassifyItems)
  140. for _, v := range allChartInfo {
  141. chartInfoMap[v.ChartClassifyId] = append(chartInfoMap[v.ChartClassifyId], v)
  142. }
  143. rootChildMap := make(map[int][]*data_manage.ChartClassifyItems)
  144. for _, v := range classifyAll {
  145. rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v)
  146. if existItems, ok := chartInfoMap[v.ChartClassifyId]; ok {
  147. v.Children = existItems
  148. } else {
  149. items := make([]*data_manage.ChartClassifyItems, 0)
  150. v.Children = items
  151. }
  152. }
  153. nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  154. for _, v := range rootList {
  155. if existItems, ok := rootChildMap[v.ChartClassifyId]; ok {
  156. v.Children = existItems
  157. } else {
  158. items := make([]*data_manage.ChartClassifyItems, 0)
  159. v.Children = items
  160. }
  161. nodeAll = append(nodeAll, v)
  162. }
  163. //
  164. //
  165. //nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  166. //for k := range rootList {
  167. // rootNode := rootList[k]
  168. // data.ChartClassifyListMakeTree(classifyAll, rootNode)
  169. // nodeAll = append(nodeAll, rootNode)
  170. //}
  171. resp.AllNodes = nodeAll
  172. // 将数据加入缓存
  173. if utils.Re == nil {
  174. data, _ := json.Marshal(resp)
  175. utils.Rc.Put(key, data, 2*time.Hour)
  176. }
  177. // 移除没有权限的图表
  178. allNodes := handleNoPermissionChart(resp.AllNodes, noPermissionChartIdMap)
  179. resp.AllNodes = allNodes
  180. br.Ret = 200
  181. br.Success = true
  182. br.Msg = "获取成功"
  183. br.Data = resp
  184. }
  185. // getChartClassifyListForMe 获取我创建的图表
  186. func getChartClassifyListForMe(adminInfo system.Admin, resp *data_manage.ChartClassifyListResp) (errMsg string, err error) {
  187. rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_DEFAULT)
  188. if err != nil && err.Error() != utils.ErrNoRow() {
  189. errMsg = "获取失败"
  190. return
  191. }
  192. classifyAll, err := data_manage.GetChartClassifyAll(utils.CHART_SOURCE_DEFAULT)
  193. if err != nil && err.Error() != utils.ErrNoRow() {
  194. errMsg = "获取失败"
  195. return
  196. }
  197. allChartInfo, err := data_manage.GetChartInfoByAdminId([]int{utils.CHART_SOURCE_DEFAULT}, adminInfo.AdminId)
  198. if err != nil && err.Error() != utils.ErrNoRow() {
  199. errMsg = "获取失败"
  200. return
  201. }
  202. chartInfoMap := make(map[int][]*data_manage.ChartClassifyItems)
  203. for _, v := range allChartInfo {
  204. chartInfoMap[v.ChartClassifyId] = append(chartInfoMap[v.ChartClassifyId], v)
  205. }
  206. rootChildMap := make(map[int][]*data_manage.ChartClassifyItems)
  207. for _, v := range classifyAll {
  208. rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v)
  209. if existItems, ok := chartInfoMap[v.ChartClassifyId]; ok {
  210. v.Children = existItems
  211. } else {
  212. items := make([]*data_manage.ChartClassifyItems, 0)
  213. v.Children = items
  214. }
  215. }
  216. nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  217. for _, v := range rootList {
  218. if existItems, ok := rootChildMap[v.ChartClassifyId]; ok {
  219. v.Children = existItems
  220. } else {
  221. items := make([]*data_manage.ChartClassifyItems, 0)
  222. v.Children = items
  223. }
  224. nodeAll = append(nodeAll, v)
  225. }
  226. resp.AllNodes = nodeAll
  227. return
  228. }
  229. // handleNoPermissionChart 图表列表返回,将没有权限的图表移除
  230. func handleNoPermissionChart(allNodes []*data_manage.ChartClassifyItems, noPermissionChartIdMap map[int]bool) (newAllNodes []*data_manage.ChartClassifyItems) {
  231. // 移除没有权限的图表
  232. newAllNodes = make([]*data_manage.ChartClassifyItems, 0)
  233. for _, node := range allNodes {
  234. // 二级分类
  235. tmpNodeInfo := *node
  236. tmpNodeList := make([]*data_manage.ChartClassifyItems, 0)
  237. if node.Children != nil {
  238. for _, chartList := range node.Children {
  239. tmpInfo := *chartList
  240. tmpList := make([]*data_manage.ChartClassifyItems, 0)
  241. if chartList.Children != nil {
  242. for _, chartInfo := range chartList.Children {
  243. thirdInfo := *chartInfo
  244. thirdList := make([]*data_manage.ChartClassifyItems, 0)
  245. // 如果指标不可见,那么就不返回该指标
  246. if _, ok := noPermissionChartIdMap[chartInfo.ChartInfoId]; ok {
  247. continue
  248. }
  249. tmpList = append(tmpList, chartInfo)
  250. if chartInfo.Children != nil {
  251. for _, thirdChart := range chartInfo.Children {
  252. // 如果指标不可见,那么就不返回该指标
  253. if _, ok := noPermissionChartIdMap[chartInfo.ChartInfoId]; ok {
  254. continue
  255. }
  256. thirdList = append(thirdList, thirdChart)
  257. }
  258. }
  259. thirdInfo.Children = thirdList
  260. tmpList = append(tmpList, &thirdInfo)
  261. }
  262. }
  263. tmpInfo.Children = tmpList
  264. tmpNodeList = append(tmpNodeList, &tmpInfo)
  265. }
  266. }
  267. tmpNodeInfo.Children = tmpNodeList
  268. newAllNodes = append(newAllNodes, &tmpNodeInfo)
  269. }
  270. return
  271. }
  272. // @Title 获取所有图表分类接口-不包含图表
  273. // @Description 获取所有图表分类接口-不包含图表
  274. // @Success 200 {object} data_manage.ChartClassifyListResp
  275. // @router /chart_classify/items [get]
  276. func (this *ChartClassifyController) ChartClassifyItems() {
  277. br := new(models.BaseResponse).Init()
  278. defer func() {
  279. this.Data["json"] = br
  280. this.ServeJSON()
  281. }()
  282. rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_DEFAULT)
  283. if err != nil {
  284. br.Msg = "获取失败"
  285. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  286. return
  287. }
  288. classifyAll, err := data_manage.GetChartClassifyAll(utils.CHART_SOURCE_DEFAULT)
  289. if err != nil {
  290. br.Msg = "获取失败"
  291. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  292. return
  293. }
  294. nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  295. for k := range rootList {
  296. rootNode := rootList[k]
  297. data.ChartClassifyItemsMakeTree(classifyAll, rootNode)
  298. nodeAll = append(nodeAll, rootNode)
  299. }
  300. language := `CN`
  301. // 指标显示的语言
  302. {
  303. configDetail, _ := system.GetConfigDetailByCode(this.SysUser.AdminId, system.ChartLanguageVar)
  304. if configDetail != nil {
  305. language = configDetail.ConfigValue
  306. } else {
  307. configDetail, _ = system.GetDefaultConfigDetailByCode(system.ChartLanguageVar)
  308. if configDetail != nil {
  309. language = configDetail.ConfigValue
  310. }
  311. }
  312. }
  313. resp := new(data_manage.ChartClassifyListResp)
  314. resp.AllNodes = nodeAll
  315. resp.Language = language
  316. br.Ret = 200
  317. br.Success = true
  318. br.Msg = "获取成功"
  319. br.Data = resp
  320. }
  321. // @Title 新增图表分类
  322. // @Description 新增图表分类接口
  323. // @Param request body data_manage.AddChartClassifyReq true "type json string"
  324. // @Success 200 Ret=200 保存成功
  325. // @router /chart_classify/add [post]
  326. func (this *ChartClassifyController) AddChartClassify() {
  327. br := new(models.BaseResponse).Init()
  328. defer func() {
  329. this.Data["json"] = br
  330. this.ServeJSON()
  331. }()
  332. var req data_manage.AddChartClassifyReq
  333. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  334. if err != nil {
  335. br.Msg = "参数解析异常!"
  336. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  337. return
  338. }
  339. if req.ChartClassifyName == "" {
  340. br.Msg = "请输入分类名称"
  341. br.IsSendEmail = false
  342. return
  343. }
  344. if req.ParentId < 0 {
  345. br.Msg = "参数错误"
  346. br.IsSendEmail = false
  347. return
  348. }
  349. count, err := data_manage.GetChartClassifyCount(req.ChartClassifyName, req.ParentId, utils.CHART_SOURCE_DEFAULT)
  350. if err != nil {
  351. br.Msg = "判断名称是否已存在失败"
  352. br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error()
  353. return
  354. }
  355. if count > 0 {
  356. br.Msg = "分类名称已存在,请重新输入"
  357. br.IsSendEmail = false
  358. return
  359. }
  360. //获取该层级下最大的排序数
  361. maxSort, err := data_manage.GetChartClassifyMaxSort(req.ParentId, utils.CHART_SOURCE_DEFAULT)
  362. classify := new(data_manage.ChartClassify)
  363. classify.ParentId = req.ParentId
  364. classify.ChartClassifyName = req.ChartClassifyName
  365. classify.HasData = 0
  366. classify.CreateTime = time.Now()
  367. classify.ModifyTime = time.Now()
  368. classify.SysUserId = this.SysUser.AdminId
  369. classify.SysUserRealName = this.SysUser.RealName
  370. classify.Level = req.Level + 1
  371. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  372. classify.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  373. classify.Sort = maxSort + 1
  374. classify.Source = utils.CHART_SOURCE_DEFAULT
  375. _, err = data_manage.AddChartClassify(classify)
  376. if err != nil {
  377. br.Msg = "保存分类失败"
  378. br.ErrMsg = "保存分类失败,Err:" + err.Error()
  379. return
  380. }
  381. br.Ret = 200
  382. br.Msg = "保存成功"
  383. br.Success = true
  384. }
  385. // @Title 修改图表分类
  386. // @Description 修改图表分类接口
  387. // @Param request body data_manage.EditChartClassifyReq true "type json string"
  388. // @Success 200 Ret=200 修改成功
  389. // @router /chart_classify/edit [post]
  390. func (this *ChartClassifyController) EditChartClassify() {
  391. br := new(models.BaseResponse).Init()
  392. defer func() {
  393. this.Data["json"] = br
  394. this.ServeJSON()
  395. }()
  396. var req data_manage.EditChartClassifyReq
  397. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  398. if err != nil {
  399. br.Msg = "参数解析异常!"
  400. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  401. return
  402. }
  403. if req.ChartClassifyName == "" {
  404. br.Msg = "请输入分类名称"
  405. br.IsSendEmail = false
  406. return
  407. }
  408. if req.ChartClassifyId <= 0 {
  409. br.Msg = "参数错误"
  410. br.IsSendEmail = false
  411. return
  412. }
  413. item, err := data_manage.GetChartClassifyById(req.ChartClassifyId)
  414. if err != nil {
  415. br.Msg = "保存失败"
  416. br.Msg = "获取分类信息失败,Err:" + err.Error()
  417. return
  418. }
  419. if item.ChartClassifyName != req.ChartClassifyName {
  420. count, err := data_manage.GetChartClassifyCount(req.ChartClassifyName, item.ParentId, utils.CHART_SOURCE_DEFAULT)
  421. if err != nil {
  422. br.Msg = "判断名称是否已存在失败"
  423. br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error()
  424. return
  425. }
  426. if count > 0 {
  427. br.Msg = "分类名称已存在,请重新输入"
  428. br.IsSendEmail = false
  429. return
  430. }
  431. err = data_manage.EditChartClassify(req.ChartClassifyId, utils.CHART_SOURCE_DEFAULT, req.ChartClassifyName)
  432. if err != nil {
  433. br.Msg = "保存失败"
  434. br.ErrMsg = "保存失败,Err:" + err.Error()
  435. return
  436. }
  437. }
  438. br.Ret = 200
  439. br.Msg = "保存成功"
  440. br.Success = true
  441. br.IsAddLog = true
  442. }
  443. // @Title 删除图表检测接口
  444. // @Description 删除图表检测接口
  445. // @Param request body data_manage.ChartClassifyDeleteCheckResp true "type json string"
  446. // @Success 200 Ret=200 检测成功
  447. // @router /chart_classify/delete/check [post]
  448. func (this *ChartClassifyController) DeleteChartClassifyCheck() {
  449. br := new(models.BaseResponse).Init()
  450. defer func() {
  451. this.Data["json"] = br
  452. this.ServeJSON()
  453. }()
  454. var req data_manage.ChartClassifyDeleteCheckReq
  455. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  456. if err != nil {
  457. br.Msg = "参数解析异常!"
  458. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  459. return
  460. }
  461. if req.ChartClassifyId < 0 && req.ChartInfoId <= 0 {
  462. br.Msg = "参数错误"
  463. br.IsSendEmail = false
  464. return
  465. }
  466. var deleteStatus int
  467. var tipsMsg string
  468. //删除分类
  469. if req.ChartClassifyId > 0 && req.ChartInfoId == 0 {
  470. //判断图表分类下,是否含有图表
  471. count, err := data_manage.GetChartInfoCountByClassifyId(req.ChartClassifyId)
  472. if err != nil {
  473. br.Msg = "删除失败"
  474. br.ErrMsg = "分类下是否含有指标失败,Err:" + err.Error()
  475. return
  476. }
  477. if count > 0 {
  478. deleteStatus = 1
  479. tipsMsg = "该分类下关联图表不可删除"
  480. }
  481. }
  482. if deleteStatus != 1 && req.ChartInfoId == 0 {
  483. classifyCount, err := data_manage.GetChartClassifyCountByClassifyId(req.ChartClassifyId)
  484. if err != nil && err.Error() != utils.ErrNoRow() {
  485. br.Msg = "删除失败"
  486. br.ErrMsg = "分类下是否含有图表失败,Err:" + err.Error()
  487. return
  488. }
  489. if classifyCount > 0 {
  490. deleteStatus = 2
  491. tipsMsg = "确认删除当前目录及包含的子目录吗"
  492. }
  493. }
  494. if deleteStatus == 0 {
  495. tipsMsg = "可删除,进行删除操作"
  496. }
  497. resp := new(data_manage.ChartClassifyDeleteCheckResp)
  498. resp.DeleteStatus = deleteStatus
  499. resp.TipsMsg = tipsMsg
  500. br.Ret = 200
  501. br.Msg = "检测成功"
  502. br.Success = true
  503. br.Data = resp
  504. }
  505. // @Title 删除图表分类/图表
  506. // @Description 删除图表分类/图表接口
  507. // @Param request body data_manage.DeleteChartClassifyReq true "type json string"
  508. // @Success 200 Ret=200 删除成功
  509. // @router /chart_classify/delete [post]
  510. func (this *ChartClassifyController) DeleteChartClassify() {
  511. br := new(models.BaseResponse).Init()
  512. defer func() {
  513. this.Data["json"] = br
  514. this.ServeJSON()
  515. }()
  516. sysUser := this.SysUser
  517. if sysUser == nil {
  518. br.Msg = "请登录"
  519. br.ErrMsg = "请登录,SysUser Is Empty"
  520. br.Ret = 408
  521. return
  522. }
  523. var req data_manage.DeleteChartClassifyReq
  524. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  525. if err != nil {
  526. br.Msg = "参数解析异常!"
  527. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  528. return
  529. }
  530. if req.ChartClassifyId < 0 && req.ChartInfoId <= 0 {
  531. br.Msg = "参数错误"
  532. br.IsSendEmail = false
  533. return
  534. }
  535. //删除分类
  536. if req.ChartClassifyId > 0 && req.ChartInfoId == 0 {
  537. //判断是否含有指标
  538. count, err := data_manage.GetChartInfoCountByClassifyId(req.ChartClassifyId)
  539. if err != nil && err.Error() != utils.ErrNoRow() {
  540. br.Msg = "删除失败"
  541. br.ErrMsg = "判断名称是否已存在失败,Err:" + err.Error()
  542. return
  543. }
  544. if count > 0 {
  545. br.Msg = "该目录下存在关联指标,不可删除"
  546. br.IsSendEmail = false
  547. return
  548. }
  549. err = data_manage.DeleteChartClassify(req.ChartClassifyId)
  550. if err != nil {
  551. br.Msg = "删除失败"
  552. br.ErrMsg = "删除失败,Err:" + err.Error()
  553. return
  554. }
  555. }
  556. resp := new(data_manage.AddChartInfoResp)
  557. //删除图表
  558. if req.ChartInfoId > 0 {
  559. chartInfo, err := data_manage.GetChartInfoById(req.ChartInfoId)
  560. if err != nil {
  561. if err.Error() == utils.ErrNoRow() {
  562. br.Msg = "图表已删除,请刷新页面"
  563. br.ErrMsg = "指标不存在,Err:" + err.Error()
  564. return
  565. } else {
  566. br.Msg = "删除失败"
  567. br.ErrMsg = "删除失败,获取指标信息失败,Err:" + err.Error()
  568. return
  569. }
  570. }
  571. if chartInfo == nil {
  572. br.Msg = "图表已删除,请刷新页面"
  573. return
  574. }
  575. //图表操作权限
  576. ok := data.CheckOpChartPermission(sysUser, chartInfo.SysUserId)
  577. if !ok {
  578. br.Msg = "没有该图表的操作权限"
  579. br.ErrMsg = "没有该图表的操作权限"
  580. return
  581. }
  582. // 获取引用该图表的MyCharts, 用于ES删除
  583. var myCond string
  584. var myPars []interface{}
  585. myCond += ` AND a.chart_info_id = ? `
  586. myPars = append(myPars, req.ChartInfoId)
  587. myCharts, e := data_manage.GetMyChartListGroupByCharyInfoIdAndAdminIdByCondition(myCond, myPars)
  588. if e != nil {
  589. br.Msg = "删除失败"
  590. br.ErrMsg = "获取引用图表的MyChats失败, Err: " + e.Error()
  591. return
  592. }
  593. myIds := make([]int, 0)
  594. for _, m := range myCharts {
  595. myIds = append(myIds, m.MyChartId)
  596. }
  597. //删除图表及关联指标
  598. err = data_manage.DeleteChartInfoAndData(req.ChartInfoId)
  599. if err != nil {
  600. br.Msg = "删除失败"
  601. br.ErrMsg = "删除失败,Err:" + err.Error()
  602. return
  603. }
  604. //删除ES
  605. {
  606. go data.EsDeleteChartInfo(req.ChartInfoId)
  607. // 删除MY ETA 图表 es数据
  608. //go data.EsDeleteMyChartInfoByChartInfoId(req.ChartInfoId)
  609. go data.EsDeleteMyChartInfoByMyChartIds(myIds)
  610. }
  611. var condition string
  612. var pars []interface{}
  613. condition += " AND chart_classify_id=? AND source = ? "
  614. pars = append(pars, chartInfo.ChartClassifyId, utils.CHART_SOURCE_DEFAULT)
  615. condition += " AND chart_info_id>? ORDER BY create_time ASC LIMIT 1 "
  616. pars = append(pars, req.ChartInfoId)
  617. nextItem, err := data_manage.GetChartInfoByCondition(condition, pars)
  618. if err != nil && err.Error() != utils.ErrNoRow() {
  619. br.Msg = "删除失败"
  620. br.ErrMsg = "获取下一级图库信息失败,Err:" + err.Error()
  621. return
  622. }
  623. if nextItem != nil {
  624. resp.UniqueCode = nextItem.UniqueCode
  625. resp.ChartInfoId = nextItem.ChartInfoId
  626. } else {
  627. var condition string
  628. var pars []interface{}
  629. condition += " AND level=3 "
  630. //pars = append(pars, chartInfo.ChartClassifyId)
  631. condition += " AND chart_classify_id>? ORDER BY chart_classify_id ASC LIMIT 1 "
  632. pars = append(pars, chartInfo.ChartClassifyId)
  633. classifyItem, err := data_manage.GetChartClassifyByCondition(condition, pars)
  634. if err != nil && err.Error() != utils.ErrNoRow() {
  635. br.Msg = "删除失败"
  636. br.ErrMsg = "获取下一级图库分类信息失败,Err:" + err.Error()
  637. return
  638. }
  639. if classifyItem != nil {
  640. nextItem, err = data_manage.GetNextChartInfo(chartInfo.ChartClassifyId)
  641. if err != nil && err.Error() != utils.ErrNoRow() {
  642. br.Msg = "删除失败"
  643. br.ErrMsg = "获取下一级图库信息失败,Err:" + err.Error()
  644. return
  645. }
  646. if nextItem != nil {
  647. resp.UniqueCode = nextItem.UniqueCode
  648. resp.ChartInfoId = nextItem.ChartInfoId
  649. }
  650. }
  651. }
  652. //新增操作日志
  653. {
  654. chartLog := new(data_manage.ChartInfoLog)
  655. chartLog.ChartName = chartInfo.ChartName
  656. chartLog.ChartInfoId = req.ChartInfoId
  657. chartLog.ChartClassifyId = chartInfo.ChartClassifyId
  658. chartLog.SysUserId = sysUser.AdminId
  659. chartLog.SysUserRealName = sysUser.RealName
  660. chartLog.UniqueCode = chartInfo.UniqueCode
  661. chartLog.CreateTime = time.Now()
  662. chartLog.Content = string(this.Ctx.Input.RequestBody)
  663. chartLog.Status = "删除图表"
  664. chartLog.Method = this.Ctx.Input.URI()
  665. go data_manage.AddChartInfoLog(chartLog)
  666. }
  667. }
  668. br.Ret = 200
  669. br.Msg = "删除成功"
  670. br.Success = true
  671. br.Data = resp
  672. br.IsAddLog = true
  673. }
  674. // ChartClassifyMove
  675. // @Title 图表分类移动接口
  676. // @Description 图表分类移动接口
  677. // @Success 200 {object} data_manage.MoveChartClassifyReq
  678. // @router /chart_classify/move [post]
  679. func (this *ChartClassifyController) ChartClassifyMove() {
  680. br := new(models.BaseResponse).Init()
  681. defer func() {
  682. this.Data["json"] = br
  683. this.ServeJSON()
  684. }()
  685. sysUser := this.SysUser
  686. if sysUser == nil {
  687. br.Msg = "请登录"
  688. br.ErrMsg = "请登录,SysUser Is Empty"
  689. br.Ret = 408
  690. return
  691. }
  692. var req data_manage.MoveChartClassifyReq
  693. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  694. if err != nil {
  695. br.Msg = "参数解析异常!"
  696. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  697. return
  698. }
  699. if req.ClassifyId <= 0 {
  700. br.Msg = "参数错误"
  701. br.ErrMsg = "分类id小于等于0"
  702. return
  703. }
  704. //判断分类是否存在
  705. chartClassifyInfo, err := data_manage.GetChartClassifyById(req.ClassifyId)
  706. if err != nil {
  707. br.Msg = "移动失败"
  708. br.ErrMsg = "获取分类信息失败,Err:" + err.Error()
  709. return
  710. }
  711. if chartClassifyInfo.Source != utils.CHART_SOURCE_DEFAULT {
  712. br.Msg = "分类异常"
  713. br.ErrMsg = "分类异常,不是ETA图库的分类"
  714. return
  715. }
  716. updateCol := make([]string, 0)
  717. //判断上级id是否一致,如果不一致的话,那么需要移动该分类层级
  718. if chartClassifyInfo.ParentId != req.ParentClassifyId && req.ParentClassifyId != 0 {
  719. parentChartClassifyInfo, err := data_manage.GetChartClassifyById(req.ParentClassifyId)
  720. if err != nil {
  721. br.Msg = "移动失败"
  722. br.ErrMsg = "获取上级分类信息失败,Err:" + err.Error()
  723. return
  724. }
  725. chartClassifyInfo.ParentId = parentChartClassifyInfo.ChartClassifyId
  726. chartClassifyInfo.Level = parentChartClassifyInfo.Level + 1
  727. chartClassifyInfo.ModifyTime = time.Now()
  728. updateCol = append(updateCol, "ParentId", "Level", "ModifyTime")
  729. }
  730. //如果有传入 上一个兄弟节点分类id
  731. if req.PrevClassifyId > 0 {
  732. //上一个兄弟节点
  733. prevClassify, err := data_manage.GetChartClassifyById(req.PrevClassifyId)
  734. if err != nil {
  735. br.Msg = "移动失败"
  736. br.ErrMsg = "获取上一个兄弟节点分类信息失败,Err:" + err.Error()
  737. return
  738. }
  739. //如果是移动在两个兄弟节点之间
  740. if req.NextClassifyId > 0 {
  741. //下一个兄弟节点
  742. nextClassify, err := data_manage.GetChartClassifyById(req.NextClassifyId)
  743. if err != nil {
  744. br.Msg = "移动失败"
  745. br.ErrMsg = "获取下一个兄弟节点分类信息失败,Err:" + err.Error()
  746. return
  747. }
  748. //如果上一个兄弟与下一个兄弟的排序权重是一致的,那么需要将下一个兄弟(以及下个兄弟的同样排序权重)的排序权重+2,自己变成上一个兄弟的排序权重+1
  749. if prevClassify.Sort == nextClassify.Sort || prevClassify.Sort == chartClassifyInfo.Sort {
  750. //变更兄弟节点的排序
  751. updateSortStr := `sort + 2`
  752. _ = data_manage.UpdateChartClassifySortByParentId(prevClassify.ParentId, prevClassify.ChartClassifyId, prevClassify.Sort, updateSortStr)
  753. } else {
  754. //如果下一个兄弟的排序权重正好是上个兄弟节点的下一层,那么需要再加一层了
  755. if nextClassify.Sort-prevClassify.Sort == 1 {
  756. //变更兄弟节点的排序
  757. updateSortStr := `sort + 1`
  758. _ = data_manage.UpdateChartClassifySortByParentId(prevClassify.ParentId, 0, prevClassify.Sort, updateSortStr)
  759. }
  760. }
  761. }
  762. chartClassifyInfo.Sort = prevClassify.Sort + 1
  763. chartClassifyInfo.ModifyTime = time.Now()
  764. updateCol = append(updateCol, "Sort", "ModifyTime")
  765. } else {
  766. firstClassify, err := data_manage.GetFirstChartClassifyByParentId(chartClassifyInfo.ParentId)
  767. if err != nil && err.Error() != utils.ErrNoRow() {
  768. br.Msg = "移动失败"
  769. br.ErrMsg = "获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + err.Error()
  770. return
  771. }
  772. //如果该分类下存在其他分类,且第一个其他分类的排序等于0,那么需要调整排序
  773. if firstClassify != nil && firstClassify.Sort == 0 {
  774. updateSortStr := ` sort + 1 `
  775. _ = data_manage.UpdateChartClassifySortByParentId(firstClassify.ParentId, firstClassify.ChartClassifyId-1, 0, updateSortStr)
  776. }
  777. chartClassifyInfo.Sort = 0 //那就是排在第一位
  778. chartClassifyInfo.ModifyTime = time.Now()
  779. updateCol = append(updateCol, "Sort", "ModifyTime")
  780. }
  781. //更新
  782. if len(updateCol) > 0 {
  783. err = chartClassifyInfo.Update(updateCol)
  784. if err != nil {
  785. br.Msg = "移动失败"
  786. br.ErrMsg = "修改失败,Err:" + err.Error()
  787. return
  788. }
  789. }
  790. br.Ret = 200
  791. br.Success = true
  792. br.Msg = "移动成功"
  793. }
  794. // getChartClassifyListForMe 获取我创建的图表分类
  795. func getChartClassifyListForMeV2(adminInfo system.Admin, resp *data_manage.ChartClassifyListResp) (errMsg string, err error) {
  796. rootList, err := data_manage.GetChartClassifyByParentId(0, utils.CHART_SOURCE_DEFAULT)
  797. if err != nil && err.Error() != utils.ErrNoRow() {
  798. errMsg = "获取失败"
  799. return
  800. }
  801. classifyAll, err := data_manage.GetChartClassifyAll(utils.CHART_SOURCE_DEFAULT)
  802. if err != nil && err.Error() != utils.ErrNoRow() {
  803. errMsg = "获取失败"
  804. return
  805. }
  806. rootChildMap := make(map[int][]*data_manage.ChartClassifyItems)
  807. for _, v := range classifyAll {
  808. rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v)
  809. }
  810. nodeAll := make([]*data_manage.ChartClassifyItems, 0)
  811. for _, v := range rootList {
  812. if existItems, ok := rootChildMap[v.ChartClassifyId]; ok {
  813. v.Children = existItems
  814. } else {
  815. items := make([]*data_manage.ChartClassifyItems, 0)
  816. v.Children = items
  817. }
  818. nodeAll = append(nodeAll, v)
  819. }
  820. resp.AllNodes = nodeAll
  821. return
  822. }
  823. // ChartClassifyChartListV2
  824. // @Title 根据图表分类获取图表列表
  825. // @Description 根据图表分类获取图表列表接口
  826. // @Param IsShowMe query bool true "是否只看我的,true、false"
  827. // @Param ChartClassifyId query bool true "图片分类id"
  828. // @Success 200 {object} data_manage.ChartClassifyListResp
  829. // @router /chart_classify/chart/list [get]
  830. func (this *ChartClassifyController) ChartClassifyChartListV2() {
  831. br := new(models.BaseResponse).Init()
  832. defer func() {
  833. this.Data["json"] = br
  834. this.ServeJSON()
  835. }()
  836. sysUser := this.SysUser
  837. if sysUser == nil {
  838. br.Msg = "请登录"
  839. br.ErrMsg = "请登录,SysUser Is Empty"
  840. br.Ret = 408
  841. return
  842. }
  843. resp := new(data_manage.ChartClassifyListResp)
  844. chartClassifyId, _ := this.GetInt("ChartClassifyId")
  845. if chartClassifyId <= 0 {
  846. br.Msg = "参数错误"
  847. return
  848. }
  849. // 获取当前账号的不可见指标
  850. noPermissionChartIdMap := make(map[int]bool)
  851. {
  852. obj := data_manage.EdbInfoNoPermissionAdmin{}
  853. confList, err := obj.GetAllChartListByAdminId(this.SysUser.AdminId)
  854. if err != nil && err.Error() != utils.ErrNoRow() {
  855. br.Msg = "获取失败"
  856. br.ErrMsg = "获取不可见指标配置数据失败,Err:" + err.Error()
  857. return
  858. }
  859. for _, v := range confList {
  860. noPermissionChartIdMap[v.ChartInfoId] = true
  861. }
  862. }
  863. isShowMe, _ := this.GetBool("IsShowMe")
  864. if isShowMe {
  865. allChartInfo, err := data_manage.GetChartInfoByAdminIdAndClassify([]int{utils.CHART_SOURCE_DEFAULT}, sysUser.AdminId, chartClassifyId)
  866. if err != nil && err.Error() != utils.ErrNoRow() {
  867. br.Msg = "获取失败"
  868. br.ErrMsg = "获取全部数据失败,Err:" + err.Error()
  869. return
  870. }
  871. // 移除没有权限的图表
  872. allNodes := handleNoPermissionChart(allChartInfo, noPermissionChartIdMap)
  873. resp.AllNodes = allNodes
  874. br.Ret = 200
  875. br.Success = true
  876. br.Msg = "获取成功"
  877. br.Data = resp
  878. fmt.Println("source my classify")
  879. return
  880. }
  881. allChartInfo, err := data_manage.GetChartInfoAllByClassifyId(utils.CHART_SOURCE_DEFAULT, chartClassifyId)
  882. if err != nil && err.Error() != utils.ErrNoRow() {
  883. br.Msg = "获取失败"
  884. br.ErrMsg = "获取图表信息失败,Err:" + err.Error()
  885. return
  886. }
  887. // 移除没有权限的图表
  888. allNodes := handleNoPermissionChart(allChartInfo, noPermissionChartIdMap)
  889. resp.AllNodes = allNodes
  890. br.Ret = 200
  891. br.Success = true
  892. br.Msg = "获取成功"
  893. br.Data = resp
  894. }