chart_common.go 21 KB

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