chart.go 24 KB

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