chart_info.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683
  1. package line_feature
  2. import (
  3. "errors"
  4. "eta/eta_mobile/models/data_manage"
  5. "eta/eta_mobile/models/data_manage/line_feature"
  6. "eta/eta_mobile/models/data_manage/line_feature/response"
  7. "eta/eta_mobile/models/system"
  8. "eta/eta_mobile/services/data"
  9. "eta/eta_mobile/utils"
  10. "github.com/shopspring/decimal"
  11. "strconv"
  12. "strings"
  13. "time"
  14. )
  15. // GetStandardDeviationData 获取标准差图表的指标数据
  16. func GetStandardDeviationData(chartInfoId int, startDate, endDate string, mappingInfo *data_manage.ChartEdbInfoMapping, calculateValue int) (edbList []*data_manage.ChartEdbInfoMapping, dataResp response.LineFeatureDataResp, err error, errMsg string) {
  17. edbList = make([]*data_manage.ChartEdbInfoMapping, 0)
  18. // 指标对应的所有数据
  19. _, edbList, err = data.GetEdbDataMapList(chartInfoId, 1, `公历`, startDate, endDate, []*data_manage.ChartEdbInfoMapping{mappingInfo}, "")
  20. if err != nil {
  21. return
  22. }
  23. if len(edbList) != 1 {
  24. errMsg = `指标异常`
  25. err = errors.New(errMsg)
  26. return
  27. }
  28. edb := edbList[0]
  29. dataList := edb.DataList.([]*data_manage.EdbDataList)
  30. newDataList := make([]data_manage.EdbDataList, 0)
  31. lenData := len(dataList)
  32. var minVal, maxVal float64
  33. if lenData >= calculateValue {
  34. tmpDataList := make([]float64, 0)
  35. for _, tmpData := range dataList {
  36. tmpDataList = append(tmpDataList, tmpData.Value)
  37. }
  38. for i := calculateValue; i <= lenData; i++ {
  39. tmpV := utils.CalculateStandardDeviation(tmpDataList[i-calculateValue : i])
  40. tmpV, _ = decimal.NewFromFloat(tmpV).Round(4).Float64()
  41. newDataList = append(newDataList, data_manage.EdbDataList{
  42. EdbDataId: i,
  43. EdbInfoId: edb.EdbInfoId,
  44. DataTime: dataList[i-1].DataTime,
  45. DataTimestamp: dataList[i-1].DataTimestamp,
  46. Value: tmpV,
  47. })
  48. if tmpV > maxVal {
  49. maxVal = tmpV
  50. }
  51. if tmpV < minVal {
  52. minVal = tmpV
  53. }
  54. }
  55. }
  56. dataResp = response.LineFeatureDataResp{
  57. MaxData: maxVal,
  58. MinData: minVal,
  59. LatestDate: edb.LatestDate,
  60. EdbInfoCategoryType: edb.EdbInfoCategoryType,
  61. ChartColor: `#00F`,
  62. ChartStyle: `spline`,
  63. PredictChartColor: `#00F`,
  64. ChartType: 0,
  65. ChartWidth: 3,
  66. EdbName: "标准差",
  67. EdbNameEn: "standard deviation",
  68. Unit: edb.Unit,
  69. UnitEn: edb.UnitEn,
  70. IsAxis: 1,
  71. DataList: newDataList,
  72. }
  73. return
  74. }
  75. // GetPercentileData 获取百分位图表的指标数据
  76. func GetPercentileData(chartInfoId int, startDate, endDate string, mappingInfo *data_manage.ChartEdbInfoMapping, calculateValue int, calculateUnit string, percentType int) (edbList []*data_manage.ChartEdbInfoMapping, dataResp response.LineFeatureDataResp, err error, errMsg string) {
  77. edbList = make([]*data_manage.ChartEdbInfoMapping, 0)
  78. moveUnitDays, ok := utils.FrequencyDaysMap[calculateUnit]
  79. if !ok {
  80. errMsg = `错误的周期`
  81. err = errors.New(errMsg)
  82. return
  83. }
  84. calculateDay := calculateValue * moveUnitDays
  85. // 指标对应的所有数据
  86. _, edbList, err = data.GetEdbDataMapList(chartInfoId, 1, `公历`, startDate, endDate, []*data_manage.ChartEdbInfoMapping{mappingInfo}, "")
  87. if err != nil {
  88. return
  89. }
  90. if len(edbList) != 1 {
  91. errMsg = `指标异常`
  92. err = errors.New(errMsg)
  93. return
  94. }
  95. edb := edbList[0]
  96. dataList := edb.DataList.([]*data_manage.EdbDataList)
  97. newDataList := make([]data_manage.EdbDataList, 0)
  98. var edbMinVal, edbMaxVal float64
  99. dataMap := make(map[time.Time]float64, 0)
  100. for _, tmpData := range dataList {
  101. currDateTime, _ := time.ParseInLocation(utils.FormatDate, tmpData.DataTime, time.Local)
  102. dataMap[currDateTime] = tmpData.Value
  103. }
  104. //百分位:对所选指标滚动地取对应时间长度的数据值,取最大值Max,最小值Min,计算Max-Min,百分位=(现值-Min)/(Max-Min),Max=Min时不予计算。
  105. if percentType == utils.PercentCalculateTypeRange {
  106. for i, tmpData := range dataList {
  107. currDateTime, _ := time.ParseInLocation(utils.FormatDate, tmpData.DataTime, time.Local)
  108. maxVal := tmpData.Value
  109. minVal := tmpData.Value
  110. for k := 0; k < calculateDay; k++ {
  111. preVal, ok2 := dataMap[currDateTime.AddDate(0, 0, -k)]
  112. if ok2 {
  113. if preVal > maxVal {
  114. maxVal = preVal
  115. }
  116. if preVal < minVal {
  117. minVal = preVal
  118. }
  119. }
  120. }
  121. if maxVal == minVal {
  122. continue
  123. }
  124. //百分位=(现值-Min)/(Max-Min)
  125. tmpV := (tmpData.Value - minVal) / (maxVal - minVal) * 100
  126. tmpV, _ = decimal.NewFromFloat(tmpV).Round(4).Float64()
  127. newDataList = append(newDataList, data_manage.EdbDataList{
  128. EdbDataId: i,
  129. EdbInfoId: edb.EdbInfoId,
  130. DataTime: dataList[i].DataTime,
  131. DataTimestamp: dataList[i].DataTimestamp,
  132. Value: tmpV,
  133. })
  134. if tmpV < edbMinVal {
  135. edbMinVal = tmpV
  136. }
  137. if tmpV > edbMaxVal {
  138. edbMaxVal = tmpV
  139. }
  140. }
  141. }
  142. // 百分位数据个数算法
  143. // 数据区间第一个和最后一个数据点的时间和数据分别为(T1,S1)(T2,S2); N=T1到T2指标数据个数, n=小于等于S2的数据个数
  144. // 个数百分位=(n-1)/(N-1)
  145. if percentType == utils.PercentCalculateTypeNum {
  146. for i, d := range dataList {
  147. // T2为当前日期
  148. s2 := decimal.NewFromFloat(d.Value)
  149. t2, _ := time.ParseInLocation(utils.FormatDate, d.DataTime, time.Local)
  150. // 计算N和n
  151. var bigN, tinyN int
  152. for k := 0; k < calculateDay; k++ {
  153. preVal, preOk := dataMap[t2.AddDate(0, 0, -k)]
  154. if !preOk {
  155. continue
  156. }
  157. bigN += 1
  158. if decimal.NewFromFloat(preVal).LessThanOrEqual(s2) {
  159. tinyN += 1
  160. }
  161. }
  162. // N=1时说明计算无效
  163. if bigN == 1 {
  164. continue
  165. }
  166. numerator := decimal.NewFromInt(int64(tinyN - 1))
  167. denominator := decimal.NewFromInt(int64(bigN - 1))
  168. percentVal, _ := numerator.Div(denominator).Mul(decimal.NewFromFloat(100)).Round(4).Float64()
  169. // 写进数组并判断指标最大最小值
  170. newDataList = append(newDataList, data_manage.EdbDataList{
  171. EdbDataId: i,
  172. EdbInfoId: edb.EdbInfoId,
  173. DataTime: dataList[i].DataTime,
  174. DataTimestamp: dataList[i].DataTimestamp,
  175. Value: percentVal,
  176. })
  177. if percentVal < edbMinVal {
  178. edbMinVal = percentVal
  179. }
  180. if percentVal > edbMaxVal {
  181. edbMaxVal = percentVal
  182. }
  183. }
  184. }
  185. dataResp = response.LineFeatureDataResp{
  186. MaxData: edbMaxVal,
  187. MinData: edbMinVal,
  188. LatestDate: edb.LatestDate,
  189. EdbInfoCategoryType: edb.EdbInfoCategoryType,
  190. ChartColor: `#00F`,
  191. ChartStyle: `spline`,
  192. PredictChartColor: `#00F`,
  193. ChartType: 0,
  194. ChartWidth: 3,
  195. EdbName: "百分位",
  196. EdbNameEn: "percentile",
  197. Unit: "%",
  198. UnitEn: "%",
  199. IsAxis: 1,
  200. DataList: newDataList,
  201. }
  202. return
  203. }
  204. // GetFrequencyDistributionData 获取频率分布的图表数据
  205. func GetFrequencyDistributionData(chartInfoId int, mappingInfo *data_manage.ChartEdbInfoMapping, dateType, stepVal int, startDate, endDate string) (edbList []*data_manage.ChartEdbInfoMapping, dataResp response.FrequencyDistributionResp, err error, errMsg string) {
  206. XDataList := make([]float64, 0)
  207. // 频度
  208. Y1DataList := make([]response.FrequencyDistributionYData, 0)
  209. // 累计频率
  210. Y2DataList := make([]response.FrequencyDistributionYData, 0)
  211. edbList = make([]*data_manage.ChartEdbInfoMapping, 0)
  212. // 指标对应的所有数据
  213. _, edbList, err = data.GetEdbDataMapList(chartInfoId, 1, `公历`, startDate, endDate, []*data_manage.ChartEdbInfoMapping{mappingInfo}, "")
  214. if err != nil {
  215. return
  216. }
  217. if len(edbList) != 1 {
  218. err = errors.New("指标异常")
  219. return
  220. }
  221. startDateTime, _ := time.ParseInLocation(utils.FormatDate, startDate, time.Local)
  222. var endDateTime time.Time
  223. if endDate != `` {
  224. endDateTime, _ = time.ParseInLocation(utils.FormatDate, endDate, time.Local)
  225. }
  226. edb := edbList[0]
  227. dataList := edb.DataList.([]*data_manage.EdbDataList)
  228. if len(dataList) <= 0 {
  229. return
  230. }
  231. // 非自定义
  232. if dateType != 8 {
  233. endDate = dataList[len(dataList)-1].DataTime
  234. endDateTime, err = time.ParseInLocation(utils.FormatDate, endDate, time.Local)
  235. if err != nil {
  236. return
  237. }
  238. //日期类型:1:最近3月;2:最近6月;3:最近1年;4:最近2年;5:最近3年;6:最近5年;7:最近10年,8:自定义时间
  239. startDateTime = utils.GetDateByDateType2(dateType, endDateTime)
  240. startDate = startDateTime.Format(utils.FormatDate)
  241. newDataList := make([]*data_manage.EdbDataList, 0)
  242. for _, v := range dataList {
  243. tmpDataTime, tmpErr := time.ParseInLocation(utils.FormatDate, v.DataTime, time.Local)
  244. if tmpErr != nil {
  245. err = tmpErr
  246. return
  247. }
  248. if tmpDataTime.Equal(startDateTime) || tmpDataTime.After(startDateTime) {
  249. newDataList = append(newDataList, v)
  250. }
  251. }
  252. dataList = newDataList
  253. }
  254. maxVal := dataList[0].Value
  255. minVal := dataList[0].Value
  256. dataValMap := make(map[float64]int)
  257. total := 0 // 数据总量
  258. for _, tmpData := range dataList {
  259. currDateTime, _ := time.ParseInLocation(utils.FormatDate, tmpData.DataTime, time.Local)
  260. if (currDateTime.Equal(startDateTime) || currDateTime.After(startDateTime)) && (endDateTime.IsZero() || currDateTime.Before(endDateTime)) {
  261. if maxVal < tmpData.Value {
  262. maxVal = tmpData.Value
  263. }
  264. if minVal > tmpData.Value {
  265. minVal = tmpData.Value
  266. }
  267. num, ok := dataValMap[tmpData.Value]
  268. if ok {
  269. dataValMap[tmpData.Value] = num + 1
  270. } else {
  271. dataValMap[tmpData.Value] = 1
  272. }
  273. total++
  274. }
  275. }
  276. if total <= 0 {
  277. errMsg = `没有数据`
  278. err = errors.New(errMsg)
  279. return
  280. }
  281. // 最大最小值 向上/下取整
  282. minVal = utils.GetFloorNewNum(minVal, 2)
  283. maxVal = utils.GetCeilNewNum(maxVal, 2)
  284. //间距
  285. spacing, _ := (decimal.NewFromFloat(maxVal).Sub(decimal.NewFromFloat(minVal))).Div(decimal.NewFromInt(int64(stepVal))).Float64()
  286. distributionDataNumMap := make(map[float64]int)
  287. for i := 1; i <= stepVal; i++ {
  288. tmpMinVal, _ := decimal.NewFromFloat(minVal).Add((decimal.NewFromFloat(spacing)).Mul(decimal.NewFromInt(int64(i - 1)))).Float64()
  289. tmpMaxVal, _ := decimal.NewFromFloat(minVal).Add((decimal.NewFromFloat(spacing)).Mul(decimal.NewFromInt(int64(i)))).Float64()
  290. XDataList = append(XDataList, tmpMinVal)
  291. distributionDataNumMap[tmpMinVal] = 0
  292. for tmpVal, num := range dataValMap {
  293. if tmpMinVal <= tmpVal {
  294. // 最后一期数据是要小于等于
  295. if i == stepVal {
  296. if tmpVal <= tmpMaxVal {
  297. distributionDataNumMap[tmpMinVal] += num
  298. }
  299. } else {
  300. if tmpVal < tmpMaxVal {
  301. distributionDataNumMap[tmpMinVal] += num
  302. }
  303. }
  304. }
  305. }
  306. }
  307. var minFrequency, maxFrequency float64
  308. tmpNum := 0
  309. for k, tmpMinVal := range XDataList {
  310. // 数量
  311. frequencyYNum := distributionDataNumMap[tmpMinVal]
  312. // 频率
  313. tmpFrequency, _ := decimal.NewFromInt(int64(frequencyYNum)).Div(decimal.NewFromInt(int64(total))).Mul(decimal.NewFromInt(100)).Round(4).Float64()
  314. Y1DataList = append(Y1DataList, response.FrequencyDistributionYData{
  315. X: tmpMinVal,
  316. Y: tmpFrequency,
  317. })
  318. if k == 0 {
  319. minFrequency = tmpFrequency
  320. maxFrequency = tmpFrequency
  321. } else {
  322. if tmpFrequency < minFrequency {
  323. minFrequency = tmpFrequency
  324. }
  325. if tmpFrequency > maxFrequency {
  326. maxFrequency = tmpFrequency
  327. }
  328. }
  329. // 累计数
  330. tmpNum += frequencyYNum
  331. // 累计频率
  332. tmpTotalFrequency, _ := decimal.NewFromInt(int64(tmpNum)).Div(decimal.NewFromInt(int64(total))).Mul(decimal.NewFromInt(100)).Round(4).Float64()
  333. Y2DataList = append(Y2DataList, response.FrequencyDistributionYData{
  334. X: tmpMinVal,
  335. Y: tmpTotalFrequency,
  336. })
  337. }
  338. newDataList := []response.FrequencyDistributionData{
  339. {
  340. Name: "频率",
  341. NameEn: "Frequency",
  342. Unit: "%",
  343. UnitEn: "%",
  344. Value: Y1DataList,
  345. Color: "#00F",
  346. IsAxis: 1,
  347. }, {
  348. Name: "累计频率",
  349. NameEn: "Total Frequency",
  350. Unit: "%",
  351. UnitEn: "%",
  352. Value: Y2DataList,
  353. Color: "#F00",
  354. IsAxis: 0,
  355. },
  356. }
  357. edbList[0].DataList = nil
  358. dataResp = response.FrequencyDistributionResp{
  359. LeftMinValue: minFrequency,
  360. LeftMaxValue: maxFrequency,
  361. RightMinValue: 0,
  362. RightMaxValue: 100,
  363. DataList: newDataList,
  364. }
  365. return
  366. }
  367. // AddChartInfo 添加图表
  368. func AddChartInfo(req data_manage.AddChartInfoReq, edbInfoMapping *data_manage.ChartEdbInfoMapping, source int, sysUser *system.Admin) (chartInfo *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  369. isSendEmail = true
  370. req.ChartName = strings.Trim(req.ChartName, " ")
  371. if req.ChartName == "" {
  372. errMsg = "请填写图表名称!"
  373. err = errors.New(errMsg)
  374. isSendEmail = false
  375. return
  376. }
  377. if req.ChartClassifyId <= 0 {
  378. errMsg = "分类参数错误!"
  379. err = errors.New(errMsg)
  380. isSendEmail = false
  381. return
  382. }
  383. chartClassify, err := data_manage.GetChartClassifyById(req.ChartClassifyId)
  384. if err != nil {
  385. if err.Error() == utils.ErrNoRow() {
  386. errMsg = "分类不存在"
  387. err = errors.New(errMsg)
  388. isSendEmail = false
  389. return
  390. }
  391. errMsg = "获取分类信息失败"
  392. err = errors.New("获取分类信息失败,Err:" + err.Error())
  393. return
  394. }
  395. if chartClassify == nil {
  396. errMsg = "分类不存在"
  397. err = errors.New(errMsg)
  398. isSendEmail = false
  399. return
  400. }
  401. edbInfoIdArr := []int{edbInfoMapping.EdbInfoId}
  402. var edbInfoIdArrStr []string
  403. for _, v := range edbInfoIdArr {
  404. edbInfoIdArrStr = append(edbInfoIdArrStr, strconv.Itoa(v))
  405. }
  406. edbInfoIdStr := strings.Join(edbInfoIdArrStr, ",")
  407. var chartInfoId int
  408. // 判断图表是否存在
  409. var condition string
  410. var pars []interface{}
  411. condition += " AND chart_name=? AND source = ? "
  412. pars = append(pars, req.ChartName, source)
  413. count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  414. if err != nil {
  415. errMsg = "判断图表名称是否存在失败"
  416. err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  417. return
  418. }
  419. if count > 0 {
  420. errMsg = "图表已存在,请重新填写"
  421. err = errors.New(errMsg)
  422. isSendEmail = false
  423. return
  424. }
  425. disableVal := data.CheckIsDisableChart(edbInfoIdArr)
  426. chartInfo = new(data_manage.ChartInfo)
  427. chartInfo.ChartName = req.ChartName
  428. chartInfo.EdbInfoIds = edbInfoIdStr
  429. chartInfo.ChartClassifyId = req.ChartClassifyId
  430. chartInfo.SysUserId = sysUser.AdminId
  431. chartInfo.SysUserRealName = sysUser.RealName
  432. chartInfo.CreateTime = time.Now()
  433. chartInfo.ModifyTime = time.Now()
  434. chartInfo.IsSetName = 0
  435. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  436. chartInfo.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
  437. chartInfo.ChartType = 9 // 相关性图
  438. chartInfo.Calendar = "公历"
  439. chartInfo.DateType = 6
  440. chartInfo.StartDate = req.StartDate
  441. chartInfo.EndDate = req.EndDate
  442. chartInfo.SeasonStartDate = req.SeasonStartDate
  443. chartInfo.SeasonEndDate = req.SeasonEndDate
  444. chartInfo.ChartImage = req.ChartImage
  445. chartInfo.LeftMin = req.LeftMin
  446. chartInfo.LeftMax = req.LeftMax
  447. chartInfo.RightMin = req.RightMin
  448. chartInfo.RightMax = req.RightMax
  449. chartInfo.Disabled = disableVal
  450. chartInfo.Source = source
  451. chartInfo.ExtraConfig = req.ExtraConfig
  452. // 指标信息
  453. mapList := make([]*data_manage.ChartEdbMapping, 0)
  454. {
  455. mapItem := new(data_manage.ChartEdbMapping)
  456. mapItem.EdbInfoId = edbInfoMapping.EdbInfoId
  457. mapItem.CreateTime = time.Now()
  458. mapItem.ModifyTime = time.Now()
  459. mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + strconv.FormatInt(time.Now().UnixNano(), 10) + "_" + strconv.Itoa(edbInfoMapping.EdbInfoId))
  460. mapItem.IsOrder = true
  461. mapItem.IsAxis = 1
  462. mapItem.EdbInfoType = 1
  463. mapItem.Source = utils.CHART_SOURCE_CORRELATION
  464. mapList = append(mapList, mapItem)
  465. }
  466. // 新增图表和指标mapping
  467. chartInfoId, e := line_feature.CreateLineFeatureChartAndEdb(chartInfo, mapList)
  468. if e != nil {
  469. errMsg = "操作失败"
  470. err = errors.New("新增相关性图表失败, Err: " + e.Error())
  471. return
  472. }
  473. //添加es数据
  474. go data.EsAddOrEditChartInfo(chartInfoId)
  475. return
  476. }
  477. // EditChartInfo 编辑图表
  478. func EditChartInfo(req data_manage.EditChartInfoReq, edbInfoMapping *data_manage.ChartEdbInfoMapping, sysUser *system.Admin) (chartItem *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  479. isSendEmail = true
  480. chartItem, err = data_manage.GetChartInfoById(req.ChartInfoId)
  481. if err != nil {
  482. if err.Error() == utils.ErrNoRow() {
  483. errMsg = "图表已被删除,请刷新页面"
  484. err = errors.New(errMsg)
  485. isSendEmail = false
  486. return
  487. }
  488. errMsg = "获取图表信息失败"
  489. err = errors.New("获取图表信息失败,Err:" + err.Error())
  490. return
  491. }
  492. if !utils.InArrayByInt([]int{utils.CHART_SOURCE_LINE_FEATURE_STANDARD_DEVIATION, utils.CHART_SOURCE_LINE_FEATURE_PERCENTILE, utils.CHART_SOURCE_LINE_FEATURE_FREQUENCY}, chartItem.Source) {
  493. errMsg = "该图不是统计分析图表!"
  494. err = errors.New(errMsg)
  495. isSendEmail = false
  496. return
  497. }
  498. req.ChartName = strings.Trim(req.ChartName, " ")
  499. if req.ChartClassifyId <= 0 {
  500. errMsg = "分类参数错误!"
  501. err = errors.New(errMsg)
  502. isSendEmail = false
  503. return
  504. }
  505. chartClassify, err := data_manage.GetChartClassifyById(req.ChartClassifyId)
  506. if err != nil {
  507. if err.Error() == utils.ErrNoRow() {
  508. errMsg = "分类不存在"
  509. err = errors.New(errMsg)
  510. isSendEmail = false
  511. return
  512. }
  513. errMsg = "获取分类信息失败"
  514. err = errors.New("获取分类信息失败,Err:" + err.Error())
  515. return
  516. }
  517. if chartClassify == nil {
  518. errMsg = "分类不存在"
  519. err = errors.New(errMsg)
  520. isSendEmail = false
  521. return
  522. }
  523. // 图表操作权限
  524. ok := data.CheckOpChartPermission(sysUser, chartItem.SysUserId)
  525. if !ok {
  526. errMsg = "没有该图表的操作权限"
  527. err = errors.New(errMsg)
  528. isSendEmail = false
  529. return
  530. }
  531. edbInfoIdArr := []int{edbInfoMapping.EdbInfoId}
  532. var edbInfoIdArrStr []string
  533. for _, v := range edbInfoIdArr {
  534. edbInfoIdArrStr = append(edbInfoIdArrStr, strconv.Itoa(v))
  535. }
  536. edbInfoIdStr := strings.Join(edbInfoIdArrStr, ",")
  537. //判断图表是否存在
  538. var condition string
  539. var pars []interface{}
  540. condition += " AND chart_info_id <> ? "
  541. pars = append(pars, req.ChartInfoId)
  542. condition += " AND chart_name=? AND source = ? "
  543. pars = append(pars, req.ChartName, chartItem.Source)
  544. count, err := data_manage.GetChartInfoCountByCondition(condition, pars)
  545. if err != nil {
  546. errMsg = "判断图表名称是否存在失败"
  547. err = errors.New("判断图表名称是否存在失败,Err:" + err.Error())
  548. return
  549. }
  550. if count > 0 {
  551. errMsg = "图表已存在,请重新填写"
  552. err = errors.New(errMsg)
  553. isSendEmail = false
  554. return
  555. }
  556. // 图表启用与否
  557. disableVal := data.CheckIsDisableChart(edbInfoIdArr)
  558. // 修改图表与指标mapping
  559. err = line_feature.EditLineFeatureChartAndMapping(&req, edbInfoIdStr, "公历", 6, disableVal, req.ExtraConfig)
  560. if err != nil {
  561. errMsg = "保存失败"
  562. err = errors.New("保存失败,Err:" + err.Error())
  563. return
  564. }
  565. resp := new(data_manage.AddChartInfoResp)
  566. resp.ChartInfoId = chartItem.ChartInfoId
  567. resp.UniqueCode = chartItem.UniqueCode
  568. resp.ChartType = req.ChartType
  569. //添加es数据
  570. go data.EsAddOrEditChartInfo(chartItem.ChartInfoId)
  571. //修改my eta es数据
  572. go data.EsAddOrEditMyChartInfoByChartInfoId(chartItem.ChartInfoId)
  573. return
  574. }
  575. // CopyChartInfo 复制图表
  576. func CopyChartInfo(configId, configSource, classifyId int, chartName string, edbInfoMapping *data_manage.ChartEdbInfoMapping, oldChartInfo *data_manage.ChartInfo, sysUser *system.Admin) (chartInfo *data_manage.ChartInfo, err error, errMsg string, isSendEmail bool) {
  577. isSendEmail = true
  578. multipleGraphConfig, err := data_manage.GetMultipleGraphConfigById(configId)
  579. if err != nil {
  580. return
  581. }
  582. multipleGraphConfig.MultipleGraphConfigId = 0
  583. err = data_manage.AddMultipleGraphConfig(multipleGraphConfig)
  584. if err != nil {
  585. return
  586. }
  587. // 添加图
  588. addChartReq := data_manage.AddChartInfoReq{
  589. ChartClassifyId: classifyId,
  590. ChartName: chartName,
  591. ChartType: utils.CHART_TYPE_CURVE,
  592. Calendar: "公历",
  593. ExtraConfig: oldChartInfo.ExtraConfig,
  594. ChartImage: oldChartInfo.ChartImage,
  595. }
  596. chartSource := oldChartInfo.Source // 默认是相关性图
  597. chartInfo, err, errMsg, isSendEmail = AddChartInfo(addChartReq, edbInfoMapping, chartSource, sysUser)
  598. if err != nil {
  599. return
  600. }
  601. // 添加关系
  602. multipleGraphConfigChartMapping := &data_manage.MultipleGraphConfigChartMapping{
  603. //Id: 0,
  604. MultipleGraphConfigId: multipleGraphConfig.MultipleGraphConfigId,
  605. ChartInfoId: chartInfo.ChartInfoId,
  606. Source: configSource,
  607. ModifyTime: time.Now(),
  608. CreateTime: time.Now(),
  609. }
  610. err = data_manage.AddMultipleGraphConfigChartMapping(multipleGraphConfigChartMapping)
  611. if err != nil {
  612. return
  613. }
  614. //添加es数据
  615. go data.EsAddOrEditChartInfo(chartInfo.ChartInfoId)
  616. return
  617. }