chart_info.go 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320
  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.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.AddCalculateQuarterV5(dataList)
  280. if tmpErr != nil {
  281. err = errors.New("获取农历数据失败,Err:" + tmpErr.Error())
  282. return
  283. }
  284. endDateTmp := dataList[len(dataList)-1].DataTime
  285. endDateForm, _ := time.Parse(utils.FormatDate, endDateTmp)
  286. quarterDataList, tErr := GetSeasonEdbInfoDataListByXDateNong(result, latestDate, seasonExtraConfig, calendarPreYear, int(endDateForm.Month()))
  287. if tErr != nil {
  288. err = errors.New("获取季节性图表数据失败,Err:" + tErr.Error())
  289. return
  290. }
  291. item.DataList = quarterDataList
  292. }
  293. } else {
  294. quarterDataList, tErr := GetSeasonEdbInfoDataListByXDate(dataList, latestDate, seasonExtraConfig)
  295. if tErr != nil {
  296. err = errors.New("获取季节性图表数据失败,Err:" + tErr.Error())
  297. return
  298. }
  299. item.DataList = quarterDataList
  300. }
  301. } else if chartType == 7 { //柱方图
  302. //item.DataList = dataList
  303. } else {
  304. item.DataList = dataList
  305. }
  306. edbList = append(edbList, item)
  307. }
  308. return
  309. }
  310. // GetSeasonEdbInfoDataListByXDate 季节性图的指标数据根据横轴展示
  311. func GetSeasonEdbInfoDataListByXDate(dataList []*models.EdbDataList, latestDate time.Time, seasonExtraConfig string) (quarterDataListSort models.QuarterDataList, err error) {
  312. xStartDate := "01-01"
  313. xEndDate := "12-31"
  314. jumpYear := 0
  315. legends := make([]models.SeasonChartLegend, 0)
  316. var seasonExtra models.SeasonExtraItem
  317. if seasonExtraConfig != "" {
  318. err = json.Unmarshal([]byte(seasonExtraConfig), &seasonExtra)
  319. if err != nil {
  320. return
  321. }
  322. }
  323. if seasonExtra.XStartDate != "" {
  324. xStartDate = seasonExtra.XStartDate
  325. xEndDate = seasonExtra.XEndDate
  326. jumpYear = seasonExtra.JumpYear
  327. legends = seasonExtra.ChartLegend
  328. }
  329. length := len(dataList)
  330. if length == 0 {
  331. return
  332. }
  333. legendMap := make(map[string]string, 0)
  334. if len(legends) > 0 {
  335. for _, v := range legends {
  336. legendMap[v.Name] = v.Value
  337. }
  338. }
  339. latestDateStr := latestDate.Format(utils.FormatDate)
  340. //判断横轴的两个时间之间是不是跨年了,如果跨年了,则横轴截止年份比起始年份+1,如果不跨年,截止年份等于起始年份
  341. //根据数据确定最早的年份,和最近年份
  342. //根据横轴的日期,汇总所有的年份
  343. startDate := dataList[0].DataTime
  344. startDateT, tmpErr := time.Parse(utils.FormatDate, startDate)
  345. if tmpErr != nil {
  346. err = tmpErr
  347. return
  348. }
  349. startYear := startDateT.Year()
  350. //获取数据的最新日期
  351. lastDate := dataList[length-1].DataTime
  352. lastDateT, tmpErr := time.Parse(utils.FormatDate, lastDate)
  353. if tmpErr != nil {
  354. err = tmpErr
  355. return
  356. }
  357. endYear := lastDateT.Year()
  358. nowYear := time.Now().Year()
  359. dataMap := make(map[string]models.QuarterXDateItem, 0)
  360. quarterDataList := make([]*models.QuarterData, 0)
  361. quarterMap := make(map[string][]*models.EdbDataList, 0)
  362. //整理出日期
  363. leftNameFlag := true
  364. for currentStartYear := startYear; currentStartYear <= endYear; currentStartYear++ {
  365. startStr := fmt.Sprintf("%d-%s", currentStartYear, xStartDate)
  366. currentEndYear := currentStartYear
  367. if jumpYear == 1 {
  368. currentEndYear = currentStartYear + 1
  369. }
  370. endStr := fmt.Sprintf("%d-%s", currentEndYear, xEndDate)
  371. name := fmt.Sprintf("%s_%s", startStr, endStr)
  372. showName := fmt.Sprintf("%d_%d", currentStartYear, currentEndYear)
  373. startT, tEr := time.Parse(utils.FormatDate, startStr)
  374. if tEr != nil {
  375. err = tEr
  376. return
  377. }
  378. endT, tEr := time.Parse(utils.FormatDate, endStr)
  379. if tEr != nil {
  380. err = tEr
  381. return
  382. }
  383. if lastDateT.Before(startT) {
  384. //如果最新的日期在起始日之前,则跳出循环
  385. break
  386. }
  387. if startT.Before(lastDateT) && endT.After(lastDateT) {
  388. if endYear > currentStartYear {
  389. leftNameFlag = false
  390. }
  391. }
  392. if endT.Year() > nowYear {
  393. //如果最新的日期比真实年份要大,则数据全部按照最大的年份补齐
  394. nowYear = endT.Year()
  395. }
  396. item := models.QuarterXDateItem{
  397. StartDate: startT,
  398. EndDate: endT,
  399. ShowName: showName,
  400. }
  401. dataMap[name] = item
  402. if lastDateT.Before(endT) {
  403. //如果最新的日期在起始日之前,则跳出循环
  404. break
  405. }
  406. }
  407. for k, v := range dataMap {
  408. if leftNameFlag {
  409. v.ChartLegend = strconv.Itoa(v.StartDate.Year())
  410. } else {
  411. v.ChartLegend = strconv.Itoa(v.EndDate.Year())
  412. }
  413. dataMap[k] = v
  414. }
  415. for _, v := range dataList {
  416. dataTimeT, _ := time.Parse(utils.FormatDate, v.DataTime)
  417. year := dataTimeT.Year()
  418. newItemDate := dataTimeT.AddDate(nowYear-year, 0, 0)
  419. for k, dateItem := range dataMap {
  420. tmpVal := models.EdbDataList{
  421. EdbDataId: v.EdbDataId,
  422. EdbInfoId: v.EdbInfoId,
  423. DataTime: v.DataTime,
  424. DataTimestamp: v.DataTimestamp,
  425. Value: v.Value,
  426. }
  427. if (dateItem.StartDate.Before(dataTimeT) && dateItem.EndDate.After(dataTimeT)) || dateItem.StartDate == dataTimeT || dateItem.EndDate == dataTimeT {
  428. if jumpYear == 1 {
  429. //计算前一年最大的日期, 只补齐数据到去年
  430. beforeYearMaxDate := fmt.Sprintf("%d-12-31", dateItem.StartDate.Year())
  431. beforeYearMaxDateT, _ := time.Parse(utils.FormatDate, beforeYearMaxDate)
  432. if dataTimeT.Before(beforeYearMaxDateT) || dataTimeT == beforeYearMaxDateT {
  433. newItemDate = dataTimeT.AddDate(nowYear-year-1, 0, 0)
  434. } else {
  435. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  436. }
  437. } else {
  438. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  439. }
  440. timestamp := newItemDate.UnixNano() / 1e6
  441. tmpVal.DataTimestamp = timestamp
  442. tmpV := &tmpVal
  443. if findVal, ok := quarterMap[k]; !ok {
  444. findVal = append(findVal, tmpV)
  445. quarterMap[k] = findVal
  446. } else {
  447. findVal = append(findVal, tmpV)
  448. quarterMap[k] = findVal
  449. }
  450. if v.DataTime == latestDateStr {
  451. dateItem.CuttingDataTimestamp = timestamp
  452. dataMap[k] = dateItem
  453. }
  454. //break
  455. }
  456. }
  457. }
  458. for k, v := range dataMap {
  459. itemList := quarterMap[k]
  460. quarterItem := new(models.QuarterData)
  461. quarterItem.Years = v.ShowName
  462. quarterItem.Year = v.ChartLegend
  463. quarterItem.ChartLegend = v.ChartLegend
  464. if le, ok := legendMap[v.ShowName]; ok {
  465. quarterItem.ChartLegend = le
  466. }
  467. quarterItem.DataList = itemList
  468. quarterItem.CuttingDataTimestamp = v.CuttingDataTimestamp
  469. //如果等于最后的实际日期,那么将切割时间戳记录
  470. if quarterItem.CuttingDataTimestamp == 0 {
  471. //如果大于最后的实际日期,那么第一个点就是切割的时间戳
  472. if latestDate.Before(v.StartDate) && len(itemList) > 0 {
  473. quarterItem.CuttingDataTimestamp = itemList[0].DataTimestamp - 100
  474. }
  475. }
  476. quarterDataList = append(quarterDataList, quarterItem)
  477. }
  478. if len(quarterDataList) > 0 {
  479. quarterDataListSort = quarterDataList
  480. sort.Sort(quarterDataListSort)
  481. }
  482. return
  483. }
  484. // GetSeasonEdbInfoDataListByXDateNong 季节性图的指标数据根据横轴选择农历时展示
  485. func GetSeasonEdbInfoDataListByXDateNong(result *models.EdbDataResult, latestDate time.Time, seasonExtraConfig string, calendarPreYear, thisMonth int) (quarterDataListSort models.QuarterDataList, err error) {
  486. xStartDate := "01-01"
  487. xEndDate := "12-31"
  488. jumpYear := 0
  489. legends := make([]models.SeasonChartLegend, 0)
  490. var seasonExtra models.SeasonExtraItem
  491. if seasonExtraConfig != "" {
  492. err = json.Unmarshal([]byte(seasonExtraConfig), &seasonExtra)
  493. if err != nil {
  494. return
  495. }
  496. }
  497. if seasonExtra.XStartDate != "" {
  498. xStartDate = seasonExtra.XStartDate
  499. xEndDate = seasonExtra.XEndDate
  500. jumpYear = seasonExtra.JumpYear
  501. legends = seasonExtra.ChartLegend
  502. }
  503. length := len(result.List)
  504. if length == 0 {
  505. return
  506. }
  507. legendMap := make(map[string]string, 0)
  508. if len(legends) > 0 {
  509. for _, v := range legends {
  510. legendMap[v.Name] = v.Value
  511. }
  512. }
  513. latestDateYear := latestDate.Year()
  514. //判断横轴的两个时间之间是不是跨年了,如果跨年了,则横轴截止年份比起始年份+1,如果不跨年,截止年份等于起始年份
  515. //根据数据确定最早的年份,和最近年份
  516. //根据横轴的日期,汇总所有的年份
  517. startYear := result.List[0].Year
  518. /*if jumpYear == 1 {
  519. if startYear != calendarPreYear {
  520. startYear = startYear - 1
  521. }
  522. }*/
  523. itemLength := len(result.List[length-1].Items)
  524. //获取数据的最新日期
  525. lastDate := result.List[length-1].Items[itemLength-1].DataTime
  526. lastDateT, tmpErr := time.Parse(utils.FormatDate, lastDate)
  527. if tmpErr != nil {
  528. err = tmpErr
  529. return
  530. }
  531. endYear := lastDateT.Year()
  532. nowYear := time.Now().Year()
  533. dataMap := make(map[string]models.QuarterXDateItem, 0)
  534. quarterDataList := make([]*models.QuarterData, 0)
  535. resultData := make([]*models.QuarterData, 0)
  536. quarterMap := make(map[string][]*models.EdbDataList, 0)
  537. //整理出日期
  538. leftNameFlag := true
  539. for currentStartYear := startYear; currentStartYear <= endYear; currentStartYear++ {
  540. startStr := fmt.Sprintf("%d-%s", currentStartYear, xStartDate)
  541. currentEndYear := currentStartYear
  542. if jumpYear == 1 {
  543. currentEndYear = currentStartYear + 1
  544. }
  545. endStr := fmt.Sprintf("%d-%s", currentEndYear, xEndDate)
  546. showName := fmt.Sprintf("%d_%d", currentStartYear, currentEndYear)
  547. startT, tEr := time.Parse(utils.FormatDate, startStr)
  548. if tEr != nil {
  549. err = tEr
  550. return
  551. }
  552. endT, tEr := time.Parse(utils.FormatDate, endStr)
  553. if tEr != nil {
  554. err = tEr
  555. return
  556. }
  557. if lastDateT.Before(startT) {
  558. //如果最新的日期在起始日之前,则跳出循环
  559. break
  560. }
  561. if startT.Before(lastDateT) && endT.After(lastDateT) {
  562. if endYear > currentStartYear {
  563. leftNameFlag = false
  564. }
  565. }
  566. if endT.Year() > nowYear {
  567. //如果最新的日期比真实年份要大,则数据全部按照最大的年份补齐
  568. nowYear = endT.Year()
  569. }
  570. item := models.QuarterXDateItem{
  571. StartDate: startT,
  572. EndDate: endT,
  573. ShowName: showName,
  574. }
  575. dataMap[showName] = item
  576. if lastDateT.Before(endT) {
  577. //如果最新的日期在起始日之前,则跳出循环
  578. break
  579. }
  580. }
  581. for k, v := range dataMap {
  582. if leftNameFlag {
  583. v.ChartLegend = strconv.Itoa(v.StartDate.Year())
  584. } else {
  585. v.ChartLegend = strconv.Itoa(v.EndDate.Year())
  586. }
  587. dataMap[k] = v
  588. }
  589. yearDataListMap := make(map[int]*models.EdbDataItems, 0)
  590. for _, lv := range result.List {
  591. yearDataListMap[lv.Year] = lv
  592. }
  593. //判断哪些点应该落在同一条时间线上
  594. maxY := 0
  595. for _, lv := range result.List {
  596. for _, item := range lv.Items {
  597. dataTimeT, _ := time.Parse(utils.FormatDate, item.DataTime)
  598. if dataTimeT.Year() > maxY {
  599. maxY = dataTimeT.Year()
  600. }
  601. }
  602. }
  603. endTmp := fmt.Sprintf("%d-%s", maxY, xEndDate)
  604. endTmpT, _ := time.Parse(utils.FormatDate, endTmp)
  605. minY := maxY
  606. if jumpYear == 1 {
  607. minY = maxY - 1
  608. }
  609. startTmp := fmt.Sprintf("%d-%s", minY, xStartDate)
  610. startTmpT, _ := time.Parse(utils.FormatDate, startTmp)
  611. for name, dateItem := range dataMap {
  612. lv, ok1 := yearDataListMap[dateItem.EndDate.Year()]
  613. if thisMonth > 11 {
  614. lv, ok1 = yearDataListMap[dateItem.EndDate.Year()-1]
  615. }
  616. if dateItem.EndDate.Year() > maxY {
  617. lv, ok1 = yearDataListMap[dateItem.StartDate.Year()]
  618. if thisMonth > 11 {
  619. lv, ok1 = yearDataListMap[dateItem.StartDate.Year()-1]
  620. }
  621. }
  622. if !ok1 {
  623. continue
  624. }
  625. for _, item := range lv.Items {
  626. tmpVal := models.EdbDataList{
  627. EdbDataId: item.EdbDataId,
  628. EdbInfoId: item.EdbInfoId,
  629. DataTime: item.DataTime,
  630. DataTimestamp: item.DataTimestamp,
  631. Value: item.Value,
  632. }
  633. dataTimeT, _ := time.Parse(utils.FormatDate, item.DataTime)
  634. year := dataTimeT.Year()
  635. newItemDate := dataTimeT.AddDate(nowYear-year, 0, 0)
  636. if dateItem.EndDate.Year() > maxY {
  637. if (dateItem.StartDate.Before(dataTimeT) && dateItem.EndDate.After(dataTimeT)) || dateItem.StartDate == dataTimeT || dateItem.EndDate == dataTimeT {
  638. if jumpYear == 1 {
  639. //计算前一年最大的日期, 只补齐数据到去年
  640. beforeYearMaxDate := fmt.Sprintf("%d-12-31", dateItem.StartDate.Year())
  641. beforeYearMaxDateT, _ := time.Parse(utils.FormatDate, beforeYearMaxDate)
  642. if dataTimeT.Before(beforeYearMaxDateT) || dataTimeT == beforeYearMaxDateT {
  643. newItemDate = dataTimeT.AddDate(nowYear-year-1, 0, 0)
  644. } else {
  645. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  646. }
  647. } else {
  648. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  649. }
  650. timestamp := newItemDate.UnixNano() / 1e6
  651. tmpVal.DataTimestamp = timestamp
  652. tmpV := &tmpVal
  653. if findVal, ok := quarterMap[name]; !ok {
  654. findVal = append(findVal, tmpV)
  655. quarterMap[name] = findVal
  656. } else {
  657. findVal = append(findVal, tmpV)
  658. quarterMap[name] = findVal
  659. }
  660. if lv.Year >= latestDateYear {
  661. // 切割的日期时间字符串
  662. cuttingDataTimeStr := latestDate.AddDate(0, 0, lv.BetweenDay).Format(utils.FormatDate)
  663. if item.DataTime == cuttingDataTimeStr {
  664. dateItem.CuttingDataTimestamp = timestamp
  665. dataMap[name] = dateItem
  666. }
  667. }
  668. }
  669. } else {
  670. if (startTmpT.Before(dataTimeT) && endTmpT.After(dataTimeT)) || startTmpT == dataTimeT || endTmpT == dataTimeT {
  671. if jumpYear == 1 {
  672. //计算前一年最大的日期, 只补齐数据到去年
  673. beforeYearMaxDate := fmt.Sprintf("%d-12-31", startTmpT.Year())
  674. beforeYearMaxDateT, _ := time.Parse(utils.FormatDate, beforeYearMaxDate)
  675. if dataTimeT.Before(beforeYearMaxDateT) || dataTimeT == beforeYearMaxDateT {
  676. newItemDate = dataTimeT.AddDate(nowYear-year-1, 0, 0)
  677. } else {
  678. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  679. }
  680. } else {
  681. newItemDate = dataTimeT.AddDate(nowYear-year, 0, 0)
  682. }
  683. timestamp := newItemDate.UnixNano() / 1e6
  684. tmpVal.DataTimestamp = timestamp
  685. tmpV := &tmpVal
  686. if findVal, ok := quarterMap[name]; !ok {
  687. findVal = append(findVal, tmpV)
  688. quarterMap[name] = findVal
  689. } else {
  690. findVal = append(findVal, tmpV)
  691. quarterMap[name] = findVal
  692. }
  693. if lv.Year >= latestDateYear {
  694. // 切割的日期时间字符串
  695. cuttingDataTimeStr := latestDate.AddDate(0, 0, lv.BetweenDay).Format(utils.FormatDate)
  696. if item.DataTime == cuttingDataTimeStr {
  697. dateItem.CuttingDataTimestamp = timestamp
  698. dataMap[name] = dateItem
  699. }
  700. }
  701. }
  702. }
  703. }
  704. }
  705. for k, v := range dataMap {
  706. itemList := quarterMap[k]
  707. quarterItem := new(models.QuarterData)
  708. quarterItem.Years = v.ShowName
  709. quarterItem.Year = v.ChartLegend
  710. quarterItem.ChartLegend = v.ChartLegend
  711. if le, ok := legendMap[v.ShowName]; ok {
  712. quarterItem.ChartLegend = le
  713. }
  714. quarterItem.DataList = itemList
  715. quarterItem.CuttingDataTimestamp = v.CuttingDataTimestamp
  716. quarterDataList = append(quarterDataList, quarterItem)
  717. }
  718. if result.List[0].Year != calendarPreYear {
  719. itemList := make([]*models.EdbDataList, 0)
  720. items := new(models.QuarterData)
  721. //items.Year = calendarPreYear
  722. items.DataList = itemList
  723. newResult := make([]*models.QuarterData, 0)
  724. newResult = append(newResult, items)
  725. newResult = append(newResult, quarterDataList...)
  726. resultData = newResult
  727. } else {
  728. resultData = quarterDataList
  729. }
  730. if len(quarterDataList) > 0 {
  731. quarterDataListSort = resultData
  732. sort.Sort(quarterDataListSort)
  733. }
  734. return
  735. }
  736. // BarChartData 柱方图的数据处理
  737. 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) {
  738. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  739. edbDataMap := make(map[int]map[string]float64)
  740. for edbInfoId, edbDataList := range edbDataListMap {
  741. edbDateData := make(map[string]float64)
  742. for _, edbData := range edbDataList {
  743. edbDateData[edbData.DataTime] = edbData.Value
  744. }
  745. edbDataMap[edbInfoId] = edbDateData
  746. }
  747. // edbIdList 指标展示顺序;x轴的指标顺序
  748. edbIdList = make([]int, 0)
  749. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  750. dateData := make(map[int]float64)
  751. if barChartInfoSort.Sort == 0 {
  752. for _, v := range mappingList {
  753. edbIdList = append(edbIdList, v.EdbInfoId)
  754. }
  755. } else {
  756. lenBarChartInfoDateList := len(barChartInfoDateList)
  757. if barChartInfoSort.DateIndex >= lenBarChartInfoDateList {
  758. err = errors.New("排序日期异常")
  759. return
  760. }
  761. notDataEdbIdList := make([]int, 0) //没有数据的指标id
  762. // 日期配置
  763. barChartInfoDate := barChartInfoDateList[barChartInfoSort.DateIndex]
  764. for edbInfoId, dataList := range edbDataListMap {
  765. if len(dataList) <= 0 {
  766. // 没有数据的指标id
  767. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  768. continue
  769. }
  770. findDate := barChartInfoDate.Date
  771. switch barChartInfoDate.Type {
  772. case 1: //最新值
  773. findDate = dataList[len(dataList)-1].DataTime
  774. case 2: //近期几天
  775. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  776. if tmpErr != nil {
  777. err = tmpErr
  778. return
  779. }
  780. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  781. lenData := len(dataList) - 1
  782. for i := lenData; i >= 0; i-- {
  783. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  784. if tmpErr != nil {
  785. err = tmpErr
  786. return
  787. }
  788. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  789. findDate = dataList[i].DataTime
  790. break
  791. }
  792. }
  793. case 3: // 固定日期
  794. //最早的日期
  795. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  796. if tmpErr != nil {
  797. err = tmpErr
  798. return
  799. }
  800. //寻找固定日期的数据
  801. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  802. if tmpErr != nil {
  803. err = tmpErr
  804. return
  805. }
  806. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  807. tmpDate := tmpDateTime.Format(utils.FormatDate)
  808. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  809. findDate = tmpDate
  810. break
  811. }
  812. }
  813. default:
  814. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  815. return
  816. }
  817. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  818. dateData[edbInfoId] = tmpValue
  819. } else {
  820. // 没有数据的指标id
  821. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  822. }
  823. }
  824. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  825. // 排序
  826. dateDataSort := utils.NewMapSorter(dateData)
  827. sort.Sort(dateDataSort)
  828. if barChartInfoSort.Sort == 1 {
  829. // 先将没有数据的指标id放在最前面
  830. if len(notDataEdbIdList) > 0 {
  831. edbIdList = append(edbIdList, notDataEdbIdList...)
  832. }
  833. for _, v := range dateDataSort {
  834. edbIdList = append(edbIdList, v.Key)
  835. }
  836. } else {
  837. for i := len(dateDataSort) - 1; i >= 0; i-- {
  838. edbIdList = append(edbIdList, dateDataSort[i].Key)
  839. }
  840. // 再将没有数据的指标id放在最后面
  841. if len(notDataEdbIdList) > 0 {
  842. edbIdList = append(edbIdList, notDataEdbIdList...)
  843. }
  844. }
  845. }
  846. yDataList = make([]models.YData, 0) //y轴的数据列表
  847. for _, barChartInfoDate := range barChartInfoDateList {
  848. var maxDate time.Time
  849. findDataList := make([]float64, 0) // 当前日期的数据值
  850. for _, edbInfoId := range edbIdList {
  851. findDate := barChartInfoDate.Date //需要的日期值
  852. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  853. if len(dataList) <= 0 {
  854. // 没有数据的指标id
  855. findDataList = append(findDataList, 0)
  856. continue
  857. }
  858. switch barChartInfoDate.Type {
  859. case 1: //最新值
  860. dataList := edbDataListMap[edbInfoId]
  861. findDate = dataList[len(dataList)-1].DataTime
  862. case 2: //近期几天
  863. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  864. if tmpErr != nil {
  865. err = tmpErr
  866. return
  867. }
  868. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  869. lenData := len(dataList) - 1
  870. for i := lenData; i >= 0; i-- {
  871. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  872. if tmpErr != nil {
  873. err = tmpErr
  874. return
  875. }
  876. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  877. findDate = dataList[i].DataTime
  878. break
  879. }
  880. }
  881. case 3: // 固定日期
  882. //最早的日期
  883. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  884. if tmpErr != nil {
  885. err = tmpErr
  886. return
  887. }
  888. //寻找固定日期的数据
  889. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  890. if tmpErr != nil {
  891. err = tmpErr
  892. return
  893. }
  894. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  895. tmpDate := tmpDateTime.Format(utils.FormatDate)
  896. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  897. findDate = tmpDate
  898. break
  899. }
  900. }
  901. default:
  902. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  903. return
  904. }
  905. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  906. if maxDate.IsZero() {
  907. maxDate = findDateTime
  908. } else {
  909. if findDateTime.After(maxDate) {
  910. maxDate = findDateTime
  911. }
  912. }
  913. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  914. findDataList = append(findDataList, tmpValue)
  915. } else {
  916. findDataList = append(findDataList, 0)
  917. }
  918. }
  919. yDate := "0000-00-00"
  920. if !maxDate.IsZero() {
  921. yDate = maxDate.Format(utils.FormatDate)
  922. }
  923. yDataList = append(yDataList, models.YData{
  924. Date: yDate,
  925. Value: findDataList,
  926. Color: barChartInfoDate.Color,
  927. Name: barChartInfoDate.Name,
  928. })
  929. }
  930. return
  931. }
  932. // ChartInfoRefreshV2 图表刷新
  933. // @author Roc
  934. // @datetime 2022-09-16 10:15:38
  935. // @description 将原来自己写的一套获取所有关联指标,然后刷新指标逻辑 改成 只获取使用的指标id,然后遍历去调用“指标刷新服务”
  936. func ChartInfoRefreshV2(chartInfoId int) (err error) {
  937. var errmsg string
  938. defer func() {
  939. if err != nil {
  940. go alarm_msg.SendAlarmMsg("ChartInfoRefresh:"+errmsg, 3)
  941. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "ChartInfoRefresh:"+errmsg, utils.EmailSendToUsers)
  942. fmt.Println("ChartInfoRefresh Err:" + errmsg)
  943. }
  944. }()
  945. edbMappingList, err := data_manage.GetChartEdbMappingList(chartInfoId)
  946. if err != nil {
  947. errmsg = "获取需要刷新的指标失败:Err:" + err.Error()
  948. return
  949. }
  950. edbIdList := make([]int, 0)
  951. for _, v := range edbMappingList {
  952. edbIdList = append(edbIdList, v.EdbInfoId)
  953. }
  954. // 批量刷新
  955. err, errmsg = EdbInfoRefreshAllFromBase(edbIdList, false)
  956. if err != nil {
  957. return
  958. }
  959. return
  960. }
  961. // GetSectionScatterChartData 柱方图的数据处理
  962. func GetSectionScatterChartData(mappingList []*models.ChartEdbInfoMapping, edbDataListMap map[int][]*models.EdbDataList, extraConfig data_manage.SectionScatterReq) (edbIdList []int, chartDataResp data_manage.SectionScatterInfoResp, err error) {
  963. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  964. edbDataMap := make(map[int]map[string]float64)
  965. for edbInfoId, edbDataList := range edbDataListMap {
  966. edbDateData := make(map[string]float64)
  967. for _, edbData := range edbDataList {
  968. edbDateData[edbData.DataTime] = edbData.Value
  969. }
  970. edbDataMap[edbInfoId] = edbDateData
  971. }
  972. // edbIdList 指标展示顺序;x轴的指标顺序
  973. edbIdList = make([]int, 0)
  974. edbMappingMap := make(map[int]*models.ChartEdbInfoMapping)
  975. for _, v := range mappingList {
  976. edbIdList = append(edbIdList, v.EdbInfoId)
  977. edbMappingMap[v.EdbInfoId] = v
  978. }
  979. //SectionScatterSeriesInfoResp
  980. dataListResp := make([]data_manage.SectionScatterSeriesItemResp, 0) //y轴的数据列表
  981. for _, seriesItem := range extraConfig.SeriesList {
  982. var maxDate time.Time
  983. // 系列中的指标数据
  984. tmpSeriesEdbInfoList := make([]data_manage.SectionScatterEdbItemResp, 0)
  985. var minXVal, maxXVal, minYVal, maxYVal float64
  986. for _, edbConf := range seriesItem.EdbInfoList {
  987. tmpItem := data_manage.SectionScatterEdbItemResp{
  988. IsShow: edbConf.IsShow,
  989. Name: edbConf.Name,
  990. NameEn: edbConf.NameEn,
  991. } //单个坐标点的数据
  992. //X轴的数据
  993. {
  994. edbInfoId := edbConf.XEdbInfoId //X轴的指标
  995. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  996. if !ok {
  997. continue
  998. }
  999. findDate := edbConf.XDate //需要的日期值
  1000. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  1001. if len(dataList) <= 0 {
  1002. // 没有数据的指标id
  1003. //findDataList = append(findDataList, 0)
  1004. continue
  1005. }
  1006. tmpItem.XEdbInfoId = edbInfoId
  1007. tmpItem.XName = edbMappingInfo.EdbName
  1008. tmpItem.XNameEn = edbMappingInfo.EdbNameEn
  1009. switch edbConf.XDateType {
  1010. case 1: //最新值
  1011. dataList := edbDataListMap[edbInfoId]
  1012. findDate = dataList[len(dataList)-1].DataTime
  1013. case 2: //近期几天
  1014. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  1015. if tmpErr != nil {
  1016. err = tmpErr
  1017. return
  1018. }
  1019. findDateTime = findDateTime.AddDate(0, 0, -edbConf.XDateValue)
  1020. lenData := len(dataList) - 1
  1021. for i := lenData; i >= 0; i-- {
  1022. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  1023. if tmpErr != nil {
  1024. err = tmpErr
  1025. return
  1026. }
  1027. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  1028. findDate = dataList[i].DataTime
  1029. break
  1030. }
  1031. }
  1032. case 3: // 固定日期
  1033. //最早的日期
  1034. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  1035. if tmpErr != nil {
  1036. err = tmpErr
  1037. return
  1038. }
  1039. //寻找固定日期的数据
  1040. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.XDate, time.Local)
  1041. if tmpErr != nil {
  1042. err = tmpErr
  1043. return
  1044. }
  1045. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  1046. tmpDate := tmpDateTime.Format(utils.FormatDate)
  1047. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  1048. findDate = tmpDate
  1049. break
  1050. }
  1051. }
  1052. default:
  1053. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.XDate))
  1054. return
  1055. }
  1056. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  1057. if maxDate.IsZero() {
  1058. maxDate = findDateTime
  1059. } else {
  1060. if findDateTime.After(maxDate) {
  1061. maxDate = findDateTime
  1062. }
  1063. }
  1064. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  1065. tmpItem.XDate = findDate
  1066. tmpItem.XValue = tmpValue
  1067. } else {
  1068. continue
  1069. }
  1070. }
  1071. //Y轴的数据
  1072. {
  1073. edbInfoId := edbConf.YEdbInfoId //Y轴的指标
  1074. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  1075. if !ok {
  1076. continue
  1077. }
  1078. findDate := edbConf.YDate //需要的日期值
  1079. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  1080. if len(dataList) <= 0 {
  1081. // 没有数据的指标id
  1082. //findDataList = append(findDataList, 0)
  1083. continue
  1084. }
  1085. tmpItem.YEdbInfoId = edbInfoId
  1086. tmpItem.YName = edbMappingInfo.EdbName
  1087. tmpItem.YNameEn = edbMappingInfo.EdbNameEn
  1088. switch edbConf.YDateType {
  1089. case 1: //最新值
  1090. dataList := edbDataListMap[edbInfoId]
  1091. findDate = dataList[len(dataList)-1].DataTime
  1092. case 2: //近期几天
  1093. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  1094. if tmpErr != nil {
  1095. err = tmpErr
  1096. return
  1097. }
  1098. findDateTime = findDateTime.AddDate(0, 0, -edbConf.YDateValue)
  1099. lenData := len(dataList) - 1
  1100. for i := lenData; i >= 0; i-- {
  1101. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  1102. if tmpErr != nil {
  1103. err = tmpErr
  1104. return
  1105. }
  1106. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  1107. findDate = dataList[i].DataTime
  1108. break
  1109. }
  1110. }
  1111. case 3: // 固定日期
  1112. //最早的日期
  1113. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  1114. if tmpErr != nil {
  1115. err = tmpErr
  1116. return
  1117. }
  1118. //寻找固定日期的数据
  1119. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.YDate, time.Local)
  1120. if tmpErr != nil {
  1121. err = tmpErr
  1122. return
  1123. }
  1124. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  1125. tmpDate := tmpDateTime.Format(utils.FormatDate)
  1126. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  1127. findDate = tmpDate
  1128. break
  1129. }
  1130. }
  1131. default:
  1132. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.YDate))
  1133. return
  1134. }
  1135. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  1136. if maxDate.IsZero() {
  1137. maxDate = findDateTime
  1138. } else {
  1139. if findDateTime.After(maxDate) {
  1140. maxDate = findDateTime
  1141. }
  1142. }
  1143. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  1144. tmpItem.YDate = findDate
  1145. tmpItem.YValue = tmpValue
  1146. } else {
  1147. continue
  1148. }
  1149. }
  1150. // 获取当前系列的X轴的最大最小值
  1151. {
  1152. if tmpItem.XValue < minXVal {
  1153. minXVal = tmpItem.XValue
  1154. }
  1155. if tmpItem.XValue > maxXVal {
  1156. maxXVal = tmpItem.XValue
  1157. }
  1158. if tmpItem.YValue < minYVal {
  1159. minYVal = tmpItem.YValue
  1160. }
  1161. if tmpItem.YValue > maxYVal {
  1162. maxYVal = tmpItem.YValue
  1163. }
  1164. }
  1165. tmpSeriesEdbInfoList = append(tmpSeriesEdbInfoList, tmpItem)
  1166. }
  1167. trendLimitData := make([]data_manage.CoordinatePoint, 0) //趋势线的前后坐标点
  1168. var trendLine, rSquare string
  1169. // 生成线性方程式
  1170. var a, b float64
  1171. {
  1172. coordinateData := make([]utils.Coordinate, 0)
  1173. for _, tmpSeriesEdbInfo := range tmpSeriesEdbInfoList {
  1174. tmpCoordinate1 := utils.Coordinate{
  1175. X: tmpSeriesEdbInfo.XValue,
  1176. Y: tmpSeriesEdbInfo.YValue,
  1177. }
  1178. coordinateData = append(coordinateData, tmpCoordinate1)
  1179. }
  1180. // 只有存在两个坐标点的时候,才能去计算线性方程和R平方
  1181. if len(coordinateData) >= 2 {
  1182. a, b = utils.GetLinearResult(coordinateData)
  1183. if !math.IsNaN(a) || !math.IsNaN(b) {
  1184. if b > 0 {
  1185. trendLine = fmt.Sprintf("y=%sx+%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  1186. } else {
  1187. trendLine = fmt.Sprintf("y=%sx%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  1188. }
  1189. minYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(minXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  1190. maxYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(maxXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  1191. }
  1192. // 计算R平方
  1193. rSquare = fmt.Sprint(utils.CalculationDecisive(coordinateData))
  1194. }
  1195. trendLimitData = append(trendLimitData, data_manage.CoordinatePoint{
  1196. X: minXVal,
  1197. Y: minYVal,
  1198. }, data_manage.CoordinatePoint{
  1199. X: maxXVal,
  1200. Y: maxYVal,
  1201. })
  1202. }
  1203. dataListResp = append(dataListResp, data_manage.SectionScatterSeriesItemResp{
  1204. Name: seriesItem.Name,
  1205. NameEn: seriesItem.NameEn,
  1206. Color: seriesItem.Color,
  1207. EdbInfoList: tmpSeriesEdbInfoList,
  1208. ShowTrendLine: seriesItem.ShowTrendLine,
  1209. ShowFitEquation: seriesItem.ShowFitEquation,
  1210. ShowRSquare: seriesItem.ShowRSquare,
  1211. TrendLine: trendLine,
  1212. RSquare: rSquare,
  1213. TrendLimitData: trendLimitData,
  1214. })
  1215. }
  1216. chartDataResp = data_manage.SectionScatterInfoResp{
  1217. XName: extraConfig.XName,
  1218. XNameEn: extraConfig.XNameEn,
  1219. XUnitName: extraConfig.XUnitName,
  1220. XUnitNameEn: extraConfig.XUnitNameEn,
  1221. YName: extraConfig.YName,
  1222. YNameEn: extraConfig.YNameEn,
  1223. YUnitName: extraConfig.YUnitName,
  1224. YUnitNameEn: extraConfig.YUnitNameEn,
  1225. XMinValue: extraConfig.XMinValue,
  1226. XMaxValue: extraConfig.XMaxValue,
  1227. YMinValue: extraConfig.YMinValue,
  1228. YMaxValue: extraConfig.YMaxValue,
  1229. DataList: dataListResp,
  1230. }
  1231. return
  1232. }