chart_info.go 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269
  1. package data
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_chart_lib/models"
  6. "eta/eta_chart_lib/models/data_manage"
  7. "eta/eta_chart_lib/services/alarm_msg"
  8. "eta/eta_chart_lib/utils"
  9. "fmt"
  10. "github.com/shopspring/decimal"
  11. "math"
  12. "sort"
  13. "strconv"
  14. "time"
  15. )
  16. // GetFrequencyEn 获取频度的英文版
  17. func GetFrequencyEn(frequency string) (frequencyEn string) {
  18. switch frequency {
  19. case "日度":
  20. frequencyEn = "day"
  21. return
  22. case "周度":
  23. frequencyEn = "week"
  24. return
  25. case "旬度":
  26. frequencyEn = "ten days"
  27. return
  28. case "月度":
  29. frequencyEn = "month"
  30. return
  31. case "季度":
  32. frequencyEn = "quarter"
  33. return
  34. case "年度":
  35. frequencyEn = "year"
  36. return
  37. }
  38. return
  39. }
  40. func GetLeadUnitEn(unit string) (unitEn string) {
  41. switch unit {
  42. case "天":
  43. unitEn = "day"
  44. return
  45. case "周":
  46. unitEn = "week"
  47. return
  48. case "月":
  49. unitEn = "month"
  50. return
  51. case "季":
  52. unitEn = "quarter"
  53. return
  54. case "年":
  55. unitEn = "year"
  56. return
  57. }
  58. return
  59. }
  60. // GetChartEdbData 获取图表的指标数据
  61. func GetChartEdbData(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*models.ChartEdbInfoMapping, extraConfigStr, seasonExtraConfig string) (edbList []*models.ChartEdbInfoMapping, xEdbIdValue []int, yDataList []models.YData, dataResp interface{}, err error, errMsg string) {
  62. edbList = make([]*models.ChartEdbInfoMapping, 0)
  63. // 指标对应的所有数据
  64. xEdbIdValue = make([]int, 0)
  65. yDataList = make([]models.YData, 0)
  66. var extraConfig interface{}
  67. switch chartType {
  68. case 7: // 柱形图
  69. var barConfig data_manage.BarChartInfoReq
  70. if extraConfigStr == `` {
  71. errMsg = "柱方图未配置"
  72. err = errors.New(errMsg)
  73. return
  74. }
  75. err = json.Unmarshal([]byte(extraConfigStr), &barConfig)
  76. if err != nil {
  77. errMsg = "柱方图配置异常"
  78. err = errors.New(errMsg)
  79. return
  80. }
  81. extraConfig = barConfig
  82. case 10: // 截面散点图
  83. var tmpExtraConfig data_manage.SectionScatterReq
  84. if extraConfigStr == `` {
  85. errMsg = "截面散点图未配置"
  86. err = errors.New(errMsg)
  87. return
  88. }
  89. err = json.Unmarshal([]byte(extraConfigStr), &tmpExtraConfig)
  90. if err != nil {
  91. errMsg = "截面散点配置异常"
  92. err = errors.New(errMsg)
  93. return
  94. }
  95. extraConfig = tmpExtraConfig
  96. default:
  97. xEdbIdValue = make([]int, 0)
  98. yDataList = make([]models.YData, 0)
  99. }
  100. // 指标对应的所有数据
  101. edbDataListMap, edbList, err := getEdbDataMapList(chartInfoId, chartType, calendar, startDate, endDate, mappingList, seasonExtraConfig)
  102. if err != nil {
  103. return
  104. }
  105. // 特殊图形数据处理
  106. switch chartType {
  107. case 7: // 柱形图
  108. barChartConf := extraConfig.(data_manage.BarChartInfoReq)
  109. xEdbIdValue, yDataList, err = BarChartData(mappingList, edbDataListMap, barChartConf.DateList, barChartConf.Sort)
  110. for _, v := range edbList {
  111. // 指标别名
  112. if barChartConf.EdbInfoIdList != nil && len(barChartConf.EdbInfoIdList) > 0 {
  113. for _, reqEdb := range barChartConf.EdbInfoIdList {
  114. if v.EdbInfoId == reqEdb.EdbInfoId {
  115. v.EdbAliasName = reqEdb.Name
  116. }
  117. }
  118. }
  119. }
  120. case 10: // 截面散点图
  121. sectionScatterConf := extraConfig.(data_manage.SectionScatterReq)
  122. xEdbIdValue, dataResp, err = GetSectionScatterChartData(mappingList, edbDataListMap, sectionScatterConf)
  123. // 这个数据没有必要返回给前端
  124. for _, v := range edbList {
  125. v.DataList = nil
  126. }
  127. }
  128. return
  129. }
  130. // GetEdbDataMapList 获取指标最后的基础数据
  131. func GetEdbDataMapList(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*models.ChartEdbInfoMapping, seasonExtraConfig string) (edbDataListMap map[int][]*models.EdbDataList, edbList []*models.ChartEdbInfoMapping, err error) {
  132. edbDataListMap, edbList, err = getEdbDataMapList(chartInfoId, chartType, calendar, startDate, endDate, mappingList, seasonExtraConfig)
  133. return
  134. }
  135. // getEdbDataMapList 获取指标最后的基础数据
  136. func getEdbDataMapList(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*models.ChartEdbInfoMapping, seasonExtraConfig string) (edbDataListMap map[int][]*models.EdbDataList, edbList []*models.ChartEdbInfoMapping, err error) {
  137. // 指标对应的所有数据
  138. edbDataListMap = make(map[int][]*models.EdbDataList)
  139. for _, v := range mappingList {
  140. //fmt.Println("v:", v.EdbInfoId)
  141. item := new(models.ChartEdbInfoMapping)
  142. item.EdbInfoId = v.EdbInfoId
  143. item.SourceName = v.SourceName
  144. item.Source = v.Source
  145. item.EdbCode = v.EdbCode
  146. item.EdbName = v.EdbName
  147. item.EdbType = v.EdbType
  148. item.EdbNameEn = v.EdbNameEn
  149. item.Frequency = v.Frequency
  150. item.FrequencyEn = GetFrequencyEn(v.Frequency)
  151. if v.Unit != `无` {
  152. item.Unit = v.Unit
  153. }
  154. item.UnitEn = v.UnitEn
  155. item.StartDate = v.StartDate
  156. item.EndDate = v.EndDate
  157. item.ModifyTime = v.ModifyTime
  158. item.EdbInfoCategoryType = v.EdbInfoCategoryType
  159. item.PredictChartColor = v.PredictChartColor
  160. if chartInfoId <= 0 {
  161. item.IsAxis = 1
  162. item.LeadValue = 0
  163. item.LeadUnit = ""
  164. //item.ChartEdbMappingId = 0
  165. item.ChartInfoId = 0
  166. item.IsOrder = false
  167. item.EdbInfoType = 1
  168. item.ChartStyle = ""
  169. item.ChartColor = ""
  170. item.ChartWidth = 0
  171. item.MaxData = v.MaxValue
  172. item.MinData = v.MinValue
  173. } else {
  174. item.IsAxis = v.IsAxis
  175. item.EdbInfoType = v.EdbInfoType
  176. item.LeadValue = v.LeadValue
  177. item.LeadUnit = v.LeadUnit
  178. item.LeadUnitEn = GetLeadUnitEn(v.LeadUnit)
  179. //item.ChartEdbMappingId = v.ChartEdbMappingId
  180. item.ChartInfoId = v.ChartInfoId
  181. item.ChartStyle = v.ChartStyle
  182. item.ChartColor = v.ChartColor
  183. item.ChartWidth = v.ChartWidth
  184. item.IsOrder = v.IsOrder
  185. item.MaxData = v.MaxData
  186. item.MinData = v.MinData
  187. }
  188. item.LatestValue = v.LatestValue
  189. item.LatestDate = v.LatestDate
  190. item.UniqueCode = v.UniqueCode
  191. item.MoveLatestDate = v.LatestDate
  192. var startDateReal string
  193. var diffSeconds int64
  194. if chartType == 2 { //季节性图
  195. startDateReal = startDate
  196. } else {
  197. if v.EdbInfoType == 0 && v.LeadUnit != "" && v.LeadValue > 0 { //领先指标
  198. var startTimeRealTemp time.Time
  199. startDateParse, _ := time.Parse(utils.FormatDate, startDate)
  200. switch v.LeadUnit {
  201. case "天":
  202. startTimeRealTemp = startDateParse.AddDate(0, 0, -v.LeadValue)
  203. case "月":
  204. startTimeRealTemp = startDateParse.AddDate(0, -v.LeadValue, 0)
  205. case "季":
  206. startTimeRealTemp = startDateParse.AddDate(0, -3*v.LeadValue, 0)
  207. case "周":
  208. startTimeRealTemp = startDateParse.AddDate(0, 0, -7*v.LeadValue)
  209. case "年":
  210. startTimeRealTemp = startDateParse.AddDate(-v.LeadValue, 0, 0)
  211. }
  212. if startTimeRealTemp.Before(startDateParse) {
  213. startDateReal = startTimeRealTemp.Format(utils.FormatDate)
  214. diffSeconds = (int64(startTimeRealTemp.UnixNano()) - int64(startDateParse.UnixNano())) / 1e6
  215. } else {
  216. startDateReal = startDate
  217. diffSeconds = 0
  218. }
  219. // 预测指标的开始日期也要偏移
  220. {
  221. day, tmpErr := utils.GetDaysBetween2Date(utils.FormatDate, startDate, startDateReal)
  222. if tmpErr != nil {
  223. err = tmpErr
  224. return
  225. }
  226. moveLatestDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, item.MoveLatestDate, time.Local)
  227. if tmpErr != nil {
  228. err = tmpErr
  229. return
  230. }
  231. item.MoveLatestDate = moveLatestDateTime.AddDate(0, 0, day).Format(utils.FormatDate)
  232. }
  233. } else {
  234. startDateReal = startDate
  235. }
  236. }
  237. //fmt.Println("line 1011 chart:", v.Source, v.EdbInfoId, startDateReal, endDate)
  238. calendarPreYear := 0
  239. if calendar == "农历" {
  240. newStartDateReal, err := time.Parse(utils.FormatDate, startDateReal)
  241. if err != nil {
  242. fmt.Println("time.Parse:" + err.Error())
  243. }
  244. calendarPreYear = newStartDateReal.Year() - 1
  245. newStartDateReal = newStartDateReal.AddDate(-1, 0, 0)
  246. startDateReal = newStartDateReal.Format(utils.FormatDate)
  247. }
  248. dataList := make([]*models.EdbDataList, 0)
  249. //fmt.Println("chart:", v.Source, v.EdbInfoId, startDateReal, endDate)
  250. switch v.EdbInfoCategoryType {
  251. case 0:
  252. dataList, err = models.GetEdbDataList(v.Source, v.SubSource, v.EdbInfoId, startDateReal, endDate)
  253. case 1:
  254. _, dataList, _, _, err, _ = GetPredictDataListByPredictEdbInfoId(v.EdbInfoId, startDateReal, endDate, false)
  255. default:
  256. err = errors.New(fmt.Sprint("获取失败,指标类型异常", v.EdbInfoCategoryType))
  257. }
  258. if err != nil {
  259. return
  260. }
  261. edbDataListMap[v.EdbInfoId] = dataList
  262. if diffSeconds != 0 && v.EdbInfoType == 0 {
  263. dataListLen := len(dataList)
  264. for i := 0; i < dataListLen; i++ {
  265. dataList[i].DataTimestamp = dataList[i].DataTimestamp - diffSeconds
  266. }
  267. }
  268. if chartType == 2 {
  269. latestDate, tmpErr := time.Parse(utils.FormatDate, v.LatestDate)
  270. if tmpErr != nil {
  271. err = errors.New(fmt.Sprint("获取最后实际数据的日期失败,Err:" + tmpErr.Error() + ";LatestDate:" + v.LatestDate))
  272. return
  273. }
  274. if calendar == "农历" {
  275. if len(dataList) <= 0 {
  276. result := new(models.EdbDataResult)
  277. item.DataList = result
  278. } else {
  279. result, tmpErr := models.AddCalculateQuarterV6(dataList)
  280. if tmpErr != nil {
  281. err = errors.New("获取农历数据失败,Err:" + tmpErr.Error())
  282. return
  283. }
  284. quarterDataList, tErr := GetSeasonEdbInfoDataListByXDateNong(result, latestDate, seasonExtraConfig, calendarPreYear)
  285. if tErr != nil {
  286. err = errors.New("获取季节性图表数据失败,Err:" + tErr.Error())
  287. return
  288. }
  289. item.DataList = quarterDataList
  290. }
  291. } else {
  292. quarterDataList, tErr := GetSeasonEdbInfoDataListByXDate(dataList, latestDate, seasonExtraConfig)
  293. if tErr != nil {
  294. err = errors.New("获取季节性图表数据失败,Err:" + tErr.Error())
  295. return
  296. }
  297. item.DataList = quarterDataList
  298. }
  299. } else if chartType == 7 { //柱方图
  300. //item.DataList = dataList
  301. } else {
  302. item.DataList = dataList
  303. }
  304. edbList = append(edbList, item)
  305. }
  306. return
  307. }
  308. // GetSeasonEdbInfoDataListByXDate 季节性图的指标数据根据横轴展示
  309. func GetSeasonEdbInfoDataListByXDate(dataList []*models.EdbDataList, latestDate time.Time, seasonExtraConfig string) (quarterDataListSort models.QuarterDataList, err error) {
  310. xStartDate := "01-01"
  311. xEndDate := "12-31"
  312. jumpYear := 0
  313. legends := make([]models.SeasonChartLegend, 0)
  314. var seasonExtra models.SeasonExtraItem
  315. if seasonExtraConfig != "" {
  316. err = json.Unmarshal([]byte(seasonExtraConfig), &seasonExtra)
  317. if err != nil {
  318. return
  319. }
  320. }
  321. if seasonExtra.XStartDate != "" {
  322. xStartDate = seasonExtra.XStartDate
  323. xEndDate = seasonExtra.XEndDate
  324. jumpYear = seasonExtra.JumpYear
  325. legends = seasonExtra.ChartLegend
  326. }
  327. length := len(dataList)
  328. if length == 0 {
  329. return
  330. }
  331. legendMap := make(map[string]string, 0)
  332. if len(legends) > 0 {
  333. for _, v := range legends {
  334. legendMap[v.Name] = v.Value
  335. }
  336. }
  337. latestDateStr := latestDate.Format(utils.FormatDate)
  338. //判断横轴的两个时间之间是不是跨年了,如果跨年了,则横轴截止年份比起始年份+1,如果不跨年,截止年份等于起始年份
  339. //根据数据确定最早的年份,和最近年份
  340. //根据横轴的日期,汇总所有的年份
  341. startDate := dataList[0].DataTime
  342. startDateT, tmpErr := time.Parse(utils.FormatDate, startDate)
  343. if tmpErr != nil {
  344. err = tmpErr
  345. return
  346. }
  347. startYear := startDateT.Year()
  348. //获取数据的最新日期
  349. lastDate := dataList[length-1].DataTime
  350. lastDateT, tmpErr := time.Parse(utils.FormatDate, lastDate)
  351. if tmpErr != nil {
  352. err = tmpErr
  353. return
  354. }
  355. endYear := lastDateT.Year()
  356. nowYear := time.Now().Year()
  357. dataMap := make(map[string]models.QuarterXDateItem, 0)
  358. quarterDataList := make([]*models.QuarterData, 0)
  359. quarterMap := make(map[string][]*models.EdbDataList, 0)
  360. //整理出日期
  361. idx := 1
  362. chartLegendMap := make(map[string]int, 0)
  363. for currentStartYear := startYear; currentStartYear <= endYear; currentStartYear++ {
  364. startStr := fmt.Sprintf("%d-%s", currentStartYear, xStartDate)
  365. currentEndYear := currentStartYear
  366. if jumpYear == 1 {
  367. currentEndYear = currentStartYear + 1
  368. }
  369. endStr := fmt.Sprintf("%d-%s", currentEndYear, xEndDate)
  370. name := fmt.Sprintf("%s_%s", startStr, endStr)
  371. showName := fmt.Sprintf("%d_%d", currentStartYear, currentEndYear)
  372. startT, tEr := time.Parse(utils.FormatDate, startStr)
  373. if tEr != nil {
  374. err = tEr
  375. return
  376. }
  377. endT, tEr := time.Parse(utils.FormatDate, endStr)
  378. if tEr != nil {
  379. err = tEr
  380. return
  381. }
  382. if lastDateT.Before(startT) {
  383. //如果最新的日期在起始日之前,则跳出循环
  384. break
  385. }
  386. if endT.Year() > nowYear {
  387. //如果最新的日期比真实年份要大,则数据全部按照最大的年份补齐
  388. nowYear = endT.Year()
  389. }
  390. item := models.QuarterXDateItem{
  391. StartDate: startT,
  392. EndDate: endT,
  393. ShowName: showName,
  394. }
  395. dataMap[name] = item
  396. chartLegendMap[name] = idx
  397. idx++
  398. if lastDateT.Before(endT) {
  399. //如果最新的日期在起始日之前,则跳出循环
  400. break
  401. }
  402. }
  403. lenYear := len(dataMap)
  404. for k, v := range dataMap {
  405. if i, ok := chartLegendMap[k]; ok {
  406. v.ChartLegend = strconv.Itoa(endYear - lenYear + i)
  407. }
  408. dataMap[k] = v
  409. }
  410. for _, v := range dataList {
  411. dataTimeT, _ := time.Parse(utils.FormatDate, v.DataTime)
  412. year := dataTimeT.Year()
  413. newItemDate := dataTimeT.AddDate(nowYear-year, 0, 0)
  414. for k, dateItem := range dataMap {
  415. tmpVal := models.EdbDataList{
  416. EdbDataId: v.EdbDataId,
  417. EdbInfoId: v.EdbInfoId,
  418. DataTime: v.DataTime,
  419. DataTimestamp: v.DataTimestamp,
  420. Value: v.Value,
  421. }
  422. if (dateItem.StartDate.Before(dataTimeT) && dateItem.EndDate.After(dataTimeT)) || dateItem.StartDate == dataTimeT || dateItem.EndDate == dataTimeT {
  423. if jumpYear == 1 {
  424. //计算前一年最大的日期, 只补齐数据到去年
  425. beforeYearMaxDate := fmt.Sprintf("%d-12-31", dateItem.StartDate.Year())
  426. beforeYearMaxDateT, _ := time.Parse(utils.FormatDate, beforeYearMaxDate)
  427. if dataTimeT.Before(beforeYearMaxDateT) || dataTimeT == beforeYearMaxDateT {
  428. newItemDate = dataTimeT.AddDate(nowYear-year-1, 0, 0)
  429. } else {
  430. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  431. }
  432. } else {
  433. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  434. }
  435. timestamp := newItemDate.UnixNano() / 1e6
  436. tmpVal.DataTimestamp = timestamp
  437. tmpV := &tmpVal
  438. if findVal, ok := quarterMap[k]; !ok {
  439. findVal = append(findVal, tmpV)
  440. quarterMap[k] = findVal
  441. } else {
  442. findVal = append(findVal, tmpV)
  443. quarterMap[k] = findVal
  444. }
  445. if v.DataTime == latestDateStr {
  446. dateItem.CuttingDataTimestamp = timestamp
  447. dataMap[k] = dateItem
  448. }
  449. //break
  450. }
  451. }
  452. }
  453. for k, v := range dataMap {
  454. itemList := quarterMap[k]
  455. quarterItem := new(models.QuarterData)
  456. quarterItem.Years = v.ShowName
  457. quarterItem.Year = v.ChartLegend
  458. quarterItem.ChartLegend = v.ChartLegend
  459. if le, ok := legendMap[v.ShowName]; ok {
  460. if le != strconv.Itoa(v.StartDate.Year()) && le != strconv.Itoa(v.EndDate.Year()) {
  461. quarterItem.ChartLegend = le
  462. }
  463. }
  464. quarterItem.DataList = itemList
  465. quarterItem.CuttingDataTimestamp = v.CuttingDataTimestamp
  466. //如果等于最后的实际日期,那么将切割时间戳记录
  467. if quarterItem.CuttingDataTimestamp == 0 {
  468. //如果大于最后的实际日期,那么第一个点就是切割的时间戳
  469. if latestDate.Before(v.StartDate) && len(itemList) > 0 {
  470. quarterItem.CuttingDataTimestamp = itemList[0].DataTimestamp - 100
  471. }
  472. }
  473. quarterDataList = append(quarterDataList, quarterItem)
  474. }
  475. if len(quarterDataList) > 0 {
  476. quarterDataListSort = quarterDataList
  477. sort.Sort(quarterDataListSort)
  478. }
  479. return
  480. }
  481. // GetSeasonEdbInfoDataListByXDateNong 季节性图的指标数据根据横轴选择农历时展示
  482. func GetSeasonEdbInfoDataListByXDateNong(result *models.EdbDataResult, latestDate time.Time, seasonExtraConfig string, calendarPreYear int) (quarterDataListSort models.QuarterDataList, err error) {
  483. xStartDate := "01-01"
  484. xEndDate := "12-31"
  485. jumpYear := 0
  486. legends := make([]models.SeasonChartLegend, 0)
  487. var seasonExtra models.SeasonExtraItem
  488. if seasonExtraConfig != "" {
  489. err = json.Unmarshal([]byte(seasonExtraConfig), &seasonExtra)
  490. if err != nil {
  491. return
  492. }
  493. }
  494. if seasonExtra.XStartDate != "" {
  495. xStartDate = seasonExtra.XStartDate
  496. xEndDate = seasonExtra.XEndDate
  497. jumpYear = seasonExtra.JumpYear
  498. legends = seasonExtra.ChartLegend
  499. }
  500. length := len(result.List)
  501. if length == 0 {
  502. return
  503. }
  504. legendMap := make(map[string]string, 0)
  505. if len(legends) > 0 {
  506. for _, v := range legends {
  507. legendMap[v.Name] = v.Value
  508. }
  509. }
  510. latestDateYear := latestDate.Year()
  511. //判断横轴的两个时间之间是不是跨年了,如果跨年了,则横轴截止年份比起始年份+1,如果不跨年,截止年份等于起始年份
  512. //根据数据确定最早的年份,和最近年份
  513. //根据横轴的日期,汇总所有的年份
  514. startYear := result.List[0].Year
  515. /*if jumpYear == 1 {
  516. if startYear != calendarPreYear {
  517. startYear = startYear - 1
  518. }
  519. }*/
  520. itemLength := len(result.List[length-1].Items)
  521. //获取数据的最新日期
  522. lastDate := result.List[length-1].Items[itemLength-1].DataTime
  523. lastDateT, tmpErr := time.Parse(utils.FormatDate, lastDate)
  524. if tmpErr != nil {
  525. err = tmpErr
  526. return
  527. }
  528. endYear := lastDateT.Year()
  529. nowYear := time.Now().Year()
  530. dataMap := make(map[string]models.QuarterXDateItem, 0)
  531. quarterDataList := make([]*models.QuarterData, 0)
  532. resultData := make([]*models.QuarterData, 0)
  533. quarterMap := make(map[string][]*models.EdbDataList, 0)
  534. //整理出日期
  535. idx := 1
  536. chartLegendMap := make(map[string]int, 0)
  537. for currentStartYear := startYear; currentStartYear <= endYear; currentStartYear++ {
  538. startStr := fmt.Sprintf("%d-%s", currentStartYear, xStartDate)
  539. currentEndYear := currentStartYear
  540. if jumpYear == 1 {
  541. currentEndYear = currentStartYear + 1
  542. }
  543. endStr := fmt.Sprintf("%d-%s", currentEndYear, xEndDate)
  544. showName := fmt.Sprintf("%d_%d", currentStartYear, currentEndYear)
  545. startT, tEr := time.Parse(utils.FormatDate, startStr)
  546. if tEr != nil {
  547. err = tEr
  548. return
  549. }
  550. endT, tEr := time.Parse(utils.FormatDate, endStr)
  551. if tEr != nil {
  552. err = tEr
  553. return
  554. }
  555. if lastDateT.Before(startT) {
  556. //如果最新的日期在起始日之前,则跳出循环
  557. break
  558. }
  559. if endT.Year() > nowYear {
  560. //如果最新的日期比真实年份要大,则数据全部按照最大的年份补齐
  561. nowYear = endT.Year()
  562. }
  563. item := models.QuarterXDateItem{
  564. StartDate: startT,
  565. EndDate: endT,
  566. ShowName: showName,
  567. }
  568. dataMap[showName] = item
  569. fmt.Println("年份" + showName + "日期" + startStr + " " + endStr)
  570. chartLegendMap[showName] = idx
  571. idx++
  572. if lastDateT.Before(endT) {
  573. //如果最新的日期在起始日之前,则跳出循环
  574. break
  575. }
  576. }
  577. lenYear := len(dataMap)
  578. for k, v := range dataMap {
  579. if i, ok := chartLegendMap[k]; ok {
  580. v.ChartLegend = strconv.Itoa(endYear - lenYear + i)
  581. }
  582. dataMap[k] = v
  583. }
  584. yearDataListMap := make(map[int]*models.EdbDataItems, 0)
  585. for _, lv := range result.List {
  586. yearDataListMap[lv.Year] = lv
  587. }
  588. //判断哪些点应该落在同一条时间线上
  589. maxY := lastDateT.Year()
  590. changeFlag := false
  591. if lastDateT.Month() >= 11 {
  592. maxY = maxY + 1
  593. }
  594. if maxY < nowYear {
  595. changeFlag = true
  596. maxY = nowYear
  597. }
  598. endTmp := fmt.Sprintf("%d-%s", maxY, xEndDate)
  599. endTmpT, _ := time.Parse(utils.FormatDate, endTmp)
  600. minY := maxY
  601. if jumpYear == 1 {
  602. minY = maxY - 1
  603. }
  604. startTmp := fmt.Sprintf("%d-%s", minY, xStartDate)
  605. startTmpT, _ := time.Parse(utils.FormatDate, startTmp)
  606. fmt.Println("横轴截取日" + startTmpT.Format(utils.FormatDate) + " " + endTmpT.Format(utils.FormatDate))
  607. for name, dateItem := range dataMap {
  608. lv, ok1 := yearDataListMap[dateItem.EndDate.Year()]
  609. if changeFlag {
  610. lv, ok1 = yearDataListMap[dateItem.StartDate.Year()]
  611. }
  612. if !ok1 {
  613. continue
  614. }
  615. for _, item := range lv.Items {
  616. tmpVal := models.EdbDataList{
  617. EdbDataId: item.EdbDataId,
  618. EdbInfoId: item.EdbInfoId,
  619. DataTime: item.DataTime,
  620. DataTimestamp: item.DataTimestamp,
  621. Value: item.Value,
  622. }
  623. dataTimeT, _ := time.Parse(utils.FormatDate, item.DataTime)
  624. if (startTmpT.Before(dataTimeT) && endTmpT.After(dataTimeT)) || startTmpT == dataTimeT || endTmpT == dataTimeT {
  625. tmpV := &tmpVal
  626. if findVal, ok := quarterMap[name]; !ok {
  627. findVal = append(findVal, tmpV)
  628. quarterMap[name] = findVal
  629. } else {
  630. findVal = append(findVal, tmpV)
  631. quarterMap[name] = findVal
  632. }
  633. if lv.Year >= latestDateYear {
  634. // 切割的日期时间字符串
  635. cuttingDataTimeStr := latestDate.AddDate(0, 0, lv.BetweenDay).Format(utils.FormatDate)
  636. if item.DataTime == cuttingDataTimeStr {
  637. dateItem.CuttingDataTimestamp = tmpVal.DataTimestamp
  638. dataMap[name] = dateItem
  639. }
  640. }
  641. }
  642. }
  643. }
  644. for k, v := range dataMap {
  645. itemList := quarterMap[k]
  646. quarterItem := new(models.QuarterData)
  647. quarterItem.Years = v.ShowName
  648. quarterItem.Year = v.ChartLegend
  649. quarterItem.ChartLegend = v.ChartLegend
  650. if le, ok := legendMap[v.ShowName]; ok {
  651. if le != strconv.Itoa(v.StartDate.Year()) && le != strconv.Itoa(v.EndDate.Year()) {
  652. quarterItem.ChartLegend = le
  653. }
  654. }
  655. quarterItem.DataList = itemList
  656. quarterItem.CuttingDataTimestamp = v.CuttingDataTimestamp
  657. //如果等于最后的实际日期,那么将切割时间戳记录
  658. if quarterItem.CuttingDataTimestamp == 0 {
  659. //如果大于最后的实际日期,那么第一个点就是切割的时间戳
  660. if latestDate.Before(v.StartDate) && len(itemList) > 0 {
  661. quarterItem.CuttingDataTimestamp = itemList[0].DataTimestamp - 100
  662. }
  663. }
  664. quarterDataList = append(quarterDataList, quarterItem)
  665. }
  666. if result.List[0].Year != calendarPreYear {
  667. itemList := make([]*models.EdbDataList, 0)
  668. items := new(models.QuarterData)
  669. //items.Year = calendarPreYear
  670. items.DataList = itemList
  671. newResult := make([]*models.QuarterData, 0)
  672. newResult = append(newResult, items)
  673. newResult = append(newResult, quarterDataList...)
  674. resultData = newResult
  675. } else {
  676. resultData = quarterDataList
  677. }
  678. if len(quarterDataList) > 0 {
  679. quarterDataListSort = resultData
  680. sort.Sort(quarterDataListSort)
  681. }
  682. return
  683. }
  684. // BarChartData 柱方图的数据处理
  685. func BarChartData(mappingList []*models.ChartEdbInfoMapping, edbDataListMap map[int][]*models.EdbDataList, barChartInfoDateList []data_manage.BarChartInfoDateReq, barChartInfoSort data_manage.BarChartInfoSortReq) (edbIdList []int, yDataList []models.YData, err error) {
  686. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  687. edbDataMap := make(map[int]map[string]float64)
  688. for edbInfoId, edbDataList := range edbDataListMap {
  689. edbDateData := make(map[string]float64)
  690. for _, edbData := range edbDataList {
  691. edbDateData[edbData.DataTime] = edbData.Value
  692. }
  693. edbDataMap[edbInfoId] = edbDateData
  694. }
  695. // edbIdList 指标展示顺序;x轴的指标顺序
  696. edbIdList = make([]int, 0)
  697. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  698. dateData := make(map[int]float64)
  699. if barChartInfoSort.Sort == 0 {
  700. for _, v := range mappingList {
  701. edbIdList = append(edbIdList, v.EdbInfoId)
  702. }
  703. } else {
  704. lenBarChartInfoDateList := len(barChartInfoDateList)
  705. if barChartInfoSort.DateIndex >= lenBarChartInfoDateList {
  706. err = errors.New("排序日期异常")
  707. return
  708. }
  709. notDataEdbIdList := make([]int, 0) //没有数据的指标id
  710. // 日期配置
  711. barChartInfoDate := barChartInfoDateList[barChartInfoSort.DateIndex]
  712. for edbInfoId, dataList := range edbDataListMap {
  713. if len(dataList) <= 0 {
  714. // 没有数据的指标id
  715. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  716. continue
  717. }
  718. findDate := barChartInfoDate.Date
  719. switch barChartInfoDate.Type {
  720. case 1: //最新值
  721. findDate = dataList[len(dataList)-1].DataTime
  722. case 2: //近期几天
  723. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  724. if tmpErr != nil {
  725. err = tmpErr
  726. return
  727. }
  728. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  729. lenData := len(dataList) - 1
  730. for i := lenData; i >= 0; i-- {
  731. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  732. if tmpErr != nil {
  733. err = tmpErr
  734. return
  735. }
  736. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  737. findDate = dataList[i].DataTime
  738. break
  739. }
  740. }
  741. case 3: // 固定日期
  742. //最早的日期
  743. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  744. if tmpErr != nil {
  745. err = tmpErr
  746. return
  747. }
  748. //寻找固定日期的数据
  749. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  750. if tmpErr != nil {
  751. err = tmpErr
  752. return
  753. }
  754. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  755. tmpDate := tmpDateTime.Format(utils.FormatDate)
  756. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  757. findDate = tmpDate
  758. break
  759. }
  760. }
  761. default:
  762. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  763. return
  764. }
  765. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  766. dateData[edbInfoId] = tmpValue
  767. } else {
  768. // 没有数据的指标id
  769. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  770. }
  771. }
  772. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  773. // 排序
  774. dateDataSort := utils.NewMapSorter(dateData)
  775. sort.Sort(dateDataSort)
  776. if barChartInfoSort.Sort == 1 {
  777. // 先将没有数据的指标id放在最前面
  778. if len(notDataEdbIdList) > 0 {
  779. edbIdList = append(edbIdList, notDataEdbIdList...)
  780. }
  781. for _, v := range dateDataSort {
  782. edbIdList = append(edbIdList, v.Key)
  783. }
  784. } else {
  785. for i := len(dateDataSort) - 1; i >= 0; i-- {
  786. edbIdList = append(edbIdList, dateDataSort[i].Key)
  787. }
  788. // 再将没有数据的指标id放在最后面
  789. if len(notDataEdbIdList) > 0 {
  790. edbIdList = append(edbIdList, notDataEdbIdList...)
  791. }
  792. }
  793. }
  794. yDataList = make([]models.YData, 0) //y轴的数据列表
  795. for _, barChartInfoDate := range barChartInfoDateList {
  796. var maxDate time.Time
  797. findDataList := make([]float64, 0) // 当前日期的数据值
  798. for _, edbInfoId := range edbIdList {
  799. findDate := barChartInfoDate.Date //需要的日期值
  800. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  801. if len(dataList) <= 0 {
  802. // 没有数据的指标id
  803. findDataList = append(findDataList, 0)
  804. continue
  805. }
  806. switch barChartInfoDate.Type {
  807. case 1: //最新值
  808. dataList := edbDataListMap[edbInfoId]
  809. findDate = dataList[len(dataList)-1].DataTime
  810. case 2: //近期几天
  811. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  812. if tmpErr != nil {
  813. err = tmpErr
  814. return
  815. }
  816. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  817. lenData := len(dataList) - 1
  818. for i := lenData; i >= 0; i-- {
  819. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  820. if tmpErr != nil {
  821. err = tmpErr
  822. return
  823. }
  824. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  825. findDate = dataList[i].DataTime
  826. break
  827. }
  828. }
  829. case 3: // 固定日期
  830. //最早的日期
  831. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  832. if tmpErr != nil {
  833. err = tmpErr
  834. return
  835. }
  836. //寻找固定日期的数据
  837. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  838. if tmpErr != nil {
  839. err = tmpErr
  840. return
  841. }
  842. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  843. tmpDate := tmpDateTime.Format(utils.FormatDate)
  844. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  845. findDate = tmpDate
  846. break
  847. }
  848. }
  849. default:
  850. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  851. return
  852. }
  853. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  854. if maxDate.IsZero() {
  855. maxDate = findDateTime
  856. } else {
  857. if findDateTime.After(maxDate) {
  858. maxDate = findDateTime
  859. }
  860. }
  861. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  862. findDataList = append(findDataList, tmpValue)
  863. } else {
  864. findDataList = append(findDataList, 0)
  865. }
  866. }
  867. yDate := "0000-00-00"
  868. if !maxDate.IsZero() {
  869. yDate = maxDate.Format(utils.FormatDate)
  870. }
  871. yDataList = append(yDataList, models.YData{
  872. Date: yDate,
  873. Value: findDataList,
  874. Color: barChartInfoDate.Color,
  875. Name: barChartInfoDate.Name,
  876. })
  877. }
  878. return
  879. }
  880. // ChartInfoRefreshV2 图表刷新
  881. // @author Roc
  882. // @datetime 2022-09-16 10:15:38
  883. // @description 将原来自己写的一套获取所有关联指标,然后刷新指标逻辑 改成 只获取使用的指标id,然后遍历去调用“指标刷新服务”
  884. func ChartInfoRefreshV2(chartInfoId int) (err error) {
  885. var errmsg string
  886. defer func() {
  887. if err != nil {
  888. go alarm_msg.SendAlarmMsg("ChartInfoRefresh:"+errmsg, 3)
  889. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "ChartInfoRefresh:"+errmsg, utils.EmailSendToUsers)
  890. fmt.Println("ChartInfoRefresh Err:" + errmsg)
  891. }
  892. }()
  893. edbMappingList, err := data_manage.GetChartEdbMappingList(chartInfoId)
  894. if err != nil {
  895. errmsg = "获取需要刷新的指标失败:Err:" + err.Error()
  896. return
  897. }
  898. edbIdList := make([]int, 0)
  899. for _, v := range edbMappingList {
  900. edbIdList = append(edbIdList, v.EdbInfoId)
  901. }
  902. // 批量刷新
  903. err, errmsg = EdbInfoRefreshAllFromBase(edbIdList, false)
  904. if err != nil {
  905. return
  906. }
  907. return
  908. }
  909. // GetSectionScatterChartData 柱方图的数据处理
  910. func GetSectionScatterChartData(mappingList []*models.ChartEdbInfoMapping, edbDataListMap map[int][]*models.EdbDataList, extraConfig data_manage.SectionScatterReq) (edbIdList []int, chartDataResp data_manage.SectionScatterInfoResp, err error) {
  911. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  912. edbDataMap := make(map[int]map[string]float64)
  913. for edbInfoId, edbDataList := range edbDataListMap {
  914. edbDateData := make(map[string]float64)
  915. for _, edbData := range edbDataList {
  916. edbDateData[edbData.DataTime] = edbData.Value
  917. }
  918. edbDataMap[edbInfoId] = edbDateData
  919. }
  920. // edbIdList 指标展示顺序;x轴的指标顺序
  921. edbIdList = make([]int, 0)
  922. edbMappingMap := make(map[int]*models.ChartEdbInfoMapping)
  923. for _, v := range mappingList {
  924. edbIdList = append(edbIdList, v.EdbInfoId)
  925. edbMappingMap[v.EdbInfoId] = v
  926. }
  927. //SectionScatterSeriesInfoResp
  928. dataListResp := make([]data_manage.SectionScatterSeriesItemResp, 0) //y轴的数据列表
  929. for _, seriesItem := range extraConfig.SeriesList {
  930. var maxDate time.Time
  931. // 系列中的指标数据
  932. tmpSeriesEdbInfoList := make([]data_manage.SectionScatterEdbItemResp, 0)
  933. var minXVal, maxXVal, minYVal, maxYVal float64
  934. for _, edbConf := range seriesItem.EdbInfoList {
  935. tmpItem := data_manage.SectionScatterEdbItemResp{
  936. IsShow: edbConf.IsShow,
  937. Name: edbConf.Name,
  938. NameEn: edbConf.NameEn,
  939. } //单个坐标点的数据
  940. //X轴的数据
  941. {
  942. edbInfoId := edbConf.XEdbInfoId //X轴的指标
  943. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  944. if !ok {
  945. continue
  946. }
  947. findDate := edbConf.XDate //需要的日期值
  948. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  949. if len(dataList) <= 0 {
  950. // 没有数据的指标id
  951. //findDataList = append(findDataList, 0)
  952. continue
  953. }
  954. tmpItem.XEdbInfoId = edbInfoId
  955. tmpItem.XName = edbMappingInfo.EdbName
  956. tmpItem.XNameEn = edbMappingInfo.EdbNameEn
  957. switch edbConf.XDateType {
  958. case 1: //最新值
  959. dataList := edbDataListMap[edbInfoId]
  960. findDate = dataList[len(dataList)-1].DataTime
  961. case 2: //近期几天
  962. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  963. if tmpErr != nil {
  964. err = tmpErr
  965. return
  966. }
  967. findDateTime = findDateTime.AddDate(0, 0, -edbConf.XDateValue)
  968. lenData := len(dataList) - 1
  969. for i := lenData; i >= 0; i-- {
  970. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  971. if tmpErr != nil {
  972. err = tmpErr
  973. return
  974. }
  975. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  976. findDate = dataList[i].DataTime
  977. break
  978. }
  979. }
  980. case 3: // 固定日期
  981. //最早的日期
  982. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  983. if tmpErr != nil {
  984. err = tmpErr
  985. return
  986. }
  987. //寻找固定日期的数据
  988. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.XDate, time.Local)
  989. if tmpErr != nil {
  990. err = tmpErr
  991. return
  992. }
  993. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  994. tmpDate := tmpDateTime.Format(utils.FormatDate)
  995. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  996. findDate = tmpDate
  997. break
  998. }
  999. }
  1000. default:
  1001. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.XDate))
  1002. return
  1003. }
  1004. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  1005. if maxDate.IsZero() {
  1006. maxDate = findDateTime
  1007. } else {
  1008. if findDateTime.After(maxDate) {
  1009. maxDate = findDateTime
  1010. }
  1011. }
  1012. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  1013. tmpItem.XDate = findDate
  1014. tmpItem.XValue = tmpValue
  1015. } else {
  1016. continue
  1017. }
  1018. }
  1019. //Y轴的数据
  1020. {
  1021. edbInfoId := edbConf.YEdbInfoId //Y轴的指标
  1022. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  1023. if !ok {
  1024. continue
  1025. }
  1026. findDate := edbConf.YDate //需要的日期值
  1027. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  1028. if len(dataList) <= 0 {
  1029. // 没有数据的指标id
  1030. //findDataList = append(findDataList, 0)
  1031. continue
  1032. }
  1033. tmpItem.YEdbInfoId = edbInfoId
  1034. tmpItem.YName = edbMappingInfo.EdbName
  1035. tmpItem.YNameEn = edbMappingInfo.EdbNameEn
  1036. switch edbConf.YDateType {
  1037. case 1: //最新值
  1038. dataList := edbDataListMap[edbInfoId]
  1039. findDate = dataList[len(dataList)-1].DataTime
  1040. case 2: //近期几天
  1041. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  1042. if tmpErr != nil {
  1043. err = tmpErr
  1044. return
  1045. }
  1046. findDateTime = findDateTime.AddDate(0, 0, -edbConf.YDateValue)
  1047. lenData := len(dataList) - 1
  1048. for i := lenData; i >= 0; i-- {
  1049. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  1050. if tmpErr != nil {
  1051. err = tmpErr
  1052. return
  1053. }
  1054. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  1055. findDate = dataList[i].DataTime
  1056. break
  1057. }
  1058. }
  1059. case 3: // 固定日期
  1060. //最早的日期
  1061. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  1062. if tmpErr != nil {
  1063. err = tmpErr
  1064. return
  1065. }
  1066. //寻找固定日期的数据
  1067. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.YDate, time.Local)
  1068. if tmpErr != nil {
  1069. err = tmpErr
  1070. return
  1071. }
  1072. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  1073. tmpDate := tmpDateTime.Format(utils.FormatDate)
  1074. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  1075. findDate = tmpDate
  1076. break
  1077. }
  1078. }
  1079. default:
  1080. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.YDate))
  1081. return
  1082. }
  1083. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  1084. if maxDate.IsZero() {
  1085. maxDate = findDateTime
  1086. } else {
  1087. if findDateTime.After(maxDate) {
  1088. maxDate = findDateTime
  1089. }
  1090. }
  1091. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  1092. tmpItem.YDate = findDate
  1093. tmpItem.YValue = tmpValue
  1094. } else {
  1095. continue
  1096. }
  1097. }
  1098. // 获取当前系列的X轴的最大最小值
  1099. {
  1100. if tmpItem.XValue < minXVal {
  1101. minXVal = tmpItem.XValue
  1102. }
  1103. if tmpItem.XValue > maxXVal {
  1104. maxXVal = tmpItem.XValue
  1105. }
  1106. if tmpItem.YValue < minYVal {
  1107. minYVal = tmpItem.YValue
  1108. }
  1109. if tmpItem.YValue > maxYVal {
  1110. maxYVal = tmpItem.YValue
  1111. }
  1112. }
  1113. tmpSeriesEdbInfoList = append(tmpSeriesEdbInfoList, tmpItem)
  1114. }
  1115. trendLimitData := make([]data_manage.CoordinatePoint, 0) //趋势线的前后坐标点
  1116. var trendLine, rSquare string
  1117. // 生成线性方程式
  1118. var a, b float64
  1119. {
  1120. coordinateData := make([]utils.Coordinate, 0)
  1121. for _, tmpSeriesEdbInfo := range tmpSeriesEdbInfoList {
  1122. tmpCoordinate1 := utils.Coordinate{
  1123. X: tmpSeriesEdbInfo.XValue,
  1124. Y: tmpSeriesEdbInfo.YValue,
  1125. }
  1126. coordinateData = append(coordinateData, tmpCoordinate1)
  1127. }
  1128. // 只有存在两个坐标点的时候,才能去计算线性方程和R平方
  1129. if len(coordinateData) >= 2 {
  1130. a, b = utils.GetLinearResult(coordinateData)
  1131. if !math.IsNaN(a) || !math.IsNaN(b) {
  1132. if b > 0 {
  1133. trendLine = fmt.Sprintf("y=%sx+%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  1134. } else {
  1135. trendLine = fmt.Sprintf("y=%sx%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  1136. }
  1137. minYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(minXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  1138. maxYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(maxXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  1139. }
  1140. // 计算R平方
  1141. rSquare = fmt.Sprint(utils.CalculationDecisive(coordinateData))
  1142. }
  1143. trendLimitData = append(trendLimitData, data_manage.CoordinatePoint{
  1144. X: minXVal,
  1145. Y: minYVal,
  1146. }, data_manage.CoordinatePoint{
  1147. X: maxXVal,
  1148. Y: maxYVal,
  1149. })
  1150. }
  1151. dataListResp = append(dataListResp, data_manage.SectionScatterSeriesItemResp{
  1152. Name: seriesItem.Name,
  1153. NameEn: seriesItem.NameEn,
  1154. Color: seriesItem.Color,
  1155. EdbInfoList: tmpSeriesEdbInfoList,
  1156. ShowTrendLine: seriesItem.ShowTrendLine,
  1157. ShowFitEquation: seriesItem.ShowFitEquation,
  1158. ShowRSquare: seriesItem.ShowRSquare,
  1159. TrendLine: trendLine,
  1160. RSquare: rSquare,
  1161. TrendLimitData: trendLimitData,
  1162. })
  1163. }
  1164. chartDataResp = data_manage.SectionScatterInfoResp{
  1165. XName: extraConfig.XName,
  1166. XNameEn: extraConfig.XNameEn,
  1167. XUnitName: extraConfig.XUnitName,
  1168. XUnitNameEn: extraConfig.XUnitNameEn,
  1169. YName: extraConfig.YName,
  1170. YNameEn: extraConfig.YNameEn,
  1171. YUnitName: extraConfig.YUnitName,
  1172. YUnitNameEn: extraConfig.YUnitNameEn,
  1173. XMinValue: extraConfig.XMinValue,
  1174. XMaxValue: extraConfig.XMaxValue,
  1175. YMinValue: extraConfig.YMinValue,
  1176. YMaxValue: extraConfig.YMaxValue,
  1177. DataList: dataListResp,
  1178. }
  1179. return
  1180. }