chart_common.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609
  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_correlation"
  15. "hongze/hongze_yb/models/tables/chart_info_log"
  16. "hongze/hongze_yb/models/tables/yb_my_chart"
  17. "hongze/hongze_yb/services/alarm_msg"
  18. "hongze/hongze_yb/services/chart"
  19. "hongze/hongze_yb/services/chart/correlation"
  20. future_goodServ "hongze/hongze_yb/services/chart/future_good"
  21. "hongze/hongze_yb/services/user"
  22. "hongze/hongze_yb/utils"
  23. "io/ioutil"
  24. "strconv"
  25. "strings"
  26. "time"
  27. )
  28. // CommonChartInfoDetailFromUniqueCode 获取图表详情(通用)
  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/common/detail [get]
  41. func CommonChartInfoDetailFromUniqueCode(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. reqMyChartClassifyId := c.DefaultQuery("MyChartClassifyId", "")
  50. myChartClassifyId, _ := strconv.Atoi(reqMyChartClassifyId)
  51. chartInfo := new(chartInfoModel.ChartInfoView)
  52. chartInfo, err := chartInfoModel.GetChartInfoViewById(chartInfoId)
  53. if err != nil {
  54. if err == utils.ErrNoRow {
  55. response.Custom(4003, "图表不存在,请刷新页面", c)
  56. return
  57. }
  58. response.FailMsg(`获取失败`, "获取图表信息失败, Err:"+err.Error(), c)
  59. return
  60. }
  61. switch chartInfo.Source {
  62. case utils.CHART_SOURCE_DEFAULT:
  63. resp, isOk, msg, errMsg := getChartInfoDetail(chartInfo, myChartClassifyId, user.GetInfoByClaims(c))
  64. if !isOk {
  65. response.FailMsg(msg, errMsg, c)
  66. return
  67. }
  68. response.OkData("获取成功", resp, c)
  69. return
  70. case utils.CHART_SOURCE_FUTURE_GOOD:
  71. resp, isOk, msg, errMsg := getFutureGoodChartInfoDetail(chartInfo, myChartClassifyId, user.GetInfoByClaims(c))
  72. if !isOk {
  73. response.FailMsg(msg, errMsg, c)
  74. return
  75. }
  76. response.OkData("获取成功", resp, c)
  77. return
  78. case utils.CHART_SOURCE_CORRELATION, utils.CHART_SOURCE_ROLLING_CORRELATION:
  79. resp, isOk, msg, errMsg := getCorrelationChartInfoDetail(chartInfo, myChartClassifyId, user.GetInfoByClaims(c))
  80. if !isOk {
  81. response.FailMsg(msg, errMsg, c)
  82. return
  83. }
  84. response.OkData("获取成功", resp, c)
  85. return
  86. default:
  87. msg := "错误的图表"
  88. errMsg := "错误的图表"
  89. response.FailMsg(msg, errMsg, c)
  90. return
  91. }
  92. }
  93. // getFutureGoodChartInfoDetail 获取商品价格曲线图表详情
  94. func getFutureGoodChartInfoDetail(chartInfo *chartInfoModel.ChartInfoView, myChartClassifyId int, userInfo user.UserInfo) (resp *chart_info.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  95. resp = new(chart_info.ChartInfoDetailResp)
  96. // 获取图表信息
  97. var err error
  98. chartInfoId := chartInfo.ChartInfoId
  99. startDate := chartInfo.StartDate
  100. endDate := chartInfo.EndDate
  101. // 兼容日期错误
  102. {
  103. if strings.Count(startDate, "-") == 1 {
  104. startDate = startDate + "-01"
  105. }
  106. if strings.Count(endDate, "-") == 1 {
  107. endDate = endDate + "-01"
  108. }
  109. }
  110. edbInfoMapping, err := chartEdbMappingModel.GetEtaEdbChartEdbMapping(chartInfoId)
  111. if err != nil {
  112. msg = "获取失败"
  113. errMsg = "获取图表,现货指标信息失败,Err:" + err.Error()
  114. return
  115. }
  116. futureGoodEdbInfoMapping, err := chartEdbMappingModel.GetFutureGoodEdbChartEdbMapping(chartInfoId)
  117. if err != nil {
  118. msg = "获取失败"
  119. errMsg = "获取图表的期货商品指标信息失败,Err:" + err.Error()
  120. return
  121. }
  122. // 商品价格曲线图的一些配置
  123. var barConfig request.BarChartInfoReq
  124. barChartInfoDateList := make([]request.BarChartInfoDateReq, 0)
  125. //barChartInfoSort := request.BarChartInfoSortReq{}
  126. if chartInfo.BarConfig == `` {
  127. msg = "商品价格曲线图未配置"
  128. errMsg = "商品价格曲线图未配置"
  129. return
  130. }
  131. err = json.Unmarshal([]byte(chartInfo.BarConfig), &barConfig)
  132. if err != nil {
  133. msg = "商品价格曲线图配置异常"
  134. errMsg = "商品价格曲线图配置异常"
  135. return
  136. }
  137. barChartInfoDateList = barConfig.DateList
  138. //barChartInfoSort = barConfig.Sort
  139. // 获取图表中的指标数据
  140. barConfigEdbInfoIdList, edbList, xEdbIdValue, xDataList, yDataList, sourceArr, err := future_goodServ.GetChartEdbData(chartInfoId, startDate, endDate, edbInfoMapping, futureGoodEdbInfoMapping, barChartInfoDateList)
  141. if err != nil {
  142. msg = "获取失败"
  143. errMsg = "获取图表,指标信息失败,Err:" + err.Error()
  144. return
  145. }
  146. if len(edbList) <= 0 {
  147. msg = "商品价格曲线图表指标异常"
  148. errMsg = "商品价格曲线图表异常"
  149. return
  150. }
  151. sourceArr = append(sourceArr, "弘则研究")
  152. chartInfo.ChartSource = strings.Join(sourceArr, ",")
  153. for _, v := range edbList {
  154. // 指标别名
  155. if barConfigEdbInfoIdList != nil && len(barConfigEdbInfoIdList) > 0 {
  156. for _, reqEdb := range barConfigEdbInfoIdList {
  157. if v.EdbInfoId == reqEdb.EdbInfoId {
  158. v.EdbAliasName = reqEdb.Name
  159. v.EdbAliasNameEn = reqEdb.NameEn
  160. }
  161. }
  162. }
  163. }
  164. baseEdbInfo := edbList[0] //现货指标
  165. chartInfo.UnitEn = baseEdbInfo.UnitEn
  166. // 访问记录-仅普通用户记录
  167. ok, _, _ := user.GetAdminByUserInfo(userInfo)
  168. if !ok {
  169. go chart.SaveChartVisitLog(userInfo, chartInfo, myChartClassifyId)
  170. }
  171. // 用户是否有收藏该图表
  172. {
  173. ob := new(yb_my_chart.YbMyChart)
  174. cond := `user_id = ? AND chart_info_id = ?`
  175. pars := make([]interface{}, 0)
  176. pars = append(pars, userInfo.UserID, chartInfo.ChartInfoId)
  177. exists, e := ob.FetchByCondition(cond, pars)
  178. if e != nil && e != utils.ErrNoRow {
  179. msg = `操作失败`
  180. errMsg = "获取用户图表失败, Err: " + e.Error()
  181. return
  182. }
  183. myChartInfo := new(responseModel.MyChartItem)
  184. if exists != nil && exists.MyChartID > 0 {
  185. myChartInfo.MyChartID = exists.MyChartID
  186. myChartInfo.MyChartClassifyID = exists.MyChartClassifyID
  187. myChartInfo.ChartInfoID = exists.ChartInfoID
  188. myChartInfo.ChartName = exists.ChartName
  189. myChartInfo.UniqueCode = exists.UniqueCode
  190. myChartInfo.ChartImage = exists.ChartImage
  191. myChartInfo.UserID = exists.UserID
  192. myChartInfo.ReportID = exists.ReportID
  193. myChartInfo.ReportChapterID = exists.ReportChapterID
  194. myChartInfo.CreateTime = utils.TimeTransferString(utils.FormatDateTime, exists.CreateTime)
  195. }
  196. resp.MyChartInfo = myChartInfo
  197. }
  198. resp.ChartInfo = chartInfo
  199. resp.EdbInfoList = edbList
  200. resp.XEdbIdValue = xEdbIdValue
  201. resp.XDataList = xDataList
  202. resp.YDataList = yDataList
  203. isOk = true
  204. return
  205. }
  206. // RefreshFutureGoodChartInfo 刷新商品价格曲线图表信息
  207. // @Tags 图库模块
  208. // @Summary 刷新图表信息
  209. // @Description 刷新图表信息
  210. // @Security ApiKeyAuth
  211. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  212. // @Accept json
  213. // @Product json
  214. // @Param data body chartInfoModel.SaveChartInfoReq true "请求参数"
  215. // @Success 200 {string} string "操作成功"
  216. // @failure 400 {string} string "操作失败"
  217. // @Router /my_chart/future_good/refreshChartInfo [post]
  218. func RefreshFutureGoodChartInfo(c *gin.Context) {
  219. // 参数校验
  220. var req chartInfoModel.RefreshChartInfoReq
  221. if c.ShouldBind(&req) != nil {
  222. response.Fail("参数异常", c)
  223. return
  224. }
  225. chartInfoId := req.ChartInfoId
  226. if chartInfoId == 0 {
  227. response.Fail("参数有误", c)
  228. return
  229. }
  230. //userInfo := user.GetInfoByClaims(c)
  231. //ok, _, err := user.GetAdminByUserInfo(userInfo)
  232. //if err != nil {
  233. // response.FailMsg("刷新失败", "RefreshChartInfo-获取系统用户信息失败"+err.Error(), c)
  234. // return
  235. //}
  236. //if !ok {
  237. // // 普通用户刷新频率限制-每个用户/图/天/2次
  238. // cacheKey := utils.HZ_CHART_LIB_DETAIL + "YB_REFRESH_LIMIT_" + strconv.Itoa(chartInfoId) + "_" + strconv.Itoa(int(userInfo.UserID))
  239. // fmt.Println("refreshCacheKey:", cacheKey)
  240. // countUserRefresh, _ := global.Redis.Get(context.TODO(), cacheKey).Int()
  241. // if countUserRefresh >= 2 {
  242. // response.Ok("目前已是最新数据", c)
  243. // return
  244. // }
  245. // countUserRefresh += 1
  246. // now := time.Now()
  247. // today := time.Date(now.Year(), now.Month(), now.Day(), 23, 59, 59, 0, time.Local)
  248. // sub := today.Sub(now)
  249. // _ = global.Redis.SetEX(context.TODO(), cacheKey, countUserRefresh, sub)
  250. //}
  251. // 图表信息校验
  252. chartInfo, err := chartInfoModel.GetChartInfoById(chartInfoId)
  253. if err != nil {
  254. if err == utils.ErrNoRow {
  255. response.Fail("图表已被删除,无需刷新", c)
  256. return
  257. }
  258. response.FailMsg("刷新失败", "刷新失败, Err:"+err.Error(), c)
  259. return
  260. }
  261. // 刷新图表
  262. if err = future_goodServ.FutureGoodChartInfoRefresh(chartInfo.ChartInfoId); err != nil {
  263. errContent := fmt.Sprint("ErrMsg: 刷新图表关联指标信息失败, " + err.Error())
  264. if global.CONFIG.Serve.RunMode == "release" {
  265. go alarm_msg.SendAlarmMsg("刷新图表报错"+time.Now().Format("2006-01-02 15:04:05")+";Err:"+errContent, 3)
  266. //go services.SendEmail("弘则研报小程序-release-刷新图表报错", errContent, utils.EmailSendToUsers)
  267. } else {
  268. global.LOG.Info(errContent)
  269. }
  270. }
  271. //清除图表缓存
  272. {
  273. key := utils.HZ_CHART_LIB_DETAIL + chartInfo.UniqueCode
  274. _ = global.Redis.Del(context.TODO(), key)
  275. }
  276. response.OkData("刷新成功", "", c)
  277. }
  278. // FutureGoodChartInfoSave 编辑商品价格图表信息
  279. // @Tags 图库模块
  280. // @Summary 编辑图表信息
  281. // @Description 编辑图表信息
  282. // @Security ApiKeyAuth
  283. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  284. // @Accept json
  285. // @Product json
  286. // @Param data body chartInfoModel.SaveChartInfoReq true "请求参数"
  287. // @Success 200 {string} string "操作成功"
  288. // @failure 400 {string} string "操作失败"
  289. // @Router /my_chart/editChartInfo [post]
  290. func FutureGoodChartInfoSave(c *gin.Context) {
  291. // 参数校验
  292. var req chartInfoModel.SaveChartInfoReq
  293. if c.ShouldBind(&req) != nil {
  294. response.Fail("参数异常", c)
  295. return
  296. }
  297. if req.ChartInfoId <= 0 {
  298. response.Fail("参数异常", c)
  299. return
  300. }
  301. // 操作权限校验
  302. userInfo := user.GetInfoByClaims(c)
  303. ok, adminInfo, err := user.GetAdminByUserInfo(userInfo)
  304. if err != nil {
  305. response.Fail("操作人信息有误", c)
  306. return
  307. }
  308. if !ok {
  309. response.Fail("非内部人员无权进行操作", c)
  310. return
  311. }
  312. // 图表信息校验
  313. chartItem, err := chartInfoModel.GetChartInfoById(req.ChartInfoId)
  314. if err != nil {
  315. if err == utils.ErrNoRow {
  316. response.Fail("图表已被删除,请刷新页面!", c)
  317. return
  318. }
  319. response.FailMsg("操作失败", "获取图表信息失败, Err:"+err.Error(), c)
  320. return
  321. }
  322. if chartItem.Source != 2 {
  323. response.Fail("该图不是商品价格曲线图", c)
  324. return
  325. }
  326. chartItem.LeftMin = req.LeftMin
  327. chartItem.LeftMax = req.LeftMax
  328. err = chartItem.Update([]string{"LeftMin", "LeftMax"})
  329. if err != nil {
  330. response.FailMsg("保存失败", "保存失败,Err:"+err.Error(), c)
  331. return
  332. }
  333. // 清除图表缓存
  334. cacheKey := utils.HZ_CHART_LIB_DETAIL + chartItem.UniqueCode
  335. _ = global.Redis.Del(context.TODO(), cacheKey)
  336. // 新增操作日志
  337. {
  338. chartLog := new(chart_info_log.ChartInfoLog)
  339. chartLog.ChartName = chartItem.ChartName
  340. chartLog.ChartInfoId = req.ChartInfoId
  341. chartLog.ChartClassifyId = chartItem.ChartClassifyId
  342. chartLog.SysUserId = int(adminInfo.AdminID)
  343. chartLog.SysUserRealName = adminInfo.RealName
  344. chartLog.UniqueCode = chartItem.UniqueCode
  345. chartLog.CreateTime = time.Now()
  346. bodyBytes, _ := ioutil.ReadAll(c.Request.Body)
  347. chartLog.Content = string(bodyBytes)
  348. chartLog.Status = "修改配置项"
  349. chartLog.Method = c.Request.URL.String()
  350. go chartLog.Create()
  351. }
  352. response.OkData("操作成功", "", c)
  353. }
  354. // getCorrelationChartInfoDetail 获取相关性图表详情
  355. func getCorrelationChartInfoDetail(chartInfo *chartInfoModel.ChartInfoView, myChartClassifyId int, userInfo user.UserInfo) (resp *chart_info.ChartInfoDetailResp, isOk bool, msg, errMsg string) {
  356. resp = new(chart_info.ChartInfoDetailResp)
  357. // 获取图表信息
  358. //var err error
  359. chartInfoId := chartInfo.ChartInfoId
  360. chartInfo.ChartSource = "弘则研究"
  361. startDate := chartInfo.StartDate
  362. endDate := chartInfo.EndDate
  363. // 兼容日期错误
  364. {
  365. if strings.Count(startDate, "-") == 1 {
  366. startDate = startDate + "-01"
  367. }
  368. if strings.Count(endDate, "-") == 1 {
  369. endDate = endDate + "-01"
  370. }
  371. }
  372. // 相关性图表信息
  373. correlationChart := new(chart_info_correlation.ChartInfoCorrelation)
  374. if e := correlationChart.GetItemById(chartInfoId); e != nil {
  375. msg = "获取失败"
  376. errMsg = "获取图表相关性信息失败, Err:" + e.Error()
  377. return
  378. }
  379. // 获取指标信息
  380. edbInfoMappingA, e := chartEdbMappingModel.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdFirst)
  381. if e != nil {
  382. msg = "获取失败"
  383. errMsg = "获取相关性图表, A指标mapping信息失败, Err:" + e.Error()
  384. return
  385. }
  386. edbInfoMappingB, e := chartEdbMappingModel.GetChartEdbMappingByEdbInfoId(correlationChart.EdbInfoIdSecond)
  387. if e != nil {
  388. msg = "获取失败"
  389. errMsg = "获取相关性图表, B指标mapping信息失败, Err:" + e.Error()
  390. return
  391. }
  392. var dataResp interface{} // 绘图数据返回(目前是滚动相关性的图)
  393. var xEdbIdValue []int
  394. var xDateTimeValue []string
  395. var yDataList []chart_info.YData
  396. switch chartInfo.Source {
  397. case utils.CHART_SOURCE_CORRELATION: // 相关性图
  398. moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.CalculateUnit]
  399. if !ok {
  400. msg = "错误的分析周期"
  401. errMsg = "相关性图表数据有误"
  402. return
  403. }
  404. startDate := time.Now().AddDate(0, 0, -correlationChart.CalculateValue*moveUnitDays).Format(utils.FormatDate)
  405. endDate := time.Now().Format(utils.FormatDate)
  406. xEdbIdValue, yDataList, e = correlation.GetChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, startDate, endDate)
  407. if e != nil {
  408. msg = "获取失败"
  409. errMsg = "获取相关性图表, 图表计算值失败, Err:" + e.Error()
  410. return
  411. }
  412. case utils.CHART_SOURCE_ROLLING_CORRELATION: // 滚动相关性图
  413. moveUnitDays, ok := utils.FrequencyDaysMap[correlationChart.BaseCalculateUnit]
  414. if !ok {
  415. msg = "错误的分析周期"
  416. errMsg = "错误的分析周期"
  417. return
  418. }
  419. startDate := time.Now().AddDate(0, 0, -correlationChart.BaseCalculateValue*moveUnitDays).Format(utils.FormatDate)
  420. endDate := time.Now().Format(utils.FormatDate)
  421. xDateTimeValue, yDataList, e = correlation.GetRollingCorrelationChartDataByEdbInfo(edbInfoMappingA, edbInfoMappingB, correlationChart.LeadValue, correlationChart.LeadUnit, correlationChart.CalculateValue, correlationChart.CalculateUnit, startDate, endDate)
  422. dataResp = chart_info.RollingCorrelationChartDataResp{
  423. XDateTimeValue: xDateTimeValue,
  424. YDataList: yDataList,
  425. }
  426. }
  427. // 完善指标信息
  428. edbList, e := correlation.GetChartEdbInfoFormat(chartInfo.ChartInfoId, edbInfoMappingA, edbInfoMappingB)
  429. if e != nil {
  430. msg = "获取失败"
  431. errMsg = "获取相关性图表, 完善指标信息失败, Err:" + e.Error()
  432. return
  433. }
  434. baseEdbInfo := edbList[0] //现货指标
  435. chartInfo.UnitEn = baseEdbInfo.UnitEn
  436. correlationInfo := new(chart_info.CorrelationInfo)
  437. correlationInfo.LeadValue = correlationChart.LeadValue
  438. correlationInfo.LeadUnit = correlationChart.LeadUnit
  439. correlationInfo.StartDate = correlationChart.StartDate.Format(utils.FormatDate)
  440. correlationInfo.EndDate = correlationChart.EndDate.Format(utils.FormatDate)
  441. correlationInfo.LeadValue = correlationChart.LeadValue
  442. correlationInfo.EdbInfoIdFirst = correlationChart.EdbInfoIdFirst
  443. correlationInfo.EdbInfoIdSecond = correlationChart.EdbInfoIdSecond
  444. // 访问记录-仅普通用户记录
  445. ok, _, _ := user.GetAdminByUserInfo(userInfo)
  446. if !ok {
  447. go chart.SaveChartVisitLog(userInfo, chartInfo, myChartClassifyId)
  448. }
  449. // 用户是否有收藏该图表
  450. {
  451. ob := new(yb_my_chart.YbMyChart)
  452. cond := `user_id = ? AND chart_info_id = ?`
  453. pars := make([]interface{}, 0)
  454. pars = append(pars, userInfo.UserID, chartInfo.ChartInfoId)
  455. exists, e := ob.FetchByCondition(cond, pars)
  456. if e != nil && e != utils.ErrNoRow {
  457. msg = `操作失败`
  458. errMsg = "获取用户图表失败, Err: " + e.Error()
  459. return
  460. }
  461. myChartInfo := new(responseModel.MyChartItem)
  462. if exists != nil && exists.MyChartID > 0 {
  463. myChartInfo.MyChartID = exists.MyChartID
  464. myChartInfo.MyChartClassifyID = exists.MyChartClassifyID
  465. myChartInfo.ChartInfoID = exists.ChartInfoID
  466. myChartInfo.ChartName = exists.ChartName
  467. myChartInfo.UniqueCode = exists.UniqueCode
  468. myChartInfo.ChartImage = exists.ChartImage
  469. myChartInfo.UserID = exists.UserID
  470. myChartInfo.ReportID = exists.ReportID
  471. myChartInfo.ReportChapterID = exists.ReportChapterID
  472. myChartInfo.CreateTime = utils.TimeTransferString(utils.FormatDateTime, exists.CreateTime)
  473. }
  474. resp.MyChartInfo = myChartInfo
  475. }
  476. resp.ChartInfo = chartInfo
  477. resp.EdbInfoList = edbList
  478. resp.XEdbIdValue = xEdbIdValue
  479. resp.YDataList = yDataList
  480. resp.CorrelationChartInfo = correlationInfo
  481. resp.DataResp = dataResp
  482. isOk = true
  483. return
  484. }
  485. // RefreshCorrelationChartInfo 刷新相关性图表信息
  486. // @Tags 图库模块
  487. // @Summary 刷新图表信息
  488. // @Description 刷新图表信息
  489. // @Security ApiKeyAuth
  490. // @Param Authorization header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
  491. // @Accept json
  492. // @Product json
  493. // @Param data body chartInfoModel.SaveChartInfoReq true "请求参数"
  494. // @Success 200 {string} string "操作成功"
  495. // @failure 400 {string} string "操作失败"
  496. // @Router /my_chart/correlation/refreshChartInfo [post]
  497. func RefreshCorrelationChartInfo(c *gin.Context) {
  498. // 参数校验
  499. var req chartInfoModel.RefreshChartInfoReq
  500. if c.ShouldBind(&req) != nil {
  501. response.Fail("参数异常", c)
  502. return
  503. }
  504. chartInfoId := req.ChartInfoId
  505. if chartInfoId == 0 {
  506. response.Fail("参数有误", c)
  507. return
  508. }
  509. userInfo := user.GetInfoByClaims(c)
  510. ok, _, err := user.GetAdminByUserInfo(userInfo)
  511. if err != nil {
  512. response.FailMsg("刷新失败", "RefreshChartInfo-获取系统用户信息失败"+err.Error(), c)
  513. return
  514. }
  515. if !ok {
  516. // 普通用户刷新频率限制-每个用户/图/天/2次
  517. cacheKey := utils.HZ_CHART_LIB_DETAIL + "YB_REFRESH_LIMIT_" + strconv.Itoa(chartInfoId) + "_" + strconv.Itoa(int(userInfo.UserID))
  518. countUserRefresh, _ := global.Redis.Get(context.TODO(), cacheKey).Int()
  519. if countUserRefresh >= 2 {
  520. response.Ok("目前已是最新数据", c)
  521. return
  522. }
  523. countUserRefresh += 1
  524. now := time.Now()
  525. today := time.Date(now.Year(), now.Month(), now.Day(), 23, 59, 59, 0, time.Local)
  526. sub := today.Sub(now)
  527. _ = global.Redis.SetEX(context.TODO(), cacheKey, countUserRefresh, sub)
  528. }
  529. // 图表信息校验
  530. chartInfo, err := chartInfoModel.GetChartInfoById(chartInfoId)
  531. if err != nil {
  532. if err == utils.ErrNoRow {
  533. response.Fail("图表已被删除,无需刷新", c)
  534. return
  535. }
  536. response.FailMsg("刷新失败", "刷新失败, Err:"+err.Error(), c)
  537. return
  538. }
  539. // 刷新图表
  540. //_ = correlation.ChartInfoRefresh(chartInfo.ChartInfoId)
  541. if err = chart.ChartInfoRefreshV2(chartInfo.ChartInfoId); err != nil {
  542. errContent := fmt.Sprint("ErrMsg: 刷新图表关联指标信息失败, " + err.Error())
  543. if global.CONFIG.Serve.RunMode == "release" {
  544. go alarm_msg.SendAlarmMsg("刷新图表报错"+time.Now().Format("2006-01-02 15:04:05")+";Err:"+errContent, 3)
  545. //go services.SendEmail("弘则研报小程序-release-刷新图表报错", errContent, utils.EmailSendToUsers)
  546. } else {
  547. global.LOG.Info(errContent)
  548. }
  549. }
  550. //清除图表缓存
  551. {
  552. key := utils.HZ_CHART_LIB_DETAIL + chartInfo.UniqueCode
  553. _ = global.Redis.Del(context.TODO(), key)
  554. }
  555. response.OkData("刷新成功", "", c)
  556. }