chart_info.go 47 KB

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