chart.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975
  1. package cross_variety
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_api/models/data_manage"
  6. cross_varietyModel "eta/eta_api/models/data_manage/cross_variety"
  7. "eta/eta_api/models/data_manage/cross_variety/request"
  8. "eta/eta_api/models/system"
  9. "eta/eta_api/services/data"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "github.com/shopspring/decimal"
  13. "strconv"
  14. "strings"
  15. "time"
  16. )
  17. // ChartInfoResp 截面散点图数据
  18. type ChartInfoResp struct {
  19. XName string `description:"x轴名称"`
  20. XNameEn string `description:"x轴名称(英文)"`
  21. XUnitName string `description:"x轴单位名称"`
  22. XUnitNameEn string `description:"x轴单位名称(英文)"`
  23. YName string `description:"y轴名称"`
  24. YNameEn string `description:"y轴名称(英文)"`
  25. YUnitName string `description:"y轴单位名称"`
  26. YUnitNameEn string `description:"y轴单位名称(英文)"`
  27. XMinValue string `description:"X轴的最小值"`
  28. XMaxValue string `description:"X轴的最大值"`
  29. YMinValue string `description:"Y轴的最小值"`
  30. YMaxValue string `description:"Y轴的最大值"`
  31. DataList []SectionScatterSeriesItemResp `description:"数据列"`
  32. }
  33. // SectionScatterSeriesItemResp 系列的返回
  34. type SectionScatterSeriesItemResp struct {
  35. Name string `description:"系列名"`
  36. NameEn string `description:"系列名(英文)"`
  37. Color string `description:"颜色"`
  38. CoordinatePointData []CoordinatePoint `description:"趋势线的前后坐标点"`
  39. }
  40. // SectionScatterEdbItemResp 截面散点的返回参数
  41. type SectionScatterEdbItemResp struct {
  42. XEdbInfoId int `description:"X轴指标id"`
  43. XDate string `description:"X轴指标实际日期"`
  44. XName string `description:"X轴指标名称"`
  45. XNameEn string `description:"X轴指标英文名称"`
  46. XValue float64 `description:"X轴实际值"`
  47. YEdbInfoId int `description:"Y轴指标id"`
  48. YDate string `description:"Y轴指标实际日期"`
  49. YName string `description:"Y轴指标名称"`
  50. YNameEn string `description:"Y轴指标英文名称"`
  51. YValue float64 `description:"Y轴实际值"`
  52. IsShow bool `description:"是否展示"`
  53. Name string `description:"标签名称"`
  54. NameEn string `description:"英文标签名称"`
  55. }
  56. // CoordinatePoint 坐标点
  57. type CoordinatePoint struct {
  58. X float64
  59. Y float64
  60. XEdbInfoId int
  61. YEdbInfoId int
  62. XDate string
  63. YDate string
  64. }
  65. // GetChartData
  66. // @Description: 获取跨品种分析图表数据
  67. // @author: Roc
  68. // @datetime 2023-11-24 09:42:59
  69. // @param chartInfoId int
  70. // @param config request.ChartConfigReq
  71. // @return edbList []*data_manage.ChartEdbInfoMapping
  72. // @return dataResp ChartInfoResp
  73. // @return err error
  74. // @return errMsg string
  75. // @return isSendEmail bool
  76. func GetChartData(chartInfoId int, config request.ChartConfigReq) (edbList []*data_manage.ChartEdbInfoMapping, dataResp ChartInfoResp, err error, errMsg string, isSendEmail bool) {
  77. moveUnitDays, ok := utils.FrequencyDaysMap[config.CalculateUnit]
  78. if !ok {
  79. errMsg = "错误的分析周期"
  80. err = errors.New(errMsg)
  81. isSendEmail = false
  82. return
  83. }
  84. isSendEmail = true
  85. // 品种map
  86. varietyMap := make(map[int]*cross_varietyModel.ChartVariety)
  87. {
  88. varietyList, tmpErr := cross_varietyModel.GetVarietyListByIdList(config.VarietyList)
  89. if tmpErr != nil {
  90. err = tmpErr
  91. return
  92. }
  93. for _, v := range varietyList {
  94. varietyMap[v.ChartVarietyId] = v
  95. }
  96. }
  97. // 标签m
  98. var xTagInfo, yTagInfo *cross_varietyModel.ChartTag
  99. {
  100. tagList, tmpErr := cross_varietyModel.GetTagListByIdList([]int{config.TagX, config.TagY})
  101. if tmpErr != nil {
  102. err = tmpErr
  103. return
  104. }
  105. for _, v := range tagList {
  106. if v.ChartTagId == config.TagX {
  107. xTagInfo = v
  108. } else if v.ChartTagId == config.TagY {
  109. yTagInfo = v
  110. }
  111. }
  112. }
  113. if xTagInfo == nil {
  114. errMsg = "找不到对应的X轴标签"
  115. err = errors.New(errMsg)
  116. return
  117. }
  118. if yTagInfo == nil {
  119. errMsg = "找不到对应的Y轴标签"
  120. err = errors.New(errMsg)
  121. return
  122. }
  123. xVarietyEdbMap, yVarietyEdbMap, edbInfoIdList, err := GetXYEdbIdList(config.TagX, config.TagY, config.VarietyList)
  124. if err != nil {
  125. return
  126. }
  127. if len(edbInfoIdList) <= 0 {
  128. errMsg = "品种未配置指标"
  129. err = errors.New(errMsg)
  130. isSendEmail = false
  131. return
  132. }
  133. mappingList, err := data_manage.GetChartEdbMappingListByEdbInfoIdList(edbInfoIdList)
  134. if err != nil {
  135. errMsg = "获取指标信息失败"
  136. err = errors.New("获取指标信息失败,ERR:" + err.Error())
  137. return
  138. }
  139. // 指标对应的所有数据
  140. chartType := 1 //1:普通图,2:季节性图
  141. calendar := "公历"
  142. edbDataListMap, edbList, err := data.GetEdbDataMapList(chartInfoId, chartType, calendar, "", "", mappingList, "")
  143. if err != nil {
  144. return
  145. }
  146. currDay := time.Now()
  147. currDay = time.Date(currDay.Year(), currDay.Month(), currDay.Day(), 0, 0, 0, 0, time.Local)
  148. dataMap := make(map[string]float64)
  149. dateMap := make(map[string]string)
  150. for dateIndex, dateConfig := range config.DateConfigList {
  151. for _, edbInfoMapping := range mappingList {
  152. // 数据会是正序的
  153. dataList, ok := edbDataListMap[edbInfoMapping.EdbInfoId]
  154. if !ok {
  155. continue
  156. }
  157. lenData := len(dataList)
  158. if lenData <= 0 {
  159. continue
  160. }
  161. // 数据的开始索引
  162. k := lenData - 1
  163. // 数据的最晚日期
  164. dataEndDateStr := dataList[k].DataTime
  165. dataEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, dataEndDateStr, time.Local)
  166. if tmpErr != nil {
  167. err = tmpErr
  168. return
  169. }
  170. // 数据开始日期
  171. endDateStr := ``
  172. var endDate time.Time
  173. var currVal float64
  174. switch dateConfig.DateType {
  175. case 1: // 1:最新日期;
  176. endDateStr = dataEndDateStr
  177. endDate = dataEndDate
  178. currVal = dataList[k].Value
  179. case 2: // 2:N天前
  180. tmpEndDate := dataEndDate.AddDate(0, 0, -dateConfig.Num)
  181. tmpEndDateStr := tmpEndDate.Format(utils.FormatDate)
  182. for i := k; i >= 0; i-- {
  183. tmpDateStr := dataList[i].DataTime
  184. // 如果正好是这一天,那么就直接break了
  185. if tmpEndDateStr == tmpDateStr {
  186. k = i
  187. endDateStr = tmpDateStr
  188. endDate = tmpEndDate
  189. currVal = dataList[i].Value
  190. break
  191. }
  192. tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
  193. if tmpErr != nil {
  194. err = tmpErr
  195. return
  196. }
  197. // 如果这期的日期晚于选择的日期,那么继续遍历
  198. if tmpDate.After(tmpEndDate) {
  199. continue
  200. }
  201. k = i
  202. endDateStr = tmpDateStr
  203. endDate = tmpDate
  204. currVal = dataList[i].Value
  205. break
  206. }
  207. }
  208. // 没有找到日期,那么就不处理
  209. if endDateStr == `` || endDate.IsZero() {
  210. continue
  211. }
  212. // 最早的日期
  213. earliestDate := endDate.AddDate(0, 0, -config.CalculateValue*moveUnitDays)
  214. earliestDateStr := earliestDate.Format(utils.FormatDate)
  215. var minVal, maxVal float64
  216. var isNotFirst bool // 是否是第一条数据
  217. for i := k; i >= 0; i-- {
  218. tmpData := dataList[i]
  219. if !isNotFirst {
  220. maxVal = tmpData.Value
  221. minVal = tmpData.Value
  222. isNotFirst = true
  223. continue
  224. }
  225. tmpDateStr := dataList[i].DataTime
  226. // 如果正好是这一天,那么就直接break了
  227. if earliestDateStr == tmpDateStr {
  228. break
  229. }
  230. tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
  231. if tmpErr != nil {
  232. err = tmpErr
  233. return
  234. }
  235. // 如果这期的日期早于选择的日期,那么继续停止遍历
  236. if tmpDate.Before(earliestDate) {
  237. continue
  238. }
  239. if tmpData.Value > maxVal {
  240. maxVal = tmpData.Value
  241. }
  242. if tmpData.Value < minVal {
  243. minVal = tmpData.Value
  244. }
  245. }
  246. // 最大值等于最小值,说明计算结果无效
  247. if maxVal == minVal {
  248. continue
  249. }
  250. //百分位=(现值-Min)/(Max-Min)
  251. tmpV := (currVal - minVal) / (maxVal - minVal) * 100
  252. tmpV, _ = decimal.NewFromFloat(tmpV).Round(4).Float64()
  253. // key的生成(日期配置下标+指标id)
  254. key := fmt.Sprint(dateIndex, "_", edbInfoMapping.EdbInfoId)
  255. dataMap[key] = tmpV
  256. dateMap[key] = endDateStr
  257. }
  258. }
  259. // 返回数据处理
  260. dataList := make([]SectionScatterSeriesItemResp, 0)
  261. var xMinVal, xMaxVal, yMinVal, yMaxVal float64
  262. var isNotFirst bool
  263. for _, varietyId := range config.VarietyList {
  264. xEdbInfoId, ok1 := xVarietyEdbMap[varietyId]
  265. if !ok1 {
  266. continue
  267. }
  268. yEdbInfoId, ok2 := yVarietyEdbMap[varietyId]
  269. if !ok2 {
  270. continue
  271. }
  272. variety, ok := varietyMap[varietyId]
  273. if !ok {
  274. continue
  275. }
  276. coordinatePointList := make([]CoordinatePoint, 0)
  277. for dateIndex, _ := range config.DateConfigList {
  278. key1 := fmt.Sprint(dateIndex, "_", xEdbInfoId)
  279. xVal, ok1 := dataMap[key1]
  280. if !ok1 {
  281. continue
  282. }
  283. key2 := fmt.Sprint(dateIndex, "_", yEdbInfoId)
  284. yVal, ok2 := dataMap[key2]
  285. if !ok2 {
  286. continue
  287. }
  288. if !isNotFirst {
  289. xMinVal = xVal
  290. xMaxVal = xVal
  291. yMinVal = yVal
  292. yMaxVal = yVal
  293. isNotFirst = true
  294. } else {
  295. if xVal < xMinVal {
  296. xMinVal = xVal
  297. }
  298. if xVal > xMaxVal {
  299. xMaxVal = xVal
  300. }
  301. if yVal < yMinVal {
  302. yMinVal = yVal
  303. }
  304. if yVal > yMaxVal {
  305. yMaxVal = yVal
  306. }
  307. }
  308. coordinatePointList = append(coordinatePointList, CoordinatePoint{
  309. X: xVal,
  310. Y: yVal,
  311. XEdbInfoId: xEdbInfoId,
  312. YEdbInfoId: yEdbInfoId,
  313. XDate: dateMap[key1],
  314. YDate: dateMap[key2],
  315. })
  316. }
  317. dataList = append(dataList, SectionScatterSeriesItemResp{
  318. Name: variety.ChartVarietyName,
  319. NameEn: variety.ChartVarietyNameEn,
  320. Color: "",
  321. CoordinatePointData: coordinatePointList,
  322. })
  323. }
  324. // 处理颜色
  325. colorMap := utils.GetColorMap()
  326. for k, _ := range dataList {
  327. if c, ok1 := colorMap[k]; ok1 {
  328. dataList[k].Color = c
  329. }
  330. }
  331. dataResp = ChartInfoResp{
  332. XName: xTagInfo.ChartTagName + "百分位",
  333. XNameEn: xTagInfo.ChartTagNameEn,
  334. XUnitName: "%",
  335. XUnitNameEn: "%",
  336. YName: yTagInfo.ChartTagName + "百分位",
  337. YNameEn: yTagInfo.ChartTagNameEn,
  338. YUnitName: "%",
  339. YUnitNameEn: "%",
  340. XMinValue: fmt.Sprint(xMinVal),
  341. XMaxValue: fmt.Sprint(xMaxVal),
  342. YMinValue: fmt.Sprint(yMinVal),
  343. YMaxValue: fmt.Sprint(yMaxVal),
  344. DataList: dataList,
  345. }
  346. // 去除返回指标中的数据信息,避免没必要的数据传输
  347. for k, _ := range edbList {
  348. edbList[k].DataList = nil
  349. }
  350. return
  351. }
  352. // CheckIsEnChart
  353. // @Description: 检测是否展示英文名称
  354. // @author: Roc
  355. // @datetime 2023-11-28 20:54:41
  356. // @param chartNameEn string
  357. // @param dataResp ChartInfoResp
  358. // @return bool
  359. func CheckIsEnChart(chartNameEn string, dataResp ChartInfoResp) bool {
  360. if chartNameEn == "" {
  361. return false
  362. }
  363. if dataResp.XNameEn == `` {
  364. return false
  365. }
  366. if dataResp.YNameEn == `` {
  367. return false
  368. }
  369. for _, v := range dataResp.DataList {
  370. if v.NameEn == `` {
  371. return false
  372. }
  373. }
  374. return true
  375. }
  376. // GetXYEdbIdList
  377. // @Description: 根据标签id和品种获取指标列表信息
  378. // @author: Roc
  379. // @datetime 2023-11-27 14:31:23
  380. // @param tagX int
  381. // @param tagY int
  382. // @param varietyList []int
  383. // @return xVarietyEdbMap map[int]int
  384. // @return yVarietyEdbMap map[int]int
  385. // @return edbInfoIdList []int
  386. // @return errMsg string
  387. // @return err error
  388. func GetXYEdbIdList(tagX, tagY int, varietyList []int) (xVarietyEdbMap, yVarietyEdbMap map[int]int, edbInfoIdList []int, err error) {
  389. edbInfoIdList = make([]int, 0)
  390. xVarietyEdbMap = make(map[int]int)
  391. yVarietyEdbMap = make(map[int]int)
  392. xList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(tagX, varietyList)
  393. if err != nil {
  394. err = errors.New("获取X轴的品种指标配置信息失败,Err:" + err.Error())
  395. return
  396. }
  397. yList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(tagY, varietyList)
  398. if err != nil {
  399. err = errors.New("获取Y轴的品种指标配置信息失败,Err:" + err.Error())
  400. return
  401. }
  402. baseVarietyIdMap := make(map[int]int)
  403. for _, v := range xList {
  404. baseVarietyIdMap[v.ChartVarietyId] = v.ChartVarietyId
  405. }
  406. // 两个标签里面的品种并集
  407. needVarietyIdMap := make(map[int]int)
  408. for _, v := range yList {
  409. if val, ok := baseVarietyIdMap[v.ChartVarietyId]; ok {
  410. // 如果在 map2 中存在相同的键,则将键和值添加到结果中
  411. needVarietyIdMap[v.ChartVarietyId] = val
  412. }
  413. }
  414. for _, v := range xList {
  415. if _, ok := needVarietyIdMap[v.ChartVarietyId]; ok {
  416. xVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
  417. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  418. }
  419. }
  420. for _, v := range yList {
  421. if _, ok := needVarietyIdMap[v.ChartVarietyId]; ok {
  422. yVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
  423. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  424. }
  425. }
  426. return
  427. }
  428. // AddChartInfo
  429. // @Description: AddChartInfo
  430. // @author: Roc
  431. // @datetime 2023-11-24 15:58:14
  432. // @param req request.AddChartReq
  433. // @param sysUser *system.Admin
  434. // @return chartInfo *data_manage.ChartInfo
  435. // @return err error
  436. // @return errMsg string
  437. // @return isSendEmail bool
  438. func AddChartInfo(req request.AddChartReq, sysUser *system.Admin) (chartInfo *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  439. isSendEmail = true
  440. source := utils.CHART_SOURCE_CROSS_HEDGING
  441. req.ChartName = strings.Trim(req.ChartName, " ")
  442. if req.ChartName == "" {
  443. errMsg = "请填写图表名称!"
  444. err = errors.New(errMsg)
  445. isSendEmail = false
  446. return
  447. }
  448. if req.TagX <= 0 {
  449. errMsg = "请选择X轴坐标的标签!"
  450. err = errors.New(errMsg)
  451. isSendEmail = false
  452. return
  453. }
  454. if req.TagY <= 0 {
  455. errMsg = "请选择Y轴坐标的标签!"
  456. err = errors.New(errMsg)
  457. isSendEmail = false
  458. return
  459. }
  460. if req.CalculateValue <= 0 {
  461. errMsg = "请设置时间长度!"
  462. err = errors.New(errMsg)
  463. isSendEmail = false
  464. return
  465. }
  466. if req.CalculateUnit == `` {
  467. errMsg = "请设置时间频度!"
  468. err = errors.New(errMsg)
  469. isSendEmail = false
  470. return
  471. }
  472. // 品种配置
  473. if len(req.VarietyList) < 0 {
  474. errMsg = "请选择品种!"
  475. err = errors.New(errMsg)
  476. isSendEmail = false
  477. return
  478. }
  479. // 日期配置
  480. dateConfigList := len(req.DateConfigList)
  481. if dateConfigList < 0 {
  482. errMsg = "请选择日期!"
  483. err = errors.New(errMsg)
  484. isSendEmail = false
  485. return
  486. }
  487. if dateConfigList > 5 {
  488. errMsg = "日期数量已达上限!"
  489. err = errors.New(errMsg)
  490. isSendEmail = false
  491. return
  492. }
  493. // 基础配置转string
  494. extraConfigByte, err := json.Marshal(req)
  495. if err != nil {
  496. return
  497. }
  498. chartClassify, err := data_manage.GetCrossVarietyChartClassifyBySysUserId(sysUser.AdminId)
  499. if err != nil {
  500. if err.Error() != utils.ErrNoRow() {
  501. errMsg = "获取分类信息失败"
  502. err = errors.New("获取分类信息失败,Err:" + err.Error())
  503. return
  504. }
  505. // 分类没有,需要创建新的分类,那么err置空
  506. err = nil
  507. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  508. chartClassify = &data_manage.ChartClassify{
  509. ChartClassifyId: 0,
  510. ChartClassifyName: sysUser.RealName,
  511. ParentId: 0,
  512. HasData: 0,
  513. CreateTime: time.Now(),
  514. ModifyTime: time.Now(),
  515. SysUserId: sysUser.AdminId,
  516. SysUserRealName: sysUser.RealName,
  517. Level: 1,
  518. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  519. Sort: 0,
  520. Source: source,
  521. }
  522. _, err = data_manage.AddChartClassify(chartClassify)
  523. }
  524. var chartInfoId int
  525. // 判断图表是否存在
  526. //var condition string
  527. //var pars []interface{}
  528. //condition += " AND chart_name=? AND source = ? "
  529. //pars = append(pars, req.ChartName, source)
  530. //count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  531. //if err != nil {
  532. // errMsg = "判断图表名称是否存在失败"
  533. // err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  534. // return
  535. //}
  536. //if count > 0 {
  537. // errMsg = "图表已存在,请重新填写"
  538. // err = errors.New(errMsg)
  539. // isSendEmail = false
  540. // return
  541. //}
  542. chartInfo = new(data_manage.ChartInfo)
  543. chartInfo.ChartName = req.ChartName
  544. //chartInfo.EdbInfoIds = edbInfoIdStr
  545. //chartInfo.ChartClassifyId = req.ChartClassifyId
  546. chartInfo.SysUserId = sysUser.AdminId
  547. chartInfo.SysUserRealName = sysUser.RealName
  548. chartInfo.ChartImage = req.ChartImage
  549. chartInfo.CreateTime = time.Now()
  550. chartInfo.ModifyTime = time.Now()
  551. chartInfo.IsSetName = 0
  552. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  553. chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
  554. chartInfo.ChartType = 9 // 相关性图
  555. chartInfo.Calendar = "公历"
  556. chartInfo.DateType = 6
  557. //chartInfo.StartDate = req.StartDate
  558. //chartInfo.EndDate = req.EndDate
  559. //chartInfo.SeasonStartDate = req.StartDate
  560. //chartInfo.SeasonEndDate = req.EndDate
  561. chartInfo.LeftMin = req.LeftMin
  562. chartInfo.LeftMax = req.LeftMax
  563. //chartInfo.RightMin = req.RightMin
  564. //chartInfo.RightMax = req.RightMax
  565. //chartInfo.Disabled = disableVal
  566. chartInfo.Source = source
  567. chartInfo.ExtraConfig = string(extraConfigByte)
  568. chartInfo.ChartImage = req.ChartImage
  569. chartInfo.ChartThemeId = req.ChartThemeId
  570. chartInfo.SourcesFrom = req.SourcesFrom
  571. chartInfo.Instructions = req.Instructions
  572. chartInfo.MarkersLines = req.MarkersLines
  573. chartInfo.MarkersAreas = req.MarkersAreas
  574. // 图表品种
  575. chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
  576. for _, varietyId := range req.VarietyList {
  577. chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
  578. Id: 0,
  579. ChartInfoId: 0,
  580. ChartVarietyId: varietyId,
  581. ModifyTime: time.Now(),
  582. CreateTime: time.Now(),
  583. })
  584. }
  585. // 图表配置
  586. chartInfoCrossVariety := &cross_varietyModel.ChartInfoCrossVariety{
  587. Id: 0,
  588. ChartInfoId: 0,
  589. ChartXTagId: req.TagX,
  590. ChartYTagId: req.TagY,
  591. CalculateValue: req.CalculateValue,
  592. CalculateUnit: req.CalculateUnit,
  593. ModifyTime: time.Now(),
  594. CreateTime: time.Now(),
  595. }
  596. // 新增图表和指标mapping
  597. chartInfoId, e := cross_varietyModel.CreateChart(chartInfo, chartClassify, chartVarietyMappingList, chartInfoCrossVariety)
  598. if e != nil {
  599. errMsg = "操作失败"
  600. err = errors.New("新增相关性图表失败, Err: " + e.Error())
  601. return
  602. }
  603. //添加es数据
  604. go data.EsAddOrEditChartInfo(chartInfoId)
  605. return
  606. }
  607. // EditChartInfo
  608. // @Description: 编辑图表
  609. // @author: Roc
  610. // @datetime 2023-11-24 15:58:31
  611. // @param req request.EditChartReq
  612. // @param sysUser *system.Admin
  613. // @return chartItem *data_manage.ChartInfo
  614. // @return err error
  615. // @return errMsg string
  616. // @return isSendEmail bool
  617. func EditChartInfo(req request.EditChartReq, sysUser *system.Admin) (chartItem *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  618. isSendEmail = true
  619. chartItem, err = data_manage.GetChartInfoById(req.ChartInfoId)
  620. if err != nil {
  621. if err.Error() == utils.ErrNoRow() {
  622. errMsg = "图表已被删除,请刷新页面"
  623. err = errors.New(errMsg)
  624. isSendEmail = false
  625. return
  626. }
  627. errMsg = "获取图表信息失败"
  628. err = errors.New("获取图表信息失败,Err:" + err.Error())
  629. return
  630. }
  631. if chartItem.Source != utils.CHART_SOURCE_CROSS_HEDGING {
  632. errMsg = "该图不是跨品种分析图表!"
  633. err = errors.New(errMsg)
  634. isSendEmail = false
  635. return
  636. }
  637. req.ChartName = strings.Trim(req.ChartName, " ")
  638. if req.ChartName == "" {
  639. errMsg = "请填写图表名称!"
  640. err = errors.New(errMsg)
  641. isSendEmail = false
  642. return
  643. }
  644. if req.TagX <= 0 {
  645. errMsg = "请选择X轴坐标的标签!"
  646. err = errors.New(errMsg)
  647. isSendEmail = false
  648. return
  649. }
  650. if req.TagY <= 0 {
  651. errMsg = "请选择Y轴坐标的标签!"
  652. err = errors.New(errMsg)
  653. isSendEmail = false
  654. return
  655. }
  656. if req.CalculateValue <= 0 {
  657. errMsg = "请设置时间长度!"
  658. err = errors.New(errMsg)
  659. isSendEmail = false
  660. return
  661. }
  662. if req.CalculateUnit == `` {
  663. errMsg = "请设置时间频度!"
  664. err = errors.New(errMsg)
  665. isSendEmail = false
  666. return
  667. }
  668. // 品种配置
  669. if len(req.VarietyList) < 0 {
  670. errMsg = "请选择品种!"
  671. err = errors.New(errMsg)
  672. isSendEmail = false
  673. return
  674. }
  675. // 日期配置
  676. dateConfigList := len(req.DateConfigList)
  677. if dateConfigList < 0 {
  678. errMsg = "请选择日期!"
  679. err = errors.New(errMsg)
  680. isSendEmail = false
  681. return
  682. }
  683. if dateConfigList > 5 {
  684. errMsg = "日期数量已达上限!"
  685. err = errors.New(errMsg)
  686. isSendEmail = false
  687. return
  688. }
  689. // 基础配置转string
  690. extraConfigByte, err := json.Marshal(req)
  691. if err != nil {
  692. return
  693. }
  694. // 图表操作权限
  695. ok := data.CheckOpChartPermission(sysUser, chartItem.SysUserId)
  696. if !ok {
  697. errMsg = "没有该图表的操作权限"
  698. err = errors.New(errMsg)
  699. isSendEmail = false
  700. return
  701. }
  702. //判断图表是否存在
  703. //var condition string
  704. //var pars []interface{}
  705. //condition += " AND chart_info_id <> ? "
  706. //pars = append(pars, req.ChartInfoId)
  707. //condition += " AND chart_name=? AND source = ? "
  708. //pars = append(pars, req.ChartName, chartItem.Source)
  709. //count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  710. //if err != nil {
  711. // errMsg = "判断图表名称是否存在失败"
  712. // err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  713. // return
  714. //}
  715. //if count > 0 {
  716. // errMsg = "图表已存在,请重新填写"
  717. // err = errors.New(errMsg)
  718. // isSendEmail = false
  719. // return
  720. //}
  721. chartItem.ChartName = req.ChartName
  722. chartItem.ExtraConfig = string(extraConfigByte)
  723. chartItem.ModifyTime = time.Now()
  724. chartUpdateCols := []string{"ChartName", "ExtraConfig", "ModifyTime"}
  725. // 跨品种分析配置
  726. chartInfoCrossVariety, err := cross_varietyModel.GetChartInfoCrossVarietyByChartInfoId(chartItem.ChartInfoId)
  727. if err != nil {
  728. return
  729. }
  730. chartInfoCrossVariety.ChartXTagId = req.TagX
  731. chartInfoCrossVariety.ChartYTagId = req.TagY
  732. chartInfoCrossVariety.CalculateValue = req.CalculateValue
  733. chartInfoCrossVariety.CalculateUnit = req.CalculateUnit
  734. chartInfoCrossVariety.ModifyTime = time.Now()
  735. chartInfoCrossVarietyUpdateCols := []string{"ChartXTagId", "ChartYTagId", "CalculateValue", "CalculateUnit", "ModifyTime"}
  736. // 图表品种
  737. chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
  738. for _, varietyId := range req.VarietyList {
  739. chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
  740. Id: 0,
  741. ChartInfoId: 0,
  742. ChartVarietyId: varietyId,
  743. ModifyTime: time.Now(),
  744. CreateTime: time.Now(),
  745. })
  746. }
  747. err = cross_varietyModel.EditChart(chartItem, chartVarietyMappingList, chartInfoCrossVariety,
  748. chartUpdateCols, chartInfoCrossVarietyUpdateCols)
  749. if err != nil {
  750. errMsg = "保存失败"
  751. err = errors.New("保存失败,Err:" + err.Error())
  752. return
  753. }
  754. resp := new(data_manage.AddChartInfoResp)
  755. resp.ChartInfoId = chartItem.ChartInfoId
  756. resp.UniqueCode = chartItem.UniqueCode
  757. //resp.ChartType = req.ChartType
  758. //添加es数据
  759. go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
  760. //修改my eta es数据
  761. go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)
  762. return
  763. }
  764. // ModifyChartEdbMapping
  765. // @Description: 修改图表的关系表
  766. // @author: Roc
  767. // @datetime 2023-12-11 17:32:31
  768. // @param tagId int
  769. func ModifyChartEdbMapping(tagId int) {
  770. errMsgList := make([]string, 0)
  771. var err error
  772. defer func() {
  773. if err != nil {
  774. errMsgList = append(errMsgList, err.Error())
  775. }
  776. if len(errMsgList) > 0 {
  777. utils.FileLog.Error("修改跨品种分析的图表关联指标失败:\n" + strings.Join(errMsgList, "\n"))
  778. }
  779. }()
  780. // 找出标签关联的图表
  781. list, err := cross_varietyModel.GeChartInfoCrossVarietyListByTagId(tagId)
  782. if err != nil {
  783. return
  784. }
  785. chartInfoIdList := make([]int, 0)
  786. tagIdList := make([]int, 0)
  787. tagIdMap := make(map[int]int, 0)
  788. for _, v := range list {
  789. chartInfoIdList = append(chartInfoIdList, v.ChartInfoId)
  790. if _, ok := tagIdMap[v.ChartXTagId]; !ok {
  791. tagIdList = append(tagIdList, v.ChartXTagId)
  792. tagIdMap[v.ChartXTagId] = 1
  793. }
  794. if _, ok := tagIdMap[v.ChartYTagId]; !ok {
  795. tagIdList = append(tagIdList, v.ChartYTagId)
  796. tagIdMap[v.ChartYTagId] = 1
  797. }
  798. }
  799. // 找出所有标签关联的品种和对应的指标
  800. chartTagVarietyList, err := cross_varietyModel.GetChartTagVarietyListByTagIdList(tagIdList)
  801. if err != nil {
  802. return
  803. }
  804. // 所有指标id
  805. allEdbInfoIdList := make([]int, 0)
  806. // 标签关联品种的指标列表
  807. tagVarietyEdbMap := make(map[string][]int)
  808. for _, v := range chartTagVarietyList {
  809. if v.EdbInfoId <= 0 {
  810. continue
  811. }
  812. key := fmt.Sprint(v.ChartTagId, "_", v.ChartVarietyId)
  813. tagVarietyEdbList, ok := tagVarietyEdbMap[key]
  814. if !ok {
  815. tagVarietyEdbList = make([]int, 0)
  816. }
  817. tagVarietyEdbMap[key] = append(tagVarietyEdbList, v.EdbInfoId)
  818. allEdbInfoIdList = append(allEdbInfoIdList, v.EdbInfoId)
  819. }
  820. edbInfoMap := make(map[int]*data_manage.EdbInfo)
  821. // 获取指标列表
  822. if len(allEdbInfoIdList) > 0 {
  823. allEdbInfoList, tmpErr := data_manage.GetEdbInfoByIdList(allEdbInfoIdList)
  824. if tmpErr != nil {
  825. err = tmpErr
  826. return
  827. }
  828. for _, v := range allEdbInfoList {
  829. edbInfoMap[v.EdbInfoId] = v
  830. }
  831. }
  832. // 图表关联的品种列表
  833. chartVarietyMappingList, err := cross_varietyModel.GetChartVarietyMappingListByChartInfoIdList(chartInfoIdList)
  834. if err != nil {
  835. return
  836. }
  837. // 图表关联的品种map
  838. chartVarietyMappingListMap := make(map[int][]int)
  839. for _, v := range chartVarietyMappingList {
  840. if v.ChartInfoId <= 0 {
  841. continue
  842. }
  843. varietyMappingList, ok := chartVarietyMappingListMap[v.ChartInfoId]
  844. if !ok {
  845. varietyMappingList = make([]int, 0)
  846. }
  847. chartVarietyMappingListMap[v.ChartInfoId] = append(varietyMappingList, v.ChartVarietyId)
  848. }
  849. // 处理图表
  850. for _, v := range list {
  851. // 获取关联的指标id
  852. edbInfoList := make([]*data_manage.EdbInfo, 0)
  853. edbInfoIdMap := make(map[int]int, 0)
  854. // 找出图表关联的品种
  855. varietyIdList, ok := chartVarietyMappingListMap[v.ChartInfoId]
  856. if ok {
  857. // 获取品种与标签的id
  858. for _, varietyId := range varietyIdList {
  859. // 先处理x轴
  860. key := fmt.Sprint(v.ChartXTagId, "_", varietyId)
  861. if tmpEdbInfoIdList, ok2 := tagVarietyEdbMap[key]; ok2 {
  862. for _, edbInfoId := range tmpEdbInfoIdList {
  863. if _, ok3 := edbInfoIdMap[edbInfoId]; !ok3 {
  864. if edbInfo, ok4 := edbInfoMap[edbInfoId]; ok4 {
  865. edbInfoList = append(edbInfoList, edbInfo)
  866. }
  867. edbInfoIdMap[edbInfoId] = edbInfoId
  868. }
  869. }
  870. }
  871. // 处理y轴
  872. key = fmt.Sprint(v.ChartYTagId, "_", varietyId)
  873. if tmpEdbInfoIdList, ok2 := tagVarietyEdbMap[key]; ok2 {
  874. for _, edbInfoId := range tmpEdbInfoIdList {
  875. if _, ok3 := edbInfoIdMap[edbInfoId]; !ok3 {
  876. if edbInfo, ok4 := edbInfoMap[edbInfoId]; ok4 {
  877. edbInfoList = append(edbInfoList, edbInfo)
  878. }
  879. edbInfoIdMap[edbInfoId] = edbInfoId
  880. }
  881. }
  882. }
  883. }
  884. }
  885. tmpErr := data_manage.ModifyChartEdbMapping(v.ChartInfoId, edbInfoList)
  886. if tmpErr != nil {
  887. errMsgList = append(errMsgList, fmt.Sprint("修改", v.ChartInfoId, "失败,err:", tmpErr))
  888. }
  889. }
  890. return
  891. }