chart.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710
  1. package models
  2. import (
  3. "errors"
  4. "eta/eta_chart_lib/models/mgo"
  5. "eta/eta_chart_lib/utils"
  6. "fmt"
  7. "github.com/beego/beego/v2/client/orm"
  8. "go.mongodb.org/mongo-driver/bson"
  9. "strconv"
  10. "time"
  11. "github.com/nosixtools/solarlunar"
  12. )
  13. type ChartInfo struct {
  14. ChartInfoId int `json:"-" orm:"column(chart_info_id);pk"`
  15. ChartName string `description:"来源名称"`
  16. ChartNameEn string `description:"英文图表名称"`
  17. ChartClassifyId int `json:"-" description:"图表分类id"`
  18. SysUserId int `json:"-"`
  19. SysUserRealName string `json:"-"`
  20. UniqueCode string `description:"图表唯一编码"`
  21. CreateTime time.Time `json:"-"`
  22. ModifyTime time.Time `json:"-"`
  23. DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间"`
  24. StartDate string `description:"自定义开始日期"`
  25. EndDate string `description:"自定义结束日期"`
  26. IsSetName int `description:"设置名称"`
  27. EdbInfoIds string `description:"指标id"`
  28. ChartType int `description:"生成样式:1:曲线图,2:季节性图"`
  29. Calendar string `description:"公历/农历"`
  30. SeasonStartDate string `description:"季节性图开始日期"`
  31. SeasonEndDate string `description:"季节性图开始日期"`
  32. ChartImage string `description:"图表图片"`
  33. BarConfig string `description:"柱方图的配置,json数据" json:"-"`
  34. XMin string `description:"图表X轴最小值"`
  35. XMax string `description:"图表X轴最大值"`
  36. LeftMin string `description:"图表左侧最小值"`
  37. LeftMax string `description:"图表左侧最大值"`
  38. RightMin string `description:"图表右侧最小值"`
  39. RightMax string `description:"图表右侧最大值"`
  40. Right2Min string `description:"图表右侧2最小值"`
  41. Right2Max string `description:"图表右侧2最大值"`
  42. MinMaxSave int `description:"是否手动保存过上下限:0-否;1-是"`
  43. Source int `description:"1:ETA图库;2:商品价格曲线"`
  44. UnitEn string `description:"英文图表单位"`
  45. Unit string `description:"图表单位"`
  46. ExtraConfig string `description:"图表额外配置,json数据" json:"-"`
  47. ChartSource string `description:"图表来源str"`
  48. ChartSourceEn string `description:"图表来源(英文)"`
  49. SeasonExtraConfig string `description:"季节性图表中的配置,json数据"`
  50. StartYear int `description:"当选择的日期类型为最近N年类型时,即date_type=20, 用start_year表示N"`
  51. ChartThemeId int `description:"图表应用主题ID"`
  52. ChartThemeStyle string `description:"图表应用主题样式"`
  53. SourcesFrom string `description:"图表来源"`
  54. Instructions string `description:"图表说明"`
  55. MarkersLines string `description:"标识线"`
  56. MarkersAreas string `description:"标识区"`
  57. }
  58. func GetChartInfoByUniqueCode(uniqueCode string) (item *ChartInfo, err error) {
  59. o := orm.NewOrmUsingDB("data")
  60. sql := ` SELECT * FROM chart_info WHERE unique_code=? `
  61. err = o.Raw(sql, uniqueCode).QueryRow(&item)
  62. return
  63. }
  64. type ChartEdbInfoMapping struct {
  65. EdbInfoId int `description:"指标id"`
  66. SourceName string `description:"来源名称"`
  67. Source int `description:"来源id"`
  68. EdbCode string `description:"指标编码"`
  69. EdbName string `description:"指标名称"`
  70. EdbAliasName string `description:"指标名称(别名)"`
  71. EdbAliasNameEn string `description:"英文指标名称(别名)"`
  72. EdbNameEn string `description:"英文指标名称"`
  73. Frequency string `description:"频率"`
  74. FrequencyEn string `description:"英文频率"`
  75. Unit string `description:"单位"`
  76. UnitEn string `description:"英文单位"`
  77. StartDate string `description:"起始日期"`
  78. EndDate string `description:"终止日期"`
  79. ModifyTime string `description:"指标最后更新时间"`
  80. ChartEdbMappingId int `description:"图表指标id" json:"-"`
  81. ChartInfoId int `description:"图表id"`
  82. MaxData float64 `description:"上限"`
  83. MinData float64 `description:"下限"`
  84. IsOrder bool `description:"true:正序,false:逆序"`
  85. IsAxis int `description:"1:左轴,0:右轴"`
  86. EdbInfoType int `description:"1:标准指标,0:领先指标"`
  87. EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
  88. LeadValue int `description:"领先值"`
  89. LeadUnit string `description:"领先单位"`
  90. LeadUnitEn string `description:"领先英文单位"`
  91. ChartStyle string `description:"图表类型"`
  92. ChartColor string `description:"颜色"`
  93. ChartWidth float64 `description:"线条大小"`
  94. DataList interface{}
  95. EdbInfoCategoryType int `description:"0:普通指标,1:预测指标"`
  96. PredictChartColor string `description:"预测数据的颜色"`
  97. ChartType int `description:"生成样式:1:曲线图,2:季节性图,3:面积图,4:柱状图,5:散点图,6:组合图"`
  98. LatestDate string `description:"数据最新日期"`
  99. LatestValue float64 `description:"数据最新值"`
  100. MoveLatestDate string `description:"移动后的数据最新日期"`
  101. UniqueCode string `description:"指标唯一编码"`
  102. MinValue float64 `json:"-" description:"最小值"`
  103. MaxValue float64 `json:"-" description:"最大值"`
  104. IsNullData bool `json:"-" description:"是否空数据"`
  105. MappingSource int `description:"1:ETA图库;2:商品价格曲线"`
  106. SubSource int `description:"子数据来源:0:经济数据库,1:日期序列"`
  107. SubSourceName string `description:"子数据来源名称"`
  108. IndicatorCode string `description:"指标代码"`
  109. StockCode string `description:"证券代码"`
  110. IsConvert int `description:"是否数据转换 0不转 1转"`
  111. ConvertType int `description:"数据转换类型 1乘 2除 3对数"`
  112. ConvertValue float64 `description:"数据转换值"`
  113. ConvertUnit string `description:"数据转换单位"`
  114. ConvertEnUnit string `description:"数据转换单位"`
  115. }
  116. func GetChartEdbMappingList(chartInfoId int) (list []*ChartEdbInfoMapping, err error) {
  117. o := orm.NewOrmUsingDB("data")
  118. aField := `a.chart_edb_mapping_id,a.chart_info_id,a.edb_info_id,a.create_time,a.modify_time,a.unique_code,a.max_data,a.min_data,a.is_order,a.is_axis,a.edb_info_type,a.lead_value,a.lead_unit,a.chart_style,a.chart_color,a.predict_chart_color,a.chart_width,a.source as mapping_source,a.edb_alias_name`
  119. sql := ` SELECT ` + aField + `,b.source_name,b.source,b.edb_code,b.edb_name,b.edb_name_en,b.frequency,b.unit,b.unit_en,b.start_date,b.end_date,b.modify_time,b.edb_type,b.latest_date,b.latest_value,b.unique_code,b.edb_info_type AS edb_info_category_type,b.sub_source,
  120. a.is_convert, a.convert_type, a.convert_value, a.convert_unit, a.convert_en_unit
  121. FROM chart_edb_mapping AS a
  122. INNER JOIN edb_info AS b ON a.edb_info_id=b.edb_info_id
  123. WHERE chart_info_id=?
  124. ORDER BY chart_edb_mapping_id ASC `
  125. _, err = o.Raw(sql, chartInfoId).QueryRows(&list)
  126. return
  127. }
  128. type EdbDataList struct {
  129. EdbDataId int `json:"-" description:" 指标数据ID"`
  130. EdbInfoId int `json:"-" description:"指标ID"`
  131. DataTime string `description:"数据日期"`
  132. DataTimestamp int64 `description:"数据日期"`
  133. Value float64 `description:"数据值"`
  134. }
  135. // GetEdbDataList 获取指标的数据(日期正序返回)
  136. func GetEdbDataList(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) {
  137. // 自有数据需要额外处理(从mongo获取)
  138. if source == utils.DATA_SOURCE_BUSINESS {
  139. return getEdbDataListByMongo(source, subSource, edbInfoId, startDate, endDate)
  140. }
  141. return getEdbDataListByMysql(source, subSource, edbInfoId, startDate, endDate)
  142. }
  143. // getEdbDataListByMysql
  144. // @Description: 从mysql中获取指标的数据(日期正序返回)
  145. // @author: Roc
  146. // @datetime 2024-05-08 16:49:48
  147. // @param source int
  148. // @param subSource int
  149. // @param edbInfoId int
  150. // @param startDate string
  151. // @param endDate string
  152. // @return list []*EdbDataList
  153. // @return err error
  154. func getEdbDataListByMysql(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) {
  155. tableName := GetEdbDataTableName(source, subSource)
  156. if tableName == "" {
  157. err = errors.New("无效的渠道:" + strconv.Itoa(source))
  158. list = make([]*EdbDataList, 0)
  159. return list, err
  160. }
  161. var pars []interface{}
  162. var condition string
  163. if startDate != "" {
  164. condition += ` AND data_time>=? `
  165. pars = append(pars, startDate)
  166. }
  167. if endDate != "" {
  168. condition += ` AND data_time<=? `
  169. pars = append(pars, endDate)
  170. }
  171. sql := `SELECT edb_data_id,edb_info_id,data_time,value,data_timestamp FROM %s WHERE edb_info_id=? `
  172. if condition != "" {
  173. sql += condition
  174. }
  175. sql += ` ORDER BY data_time ASC `
  176. sql = fmt.Sprintf(sql, tableName)
  177. o := orm.NewOrmUsingDB("data")
  178. _, err = o.Raw(sql, edbInfoId, pars).QueryRows(&list)
  179. return
  180. }
  181. // getEdbDataListByMongo
  182. // @Description: 从mongo中获取指标的数据(日期正序返回)
  183. // @author: Roc
  184. // @datetime 2024-05-08 16:49:59
  185. // @param source int
  186. // @param subSource int
  187. // @param edbInfoId int
  188. // @param startDate string
  189. // @param endDate string
  190. // @return list []*EdbDataList
  191. // @return err error
  192. func getEdbDataListByMongo(source, subSource, edbInfoId int, startDate, endDate string) (list []*EdbDataList, err error) {
  193. list = make([]*EdbDataList, 0)
  194. mogDataObj := mgo.EdbDataBusiness{}
  195. // 构建查询条件
  196. queryConditions := bson.M{
  197. "edb_info_id": edbInfoId,
  198. }
  199. // 数据开始日期
  200. if startDate != `` {
  201. startDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, startDate, time.Local)
  202. if tmpErr != nil {
  203. err = tmpErr
  204. return
  205. }
  206. queryConditions["data_time"] = bson.M{"$gte": startDateTime}
  207. }
  208. // 数据结束日期
  209. if endDate != "" {
  210. endDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, endDate, time.Local)
  211. if tmpErr != nil {
  212. err = tmpErr
  213. return
  214. }
  215. queryConditions["data_time"] = bson.M{"$lte": endDateTime}
  216. }
  217. // 获取列表数据
  218. tmpDataList, tmpErr := mogDataObj.GetAllDataList(queryConditions, []string{"data_time"})
  219. if tmpErr != nil {
  220. err = tmpErr
  221. return
  222. }
  223. for k, v := range tmpDataList {
  224. list = append(list, &EdbDataList{
  225. EdbDataId: k + 1,
  226. EdbInfoId: v.EdbInfoId,
  227. DataTime: v.DataTime.Format(utils.FormatDate),
  228. DataTimestamp: v.DataTimestamp,
  229. Value: v.Value,
  230. })
  231. }
  232. return
  233. }
  234. type EdbDataItemList struct {
  235. EdbDataId int `json:"-" description:" 指标数据ID"`
  236. EdbInfoId int `json:"-" description:"指标ID"`
  237. DataTime string `description:"数据日期"`
  238. DataTimestamp int64 `description:"数据日期"`
  239. Value float64 `description:"数据值"`
  240. }
  241. type EdbDataItems struct {
  242. Items []*EdbDataList
  243. Year int
  244. BetweenDay int `json:"-" description:"公历与农历之间相差的天数"`
  245. CuttingDataTimestamp int64 `description:"切割的时间戳"`
  246. }
  247. type EdbDataResult struct {
  248. List []*EdbDataItems
  249. }
  250. type QuarterData struct {
  251. Year string
  252. DataList []*EdbDataList
  253. CuttingDataTimestamp int64 `description:"切割的时间戳"`
  254. ChartLegend string
  255. Years string
  256. }
  257. type QuarterXDateItem struct {
  258. StartDate time.Time
  259. EndDate time.Time
  260. ShowName string
  261. ChartLegend string
  262. CuttingDataTimestamp int64 `description:"切割的时间戳"`
  263. }
  264. type ChartInfoDetailResp struct {
  265. ChartInfo *ChartInfo
  266. EdbInfoList []*ChartEdbInfoMapping
  267. XEdbIdValue []int `description:"柱方图的x轴数据,指标id"`
  268. YDataList []YData `description:"柱方图的y轴数据"`
  269. XDataList []XData `description:"商品价格曲线的X轴数据"`
  270. CorrelationChartInfo *CorrelationInfo `description:"相关性图表信息"`
  271. DataResp interface{} `description:"图表数据,根据图的类型而定的,没有确定的数据格式"`
  272. WaterMark string `description:"水印"`
  273. }
  274. // XData 商品价格曲线的的x轴数据
  275. type XData struct {
  276. Name string `description:"别名"`
  277. NameEn string `description:"英文别名"`
  278. }
  279. // YData 柱方图的y轴数据
  280. type YData struct {
  281. Date string `description:"数据日期"`
  282. ConfigDate time.Time `description:"配置的日期" json:"-"`
  283. Color string `description:"数据颜色"`
  284. Name string `description:"别名"`
  285. NameEn string `description:"英文别名"`
  286. Value []float64 `description:"每个指标的值"`
  287. NoDataEdbList []int `description:"没有数据的指标列表"`
  288. XEdbInfoIdList []int `description:"对应X轴的指标id列表"`
  289. EdbValMap map[int]float64 `description:"指标与值的对应" json:"-"`
  290. M []int `description:"对应开始日期的间隔值" json:"-"`
  291. NameList []string `description:"每个值对应的名称"`
  292. EnNameList []string `description:"每个值对应的英文名称"`
  293. Unit string `description:"中文单位名称"`
  294. UnitEn string `description:"英文单位名称"`
  295. }
  296. // RollingCorrelationChartDataResp 滚动相关性图
  297. type RollingCorrelationChartDataResp struct {
  298. XDateTimeValue []string `description:"滚动相关性图的x轴数据,日期数据"`
  299. YDataList []YData `description:"滚动相关性图的y轴数据"`
  300. }
  301. // 指标季度数据计算(公历转农历)
  302. func AddCalculateQuarterV5(dataList []*EdbDataList) (result *EdbDataResult, err error) {
  303. var errMsg string
  304. defer func() {
  305. if errMsg != "" {
  306. fmt.Println("errMsg:", errMsg)
  307. }
  308. }()
  309. if len(dataList) <= 0 {
  310. result.List = make([]*EdbDataItems, 0)
  311. return
  312. }
  313. endDate := dataList[len(dataList)-1].DataTime
  314. endDateForm, err := time.Parse(utils.FormatDate, endDate)
  315. if err != nil {
  316. return result, err
  317. }
  318. thisMonth := int(endDateForm.Month())
  319. result = new(EdbDataResult)
  320. var yearArr []int
  321. yearMap := make(map[int]int)
  322. var cureentDate time.Time
  323. if thisMonth < 11 {
  324. for k, v := range dataList {
  325. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  326. if err != nil {
  327. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  328. return result, err
  329. }
  330. if k == len(dataList)-1 {
  331. cureentDate = dateTime
  332. }
  333. year := dateTime.Year()
  334. if _, ok := yearMap[year]; !ok {
  335. yearArr = append(yearArr, year)
  336. }
  337. yearMap[year] = year
  338. }
  339. } else {
  340. for k, v := range dataList {
  341. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  342. if err != nil {
  343. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  344. return result, err
  345. }
  346. if k == len(dataList)-1 {
  347. cureentDate = dateTime
  348. }
  349. year := dateTime.Year() + 1
  350. if _, ok := yearMap[year]; !ok {
  351. yearArr = append(yearArr, year)
  352. }
  353. yearMap[year] = year
  354. }
  355. }
  356. //排序
  357. fmt.Println("yearArr:", yearArr)
  358. thisYear := cureentDate.Year()
  359. //thisMonth := int(cureentDate.Month())
  360. fmt.Println("thisMonth:", thisMonth)
  361. for ky, vy := range yearArr {
  362. fmt.Println("line 432:", ky, vy, thisYear, thisMonth)
  363. if thisMonth < 11 {
  364. currentYearCjnl := strconv.Itoa(thisYear) + "-01-01" //当前年份春节农历
  365. currentYearCjgl := solarlunar.LunarToSolar(currentYearCjnl, false) //当前年份春节公历
  366. currentYearCjglDate, err := time.Parse(utils.FormatDate, currentYearCjgl)
  367. if err != nil {
  368. errMsg = "生成当前春节失败,Err:" + err.Error()
  369. return result, err
  370. }
  371. preYear := vy
  372. preYearCjnl := strconv.Itoa(preYear) + "-01-01" //之前年份春节农历
  373. preYearCjgl := solarlunar.LunarToSolar(preYearCjnl, false) //之前年份春节公历
  374. preYearCjglDate, err := time.Parse(utils.FormatDate, preYearCjgl)
  375. if err != nil {
  376. errMsg = "生成历史年份春节失败,Err:" + err.Error()
  377. return result, err
  378. }
  379. day := currentYearCjglDate.Sub(preYearCjglDate).Hours() / float64(24)
  380. items := new(EdbDataItems)
  381. items.BetweenDay = int(day) //公历日期换算成农历,需要减除的天数
  382. items.Year = preYear
  383. for _, v := range dataList {
  384. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  385. if err != nil {
  386. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  387. return result, err
  388. }
  389. newDate := dateTime.AddDate(0, 0, int(day))
  390. selectDateStr := strconv.Itoa(thisYear) + "-11" + "-30"
  391. selectDate, _ := time.Parse(utils.FormatDate, selectDateStr)
  392. if newDate.Before(selectDate) || newDate == selectDate {
  393. timestamp := newDate.UnixNano() / 1e6
  394. item := new(EdbDataList)
  395. item.DataTime = newDate.Format(utils.FormatDate)
  396. item.EdbInfoId = v.EdbInfoId
  397. item.Value = v.Value
  398. item.EdbDataId = v.EdbDataId
  399. item.DataTimestamp = timestamp
  400. items.Items = append(items.Items, item)
  401. }
  402. }
  403. result.List = append(result.List, items)
  404. } else {
  405. nextYear := thisYear + 1
  406. nextYearCjnl := strconv.Itoa(nextYear) + "-01-01" //当前年份春节农历
  407. nextYearCjgl := solarlunar.LunarToSolar(nextYearCjnl, false) //当前年份春节公历
  408. nextYearCjglDate, err := time.Parse(utils.FormatDate, nextYearCjgl)
  409. if err != nil {
  410. errMsg = "生成当前春节失败,Err:" + err.Error()
  411. return result, err
  412. }
  413. preYear := vy
  414. preYearCjnl := strconv.Itoa(preYear) + "-01-01" //之前年份春节农历
  415. preYearCjgl := solarlunar.LunarToSolar(preYearCjnl, false) //之前年份春节公历
  416. preYearCjglDate, err := time.Parse(utils.FormatDate, preYearCjgl)
  417. if err != nil {
  418. errMsg = "生成历史年份春节失败,Err:" + err.Error()
  419. return result, err
  420. }
  421. day := nextYearCjglDate.Sub(preYearCjglDate).Hours() / float64(24)
  422. fmt.Println("day:", day, nextYearCjglDate, preYearCjglDate)
  423. items := new(EdbDataItems)
  424. items.BetweenDay = int(day) //公历日期换算成农历,需要减除的天数
  425. items.Year = preYear - 1
  426. fmt.Println("preYear:", preYear, "ky:", ky, "yearArrLen:", len(yearArr))
  427. if ky+1 < len(yearArr) {
  428. for _, v := range dataList {
  429. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  430. if err != nil {
  431. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  432. return result, err
  433. }
  434. newDate := dateTime.AddDate(0, 0, int(day))
  435. selectDateStr := strconv.Itoa(nextYear) + "-05" + "-31"
  436. selectDate, _ := time.Parse(utils.FormatDate, selectDateStr)
  437. if newDate.Before(selectDate) || newDate == selectDate {
  438. timestamp := newDate.UnixNano() / 1e6
  439. item := new(EdbDataList)
  440. item.DataTime = newDate.Format(utils.FormatDate)
  441. item.EdbInfoId = v.EdbInfoId
  442. item.Value = v.Value
  443. item.EdbDataId = v.EdbDataId
  444. item.DataTimestamp = timestamp
  445. items.Items = append(items.Items, item)
  446. }
  447. }
  448. result.List = append(result.List, items)
  449. } else {
  450. for _, v := range dataList {
  451. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  452. if err != nil {
  453. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  454. return result, err
  455. }
  456. timestamp := dateTime.UnixNano() / 1e6
  457. item := new(EdbDataList)
  458. item.DataTime = dateTime.Format(utils.FormatDate)
  459. item.EdbInfoId = v.EdbInfoId
  460. item.Value = v.Value
  461. item.EdbDataId = v.EdbDataId
  462. item.DataTimestamp = timestamp
  463. items.Items = append(items.Items, item)
  464. }
  465. result.List = append(result.List, items)
  466. }
  467. }
  468. }
  469. return
  470. }
  471. // AddCalculateQuarterV6 指标季度数据计算(季节性图表)
  472. func AddCalculateQuarterV6(dataList []*EdbDataList) (result *EdbDataResult, err error) {
  473. var errMsg string
  474. defer func() {
  475. if errMsg != "" {
  476. fmt.Println("errMsg:", errMsg)
  477. }
  478. }()
  479. endDate := dataList[len(dataList)-1].DataTime
  480. endDateForm, err := time.Parse(utils.FormatDate, endDate)
  481. if err != nil {
  482. return result, err
  483. }
  484. thisMonth := int(endDateForm.Month())
  485. result = new(EdbDataResult)
  486. var yearArr []int
  487. yearMap := make(map[int]int)
  488. var cureentDate time.Time
  489. if thisMonth < 11 {
  490. for k, v := range dataList {
  491. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  492. if err != nil {
  493. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  494. return result, err
  495. }
  496. if k == len(dataList)-1 {
  497. cureentDate = dateTime
  498. }
  499. year := dateTime.Year()
  500. if _, ok := yearMap[year]; !ok {
  501. yearArr = append(yearArr, year)
  502. }
  503. yearMap[year] = year
  504. }
  505. } else {
  506. for k, v := range dataList {
  507. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  508. if err != nil {
  509. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  510. return result, err
  511. }
  512. if k == len(dataList)-1 {
  513. cureentDate = dateTime
  514. }
  515. year := dateTime.Year() + 1
  516. if _, ok := yearMap[year]; !ok {
  517. yearArr = append(yearArr, year)
  518. }
  519. yearMap[year] = year
  520. }
  521. }
  522. //排序
  523. fmt.Println("yearArr:", yearArr)
  524. thisYear := cureentDate.Year()
  525. //thisMonth := int(cureentDate.Month())
  526. fmt.Println("thisMonth:", thisMonth)
  527. for ky, vy := range yearArr {
  528. fmt.Printf("line 432:ky:%d, vy:%d, thisYear:%d, thisMonth:%d", ky, vy, thisYear, thisMonth)
  529. fmt.Println("")
  530. if thisMonth < 11 {
  531. currentYearCjnl := strconv.Itoa(thisYear) + "-01-01" //当前年份春节农历
  532. currentYearCjgl := solarlunar.LunarToSolar(currentYearCjnl, false) //当前年份春节公历
  533. currentYearCjglDate, err := time.Parse(utils.FormatDate, currentYearCjgl)
  534. if err != nil {
  535. errMsg = "生成当前春节失败,Err:" + err.Error()
  536. return result, err
  537. }
  538. preYear := vy
  539. preYearCjnl := strconv.Itoa(preYear) + "-01-01" //之前年份春节农历
  540. preYearCjgl := solarlunar.LunarToSolar(preYearCjnl, false) //之前年份春节公历
  541. preYearCjglDate, err := time.Parse(utils.FormatDate, preYearCjgl)
  542. if err != nil {
  543. errMsg = "生成历史年份春节失败,Err:" + err.Error()
  544. return result, err
  545. }
  546. day := currentYearCjglDate.Sub(preYearCjglDate).Hours() / float64(24)
  547. fmt.Println("day:", day, currentYearCjglDate, preYearCjglDate)
  548. items := new(EdbDataItems)
  549. items.BetweenDay = int(day) //公历日期换算成农历,需要减除的天数
  550. items.Year = preYear
  551. for _, v := range dataList {
  552. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  553. if err != nil {
  554. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  555. return result, err
  556. }
  557. newDate := dateTime.AddDate(0, 0, int(day))
  558. timestamp := newDate.UnixNano() / 1e6
  559. item := new(EdbDataList)
  560. item.DataTime = newDate.Format(utils.FormatDate)
  561. item.EdbInfoId = v.EdbInfoId
  562. item.Value = v.Value
  563. item.EdbDataId = v.EdbDataId
  564. item.DataTimestamp = timestamp
  565. items.Items = append(items.Items, item)
  566. }
  567. result.List = append(result.List, items)
  568. } else {
  569. nextYear := thisYear + 1
  570. nextYearCjnl := strconv.Itoa(nextYear) + "-01-01" //当前年份春节农历
  571. nextYearCjgl := solarlunar.LunarToSolar(nextYearCjnl, false) //当前年份春节公历
  572. nextYearCjglDate, err := time.Parse(utils.FormatDate, nextYearCjgl)
  573. if err != nil {
  574. errMsg = "生成当前春节失败,Err:" + err.Error()
  575. return result, err
  576. }
  577. preYear := vy
  578. preYearCjnl := strconv.Itoa(preYear) + "-01-01" //之前年份春节农历
  579. preYearCjgl := solarlunar.LunarToSolar(preYearCjnl, false) //之前年份春节公历
  580. preYearCjglDate, err := time.Parse(utils.FormatDate, preYearCjgl)
  581. if err != nil {
  582. errMsg = "生成历史年份春节失败,Err:" + err.Error()
  583. return result, err
  584. }
  585. day := nextYearCjglDate.Sub(preYearCjglDate).Hours() / float64(24)
  586. fmt.Println("day:", day, nextYearCjglDate, preYearCjglDate)
  587. items := new(EdbDataItems)
  588. items.BetweenDay = int(day) //公历日期换算成农历,需要减除的天数
  589. items.Year = preYear
  590. fmt.Println("preYear:", preYear, "ky:", ky, "yearArrLen:", len(yearArr))
  591. //if ky+1 < len(yearArr) {
  592. for _, v := range dataList {
  593. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  594. if err != nil {
  595. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  596. return result, err
  597. }
  598. newDate := dateTime.AddDate(0, 0, int(day))
  599. timestamp := newDate.UnixNano() / 1e6
  600. item := new(EdbDataList)
  601. item.DataTime = newDate.Format(utils.FormatDate)
  602. item.EdbInfoId = v.EdbInfoId
  603. item.Value = v.Value
  604. item.EdbDataId = v.EdbDataId
  605. item.DataTimestamp = timestamp
  606. items.Items = append(items.Items, item)
  607. }
  608. result.List = append(result.List, items)
  609. /*} else {
  610. for _, v := range dataList {
  611. dateTime, err := time.Parse(utils.FormatDate, v.DataTime)
  612. if err != nil {
  613. errMsg = "time.Parse Err:" + err.Error() + ";DataTime:" + v.DataTime
  614. return result, err
  615. }
  616. timestamp := dateTime.UnixNano() / 1e6
  617. item := new(EdbDataList)
  618. item.DataTime = dateTime.Format(utils.FormatDate)
  619. item.EdbInfoId = v.EdbInfoId
  620. item.Value = v.Value
  621. item.EdbDataId = v.EdbDataId
  622. item.DataTimestamp = timestamp
  623. items.Items = append(items.Items, item)
  624. }
  625. result.List = append(result.List, items)
  626. }*/
  627. }
  628. }
  629. return
  630. }
  631. type CorrelationInfo struct {
  632. LeadValue int `description:"领先值"`
  633. LeadUnit string `description:"领先单位"`
  634. StartDate string `description:"开始日期"`
  635. EndDate string `description:"结束日期"`
  636. EdbInfoIdFirst int `description:"A指标ID"`
  637. EdbInfoIdSecond int `description:"B指标ID"`
  638. PeriodData string `description:"X轴-期数数据"`
  639. CorrelationData string `description:"Y轴-相关性系数"`
  640. }
  641. type SeasonExtraItem struct {
  642. ChartLegend []SeasonChartLegend `description:"自定义的图例名称"`
  643. XStartDate string `description:"横坐标显示的起始日"`
  644. XEndDate string `description:"横坐标显示的截止日"`
  645. JumpYear int `description:"横坐标日期是否跨年,1跨年,0不跨年"`
  646. }
  647. type SeasonChartLegend struct {
  648. Name string
  649. Value string
  650. }
  651. type QuarterDataList []*QuarterData
  652. func (m QuarterDataList) Len() int {
  653. return len(m)
  654. }
  655. func (m QuarterDataList) Less(i, j int) bool {
  656. return m[i].Years < m[j].Years
  657. }
  658. func (m QuarterDataList) Swap(i, j int) {
  659. m[i], m[j] = m[j], m[i]
  660. }