chart_info.go 47 KB

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