chart.go 24 KB

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