chart_info.go 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084
  1. package data
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "github.com/shopspring/decimal"
  7. "hongze/hongze_ETA_mobile_api/models/company"
  8. "hongze/hongze_ETA_mobile_api/models/data_manage"
  9. "hongze/hongze_ETA_mobile_api/models/system"
  10. "hongze/hongze_ETA_mobile_api/services/alarm_msg"
  11. "hongze/hongze_ETA_mobile_api/utils"
  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) (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)
  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. case 10: // 截面散点图
  118. sectionScatterConf := extraConfig.(data_manage.SectionScatterReq)
  119. xEdbIdValue, dataResp, err = GetSectionScatterChartData(mappingList, edbDataListMap, sectionScatterConf)
  120. // 这个数据没有必要返回给前端
  121. for _, v := range edbList {
  122. v.DataList = nil
  123. }
  124. }
  125. return
  126. }
  127. // GetEdbDataMapList 获取指标最后的基础数据
  128. func GetEdbDataMapList(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*data_manage.ChartEdbInfoMapping) (edbDataListMap map[int][]*data_manage.EdbDataList, edbList []*data_manage.ChartEdbInfoMapping, err error) {
  129. edbDataListMap, edbList, err = getEdbDataMapList(chartInfoId, chartType, calendar, startDate, endDate, mappingList)
  130. return
  131. }
  132. // getEdbDataMapList 获取指标最后的基础数据
  133. func getEdbDataMapList(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*data_manage.ChartEdbInfoMapping) (edbDataListMap map[int][]*data_manage.EdbDataList, edbList []*data_manage.ChartEdbInfoMapping, err error) {
  134. // 指标对应的所有数据
  135. edbDataListMap = make(map[int][]*data_manage.EdbDataList)
  136. for _, v := range mappingList {
  137. //fmt.Println("v:", v.EdbInfoId)
  138. item := new(data_manage.ChartEdbInfoMapping)
  139. item.EdbInfoId = v.EdbInfoId
  140. item.SourceName = v.SourceName
  141. item.Source = v.Source
  142. item.EdbCode = v.EdbCode
  143. item.EdbName = v.EdbName
  144. item.EdbNameEn = v.EdbNameEn
  145. item.Frequency = v.Frequency
  146. item.EdbType = v.EdbType
  147. item.FrequencyEn = GetFrequencyEn(v.Frequency)
  148. if v.Unit != `无` {
  149. item.Unit = v.Unit
  150. }
  151. item.UnitEn = v.UnitEn
  152. item.StartDate = v.StartDate
  153. item.EndDate = v.EndDate
  154. item.ModifyTime = v.ModifyTime
  155. item.EdbInfoCategoryType = v.EdbInfoCategoryType
  156. item.PredictChartColor = v.PredictChartColor
  157. item.ClassifyId = v.ClassifyId
  158. if chartInfoId <= 0 {
  159. item.IsAxis = 1
  160. item.LeadValue = 0
  161. item.LeadUnit = ""
  162. item.ChartEdbMappingId = 0
  163. item.ChartInfoId = 0
  164. item.IsOrder = false
  165. item.EdbInfoType = 1
  166. item.ChartStyle = ""
  167. item.ChartColor = ""
  168. item.ChartWidth = 0
  169. item.MaxData = v.MaxValue
  170. item.MinData = v.MinValue
  171. } else {
  172. item.IsAxis = v.IsAxis
  173. item.EdbInfoType = v.EdbInfoType
  174. item.LeadValue = v.LeadValue
  175. item.LeadUnit = v.LeadUnit
  176. item.LeadUnitEn = GetLeadUnitEn(v.LeadUnit)
  177. item.ChartEdbMappingId = v.ChartEdbMappingId
  178. item.ChartInfoId = v.ChartInfoId
  179. item.ChartStyle = v.ChartStyle
  180. item.ChartColor = v.ChartColor
  181. item.ChartWidth = v.ChartWidth
  182. item.IsOrder = v.IsOrder
  183. item.MaxData = v.MaxData
  184. item.MinData = v.MinData
  185. }
  186. item.LatestValue = v.LatestValue
  187. item.LatestDate = v.LatestDate
  188. item.UniqueCode = v.UniqueCode
  189. var startDateReal string
  190. var diffSeconds int64
  191. if chartType == 2 { //季节性图
  192. startDateReal = startDate
  193. } else {
  194. if v.EdbInfoType == 0 && v.LeadUnit != "" && v.LeadValue > 0 { //领先指标
  195. var startTimeRealTemp time.Time
  196. startDateParse, _ := time.Parse(utils.FormatDate, startDate)
  197. switch v.LeadUnit {
  198. case "天":
  199. startTimeRealTemp = startDateParse.AddDate(0, 0, -v.LeadValue)
  200. case "月":
  201. startTimeRealTemp = startDateParse.AddDate(0, -v.LeadValue, 0)
  202. case "季":
  203. startTimeRealTemp = startDateParse.AddDate(0, -3*v.LeadValue, 0)
  204. case "周":
  205. startTimeRealTemp = startDateParse.AddDate(0, 0, -7*v.LeadValue)
  206. case "年":
  207. startTimeRealTemp = startDateParse.AddDate(-v.LeadValue, 0, 0)
  208. }
  209. if startTimeRealTemp.Before(startDateParse) {
  210. startDateReal = startTimeRealTemp.Format(utils.FormatDate)
  211. diffSeconds = (int64(startTimeRealTemp.UnixNano()) - int64(startDateParse.UnixNano())) / 1e6
  212. } else {
  213. startDateReal = startDate
  214. diffSeconds = 0
  215. }
  216. // 预测指标的开始日期也要偏移
  217. {
  218. day, tmpErr := utils.GetDaysBetween2Date(utils.FormatDate, startDate, startDateReal)
  219. if tmpErr != nil {
  220. err = tmpErr
  221. return
  222. }
  223. latestDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, item.LatestDate, time.Local)
  224. if tmpErr != nil {
  225. err = tmpErr
  226. return
  227. }
  228. item.LatestDate = latestDateTime.AddDate(0, 0, day).Format(utils.FormatDate)
  229. }
  230. } else {
  231. startDateReal = startDate
  232. }
  233. }
  234. //fmt.Println("line 1011 chart:", v.Source, v.EdbInfoId, startDateReal, endDate)
  235. calendarPreYear := 0
  236. if calendar == "农历" {
  237. newStartDateReal, err := time.Parse(utils.FormatDate, startDateReal)
  238. if err != nil {
  239. fmt.Println("time.Parse:" + err.Error())
  240. }
  241. calendarPreYear = newStartDateReal.Year() - 1
  242. newStartDateReal = newStartDateReal.AddDate(-1, 0, 0)
  243. startDateReal = newStartDateReal.Format(utils.FormatDate)
  244. }
  245. dataList := make([]*data_manage.EdbDataList, 0)
  246. //fmt.Println("chart:", v.Source, v.EdbInfoId, startDateReal, endDate)
  247. //var newEdbInfo *data_manage.EdbInfo
  248. switch v.EdbInfoCategoryType {
  249. case 0:
  250. dataList, err = data_manage.GetEdbDataList(v.Source, v.EdbInfoId, startDateReal, endDate)
  251. case 1:
  252. _, dataList, _, _, err, _ = GetPredictDataListByPredictEdbInfoId(v.EdbInfoId, startDateReal, endDate, false)
  253. default:
  254. err = errors.New(fmt.Sprint("获取失败,指标类型异常", v.EdbInfoCategoryType))
  255. }
  256. if err != nil {
  257. return
  258. }
  259. edbDataListMap[v.EdbInfoId] = dataList
  260. if diffSeconds != 0 && v.EdbInfoType == 0 {
  261. dataListLen := len(dataList)
  262. for i := 0; i < dataListLen; i++ {
  263. dataList[i].DataTimestamp = dataList[i].DataTimestamp - diffSeconds
  264. }
  265. }
  266. if chartType == 2 {
  267. latestDateStr := v.LatestDate //实际数据的截止日期
  268. latestDate, tmpErr := time.Parse(utils.FormatDate, v.LatestDate)
  269. if tmpErr != nil {
  270. //item.DataList = dataList
  271. item.IsNullData = true
  272. edbList = append(edbList, item)
  273. continue
  274. err = errors.New(fmt.Sprint("获取最后实际数据的日期失败,Err:" + tmpErr.Error() + ";LatestDate:" + v.LatestDate))
  275. return
  276. }
  277. latestDateYear := latestDate.Year() //实际数据截止年份
  278. if calendar == "农历" {
  279. if len(dataList) <= 0 {
  280. result := new(data_manage.EdbDataResult)
  281. item.DataList = result
  282. } else {
  283. result, tmpErr := data_manage.AddCalculateQuarterV4(dataList)
  284. if tmpErr != nil {
  285. err = errors.New("获取农历数据失败,Err:" + tmpErr.Error())
  286. return
  287. }
  288. // 处理季节图的截止日期
  289. for k, edbDataItems := range result.List {
  290. var cuttingDataTimestamp int64
  291. // 切割的日期时间字符串
  292. cuttingDataTimeStr := latestDate.AddDate(0, 0, edbDataItems.BetweenDay).Format(utils.FormatDate)
  293. //如果等于最后的实际日期,那么遍历找到该日期对应的时间戳,并将其赋值为 切割时间戳
  294. if edbDataItems.Year >= latestDateYear {
  295. for _, tmpData := range edbDataItems.Items {
  296. if tmpData.DataTime == cuttingDataTimeStr {
  297. cuttingDataTimestamp = tmpData.DataTimestamp
  298. break
  299. }
  300. }
  301. }
  302. edbDataItems.CuttingDataTimestamp = cuttingDataTimestamp
  303. result.List[k] = edbDataItems
  304. }
  305. if result.List[0].Year != calendarPreYear {
  306. itemList := make([]*data_manage.EdbDataList, 0)
  307. items := new(data_manage.EdbDataItems)
  308. //items.Year = calendarPreYear
  309. items.Items = itemList
  310. newResult := new(data_manage.EdbDataResult)
  311. newResult.List = append(newResult.List, items)
  312. newResult.List = append(newResult.List, result.List...)
  313. item.DataList = newResult
  314. } else {
  315. item.DataList = result
  316. }
  317. }
  318. } else {
  319. currentYear := time.Now().Year()
  320. quarterDataList := make([]*data_manage.QuarterData, 0)
  321. quarterMap := make(map[int][]*data_manage.EdbDataList)
  322. var quarterArr []int
  323. for _, v := range dataList {
  324. itemDate, tmpErr := time.Parse(utils.FormatDate, v.DataTime)
  325. if tmpErr != nil {
  326. err = errors.New("季度指标日期转换,Err:" + tmpErr.Error() + ";DataTime:" + v.DataTime)
  327. return
  328. }
  329. year := itemDate.Year()
  330. newItemDate := itemDate.AddDate(currentYear-year, 0, 0)
  331. timestamp := newItemDate.UnixNano() / 1e6
  332. v.DataTimestamp = timestamp
  333. if findVal, ok := quarterMap[year]; !ok {
  334. quarterArr = append(quarterArr, year)
  335. findVal = append(findVal, v)
  336. quarterMap[year] = findVal
  337. } else {
  338. findVal = append(findVal, v)
  339. quarterMap[year] = findVal
  340. }
  341. }
  342. for _, v := range quarterArr {
  343. itemList := quarterMap[v]
  344. quarterItem := new(data_manage.QuarterData)
  345. quarterItem.Year = v
  346. quarterItem.DataList = itemList
  347. //如果等于最后的实际日期,那么将切割时间戳记录
  348. if v == latestDateYear {
  349. var cuttingDataTimestamp int64
  350. for _, tmpData := range itemList {
  351. if tmpData.DataTime == latestDateStr {
  352. cuttingDataTimestamp = tmpData.DataTimestamp
  353. break
  354. }
  355. }
  356. quarterItem.CuttingDataTimestamp = cuttingDataTimestamp
  357. } else if v > latestDateYear {
  358. //如果大于最后的实际日期,那么第一个点就是切割的时间戳
  359. if len(itemList) > 0 {
  360. quarterItem.CuttingDataTimestamp = itemList[0].DataTimestamp - 100
  361. }
  362. }
  363. quarterDataList = append(quarterDataList, quarterItem)
  364. }
  365. item.DataList = quarterDataList
  366. }
  367. } else if chartType == 7 { //柱方图
  368. //item.DataList = dataList
  369. } else {
  370. item.DataList = dataList
  371. }
  372. edbList = append(edbList, item)
  373. }
  374. return
  375. }
  376. // BarChartData 柱方图的数据处理
  377. 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) {
  378. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  379. edbDataMap := make(map[int]map[string]float64)
  380. for edbInfoId, edbDataList := range edbDataListMap {
  381. edbDateData := make(map[string]float64)
  382. for _, edbData := range edbDataList {
  383. edbDateData[edbData.DataTime] = edbData.Value
  384. }
  385. edbDataMap[edbInfoId] = edbDateData
  386. }
  387. // edbIdList 指标展示顺序;x轴的指标顺序
  388. edbIdList = make([]int, 0)
  389. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  390. dateData := make(map[int]float64)
  391. if barChartInfoSort.Sort == 0 {
  392. for _, v := range mappingList {
  393. edbIdList = append(edbIdList, v.EdbInfoId)
  394. }
  395. } else {
  396. lenBarChartInfoDateList := len(barChartInfoDateList)
  397. if barChartInfoSort.DateIndex >= lenBarChartInfoDateList {
  398. err = errors.New("排序日期异常")
  399. return
  400. }
  401. notDataEdbIdList := make([]int, 0) //没有数据的指标id
  402. // 日期配置
  403. barChartInfoDate := barChartInfoDateList[barChartInfoSort.DateIndex]
  404. for edbInfoId, dataList := range edbDataListMap {
  405. if len(dataList) <= 0 {
  406. // 没有数据的指标id
  407. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  408. continue
  409. }
  410. findDate := barChartInfoDate.Date
  411. switch barChartInfoDate.Type {
  412. case 1: //最新值
  413. findDate = dataList[len(dataList)-1].DataTime
  414. case 2: //近期几天
  415. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  416. if tmpErr != nil {
  417. err = tmpErr
  418. return
  419. }
  420. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  421. lenData := len(dataList) - 1
  422. for i := lenData; i >= 0; i-- {
  423. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  424. if tmpErr != nil {
  425. err = tmpErr
  426. return
  427. }
  428. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  429. findDate = dataList[i].DataTime
  430. break
  431. }
  432. }
  433. case 3: // 固定日期
  434. //最早的日期
  435. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  436. if tmpErr != nil {
  437. err = tmpErr
  438. return
  439. }
  440. //寻找固定日期的数据
  441. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  442. if tmpErr != nil {
  443. err = tmpErr
  444. return
  445. }
  446. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  447. tmpDate := tmpDateTime.Format(utils.FormatDate)
  448. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  449. findDate = tmpDate
  450. break
  451. }
  452. }
  453. default:
  454. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  455. return
  456. }
  457. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  458. dateData[edbInfoId] = tmpValue
  459. } else {
  460. // 没有数据的指标id
  461. notDataEdbIdList = append(notDataEdbIdList, edbInfoId)
  462. }
  463. }
  464. //Sort int `description:"排序类型,0:默认,1:升序,2:降序"`
  465. // 排序
  466. dateDataSort := utils.NewMapSorter(dateData)
  467. sort.Sort(dateDataSort)
  468. if barChartInfoSort.Sort == 1 {
  469. // 先将没有数据的指标id放在最前面
  470. if len(notDataEdbIdList) > 0 {
  471. edbIdList = append(edbIdList, notDataEdbIdList...)
  472. }
  473. for _, v := range dateDataSort {
  474. edbIdList = append(edbIdList, v.Key)
  475. }
  476. } else {
  477. for i := len(dateDataSort) - 1; i >= 0; i-- {
  478. edbIdList = append(edbIdList, dateDataSort[i].Key)
  479. }
  480. // 再将没有数据的指标id放在最后面
  481. if len(notDataEdbIdList) > 0 {
  482. edbIdList = append(edbIdList, notDataEdbIdList...)
  483. }
  484. }
  485. }
  486. yDataList = make([]data_manage.YData, 0) //y轴的数据列表
  487. for _, barChartInfoDate := range barChartInfoDateList {
  488. var maxDate time.Time
  489. findDataList := make([]float64, 0) // 当前日期的数据值
  490. for _, edbInfoId := range edbIdList {
  491. findDate := barChartInfoDate.Date //需要的日期值
  492. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  493. if len(dataList) <= 0 {
  494. // 没有数据的指标id
  495. findDataList = append(findDataList, 0)
  496. continue
  497. }
  498. switch barChartInfoDate.Type {
  499. case 1: //最新值
  500. dataList := edbDataListMap[edbInfoId]
  501. findDate = dataList[len(dataList)-1].DataTime
  502. case 2: //近期几天
  503. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  504. if tmpErr != nil {
  505. err = tmpErr
  506. return
  507. }
  508. findDateTime = findDateTime.AddDate(0, 0, -barChartInfoDate.Value)
  509. lenData := len(dataList) - 1
  510. for i := lenData; i >= 0; i-- {
  511. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  512. if tmpErr != nil {
  513. err = tmpErr
  514. return
  515. }
  516. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  517. findDate = dataList[i].DataTime
  518. break
  519. }
  520. }
  521. case 3: // 固定日期
  522. //最早的日期
  523. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  524. if tmpErr != nil {
  525. err = tmpErr
  526. return
  527. }
  528. //寻找固定日期的数据
  529. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
  530. if tmpErr != nil {
  531. err = tmpErr
  532. return
  533. }
  534. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  535. tmpDate := tmpDateTime.Format(utils.FormatDate)
  536. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  537. findDate = tmpDate
  538. break
  539. }
  540. }
  541. default:
  542. err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
  543. return
  544. }
  545. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  546. if maxDate.IsZero() {
  547. maxDate = findDateTime
  548. } else {
  549. if findDateTime.After(maxDate) {
  550. maxDate = findDateTime
  551. }
  552. }
  553. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  554. tmpValue, _ = decimal.NewFromFloat(tmpValue).Round(4).Float64()
  555. findDataList = append(findDataList, tmpValue)
  556. } else {
  557. findDataList = append(findDataList, 0)
  558. }
  559. }
  560. yDate := "0000-00-00"
  561. if !maxDate.IsZero() {
  562. yDate = maxDate.Format(utils.FormatDate)
  563. }
  564. yDataList = append(yDataList, data_manage.YData{
  565. Date: yDate,
  566. Value: findDataList,
  567. Color: barChartInfoDate.Color,
  568. Name: barChartInfoDate.Name,
  569. })
  570. }
  571. return
  572. }
  573. func CheckIsEnChart(chartNameEn string, edbList []*data_manage.ChartEdbInfoMapping, source, chartType int) bool {
  574. // 相关性图表不判断指标
  575. if source == utils.CHART_SOURCE_CORRELATION && chartNameEn != "" {
  576. return true
  577. }
  578. if chartNameEn == "" {
  579. return false
  580. }
  581. // 截面散点图的话,肯定是有英文配置的
  582. if chartType == utils.CHART_TYPE_SECTION_SCATTER {
  583. return true
  584. }
  585. for _, v := range edbList {
  586. if v.EdbNameEn == "" {
  587. return false
  588. }
  589. if v.Unit != "无" && v.Unit != "" && v.UnitEn == "" {
  590. return false
  591. }
  592. }
  593. return true
  594. }
  595. func CheckIsEnEdb(edbNameEn, unit, unitEn string) bool {
  596. if edbNameEn == "" {
  597. return false
  598. }
  599. if unit != "无" && unit != "" && unitEn == "" {
  600. return false
  601. }
  602. return true
  603. }
  604. // CheckIsDisableChart 根据指标Id判断是否是禁用图表
  605. // return int 0:启用,1:禁用
  606. func CheckIsDisableChart(edbInfoIdArr []int) (isDisable int) {
  607. crmConfig, err := company.GetConfigDetailByCode("chart_disabled_edb")
  608. if err != nil {
  609. return
  610. }
  611. //没有配置,立马返回
  612. if crmConfig.ConfigValue == `` {
  613. return
  614. }
  615. confEdbInfoIdList := strings.Split(crmConfig.ConfigValue, ",")
  616. for _, edbInfoId := range edbInfoIdArr {
  617. for _, confEdbInfoId := range confEdbInfoIdList {
  618. if strconv.Itoa(edbInfoId) == confEdbInfoId {
  619. isDisable = 1
  620. return
  621. }
  622. }
  623. }
  624. return
  625. }
  626. // CheckChartExtraConfig 校验图表额外配置的信息,并且获取相关联的指标id
  627. func CheckChartExtraConfig(chartType int, extraConfigStr string) (edbIdList []int, err error, errMsg string) {
  628. switch chartType {
  629. case 10: //截面散点
  630. var extraConfig data_manage.SectionScatterReq
  631. err = json.Unmarshal([]byte(extraConfigStr), &extraConfig)
  632. if err != nil {
  633. return
  634. }
  635. // 判断是否有配置日期序列
  636. if len(extraConfig.SeriesList) <= 0 {
  637. errMsg = `请配置序列`
  638. err = errors.New(errMsg)
  639. return
  640. }
  641. // 判断是否有填写指标
  642. if len(extraConfig.SeriesList[0].EdbInfoList) <= 0 {
  643. errMsg = `请选择指标`
  644. err = errors.New(errMsg)
  645. return
  646. }
  647. // 遍历指标列表获取指标id
  648. edbIdMap := make(map[int]int)
  649. for _, v := range extraConfig.SeriesList[0].EdbInfoList {
  650. // X 轴的指标id
  651. if _, ok := edbIdMap[v.XEdbInfoId]; !ok {
  652. edbIdMap[v.XEdbInfoId] = v.XEdbInfoId
  653. edbIdList = append(edbIdList, v.XEdbInfoId)
  654. }
  655. // Y 轴的指标id
  656. if _, ok := edbIdMap[v.YEdbInfoId]; !ok {
  657. edbIdMap[v.YEdbInfoId] = v.YEdbInfoId
  658. edbIdList = append(edbIdList, v.YEdbInfoId)
  659. }
  660. }
  661. }
  662. return
  663. }
  664. // GetSectionScatterChartData 截面散点图的数据处理
  665. func GetSectionScatterChartData(mappingList []*data_manage.ChartEdbInfoMapping, edbDataListMap map[int][]*data_manage.EdbDataList, extraConfig data_manage.SectionScatterReq) (edbIdList []int, chartDataResp data_manage.SectionScatterInfoResp, err error) {
  666. // 指标数据数组(10086:{"2022-12-02":100.01,"2022-12-01":102.3})
  667. edbDataMap := make(map[int]map[string]float64)
  668. for edbInfoId, edbDataList := range edbDataListMap {
  669. edbDateData := make(map[string]float64)
  670. for _, edbData := range edbDataList {
  671. edbDateData[edbData.DataTime] = edbData.Value
  672. }
  673. edbDataMap[edbInfoId] = edbDateData
  674. }
  675. // edbIdList 指标展示顺序;x轴的指标顺序
  676. edbIdList = make([]int, 0)
  677. edbMappingMap := make(map[int]*data_manage.ChartEdbInfoMapping)
  678. for _, v := range mappingList {
  679. edbIdList = append(edbIdList, v.EdbInfoId)
  680. edbMappingMap[v.EdbInfoId] = v
  681. }
  682. //SectionScatterSeriesInfoResp
  683. dataListResp := make([]data_manage.SectionScatterSeriesItemResp, 0) //y轴的数据列表
  684. for _, seriesItem := range extraConfig.SeriesList {
  685. var maxDate time.Time
  686. // 系列中的指标数据
  687. tmpSeriesEdbInfoList := make([]data_manage.SectionScatterEdbItemResp, 0)
  688. var minXVal, maxXVal, minYVal, maxYVal float64
  689. for _, edbConf := range seriesItem.EdbInfoList {
  690. tmpItem := data_manage.SectionScatterEdbItemResp{
  691. IsShow: edbConf.IsShow,
  692. Name: edbConf.Name,
  693. NameEn: edbConf.NameEn,
  694. } //单个坐标点的数据
  695. //X轴的数据
  696. {
  697. edbInfoId := edbConf.XEdbInfoId //X轴的指标
  698. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  699. if !ok {
  700. continue
  701. }
  702. findDate := edbConf.XDate //需要的日期值
  703. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  704. if len(dataList) <= 0 {
  705. // 没有数据的指标id
  706. //findDataList = append(findDataList, 0)
  707. continue
  708. }
  709. tmpItem.XEdbInfoId = edbInfoId
  710. tmpItem.XName = edbMappingInfo.EdbName
  711. tmpItem.XNameEn = edbMappingInfo.EdbNameEn
  712. switch edbConf.XDateType {
  713. case 1: //最新值
  714. findDate = dataList[len(dataList)-1].DataTime
  715. case 2: //近期几天
  716. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  717. if tmpErr != nil {
  718. err = tmpErr
  719. return
  720. }
  721. findDateTime = findDateTime.AddDate(0, 0, -edbConf.XDateValue)
  722. lenData := len(dataList) - 1
  723. for i := lenData; i >= 0; i-- {
  724. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  725. if tmpErr != nil {
  726. err = tmpErr
  727. return
  728. }
  729. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  730. findDate = dataList[i].DataTime
  731. break
  732. }
  733. }
  734. case 3: // 固定日期
  735. //最早的日期
  736. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  737. if tmpErr != nil {
  738. err = tmpErr
  739. return
  740. }
  741. //寻找固定日期的数据
  742. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.XDate, time.Local)
  743. if tmpErr != nil {
  744. err = tmpErr
  745. return
  746. }
  747. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  748. tmpDate := tmpDateTime.Format(utils.FormatDate)
  749. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  750. findDate = tmpDate
  751. break
  752. }
  753. }
  754. default:
  755. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.XDate))
  756. return
  757. }
  758. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  759. if maxDate.IsZero() {
  760. maxDate = findDateTime
  761. } else {
  762. if findDateTime.After(maxDate) {
  763. maxDate = findDateTime
  764. }
  765. }
  766. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  767. tmpItem.XDate = findDate
  768. tmpItem.XValue = tmpValue
  769. } else {
  770. continue
  771. }
  772. }
  773. //Y轴的数据
  774. {
  775. edbInfoId := edbConf.YEdbInfoId //Y轴的指标
  776. edbMappingInfo, ok := edbMappingMap[edbInfoId]
  777. if !ok {
  778. continue
  779. }
  780. findDate := edbConf.YDate //需要的日期值
  781. dataList := edbDataListMap[edbInfoId] //指标的所有数据值
  782. if len(dataList) <= 0 {
  783. // 没有数据的指标id
  784. //findDataList = append(findDataList, 0)
  785. continue
  786. }
  787. tmpItem.YEdbInfoId = edbInfoId
  788. tmpItem.YName = edbMappingInfo.EdbName
  789. tmpItem.YNameEn = edbMappingInfo.EdbNameEn
  790. switch edbConf.YDateType {
  791. case 1: //最新值
  792. findDate = dataList[len(dataList)-1].DataTime
  793. case 2: //近期几天
  794. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[len(dataList)-1].DataTime, time.Local)
  795. if tmpErr != nil {
  796. err = tmpErr
  797. return
  798. }
  799. findDateTime = findDateTime.AddDate(0, 0, -edbConf.YDateValue)
  800. lenData := len(dataList) - 1
  801. for i := lenData; i >= 0; i-- {
  802. currDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[i].DataTime, time.Local)
  803. if tmpErr != nil {
  804. err = tmpErr
  805. return
  806. }
  807. if currDateTime.Equal(findDateTime) || currDateTime.Before(findDateTime) {
  808. findDate = dataList[i].DataTime
  809. break
  810. }
  811. }
  812. case 3: // 固定日期
  813. //最早的日期
  814. minDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  815. if tmpErr != nil {
  816. err = tmpErr
  817. return
  818. }
  819. //寻找固定日期的数据
  820. findDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, edbConf.YDate, time.Local)
  821. if tmpErr != nil {
  822. err = tmpErr
  823. return
  824. }
  825. for tmpDateTime := findDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
  826. tmpDate := tmpDateTime.Format(utils.FormatDate)
  827. if _, ok := edbDataMap[edbInfoId][tmpDate]; ok { //如果能找到数据,那么就返回
  828. findDate = tmpDate
  829. break
  830. }
  831. }
  832. default:
  833. err = errors.New(fmt.Sprint("日期类型异常,Type:", edbConf.YDate))
  834. return
  835. }
  836. findDateTime, _ := time.ParseInLocation(utils.FormatDate, findDate, time.Local)
  837. if maxDate.IsZero() {
  838. maxDate = findDateTime
  839. } else {
  840. if findDateTime.After(maxDate) {
  841. maxDate = findDateTime
  842. }
  843. }
  844. if tmpValue, ok := edbDataMap[edbInfoId][findDate]; ok {
  845. tmpItem.YDate = findDate
  846. tmpItem.YValue = tmpValue
  847. } else {
  848. continue
  849. }
  850. }
  851. // 获取当前系列的X轴的最大最小值
  852. {
  853. if tmpItem.XValue < minXVal {
  854. minXVal = tmpItem.XValue
  855. }
  856. if tmpItem.XValue > maxXVal {
  857. maxXVal = tmpItem.XValue
  858. }
  859. if tmpItem.YValue < minYVal {
  860. minYVal = tmpItem.YValue
  861. }
  862. if tmpItem.YValue > maxYVal {
  863. maxYVal = tmpItem.YValue
  864. }
  865. }
  866. tmpSeriesEdbInfoList = append(tmpSeriesEdbInfoList, tmpItem)
  867. }
  868. trendLimitData := make([]data_manage.CoordinatePoint, 0) //趋势线的前后坐标点
  869. var trendLine, rSquare string
  870. // 生成线性方程式
  871. var a, b float64
  872. {
  873. coordinateData := make([]utils.Coordinate, 0)
  874. for _, tmpSeriesEdbInfo := range tmpSeriesEdbInfoList {
  875. tmpCoordinate1 := utils.Coordinate{
  876. X: tmpSeriesEdbInfo.XValue,
  877. Y: tmpSeriesEdbInfo.YValue,
  878. }
  879. coordinateData = append(coordinateData, tmpCoordinate1)
  880. }
  881. // 只有存在两个坐标点的时候,才能去计算线性方程和R平方
  882. if len(coordinateData) >= 2 {
  883. a, b = utils.GetLinearResult(coordinateData)
  884. if !math.IsNaN(a) && !math.IsNaN(b) {
  885. if b > 0 {
  886. trendLine = fmt.Sprintf("y=%sx+%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  887. } else {
  888. trendLine = fmt.Sprintf("y=%sx%s", utils.SubFloatToString(a, 4), utils.SubFloatToString(b, 4))
  889. }
  890. minYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(minXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  891. maxYVal, _ = decimal.NewFromFloat(a).Mul(decimal.NewFromFloat(maxXVal)).Add(decimal.NewFromFloat(b)).Round(4).Float64()
  892. }
  893. // 计算R平方
  894. rSquare = fmt.Sprint(utils.CalculationDecisive(coordinateData))
  895. }
  896. trendLimitData = append(trendLimitData, data_manage.CoordinatePoint{
  897. X: minXVal,
  898. Y: minYVal,
  899. }, data_manage.CoordinatePoint{
  900. X: maxXVal,
  901. Y: maxYVal,
  902. })
  903. }
  904. dataListResp = append(dataListResp, data_manage.SectionScatterSeriesItemResp{
  905. Name: seriesItem.Name,
  906. NameEn: seriesItem.NameEn,
  907. Color: seriesItem.Color,
  908. EdbInfoList: tmpSeriesEdbInfoList,
  909. ShowTrendLine: seriesItem.ShowTrendLine,
  910. ShowFitEquation: seriesItem.ShowFitEquation,
  911. ShowRSquare: seriesItem.ShowRSquare,
  912. TrendLine: trendLine,
  913. RSquare: rSquare,
  914. TrendLimitData: trendLimitData,
  915. })
  916. }
  917. chartDataResp = data_manage.SectionScatterInfoResp{
  918. XName: extraConfig.XName,
  919. XNameEn: extraConfig.XNameEn,
  920. XUnitName: extraConfig.XUnitName,
  921. XUnitNameEn: extraConfig.XUnitNameEn,
  922. YName: extraConfig.YName,
  923. YNameEn: extraConfig.YNameEn,
  924. YUnitName: extraConfig.YUnitName,
  925. YUnitNameEn: extraConfig.YUnitNameEn,
  926. XMinValue: extraConfig.XMinValue,
  927. XMaxValue: extraConfig.XMaxValue,
  928. YMinValue: extraConfig.YMinValue,
  929. YMaxValue: extraConfig.YMaxValue,
  930. DataList: dataListResp,
  931. }
  932. return
  933. }
  934. // ChartInfoRefreshV2 图表刷新
  935. // @author Roc
  936. // @datetime 2022-09-16 10:15:38
  937. // @description 将原来自己写的一套获取所有关联指标,然后刷新指标逻辑 改成 只获取使用的指标id,然后遍历去调用“指标刷新服务”
  938. func ChartInfoRefreshV2(chartInfoId int) (err error) {
  939. var errmsg string
  940. defer func() {
  941. if err != nil {
  942. go alarm_msg.SendAlarmMsg("ChartInfoRefresh:"+errmsg, 3)
  943. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "ChartInfoRefresh:"+errmsg, utils.EmailSendToUsers)
  944. fmt.Println("ChartInfoRefresh Err:" + errmsg)
  945. }
  946. }()
  947. edbMappingList, err := data_manage.GetChartEdbMappingList(chartInfoId)
  948. if err != nil {
  949. errmsg = "获取需要刷新的指标失败:Err:" + err.Error()
  950. return
  951. }
  952. edbIdList := make([]int, 0)
  953. for _, v := range edbMappingList {
  954. edbIdList = append(edbIdList, v.EdbInfoId)
  955. }
  956. // 批量刷新
  957. err = EdbInfoRefreshAllFromBaseV3(edbIdList, false)
  958. if err != nil {
  959. return
  960. }
  961. return
  962. }
  963. // BatchChartInfoRefreshV2 图表批量刷新
  964. // @author Roc
  965. // @datetime 2022-09-16 10:15:38
  966. // @description 将原来自己写的一套获取所有关联指标,然后刷新指标逻辑 改成 只获取使用的指标id,然后遍历去调用“指标刷新服务”
  967. func BatchChartInfoRefreshV2(chartInfoList []*data_manage.ChartInfo) (err error) {
  968. if len(chartInfoList) <= 0 {
  969. return
  970. }
  971. chartInfoIdSlice := make([]int, 0)
  972. for _, chartInfo := range chartInfoList {
  973. chartInfoIdSlice = append(chartInfoIdSlice, chartInfo.ChartInfoId)
  974. }
  975. var errmsg string
  976. defer func() {
  977. if err != nil {
  978. go alarm_msg.SendAlarmMsg("ChartInfoRefresh:"+errmsg, 3)
  979. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "ChartInfoRefresh:"+errmsg, utils.EmailSendToUsers)
  980. fmt.Println("ChartInfoRefresh Err:" + errmsg)
  981. }
  982. }()
  983. edbIdList := make([]int, 0)
  984. edbInfoIdMap := make(map[int]int)
  985. edbMappingList, err := data_manage.GetChartEdbMappingListByIdList(chartInfoIdSlice)
  986. for _, v := range edbMappingList {
  987. if _, ok := edbInfoIdMap[v.EdbInfoId]; !ok {
  988. edbIdList = append(edbIdList, v.EdbInfoId)
  989. edbInfoIdMap[v.EdbInfoId] = v.EdbInfoId
  990. }
  991. }
  992. // 批量刷新
  993. err = EdbInfoRefreshAllFromBaseV3(edbIdList, false)
  994. if err != nil {
  995. return
  996. }
  997. return
  998. }