chart_info.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670
  1. package chart
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "github.com/gin-gonic/gin"
  7. "hongze/hongze_yb/controller/response"
  8. "hongze/hongze_yb/global"
  9. "hongze/hongze_yb/models/request"
  10. responseModel "hongze/hongze_yb/models/response"
  11. "hongze/hongze_yb/models/response/chart_info"
  12. chartEdbMappingModel "hongze/hongze_yb/models/tables/chart_edb_mapping"
  13. chartInfoModel "hongze/hongze_yb/models/tables/chart_info"
  14. "hongze/hongze_yb/models/tables/chart_info_log"
  15. edbInfoModel "hongze/hongze_yb/models/tables/edb_info"
  16. "hongze/hongze_yb/models/tables/yb_my_chart"
  17. "hongze/hongze_yb/services"
  18. "hongze/hongze_yb/services/alarm_msg"
  19. "hongze/hongze_yb/services/chart"
  20. "hongze/hongze_yb/services/chart/cross_variety"
  21. "hongze/hongze_yb/services/user"
  22. "hongze/hongze_yb/utils"
  23. "sort"
  24. "strconv"
  25. "strings"
  26. "time"
  27. )
  28. // GetChartInfoDetail 获取图表详情
  29. // @Tags 图库模块
  30. // @Summary 获取图表详情
  31. // @Description 获取图表详情
  32. // @Security ApiKeyAuth
  33. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  34. // @Accept json
  35. // @Product json
  36. // @Param DateType query string false "时间段:1-00年至今; 2-10年至今; 3-15年至今; 4-21年至今; 5-指定区间; 6-指定年月至今; 7-18年至今; 8-19年至今; 9-20年至今"
  37. // @Param ClassifyId query string false "图表分类ID"
  38. // @Success 200 {object} chart_info.ChartInfoDetailResp
  39. // @failure 400 {string} string "图表详情获取失败"
  40. // @Router /my_chart/getChartInfoDetail [get]
  41. func GetChartInfoDetail(c *gin.Context) {
  42. // 图表ID
  43. reqChartInfoId := c.DefaultQuery("ChartInfoId", "")
  44. if reqChartInfoId == "" {
  45. response.Fail("参数有误:图表ID", c)
  46. return
  47. }
  48. chartInfoId, _ := strconv.Atoi(reqChartInfoId)
  49. reqStartDate := c.DefaultQuery("StartDate", "")
  50. reqEndDate := c.DefaultQuery("EndDate", "")
  51. // 图表样式类型
  52. reqChartType := c.DefaultQuery("ChartType", "")
  53. chartType, _ := strconv.Atoi(reqChartType)
  54. /*// 季节性图表时间
  55. reqSeasonStartDate := c.DefaultQuery("SeasonStartDate", "")
  56. reqSeasonEndDate := c.DefaultQuery("SeasonEndDate", "")*/
  57. // 指标ID
  58. edbInfoId := c.DefaultQuery("EdbInfoId", "")
  59. // 公历/农历
  60. reqCalendar := c.DefaultQuery("Calendar", "")
  61. // 获取图表信息
  62. var err error
  63. chartInfo := new(chartInfoModel.ChartInfoView)
  64. chartInfo, err = chartInfoModel.GetChartInfoViewById(chartInfoId)
  65. if err != nil {
  66. if err == utils.ErrNoRow {
  67. response.Custom(4003, "图表不存在,请刷新页面", c)
  68. return
  69. }
  70. response.FailMsg("获取失败", "获取图表信息失败, Err:"+err.Error(), c)
  71. return
  72. }
  73. var (
  74. edbList []*chartEdbMappingModel.ChartEdbInfoMappingList
  75. xEdbIdValue []int
  76. yDataList []chart_info.YData
  77. sourceArr []string
  78. dataResp interface{}
  79. )
  80. // todo 特殊处理平衡表
  81. if chartInfo.Source == utils.CHART_SOURCE_BALANCE_EXCEL {
  82. //调用接口
  83. chartData, e := services.GetBalanceChartDetail(chartInfo.UniqueCode)
  84. if e != nil {
  85. response.FailMsg("获取失败", "获取图表信息失败, Err:"+e.Error(), c)
  86. return
  87. }
  88. chartDataResp := new(chart_info.ChartInfoDetailResp)
  89. // 兼容返回值类型
  90. chartDataString, _ := json.Marshal(chartData)
  91. err = json.Unmarshal(chartDataString, chartDataResp)
  92. if err != nil {
  93. response.FailMsg("获取失败", "获取图表信息失败, Err:"+err.Error(), c)
  94. return
  95. }
  96. chartInfo = chartDataResp.ChartInfo
  97. yDataList = chartDataResp.YDataList
  98. edbList = chartDataResp.EdbInfoList
  99. xEdbIdValue = chartDataResp.XEdbIdValue
  100. sourceArr = append(sourceArr, "平衡表")
  101. dataResp = chartDataResp.DataResp
  102. } else {
  103. // 获取主题样式
  104. chartTheme, err := chart.GetChartThemeConfig(chartInfo.ChartThemeId, chartInfo.Source, chartInfo.ChartType)
  105. if err != nil {
  106. response.FailMsg("获取失败", "获取主题信息失败, Err:"+err.Error(), c)
  107. return
  108. }
  109. chartInfo.ChartThemeStyle = chartTheme.Config
  110. chartInfo.ChartThemeId = chartTheme.ChartThemeID
  111. chartType = chartInfo.ChartType
  112. startDate := chartInfo.StartDate
  113. endDate := chartInfo.EndDate
  114. startYear := chartInfo.StartYear
  115. calendar := chartInfo.Calendar
  116. if reqCalendar != "" {
  117. calendar = reqCalendar
  118. }
  119. // 时段筛选
  120. reqDateType := c.DefaultQuery("DateType", "")
  121. dateType := chartInfo.DateType
  122. if reqDateType != "" {
  123. dateType, _ = strconv.Atoi(reqDateType)
  124. }
  125. if dateType <= 0 {
  126. dateType = 3 // 默认同后台15年至今
  127. }
  128. if reqStartDate != "" {
  129. startDate = reqStartDate
  130. }
  131. if reqEndDate != "" {
  132. endDate = reqEndDate
  133. }
  134. // 获取图表指标映射
  135. mappingList := make([]*chartEdbMappingModel.ChartEdbInfoMapping, 0)
  136. if chartInfoId > 0 {
  137. mappingList, err = chartEdbMappingModel.GetMappingListByChartInfoId(chartInfoId)
  138. if err != nil {
  139. response.FailMsg("获取失败", "获取图表指标信息失败4001, Err:"+err.Error(), c)
  140. return
  141. }
  142. } else {
  143. if edbInfoId != "" {
  144. mappingList, err = chartEdbMappingModel.GetMappingListByEdbInfoId(edbInfoId)
  145. if err != nil {
  146. response.FailMsg("获取失败", "获取图表指标信息失败4002, Err:"+err.Error(), c)
  147. return
  148. }
  149. }
  150. }
  151. yearMax := 0
  152. if dateType == utils.DateTypeNYears {
  153. for _, v := range mappingList {
  154. if v.LatestDate.Year() > yearMax {
  155. yearMax = v.LatestDate.Year()
  156. }
  157. }
  158. }
  159. startDate, endDate = utils.GetDateByDateTypeV2(dateType, startDate, endDate, startYear, yearMax)
  160. // 图表额外数据参数
  161. extraConfigStr := chartInfo.ExtraConfig
  162. // 柱方图的一些配置
  163. var barConfig request.BarChartInfoReq
  164. if chartInfo != nil && chartInfo.ChartType == 7 {
  165. if chartInfo.BarConfig == `` {
  166. response.FailMsg("柱方图未配置", "柱方图未配置", c)
  167. return
  168. }
  169. err = json.Unmarshal([]byte(chartInfo.BarConfig), &barConfig)
  170. if err != nil {
  171. response.FailMsg("柱方图配置异常", "柱方图配置异常", c)
  172. return
  173. }
  174. extraConfigStr = chartInfo.BarConfig
  175. }
  176. var errMsg string
  177. // 获取图表中的指标数据
  178. edbList, xEdbIdValue, yDataList, sourceArr, dataResp, err, errMsg = chart.GetChartEdbData(chartInfoId, chartType, calendar, startDate, endDate, mappingList, extraConfigStr, chartInfo.SeasonExtraConfig)
  179. if err != nil {
  180. msg := `获取失败`
  181. if errMsg != `` {
  182. msg = errMsg
  183. }
  184. response.FailMsg(msg, "获取图表,指标信息失败, Err:"+err.Error(), c)
  185. return
  186. }
  187. for _, v := range edbList {
  188. // 指标别名
  189. if barConfig.EdbInfoIdList != nil && len(barConfig.EdbInfoIdList) > 0 {
  190. for _, reqEdb := range barConfig.EdbInfoIdList {
  191. if v.EdbInfoId == reqEdb.EdbInfoId {
  192. v.EdbAliasName = reqEdb.Name
  193. }
  194. }
  195. }
  196. }
  197. // 图表的指标来源
  198. sourceNameList, sourceNameEnList := chart.GetEdbSourceByEdbInfoIdList(edbList)
  199. chartInfo.ChartSource = strings.Join(sourceNameList, ",")
  200. chartInfo.ChartSourceEn = strings.Join(sourceNameEnList, ",")
  201. }
  202. sourceArr = append(sourceArr, "弘则研究")
  203. chartInfo.ChartSource = strings.Join(sourceArr, ",")
  204. // 访问记录-仅普通用户记录
  205. userInfo := user.GetInfoByClaims(c)
  206. ok, _, _ := user.GetAdminByUserInfo(userInfo)
  207. if !ok {
  208. reqMyChartClassifyId := c.DefaultQuery("MyChartClassifyId", "")
  209. myChartClassifyId, _ := strconv.Atoi(reqMyChartClassifyId)
  210. go chart.SaveChartVisitLog(userInfo, chartInfo, myChartClassifyId)
  211. }
  212. // 用户是否有收藏该图表
  213. ob := new(yb_my_chart.YbMyChart)
  214. cond := `user_id = ? AND chart_info_id = ?`
  215. pars := make([]interface{}, 0)
  216. pars = append(pars, userInfo.UserID, chartInfo.ChartInfoId)
  217. exists, e := ob.FetchByCondition(cond, pars)
  218. if e != nil && e != utils.ErrNoRow {
  219. response.FailMsg("操作失败", "获取用户图表失败, Err: "+e.Error(), c)
  220. return
  221. }
  222. myChartInfo := new(responseModel.MyChartItem)
  223. if exists != nil && exists.MyChartID > 0 {
  224. myChartInfo.MyChartID = exists.MyChartID
  225. myChartInfo.MyChartClassifyID = exists.MyChartClassifyID
  226. myChartInfo.ChartInfoID = exists.ChartInfoID
  227. myChartInfo.ChartName = exists.ChartName
  228. myChartInfo.UniqueCode = exists.UniqueCode
  229. myChartInfo.ChartImage = exists.ChartImage
  230. myChartInfo.UserID = exists.UserID
  231. myChartInfo.ReportID = exists.ReportID
  232. myChartInfo.ReportChapterID = exists.ReportChapterID
  233. myChartInfo.CreateTime = utils.TimeTransferString(utils.FormatDateTime, exists.CreateTime)
  234. }
  235. resp := new(chart_info.ChartInfoDetailResp)
  236. resp.ChartInfo = chartInfo
  237. resp.EdbInfoList = edbList
  238. resp.XEdbIdValue = xEdbIdValue
  239. resp.YDataList = yDataList
  240. resp.MyChartInfo = myChartInfo
  241. resp.DataResp = dataResp
  242. response.OkData("获取成功", resp, c)
  243. }
  244. func getChartInfoDetail(chartInfo *chartInfoModel.ChartInfoView, myChartClassifyId int, userInfo user.UserInfo) (resp *chart_info.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  245. // 获取图表信息
  246. var err error
  247. chartType := chartInfo.ChartType
  248. calendar := chartInfo.Calendar
  249. startDate := chartInfo.StartDate
  250. endDate := chartInfo.EndDate
  251. startYear := chartInfo.StartYear
  252. // 时段筛选
  253. dateType := chartInfo.DateType
  254. if dateType <= 0 {
  255. dateType = 3 // 默认同后台15年至今
  256. }
  257. // 获取图表指标映射
  258. mappingList := make([]*chartEdbMappingModel.ChartEdbInfoMapping, 0)
  259. mappingList, err = chartEdbMappingModel.GetMappingListByChartInfoId(chartInfo.ChartInfoId)
  260. if err != nil {
  261. msg = `获取失败`
  262. errMsg = "获取图表指标信息失败4001, Err:" + err.Error()
  263. return
  264. }
  265. yearMax := 0
  266. if dateType == utils.DateTypeNYears {
  267. for _, v := range mappingList {
  268. if v.LatestDate.Year() > yearMax {
  269. yearMax = v.LatestDate.Year()
  270. }
  271. }
  272. }
  273. startDate, endDate = utils.GetDateByDateTypeV2(dateType, startDate, endDate, startYear, yearMax)
  274. //fmt.Println("start_date:", startDate)
  275. //fmt.Println("end_date:", endDate)
  276. // 图表额外数据参数
  277. extraConfigStr := chartInfo.ExtraConfig
  278. // 柱方图的一些配置
  279. var barConfig request.BarChartInfoReq
  280. if chartInfo != nil && chartInfo.ChartType == 7 {
  281. if chartInfo.BarConfig == `` {
  282. msg = "柱方图未配置"
  283. errMsg = "柱方图未配置"
  284. return
  285. }
  286. err := json.Unmarshal([]byte(chartInfo.BarConfig), &barConfig)
  287. if err != nil {
  288. msg = "柱方图配置异常"
  289. errMsg = "柱方图配置异常"
  290. return
  291. }
  292. extraConfigStr = chartInfo.BarConfig
  293. }
  294. // 获取图表中的指标数据
  295. edbList, xEdbIdValue, yDataList, sourceArr, dataResp, err, tmpErrMsg := chart.GetChartEdbData(chartInfo.ChartInfoId, chartType, calendar, startDate, endDate, mappingList, extraConfigStr, chartInfo.SeasonExtraConfig)
  296. if err != nil {
  297. msg = `获取失败`
  298. if tmpErrMsg != `` {
  299. msg = tmpErrMsg
  300. }
  301. errMsg = "获取图表,指标信息失败, Err:" + err.Error()
  302. return
  303. }
  304. for _, v := range edbList {
  305. // 指标别名
  306. if barConfig.EdbInfoIdList != nil && len(barConfig.EdbInfoIdList) > 0 {
  307. for _, reqEdb := range barConfig.EdbInfoIdList {
  308. if v.EdbInfoId == reqEdb.EdbInfoId {
  309. v.EdbAliasName = reqEdb.Name
  310. }
  311. }
  312. }
  313. }
  314. sourceArr = append(sourceArr, "弘则研究")
  315. chartInfo.ChartSource = strings.Join(sourceArr, ",")
  316. // 访问记录-仅普通用户记录
  317. ok, _, _ := user.GetAdminByUserInfo(userInfo)
  318. if !ok {
  319. go chart.SaveChartVisitLog(userInfo, chartInfo, myChartClassifyId)
  320. }
  321. // 用户是否有收藏该图表
  322. ob := new(yb_my_chart.YbMyChart)
  323. cond := `user_id = ? AND chart_info_id = ?`
  324. pars := make([]interface{}, 0)
  325. pars = append(pars, userInfo.UserID, chartInfo.ChartInfoId)
  326. exists, e := ob.FetchByCondition(cond, pars)
  327. if e != nil && e != utils.ErrNoRow {
  328. msg = `操作失败`
  329. errMsg = "获取用户图表失败, Err: " + e.Error()
  330. return
  331. }
  332. myChartInfo := new(responseModel.MyChartItem)
  333. if exists != nil && exists.MyChartID > 0 {
  334. myChartInfo.MyChartID = exists.MyChartID
  335. myChartInfo.MyChartClassifyID = exists.MyChartClassifyID
  336. myChartInfo.ChartInfoID = exists.ChartInfoID
  337. myChartInfo.ChartName = exists.ChartName
  338. myChartInfo.UniqueCode = exists.UniqueCode
  339. myChartInfo.ChartImage = exists.ChartImage
  340. myChartInfo.UserID = exists.UserID
  341. myChartInfo.ReportID = exists.ReportID
  342. myChartInfo.ReportChapterID = exists.ReportChapterID
  343. myChartInfo.CreateTime = utils.TimeTransferString(utils.FormatDateTime, exists.CreateTime)
  344. }
  345. resp = new(chart_info.ChartInfoDetailResp)
  346. resp.ChartInfo = chartInfo
  347. resp.EdbInfoList = edbList
  348. resp.XEdbIdValue = xEdbIdValue
  349. resp.YDataList = yDataList
  350. resp.MyChartInfo = myChartInfo
  351. resp.DataResp = dataResp
  352. isOk = true
  353. return
  354. }
  355. // RefreshChartInfo 刷新图表信息
  356. // @Tags 图库模块
  357. // @Summary 刷新图表信息
  358. // @Description 刷新图表信息
  359. // @Security ApiKeyAuth
  360. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  361. // @Accept json
  362. // @Product json
  363. // @Param data body chartInfoModel.SaveChartInfoReq true "请求参数"
  364. // @Success 200 {string} string "操作成功"
  365. // @failure 400 {string} string "操作失败"
  366. // @Router /my_chart/refreshChartInfo [post]
  367. func RefreshChartInfo(c *gin.Context) {
  368. // 参数校验
  369. var req chartInfoModel.RefreshChartInfoReq
  370. if c.ShouldBind(&req) != nil {
  371. response.Fail("参数异常", c)
  372. return
  373. }
  374. chartInfoId := req.ChartInfoId
  375. if chartInfoId == 0 {
  376. response.Fail("参数有误", c)
  377. return
  378. }
  379. userInfo := user.GetInfoByClaims(c)
  380. ok, _, err := user.GetAdminByUserInfo(userInfo)
  381. if err != nil {
  382. response.FailMsg("刷新失败", "RefreshChartInfo-获取系统用户信息失败"+err.Error(), c)
  383. return
  384. }
  385. if !ok {
  386. // 普通用户刷新频率限制-每个用户/图/天/2次
  387. cacheKey := utils.HZ_CHART_LIB_DETAIL + "YB_REFRESH_LIMIT_" + strconv.Itoa(chartInfoId) + "_" + strconv.Itoa(int(userInfo.UserID))
  388. fmt.Println("refreshCacheKey:", cacheKey)
  389. countUserRefresh, _ := global.Redis.Get(context.TODO(), cacheKey).Int()
  390. if countUserRefresh >= 2 {
  391. response.Ok("目前已是最新数据", c)
  392. return
  393. }
  394. countUserRefresh += 1
  395. now := time.Now()
  396. today := time.Date(now.Year(), now.Month(), now.Day(), 23, 59, 59, 0, time.Local)
  397. sub := today.Sub(now)
  398. _ = global.Redis.SetEX(context.TODO(), cacheKey, countUserRefresh, sub)
  399. }
  400. // 图表信息校验
  401. chartInfo, err := chartInfoModel.GetChartInfoById(chartInfoId)
  402. if err != nil {
  403. if err == utils.ErrNoRow {
  404. response.Fail("图表已被删除,无需刷新", c)
  405. return
  406. }
  407. response.FailMsg("刷新失败", "刷新失败, Err:"+err.Error(), c)
  408. return
  409. }
  410. if chartInfo.Source == utils.CHART_SOURCE_CROSS_HEDGING {
  411. var config request.ChartConfigReq
  412. err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &config)
  413. if err != nil {
  414. response.FailMsg("解析跨品种分析配置失败", "解析跨品种分析配置失败, Err:"+err.Error(), c)
  415. return
  416. }
  417. // 获取关联的指标信息
  418. _, _, edbInfoIdList, err := cross_variety.GetXYEdbIdList(config.TagX, config.TagY, config.VarietyList)
  419. if err != nil {
  420. response.FailMsg("获取指标信息失败", "刷新失败,获取指标信息失败, Err:"+err.Error(), c)
  421. return
  422. }
  423. // 刷新图表
  424. //_ = correlation.ChartInfoRefresh(chartInfo.ChartInfoId)
  425. err, errMsg := chart.EdbInfoRefreshAllFromBase(edbInfoIdList, false)
  426. if err != nil {
  427. errContent := fmt.Sprint("刷新图表关联指标信息失败 errMsg:", errMsg, "err: "+err.Error())
  428. if global.CONFIG.Serve.RunMode == "release" {
  429. go alarm_msg.SendAlarmMsg("刷新图表报错"+time.Now().Format("2006-01-02 15:04:05")+";Err:"+errContent, 3)
  430. //go services.SendEmail("弘则研报小程序-release-刷新图表报错", errContent, utils.EmailSendToUsers)
  431. } else {
  432. global.LOG.Info(errContent)
  433. }
  434. response.FailMsg("刷新失败", "刷新失败,获取指标信息失败, Err:"+err.Error(), c)
  435. return
  436. }
  437. } else {
  438. // 刷新图表
  439. if err = chart.ChartInfoRefreshV2(chartInfo.ChartInfoId); err != nil {
  440. errContent := fmt.Sprint("ErrMsg: 刷新图表关联指标信息失败, " + err.Error())
  441. if global.CONFIG.Serve.RunMode == "release" {
  442. go alarm_msg.SendAlarmMsg("刷新图表报错"+time.Now().Format("2006-01-02 15:04:05")+";Err:"+errContent, 3)
  443. //go services.SendEmail("弘则研报小程序-release-刷新图表报错", errContent, utils.EmailSendToUsers)
  444. } else {
  445. global.LOG.Info(errContent)
  446. }
  447. }
  448. }
  449. //清除图表缓存
  450. {
  451. key := utils.HZ_CHART_LIB_DETAIL + chartInfo.UniqueCode
  452. _ = global.Redis.Del(context.TODO(), key)
  453. }
  454. response.OkData("刷新成功", "", c)
  455. }
  456. // EditChartInfo 编辑图表信息
  457. // @Tags 图库模块
  458. // @Summary 编辑图表信息
  459. // @Description 编辑图表信息
  460. // @Security ApiKeyAuth
  461. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  462. // @Accept json
  463. // @Product json
  464. // @Param data body chartInfoModel.SaveChartInfoReq true "请求参数"
  465. // @Success 200 {string} string "操作成功"
  466. // @failure 400 {string} string "操作失败"
  467. // @Router /my_chart/editChartInfo [post]
  468. func EditChartInfo(c *gin.Context) {
  469. // 参数校验
  470. var req chartInfoModel.SaveChartInfoReq
  471. if c.ShouldBind(&req) != nil {
  472. response.Fail("参数异常", c)
  473. return
  474. }
  475. if req.ChartInfoId <= 0 {
  476. response.Fail("参数异常", c)
  477. return
  478. }
  479. // 操作权限校验
  480. userInfo := user.GetInfoByClaims(c)
  481. ok, adminInfo, err := user.GetAdminByUserInfo(userInfo)
  482. if err != nil {
  483. response.Fail("操作人信息有误", c)
  484. return
  485. }
  486. if !ok {
  487. response.Fail("非内部人员无权进行操作", c)
  488. return
  489. }
  490. // 图表信息校验
  491. chartItem, err := chartInfoModel.GetChartInfoById(req.ChartInfoId)
  492. if err != nil {
  493. if err == utils.ErrNoRow {
  494. response.Fail("图表已被删除,请刷新页面!", c)
  495. return
  496. }
  497. response.FailMsg("操作失败", "获取图表信息失败, Err:"+err.Error(), c)
  498. return
  499. }
  500. // 图表关联指标id
  501. chartEdbInfoList := make([]*chartInfoModel.ChartSaveItem, 0)
  502. // 关联指标
  503. var edbInfoIdArr []int
  504. // 非散点截面图的直接通过前端获取,散点截面图需要额外从配置中获取
  505. if chartItem.ChartType != 10 {
  506. //edbList := make([]*data_manage.EdbInfo, 0)
  507. for _, v := range req.ChartEdbInfoList {
  508. edbInfoId := v.EdbInfoId
  509. edbInfo, err := edbInfoModel.GetEdbInfoById(edbInfoId)
  510. if err != nil {
  511. if err == utils.ErrNoRow {
  512. response.FailMsg("操作失败", "图表不存在,ChartInfoId:"+strconv.Itoa(v.EdbInfoId), c)
  513. return
  514. }
  515. response.FailMsg("操作失败", "获取图表的指标信息失败,Err:"+err.Error(), c)
  516. return
  517. }
  518. if edbInfo == nil {
  519. response.FailMsg("操作失败", "指标不存在,ChartInfoId:"+strconv.Itoa(v.EdbInfoId), c)
  520. return
  521. }
  522. if edbInfo.EdbInfoId <= 0 {
  523. response.FailMsg("指标已被删除,请重新选择!", "指标不存在,ChartInfoId:"+strconv.Itoa(v.EdbInfoId), c)
  524. return
  525. }
  526. edbInfoIdArr = append(edbInfoIdArr, edbInfoId)
  527. edbInfo.EdbNameSource = edbInfo.EdbName
  528. //edbList = append(edbList, edbInfo)
  529. }
  530. chartEdbInfoList = req.ChartEdbInfoList
  531. } else {
  532. if req.ExtraConfig == `` {
  533. response.FailMsg("请选择指标!", "请选择指标!, Err:"+err.Error(), c)
  534. return
  535. }
  536. // 校验配置的指标列表
  537. tmpEdbInfoIdArr, err, errMsg := chart.CheckChartExtraConfig(chartItem.ChartType, req.ExtraConfig)
  538. if err != nil {
  539. response.FailMsg(errMsg, "修改失败!, Err:"+err.Error(), c)
  540. return
  541. }
  542. edbInfoIdArr = tmpEdbInfoIdArr
  543. lenEdbInfoIdArr := len(edbInfoIdArr)
  544. if lenEdbInfoIdArr > 0 {
  545. tmpEdbList, err := edbInfoModel.GetEdbInfoByIdList(edbInfoIdArr)
  546. if err != nil {
  547. response.FailMsg("指标异常!", "指标异常!, Err:"+err.Error(), c)
  548. return
  549. }
  550. if len(tmpEdbList) != lenEdbInfoIdArr {
  551. response.FailMsg("指标异常!", fmt.Sprint("查找出来的指标数量与选择的指标数量不符!查找出来的指标数量:", len(tmpEdbList), ";用户选择的指标数量:", lenEdbInfoIdArr), c)
  552. return
  553. }
  554. for _, v := range tmpEdbList {
  555. chartEdbInfoList = append(chartEdbInfoList, &chartInfoModel.ChartSaveItem{
  556. EdbInfoId: v.EdbInfoId,
  557. MaxData: 0,
  558. MinData: 0,
  559. IsOrder: false,
  560. IsAxis: 1,
  561. EdbInfoType: 1,
  562. LeadValue: 0,
  563. LeadUnit: "",
  564. ChartStyle: "",
  565. ChartColor: "",
  566. //PredictChartColor: "",
  567. ChartWidth: 0,
  568. Source: utils.CHART_SOURCE_DEFAULT,
  569. })
  570. }
  571. }
  572. }
  573. if len(edbInfoIdArr) <= 0 {
  574. response.Fail("请选择指标!", c)
  575. return
  576. }
  577. if chartItem.ChartType == 2 && len(req.ChartEdbInfoList) > 1 {
  578. response.Fail("您选择的图表样式为季节性图表,只支持单指标画图!", c)
  579. return
  580. }
  581. if req.Calendar == "" {
  582. req.Calendar = "公历"
  583. }
  584. sort.Ints(edbInfoIdArr)
  585. var edbInfoIdArrStr []string
  586. for _, v := range edbInfoIdArr {
  587. edbInfoIdArrStr = append(edbInfoIdArrStr, strconv.Itoa(v))
  588. }
  589. edbInfoIdStr := strings.Join(edbInfoIdArrStr, ",")
  590. err = chart.ModifyChartInfoAndMapping(edbInfoIdStr, &req, chartItem.ChartType, chartEdbInfoList)
  591. if err != nil {
  592. response.Fail("图表保存失败, Err:"+err.Error(), c)
  593. return
  594. }
  595. // 清除图表缓存
  596. cacheKey := utils.HZ_CHART_LIB_DETAIL + chartItem.UniqueCode
  597. _ = global.Redis.Del(context.TODO(), cacheKey)
  598. // 新增操作日志
  599. {
  600. chartLog := new(chart_info_log.ChartInfoLog)
  601. chartLog.ChartName = chartItem.ChartName
  602. chartLog.ChartInfoId = req.ChartInfoId
  603. chartLog.ChartClassifyId = chartItem.ChartClassifyId
  604. chartLog.SysUserId = int(adminInfo.AdminID)
  605. chartLog.SysUserRealName = adminInfo.RealName
  606. chartLog.UniqueCode = chartItem.UniqueCode
  607. chartLog.CreateTime = time.Now()
  608. bodyBytes, _ := json.Marshal(req)
  609. chartLog.Content = string(bodyBytes)
  610. chartLog.Status = "修改配置项"
  611. chartLog.Method = c.Request.URL.String()
  612. go chartLog.Create()
  613. }
  614. response.OkData("操作成功", "", c)
  615. }