chart.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968
  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. dataResp = ChartInfoResp{
  325. XName: xTagInfo.ChartTagName + "百分位",
  326. XNameEn: xTagInfo.ChartTagNameEn,
  327. XUnitName: "%",
  328. XUnitNameEn: "%",
  329. YName: yTagInfo.ChartTagName + "百分位",
  330. YNameEn: yTagInfo.ChartTagNameEn,
  331. YUnitName: "%",
  332. YUnitNameEn: "%",
  333. XMinValue: fmt.Sprint(xMinVal),
  334. XMaxValue: fmt.Sprint(xMaxVal),
  335. YMinValue: fmt.Sprint(yMinVal),
  336. YMaxValue: fmt.Sprint(yMaxVal),
  337. DataList: dataList,
  338. }
  339. // 去除返回指标中的数据信息,避免没必要的数据传输
  340. for k, _ := range edbList {
  341. edbList[k].DataList = nil
  342. }
  343. return
  344. }
  345. // CheckIsEnChart
  346. // @Description: 检测是否展示英文名称
  347. // @author: Roc
  348. // @datetime 2023-11-28 20:54:41
  349. // @param chartNameEn string
  350. // @param dataResp ChartInfoResp
  351. // @return bool
  352. func CheckIsEnChart(chartNameEn string, dataResp ChartInfoResp) bool {
  353. if chartNameEn == "" {
  354. return false
  355. }
  356. if dataResp.XNameEn == `` {
  357. return false
  358. }
  359. if dataResp.YNameEn == `` {
  360. return false
  361. }
  362. for _, v := range dataResp.DataList {
  363. if v.NameEn == `` {
  364. return false
  365. }
  366. }
  367. return true
  368. }
  369. // GetXYEdbIdList
  370. // @Description: 根据标签id和品种获取指标列表信息
  371. // @author: Roc
  372. // @datetime 2023-11-27 14:31:23
  373. // @param tagX int
  374. // @param tagY int
  375. // @param varietyList []int
  376. // @return xVarietyEdbMap map[int]int
  377. // @return yVarietyEdbMap map[int]int
  378. // @return edbInfoIdList []int
  379. // @return errMsg string
  380. // @return err error
  381. func GetXYEdbIdList(tagX, tagY int, varietyList []int) (xVarietyEdbMap, yVarietyEdbMap map[int]int, edbInfoIdList []int, err error) {
  382. edbInfoIdList = make([]int, 0)
  383. xVarietyEdbMap = make(map[int]int)
  384. yVarietyEdbMap = make(map[int]int)
  385. xList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(tagX, varietyList)
  386. if err != nil {
  387. err = errors.New("获取X轴的品种指标配置信息失败,Err:" + err.Error())
  388. return
  389. }
  390. yList, err := cross_varietyModel.GetChartTagVarietyListByTagAndVariety(tagY, varietyList)
  391. if err != nil {
  392. err = errors.New("获取Y轴的品种指标配置信息失败,Err:" + err.Error())
  393. return
  394. }
  395. baseVarietyIdMap := make(map[int]int)
  396. for _, v := range xList {
  397. baseVarietyIdMap[v.ChartVarietyId] = v.ChartVarietyId
  398. }
  399. // 两个标签里面的品种并集
  400. needVarietyIdMap := make(map[int]int)
  401. for _, v := range yList {
  402. if val, ok := baseVarietyIdMap[v.ChartVarietyId]; ok {
  403. // 如果在 map2 中存在相同的键,则将键和值添加到结果中
  404. needVarietyIdMap[v.ChartVarietyId] = val
  405. }
  406. }
  407. for _, v := range xList {
  408. if _, ok := needVarietyIdMap[v.ChartVarietyId]; ok {
  409. xVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
  410. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  411. }
  412. }
  413. for _, v := range yList {
  414. if _, ok := needVarietyIdMap[v.ChartVarietyId]; ok {
  415. yVarietyEdbMap[v.ChartVarietyId] = v.EdbInfoId
  416. edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
  417. }
  418. }
  419. return
  420. }
  421. // AddChartInfo
  422. // @Description: AddChartInfo
  423. // @author: Roc
  424. // @datetime 2023-11-24 15:58:14
  425. // @param req request.AddChartReq
  426. // @param sysUser *system.Admin
  427. // @return chartInfo *data_manage.ChartInfo
  428. // @return err error
  429. // @return errMsg string
  430. // @return isSendEmail bool
  431. func AddChartInfo(req request.AddChartReq, sysUser *system.Admin) (chartInfo *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  432. isSendEmail = true
  433. source := utils.CHART_SOURCE_CROSS_HEDGING
  434. req.ChartName = strings.Trim(req.ChartName, " ")
  435. if req.ChartName == "" {
  436. errMsg = "请填写图表名称!"
  437. err = errors.New(errMsg)
  438. isSendEmail = false
  439. return
  440. }
  441. if req.TagX <= 0 {
  442. errMsg = "请选择X轴坐标的标签!"
  443. err = errors.New(errMsg)
  444. isSendEmail = false
  445. return
  446. }
  447. if req.TagY <= 0 {
  448. errMsg = "请选择Y轴坐标的标签!"
  449. err = errors.New(errMsg)
  450. isSendEmail = false
  451. return
  452. }
  453. if req.CalculateValue <= 0 {
  454. errMsg = "请设置时间长度!"
  455. err = errors.New(errMsg)
  456. isSendEmail = false
  457. return
  458. }
  459. if req.CalculateUnit == `` {
  460. errMsg = "请设置时间频度!"
  461. err = errors.New(errMsg)
  462. isSendEmail = false
  463. return
  464. }
  465. // 品种配置
  466. if len(req.VarietyList) < 0 {
  467. errMsg = "请选择品种!"
  468. err = errors.New(errMsg)
  469. isSendEmail = false
  470. return
  471. }
  472. // 日期配置
  473. dateConfigList := len(req.DateConfigList)
  474. if dateConfigList < 0 {
  475. errMsg = "请选择日期!"
  476. err = errors.New(errMsg)
  477. isSendEmail = false
  478. return
  479. }
  480. if dateConfigList > 5 {
  481. errMsg = "日期数量已达上限!"
  482. err = errors.New(errMsg)
  483. isSendEmail = false
  484. return
  485. }
  486. // 基础配置转string
  487. extraConfigByte, err := json.Marshal(req)
  488. if err != nil {
  489. return
  490. }
  491. chartClassify, err := data_manage.GetCrossVarietyChartClassifyBySysUserId(sysUser.AdminId)
  492. if err != nil {
  493. if err.Error() != utils.ErrNoRow() {
  494. errMsg = "获取分类信息失败"
  495. err = errors.New("获取分类信息失败,Err:" + err.Error())
  496. return
  497. }
  498. // 分类没有,需要创建新的分类,那么err置空
  499. err = nil
  500. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  501. chartClassify = &data_manage.ChartClassify{
  502. ChartClassifyId: 0,
  503. ChartClassifyName: sysUser.RealName,
  504. ParentId: 0,
  505. HasData: 0,
  506. CreateTime: time.Now(),
  507. ModifyTime: time.Now(),
  508. SysUserId: sysUser.AdminId,
  509. SysUserRealName: sysUser.RealName,
  510. Level: 1,
  511. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  512. Sort: 0,
  513. Source: source,
  514. }
  515. _, err = data_manage.AddChartClassify(chartClassify)
  516. }
  517. var chartInfoId int
  518. // 判断图表是否存在
  519. //var condition string
  520. //var pars []interface{}
  521. //condition += " AND chart_name=? AND source = ? "
  522. //pars = append(pars, req.ChartName, source)
  523. //count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  524. //if err != nil {
  525. // errMsg = "判断图表名称是否存在失败"
  526. // err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  527. // return
  528. //}
  529. //if count > 0 {
  530. // errMsg = "图表已存在,请重新填写"
  531. // err = errors.New(errMsg)
  532. // isSendEmail = false
  533. // return
  534. //}
  535. chartInfo = new(data_manage.ChartInfo)
  536. chartInfo.ChartName = req.ChartName
  537. //chartInfo.EdbInfoIds = edbInfoIdStr
  538. //chartInfo.ChartClassifyId = req.ChartClassifyId
  539. chartInfo.SysUserId = sysUser.AdminId
  540. chartInfo.SysUserRealName = sysUser.RealName
  541. chartInfo.ChartImage = req.ChartImage
  542. chartInfo.CreateTime = time.Now()
  543. chartInfo.ModifyTime = time.Now()
  544. chartInfo.IsSetName = 0
  545. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  546. chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
  547. chartInfo.ChartType = 9 // 相关性图
  548. chartInfo.Calendar = "公历"
  549. chartInfo.DateType = 6
  550. //chartInfo.StartDate = req.StartDate
  551. //chartInfo.EndDate = req.EndDate
  552. //chartInfo.SeasonStartDate = req.StartDate
  553. //chartInfo.SeasonEndDate = req.EndDate
  554. chartInfo.LeftMin = req.LeftMin
  555. chartInfo.LeftMax = req.LeftMax
  556. //chartInfo.RightMin = req.RightMin
  557. //chartInfo.RightMax = req.RightMax
  558. //chartInfo.Disabled = disableVal
  559. chartInfo.Source = source
  560. chartInfo.ExtraConfig = string(extraConfigByte)
  561. chartInfo.ChartImage = req.ChartImage
  562. chartInfo.ChartThemeId = req.ChartThemeId
  563. chartInfo.SourcesFrom = req.SourcesFrom
  564. chartInfo.Instructions = req.Instructions
  565. chartInfo.MarkersLines = req.MarkersLines
  566. chartInfo.MarkersAreas = req.MarkersAreas
  567. // 图表品种
  568. chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
  569. for _, varietyId := range req.VarietyList {
  570. chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
  571. Id: 0,
  572. ChartInfoId: 0,
  573. ChartVarietyId: varietyId,
  574. ModifyTime: time.Now(),
  575. CreateTime: time.Now(),
  576. })
  577. }
  578. // 图表配置
  579. chartInfoCrossVariety := &cross_varietyModel.ChartInfoCrossVariety{
  580. Id: 0,
  581. ChartInfoId: 0,
  582. ChartXTagId: req.TagX,
  583. ChartYTagId: req.TagY,
  584. CalculateValue: req.CalculateValue,
  585. CalculateUnit: req.CalculateUnit,
  586. ModifyTime: time.Now(),
  587. CreateTime: time.Now(),
  588. }
  589. // 新增图表和指标mapping
  590. chartInfoId, e := cross_varietyModel.CreateChart(chartInfo, chartClassify, chartVarietyMappingList, chartInfoCrossVariety)
  591. if e != nil {
  592. errMsg = "操作失败"
  593. err = errors.New("新增相关性图表失败, Err: " + e.Error())
  594. return
  595. }
  596. //添加es数据
  597. go data.EsAddOrEditChartInfo(chartInfoId)
  598. return
  599. }
  600. // EditChartInfo
  601. // @Description: 编辑图表
  602. // @author: Roc
  603. // @datetime 2023-11-24 15:58:31
  604. // @param req request.EditChartReq
  605. // @param sysUser *system.Admin
  606. // @return chartItem *data_manage.ChartInfo
  607. // @return err error
  608. // @return errMsg string
  609. // @return isSendEmail bool
  610. func EditChartInfo(req request.EditChartReq, sysUser *system.Admin) (chartItem *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  611. isSendEmail = true
  612. chartItem, err = data_manage.GetChartInfoById(req.ChartInfoId)
  613. if err != nil {
  614. if err.Error() == utils.ErrNoRow() {
  615. errMsg = "图表已被删除,请刷新页面"
  616. err = errors.New(errMsg)
  617. isSendEmail = false
  618. return
  619. }
  620. errMsg = "获取图表信息失败"
  621. err = errors.New("获取图表信息失败,Err:" + err.Error())
  622. return
  623. }
  624. if chartItem.Source != utils.CHART_SOURCE_CROSS_HEDGING {
  625. errMsg = "该图不是跨品种分析图表!"
  626. err = errors.New(errMsg)
  627. isSendEmail = false
  628. return
  629. }
  630. req.ChartName = strings.Trim(req.ChartName, " ")
  631. if req.ChartName == "" {
  632. errMsg = "请填写图表名称!"
  633. err = errors.New(errMsg)
  634. isSendEmail = false
  635. return
  636. }
  637. if req.TagX <= 0 {
  638. errMsg = "请选择X轴坐标的标签!"
  639. err = errors.New(errMsg)
  640. isSendEmail = false
  641. return
  642. }
  643. if req.TagY <= 0 {
  644. errMsg = "请选择Y轴坐标的标签!"
  645. err = errors.New(errMsg)
  646. isSendEmail = false
  647. return
  648. }
  649. if req.CalculateValue <= 0 {
  650. errMsg = "请设置时间长度!"
  651. err = errors.New(errMsg)
  652. isSendEmail = false
  653. return
  654. }
  655. if req.CalculateUnit == `` {
  656. errMsg = "请设置时间频度!"
  657. err = errors.New(errMsg)
  658. isSendEmail = false
  659. return
  660. }
  661. // 品种配置
  662. if len(req.VarietyList) < 0 {
  663. errMsg = "请选择品种!"
  664. err = errors.New(errMsg)
  665. isSendEmail = false
  666. return
  667. }
  668. // 日期配置
  669. dateConfigList := len(req.DateConfigList)
  670. if dateConfigList < 0 {
  671. errMsg = "请选择日期!"
  672. err = errors.New(errMsg)
  673. isSendEmail = false
  674. return
  675. }
  676. if dateConfigList > 5 {
  677. errMsg = "日期数量已达上限!"
  678. err = errors.New(errMsg)
  679. isSendEmail = false
  680. return
  681. }
  682. // 基础配置转string
  683. extraConfigByte, err := json.Marshal(req)
  684. if err != nil {
  685. return
  686. }
  687. // 图表操作权限
  688. ok := data.CheckOpChartPermission(sysUser, chartItem.SysUserId)
  689. if !ok {
  690. errMsg = "没有该图表的操作权限"
  691. err = errors.New(errMsg)
  692. isSendEmail = false
  693. return
  694. }
  695. //判断图表是否存在
  696. //var condition string
  697. //var pars []interface{}
  698. //condition += " AND chart_info_id <> ? "
  699. //pars = append(pars, req.ChartInfoId)
  700. //condition += " AND chart_name=? AND source = ? "
  701. //pars = append(pars, req.ChartName, chartItem.Source)
  702. //count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  703. //if err != nil {
  704. // errMsg = "判断图表名称是否存在失败"
  705. // err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  706. // return
  707. //}
  708. //if count > 0 {
  709. // errMsg = "图表已存在,请重新填写"
  710. // err = errors.New(errMsg)
  711. // isSendEmail = false
  712. // return
  713. //}
  714. chartItem.ChartName = req.ChartName
  715. chartItem.ExtraConfig = string(extraConfigByte)
  716. chartItem.ModifyTime = time.Now()
  717. chartUpdateCols := []string{"ChartName", "ExtraConfig", "ModifyTime"}
  718. // 跨品种分析配置
  719. chartInfoCrossVariety, err := cross_varietyModel.GetChartInfoCrossVarietyByChartInfoId(chartItem.ChartInfoId)
  720. if err != nil {
  721. return
  722. }
  723. chartInfoCrossVariety.ChartXTagId = req.TagX
  724. chartInfoCrossVariety.ChartYTagId = req.TagY
  725. chartInfoCrossVariety.CalculateValue = req.CalculateValue
  726. chartInfoCrossVariety.CalculateUnit = req.CalculateUnit
  727. chartInfoCrossVariety.ModifyTime = time.Now()
  728. chartInfoCrossVarietyUpdateCols := []string{"ChartXTagId", "ChartYTagId", "CalculateValue", "CalculateUnit", "ModifyTime"}
  729. // 图表品种
  730. chartVarietyMappingList := make([]*cross_varietyModel.ChartVarietyMapping, 0)
  731. for _, varietyId := range req.VarietyList {
  732. chartVarietyMappingList = append(chartVarietyMappingList, &cross_varietyModel.ChartVarietyMapping{
  733. Id: 0,
  734. ChartInfoId: 0,
  735. ChartVarietyId: varietyId,
  736. ModifyTime: time.Now(),
  737. CreateTime: time.Now(),
  738. })
  739. }
  740. err = cross_varietyModel.EditChart(chartItem, chartVarietyMappingList, chartInfoCrossVariety,
  741. chartUpdateCols, chartInfoCrossVarietyUpdateCols)
  742. if err != nil {
  743. errMsg = "保存失败"
  744. err = errors.New("保存失败,Err:" + err.Error())
  745. return
  746. }
  747. resp := new(data_manage.AddChartInfoResp)
  748. resp.ChartInfoId = chartItem.ChartInfoId
  749. resp.UniqueCode = chartItem.UniqueCode
  750. //resp.ChartType = req.ChartType
  751. //添加es数据
  752. go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
  753. //修改my eta es数据
  754. go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)
  755. return
  756. }
  757. // ModifyChartEdbMapping
  758. // @Description: 修改图表的关系表
  759. // @author: Roc
  760. // @datetime 2023-12-11 17:32:31
  761. // @param tagId int
  762. func ModifyChartEdbMapping(tagId int) {
  763. errMsgList := make([]string, 0)
  764. var err error
  765. defer func() {
  766. if err != nil {
  767. errMsgList = append(errMsgList, err.Error())
  768. }
  769. if len(errMsgList) > 0 {
  770. utils.FileLog.Error("修改跨品种分析的图表关联指标失败:\n" + strings.Join(errMsgList, "\n"))
  771. }
  772. }()
  773. // 找出标签关联的图表
  774. list, err := cross_varietyModel.GeChartInfoCrossVarietyListByTagId(tagId)
  775. if err != nil {
  776. return
  777. }
  778. chartInfoIdList := make([]int, 0)
  779. tagIdList := make([]int, 0)
  780. tagIdMap := make(map[int]int, 0)
  781. for _, v := range list {
  782. chartInfoIdList = append(chartInfoIdList, v.ChartInfoId)
  783. if _, ok := tagIdMap[v.ChartXTagId]; !ok {
  784. tagIdList = append(tagIdList, v.ChartXTagId)
  785. tagIdMap[v.ChartXTagId] = 1
  786. }
  787. if _, ok := tagIdMap[v.ChartYTagId]; !ok {
  788. tagIdList = append(tagIdList, v.ChartYTagId)
  789. tagIdMap[v.ChartYTagId] = 1
  790. }
  791. }
  792. // 找出所有标签关联的品种和对应的指标
  793. chartTagVarietyList, err := cross_varietyModel.GetChartTagVarietyListByTagIdList(tagIdList)
  794. if err != nil {
  795. return
  796. }
  797. // 所有指标id
  798. allEdbInfoIdList := make([]int, 0)
  799. // 标签关联品种的指标列表
  800. tagVarietyEdbMap := make(map[string][]int)
  801. for _, v := range chartTagVarietyList {
  802. if v.EdbInfoId <= 0 {
  803. continue
  804. }
  805. key := fmt.Sprint(v.ChartTagId, "_", v.ChartVarietyId)
  806. tagVarietyEdbList, ok := tagVarietyEdbMap[key]
  807. if !ok {
  808. tagVarietyEdbList = make([]int, 0)
  809. }
  810. tagVarietyEdbMap[key] = append(tagVarietyEdbList, v.EdbInfoId)
  811. allEdbInfoIdList = append(allEdbInfoIdList, v.EdbInfoId)
  812. }
  813. edbInfoMap := make(map[int]*data_manage.EdbInfo)
  814. // 获取指标列表
  815. if len(allEdbInfoIdList) > 0 {
  816. allEdbInfoList, tmpErr := data_manage.GetEdbInfoByIdList(allEdbInfoIdList)
  817. if tmpErr != nil {
  818. err = tmpErr
  819. return
  820. }
  821. for _, v := range allEdbInfoList {
  822. edbInfoMap[v.EdbInfoId] = v
  823. }
  824. }
  825. // 图表关联的品种列表
  826. chartVarietyMappingList, err := cross_varietyModel.GetChartVarietyMappingListByChartInfoIdList(chartInfoIdList)
  827. if err != nil {
  828. return
  829. }
  830. // 图表关联的品种map
  831. chartVarietyMappingListMap := make(map[int][]int)
  832. for _, v := range chartVarietyMappingList {
  833. if v.ChartInfoId <= 0 {
  834. continue
  835. }
  836. varietyMappingList, ok := chartVarietyMappingListMap[v.ChartInfoId]
  837. if !ok {
  838. varietyMappingList = make([]int, 0)
  839. }
  840. chartVarietyMappingListMap[v.ChartInfoId] = append(varietyMappingList, v.ChartVarietyId)
  841. }
  842. // 处理图表
  843. for _, v := range list {
  844. // 获取关联的指标id
  845. edbInfoList := make([]*data_manage.EdbInfo, 0)
  846. edbInfoIdMap := make(map[int]int, 0)
  847. // 找出图表关联的品种
  848. varietyIdList, ok := chartVarietyMappingListMap[v.ChartInfoId]
  849. if ok {
  850. // 获取品种与标签的id
  851. for _, varietyId := range varietyIdList {
  852. // 先处理x轴
  853. key := fmt.Sprint(v.ChartXTagId, "_", varietyId)
  854. if tmpEdbInfoIdList, ok2 := tagVarietyEdbMap[key]; ok2 {
  855. for _, edbInfoId := range tmpEdbInfoIdList {
  856. if _, ok3 := edbInfoIdMap[edbInfoId]; !ok3 {
  857. if edbInfo, ok4 := edbInfoMap[edbInfoId]; ok4 {
  858. edbInfoList = append(edbInfoList, edbInfo)
  859. }
  860. edbInfoIdMap[edbInfoId] = edbInfoId
  861. }
  862. }
  863. }
  864. // 处理y轴
  865. key = fmt.Sprint(v.ChartYTagId, "_", varietyId)
  866. if tmpEdbInfoIdList, ok2 := tagVarietyEdbMap[key]; ok2 {
  867. for _, edbInfoId := range tmpEdbInfoIdList {
  868. if _, ok3 := edbInfoIdMap[edbInfoId]; !ok3 {
  869. if edbInfo, ok4 := edbInfoMap[edbInfoId]; ok4 {
  870. edbInfoList = append(edbInfoList, edbInfo)
  871. }
  872. edbInfoIdMap[edbInfoId] = edbInfoId
  873. }
  874. }
  875. }
  876. }
  877. }
  878. tmpErr := data_manage.ModifyChartEdbMapping(v.ChartInfoId, edbInfoList)
  879. if tmpErr != nil {
  880. errMsgList = append(errMsgList, fmt.Sprint("修改", v.ChartInfoId, "失败,err:", tmpErr))
  881. }
  882. }
  883. return
  884. }