chart_info_show.go 49 KB

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