base_calculate.go 52 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847
  1. package models
  2. import (
  3. "errors"
  4. "eta/eta_index_lib/utils"
  5. "fmt"
  6. "github.com/shopspring/decimal"
  7. "sort"
  8. "strconv"
  9. "time"
  10. )
  11. // EdbInfoData
  12. // @Description: 指标数据
  13. type EdbInfoData struct {
  14. EdbDataId int `description:"数据ID"`
  15. DataTime time.Time `description:"数据日期"`
  16. Value float64 `description:"数据"`
  17. }
  18. // BaseCalculateBySearchData
  19. // @Description: 基础的计算公式
  20. type BaseCalculateBySearchData struct {
  21. DataList []*EdbInfoSearchData
  22. Frequency string `description:"需要转换的频度"`
  23. Formula interface{}
  24. Calendar string `description:"公历/农历"`
  25. MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
  26. MoveFrequency string `description:"移动频度"`
  27. FromFrequency string `description:"来源的频度"`
  28. Source int `description:"1:累计值转月;2:累计值转季;3:同比值;4:同差值;5:N数值移动平均数计算;6:环比值;7:环差值;8:升频;9:降频;10:时间移位;11:超季节性;12:年化;13:累计值;14:累计值年初至今;15:指数修匀;16:日均值"`
  29. }
  30. // BaseCalculate
  31. // @Description: 基础的计算公式
  32. type BaseCalculate struct {
  33. DataList []*EdbInfoData
  34. Frequency string `description:"需要转换的频度"`
  35. Formula interface{}
  36. Calendar string `description:"公历/农历"`
  37. MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
  38. MoveFrequency string `description:"移动频度"`
  39. FromFrequency string `description:"来源的频度"`
  40. Source int `description:"1:累计值转月;2:累计值转季;3:同比值;4:同差值;5:N数值移动平均数计算;6:环比值;7:环差值;8:升频;9:降频;10:时间移位;11:超季节性;12:年化;13:累计值;14:累计值年初至今;15:指数修匀;16:日均值"`
  41. }
  42. type BaseCalculateResp struct {
  43. DataMap map[string]float64
  44. DateList []string
  45. }
  46. // Ljzzy
  47. // @Description: 累计值转月
  48. // @author: Roc
  49. // @receiver obj
  50. // @datetime2023-11-02 18:05:19
  51. // @return dateDataMap map[time.Time]float64
  52. // @return err error
  53. func (obj BaseCalculate) Ljzzy() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  54. dateDataMap = make(map[time.Time]float64)
  55. dataList := obj.DataList // 升序
  56. // 数据处理
  57. yearMap := make(map[int]map[int]*EdbInfoData)
  58. dataLen := len(dataList)
  59. for i := 0; i < dataLen; i++ {
  60. item := dataList[i]
  61. //日其中获取年
  62. itemDate := item.DataTime
  63. year := itemDate.Year()
  64. month := int(itemDate.Month())
  65. if monthMap, yok := yearMap[year]; yok {
  66. monthMap[month] = item
  67. yearMap[year] = monthMap
  68. } else {
  69. monthMap = make(map[int]*EdbInfoData)
  70. monthMap[month] = item
  71. yearMap[year] = monthMap
  72. }
  73. }
  74. // 开始计算
  75. for yk, yv := range yearMap {
  76. _, oneMonthOk := yv[1]
  77. _, twoMonthOk := yv[2]
  78. if !oneMonthOk && !twoMonthOk {
  79. continue
  80. }
  81. for i := 1; i <= 12; i++ {
  82. //fmt.Println(yk, i, yv[i])
  83. dataCurrentItem := yv[i]
  84. if dataCurrentItem == nil {
  85. continue
  86. }
  87. var date time.Time
  88. var val float64
  89. // 一二月单独处理
  90. if i == 1 || i == 2 {
  91. if _, mok := yv[1]; mok { //1月有值
  92. if i == 1 {
  93. date = dataCurrentItem.DataTime
  94. val, _ = decimal.NewFromFloat(dataCurrentItem.Value).Float64() //a.Div(b).Float64()
  95. }
  96. if i == 2 {
  97. dataOneItem := yv[1]
  98. date = dataCurrentItem.DataTime
  99. twoMonth := decimal.NewFromFloat(dataCurrentItem.Value)
  100. oneMonth := decimal.NewFromFloat(dataOneItem.Value)
  101. val, _ = twoMonth.Sub(oneMonth).Float64()
  102. }
  103. } else { //1月无值
  104. dataTwoItem := yv[2]
  105. if i == 1 {
  106. dateStr := strconv.Itoa(yk) + "-01-31"
  107. a := decimal.NewFromFloat(dataTwoItem.Value)
  108. b := decimal.NewFromFloat(2.0)
  109. val, _ = a.Div(b).Float64()
  110. date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
  111. if err != nil {
  112. return
  113. }
  114. }
  115. if i == 2 {
  116. //1月无值:1月=2月/2 (不管怎样,都要给1月赋值)
  117. {
  118. dateStr := strconv.Itoa(yk) + "-01-31"
  119. a := decimal.NewFromFloat(dataTwoItem.Value)
  120. b := decimal.NewFromFloat(2.0)
  121. val, _ = a.Div(b).Float64()
  122. date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
  123. if err != nil {
  124. return
  125. }
  126. dateDataMap[date] = val
  127. }
  128. //end 1月无值
  129. // 这是正常二月份的值
  130. date = dataCurrentItem.DataTime
  131. a := decimal.NewFromFloat(dataTwoItem.Value)
  132. b := decimal.NewFromFloat(2.0)
  133. val, _ = a.Div(b).Float64()
  134. }
  135. }
  136. } else {
  137. dataPreItem := yv[i-1]
  138. if dataCurrentItem != nil && dataPreItem != nil {
  139. date = dataCurrentItem.DataTime
  140. //val = dataCurrentItem.Value - dataPreItem.Value
  141. a := decimal.NewFromFloat(dataCurrentItem.Value)
  142. b := decimal.NewFromFloat(dataPreItem.Value)
  143. val, _ = a.Sub(b).Float64()
  144. }
  145. }
  146. // 如果没有日期,那么就退出当前循环,进入下一个循环
  147. if date.IsZero() {
  148. continue
  149. }
  150. dateDataMap[date] = val
  151. }
  152. }
  153. return
  154. }
  155. // Ljzzj
  156. // @Description: 累计值转季度
  157. // @author: Roc
  158. // @receiver obj
  159. // @datetime2023-11-02 18:05:05
  160. // @return dateDataMap map[time.Time]float64
  161. // @return err error
  162. func (obj BaseCalculate) Ljzzj() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  163. dateDataMap = make(map[time.Time]float64)
  164. dataList := obj.DataList // 升序
  165. // 数据处理
  166. yearMap := make(map[int]map[int]*EdbInfoData)
  167. dataLen := len(dataList)
  168. for i := 0; i < dataLen; i++ {
  169. item := dataList[i]
  170. //日其中获取年
  171. itemDate := item.DataTime
  172. year := itemDate.Year()
  173. quarter := utils.MonthQuarterMap[int(itemDate.Month())]
  174. if quarterMap, yok := yearMap[year]; yok {
  175. quarterMap[quarter] = item
  176. yearMap[year] = quarterMap
  177. } else {
  178. quarterMap = make(map[int]*EdbInfoData)
  179. quarterMap[quarter] = item
  180. yearMap[year] = quarterMap
  181. }
  182. }
  183. // 开始计算
  184. for yk, yv := range yearMap {
  185. _, oneQuarterOk := yv[1]
  186. _, twoQuarterOk := yv[2]
  187. if !oneQuarterOk && !twoQuarterOk {
  188. continue
  189. }
  190. for i := 1; i <= 4; i++ {
  191. //fmt.Println(yk, i, yv[i])
  192. dataCurrentItem := yv[i]
  193. if dataCurrentItem == nil {
  194. continue
  195. }
  196. var date time.Time
  197. var val float64
  198. // 一二季度单独处理
  199. if i == 1 || i == 2 {
  200. if _, mok := yv[1]; mok { //第1个季度有值
  201. if i == 1 {
  202. date = dataCurrentItem.DataTime
  203. val, _ = decimal.NewFromFloat(dataCurrentItem.Value).Float64() //a.Div(b).Float64()
  204. }
  205. if i == 2 {
  206. dataOneItem := yv[1]
  207. date = dataCurrentItem.DataTime
  208. twoQuarter := decimal.NewFromFloat(dataCurrentItem.Value)
  209. oneQuarter := decimal.NewFromFloat(dataOneItem.Value)
  210. val, _ = twoQuarter.Sub(oneQuarter).Float64()
  211. }
  212. } else { //第1个季度无值
  213. dataTwoItem := yv[2]
  214. if i == 1 {
  215. dateStr := strconv.Itoa(yk) + "-03-31"
  216. a := decimal.NewFromFloat(dataTwoItem.Value)
  217. b := decimal.NewFromFloat(2.0)
  218. val, _ = a.Div(b).Float64()
  219. date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
  220. if err != nil {
  221. return
  222. }
  223. }
  224. if i == 2 {
  225. //第1个季度无值:第1个季度=第2个季度/2 (不管怎样,都要给1季度赋值)
  226. {
  227. dateStr := strconv.Itoa(yk) + "-03-31"
  228. a := decimal.NewFromFloat(dataTwoItem.Value)
  229. b := decimal.NewFromFloat(2.0)
  230. val, _ = a.Div(b).Float64()
  231. date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
  232. if err != nil {
  233. return
  234. }
  235. dateDataMap[date] = val
  236. }
  237. //end 第1个季度无值
  238. date = dataCurrentItem.DataTime
  239. a := decimal.NewFromFloat(dataTwoItem.Value)
  240. b := decimal.NewFromFloat(2.0)
  241. val, _ = a.Div(b).Float64()
  242. }
  243. }
  244. } else {
  245. dataPreItem := yv[i-1]
  246. if dataCurrentItem != nil && dataPreItem != nil {
  247. date = dataCurrentItem.DataTime
  248. //val = dataCurrentItem.Value - dataPreItem.Value
  249. a := decimal.NewFromFloat(dataCurrentItem.Value)
  250. b := decimal.NewFromFloat(dataPreItem.Value)
  251. val, _ = a.Sub(b).Float64()
  252. }
  253. }
  254. // 如果没有日期,那么就退出当前循环,进入下一个循环
  255. if date.IsZero() {
  256. continue
  257. }
  258. dateDataMap[date] = val
  259. }
  260. }
  261. return
  262. }
  263. // Tbz
  264. // @Description: 同比值计算
  265. // @author: Roc
  266. // @receiver obj
  267. // @datetime2023-11-02 18:04:59
  268. // @return dateDataMap map[time.Time]float64
  269. // @return err error
  270. func (obj BaseCalculate) Tbz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  271. dateDataMap = make(map[time.Time]float64)
  272. dataList := obj.DataList
  273. frequency := obj.FromFrequency
  274. //数据处理
  275. // 数据降序
  276. dataList = reverseSliceByDesc(dataList)
  277. var dateArr []time.Time
  278. dataMap := make(map[string]*EdbInfoData) // 避免因为时间戳导致的日期不对,还是用string来表示比较合适
  279. for _, v := range dataList {
  280. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  281. dateArr = append(dateArr, v.DataTime)
  282. dataMap[v.DataTime.Format(utils.FormatDate)] = v
  283. }
  284. // 开始计算
  285. for _, currentDate := range dateArr {
  286. //当前日期
  287. currentDateStr := currentDate.Format(utils.FormatDate)
  288. currentItem, ok := dataMap[currentDateStr]
  289. if !ok {
  290. continue
  291. }
  292. // 找到的数据
  293. var findItem *EdbInfoData
  294. //上一年的日期
  295. preDate := currentDate.AddDate(-1, 0, 0)
  296. preDateStr := preDate.Format(utils.FormatDate)
  297. if findItem, ok = dataMap[preDateStr]; !ok { //上一年同期没找到
  298. if frequency == "月度" { //向上和向下,各找一个月
  299. for i := 0; i <= 35; i++ {
  300. nextDateDay := preDate.AddDate(0, 0, i)
  301. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  302. if findItem, ok = dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  303. break
  304. } else {
  305. preDateDay := preDate.AddDate(0, 0, -i)
  306. preDateDayStr := preDateDay.Format(utils.FormatDate)
  307. if findItem, ok = dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  308. break
  309. }
  310. }
  311. }
  312. } else if frequency == "季度" || frequency == "年度" {
  313. if findItem, ok = dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  314. break
  315. }
  316. } else {
  317. nextDateDay := preDate.AddDate(0, 0, 1)
  318. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  319. preDateDay := preDate.AddDate(0, 0, -1)
  320. preDateDayStr := preDateDay.Format(utils.FormatDate)
  321. for i := 0; i < 35; i++ {
  322. if i >= 1 {
  323. nextDateDay = nextDateDay.AddDate(0, 0, i)
  324. nextDateDayStr = nextDateDay.Format(utils.FormatDate)
  325. }
  326. if findItem, ok = dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  327. break
  328. } else {
  329. if i >= 1 {
  330. preDateDay = preDateDay.AddDate(0, 0, -i)
  331. preDateDayStr = preDateDay.Format(utils.FormatDate)
  332. }
  333. if findItem, ok = dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  334. break
  335. }
  336. }
  337. }
  338. }
  339. }
  340. // 如果没找到数据
  341. if findItem == nil {
  342. continue
  343. }
  344. dateDataMap[currentDate] = tbzDiv(currentItem.Value, findItem.Value)
  345. }
  346. return
  347. }
  348. // tbzDiv
  349. // @Description: 同比值计算
  350. // @author: Roc
  351. // @datetime2023-11-02 16:29:14
  352. // @param a float64
  353. // @param b float64
  354. // @return float64
  355. func tbzDiv(a, b float64) float64 {
  356. var val float64
  357. if b != 0 {
  358. af := decimal.NewFromFloat(a)
  359. bf := decimal.NewFromFloat(b)
  360. val, _ = af.Div(bf).Sub(decimal.NewFromFloat(1)).RoundCeil(4).Float64()
  361. }
  362. return val
  363. }
  364. // Tcz
  365. // @Description: 计算同差值
  366. // @author: Roc
  367. // @receiver obj
  368. // @datetime2023-11-02 18:04:51
  369. // @return dateDataMap map[time.Time]float64
  370. // @return err error
  371. func (obj BaseCalculate) Tcz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  372. dateDataMap = make(map[time.Time]float64)
  373. dataList := obj.DataList // 降序
  374. frequency := obj.FromFrequency
  375. // 数据处理
  376. // 数据降序
  377. dataList = reverseSliceByDesc(dataList)
  378. var dateArr []time.Time
  379. dataMap := make(map[string]*EdbInfoData)
  380. for _, v := range dataList {
  381. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  382. dateArr = append(dateArr, v.DataTime)
  383. dataMap[v.DataTime.Format(utils.FormatDate)] = v
  384. }
  385. // 开始计算
  386. for _, currentDate := range dateArr {
  387. //当前日期
  388. currentItem, ok := dataMap[currentDate.Format(utils.FormatDate)]
  389. if !ok {
  390. continue
  391. }
  392. // 找到的数据
  393. var findItem *EdbInfoData
  394. //上一年的日期
  395. preDate := currentDate.AddDate(-1, 0, 0)
  396. preDateStr := preDate.Format(utils.FormatDate)
  397. if findItem, ok = dataMap[preDateStr]; !ok {
  398. //上一年同期没找到
  399. if frequency == "月度" { //向上和向下,各找一个月
  400. for i := 0; i <= 35; i++ {
  401. nextDateDay := preDate.AddDate(0, 0, i)
  402. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  403. if findItem, ok = dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  404. break
  405. } else {
  406. preDateDay := preDate.AddDate(0, 0, -i)
  407. preDateDayStr := preDateDay.Format(utils.FormatDate)
  408. if findItem, ok = dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  409. break
  410. }
  411. }
  412. }
  413. } else if frequency == "季度" || frequency == "年度" {
  414. if findItem, ok = dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  415. break
  416. }
  417. } else {
  418. for i := 0; i < 35; i++ {
  419. nextDateDay := preDate.AddDate(0, 0, i)
  420. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  421. if findItem, ok = dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  422. break
  423. } else {
  424. preDateDay := preDate.AddDate(0, 0, -i)
  425. preDateDayStr := preDateDay.Format(utils.FormatDate)
  426. if findItem, ok = dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  427. break
  428. }
  429. }
  430. }
  431. }
  432. }
  433. // 如果没找到数据
  434. if findItem == nil {
  435. continue
  436. }
  437. dateDataMap[currentDate] = tczSub(currentItem.Value, findItem.Value)
  438. }
  439. return
  440. }
  441. // tczSub
  442. // @Description: 计算同差值
  443. // @author: Roc
  444. // @datetime2023-11-02 18:01:46
  445. // @param a float64
  446. // @param b float64
  447. // @return float64
  448. func tczSub(a, b float64) float64 {
  449. af := decimal.NewFromFloat(a)
  450. bf := decimal.NewFromFloat(b)
  451. val, _ := af.Sub(bf).RoundCeil(4).Float64()
  452. return val
  453. }
  454. // Nszydpjjs
  455. // @Description: N数值移动平均数计算
  456. // @author: Roc
  457. // @receiver obj
  458. // @datetime2023-11-02 18:17:38
  459. // @return dateDataMap map[time.Time]float64
  460. // @return err error
  461. func (obj BaseCalculate) Nszydpjjs() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  462. dateDataMap = make(map[time.Time]float64)
  463. fromDataList := obj.DataList
  464. nVal := fmt.Sprint(obj.Formula)
  465. formulaInt, err := strconv.Atoi(nVal)
  466. if err != nil {
  467. return
  468. }
  469. // 数据处理
  470. // 数据降序
  471. fromDataList = reverseSliceByDesc(fromDataList)
  472. var fromDateArr []time.Time
  473. fromDataMap := make(map[time.Time]*EdbInfoData)
  474. for _, v := range fromDataList {
  475. fromDateArr = append(fromDateArr, v.DataTime)
  476. fromDataMap[v.DataTime] = v
  477. }
  478. arrLen := len(fromDateArr)
  479. for ak, currentDate := range fromDateArr {
  480. //处理第一个值
  481. var valArr []float64
  482. if findItem, ok := fromDataMap[currentDate]; ok {
  483. valArr = append(valArr, findItem.Value)
  484. } else {
  485. continue
  486. }
  487. if ak+1 != arrLen {
  488. //处理除第一个值之外的N-1个值
  489. for i := 1; i < formulaInt; i++ {
  490. arrIndex := ak + i
  491. if arrIndex >= arrLen {
  492. break
  493. }
  494. arrVal := fromDateArr[arrIndex]
  495. if findItem, ok := fromDataMap[arrVal]; ok {
  496. valArr = append(valArr, findItem.Value)
  497. } else {
  498. continue
  499. }
  500. }
  501. }
  502. valArrLen := len(valArr)
  503. //var totalVal float64
  504. totalVal := decimal.NewFromFloat(0.00)
  505. for _, v := range valArr {
  506. newDecimal := decimal.NewFromFloat(v)
  507. totalVal = totalVal.Add(newDecimal)
  508. }
  509. af := totalVal //decimal.NewFromFloat(totalVal)
  510. bf := decimal.NewFromFloat(float64(valArrLen))
  511. val, _ := af.Div(bf).RoundCeil(4).Float64()
  512. dateDataMap[currentDate] = val
  513. }
  514. return
  515. }
  516. // Hbz
  517. // @Description: 环比值计算
  518. // @author: Roc
  519. // @receiver obj
  520. // @datetime2023-11-02 18:28:24
  521. // @return dateDataMap map[time.Time]float64
  522. // @return err error
  523. func (obj BaseCalculate) Hbz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  524. dateDataMap = make(map[time.Time]float64)
  525. dataList := obj.DataList
  526. nVal := fmt.Sprint(obj.Formula)
  527. formulaInt, err := strconv.Atoi(nVal)
  528. if err != nil {
  529. return
  530. }
  531. // 数据处理
  532. // 数据降序
  533. dataList = reverseSliceByDesc(dataList)
  534. var dateArr []time.Time
  535. for _, v := range dataList {
  536. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  537. dateArr = append(dateArr, v.DataTime)
  538. }
  539. // 数据计算
  540. dataLen := len(dataList)
  541. //fmt.Println("dataLen:", dataLen)
  542. for i := 0; i < dataLen; i++ {
  543. j := i + formulaInt
  544. if j < dataLen {
  545. //当期
  546. currentItem := dataList[i]
  547. preItem := dataList[j]
  548. // 当期和上期没有数据
  549. if currentItem == nil || preItem == nil {
  550. continue
  551. }
  552. // 开始计算
  553. val, ok := hbzDiv(currentItem.Value, preItem.Value)
  554. // 计算失败
  555. if !ok {
  556. continue
  557. }
  558. dateDataMap[currentItem.DataTime] = val
  559. }
  560. }
  561. return
  562. }
  563. // hbzDiv
  564. // @Description: 环比值计算,current:当期,pre:上期 公式: (当期-上期)/上期
  565. // @author: Roc
  566. // @datetime2023-11-02 18:20:11
  567. // @param current float64
  568. // @param pre float64
  569. // @return val float64
  570. // @return ok bool
  571. func hbzDiv(current, pre float64) (val float64, ok bool) {
  572. if pre == 0 {
  573. return
  574. }
  575. currentVal := decimal.NewFromFloat(current)
  576. preVal := decimal.NewFromFloat(pre)
  577. val, _ = currentVal.Sub(preVal).Div(preVal).RoundCeil(4).Float64()
  578. //valStr := decimal.NewFromFloat(val).RoundCeil(4).String() //utils.SubFloatToString(val, 4)
  579. ok = true
  580. return
  581. }
  582. // Hcz
  583. // @Description: 计算环差值数据
  584. // @author: Roc
  585. // @receiver obj
  586. // @datetime2023-11-02 18:33:20
  587. // @return dateDataMap map[time.Time]float64
  588. // @return err error
  589. func (obj BaseCalculate) Hcz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  590. dateDataMap = make(map[time.Time]float64)
  591. dataList := obj.DataList
  592. nVal := fmt.Sprint(obj.Formula)
  593. formulaInt, err := strconv.Atoi(nVal)
  594. if err != nil {
  595. return
  596. }
  597. // 数据处理
  598. // 数据降序
  599. dataList = reverseSliceByDesc(dataList)
  600. var dateArr []time.Time
  601. for _, v := range dataList {
  602. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  603. dateArr = append(dateArr, v.DataTime)
  604. }
  605. dataLen := len(dataList)
  606. fmt.Println("dataLen:", dataLen)
  607. for i := 0; i < dataLen; i++ {
  608. j := i + formulaInt
  609. if j < dataLen {
  610. //当期
  611. currentItem := dataList[i]
  612. preItem := dataList[j]
  613. // 当期和上期没有数据
  614. if currentItem == nil || preItem == nil {
  615. continue
  616. }
  617. // 开始计算
  618. val := hczDiv(currentItem.Value, preItem.Value)
  619. dateDataMap[currentItem.DataTime] = val
  620. }
  621. }
  622. return
  623. }
  624. // hczDiv
  625. // @Description: 环差值计算,current:当期,pre:上期 公式:当期-上期
  626. // @author: Roc
  627. // @datetime2023-11-02 18:33:07
  628. // @param current float64
  629. // @param pre float64
  630. // @return float64
  631. func hczDiv(current, pre float64) float64 {
  632. currentVal := decimal.NewFromFloat(current)
  633. preVal := decimal.NewFromFloat(pre)
  634. val, _ := currentVal.Sub(preVal).RoundCeil(4).Float64()
  635. //valStr := decimal.NewFromFloat(val).RoundCeil(4).String() //utils.SubFloatToString(val, 4)
  636. return val
  637. }
  638. // UpFrequency
  639. // @Description: 升频计算
  640. // @author: Roc
  641. // @receiver obj
  642. // @datetime2023-11-02 18:43:03
  643. // @return dateDataMap map[time.Time]float64
  644. // @return err error
  645. func (obj BaseCalculate) UpFrequency() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  646. dateDataMap = make(map[time.Time]float64)
  647. dataList := obj.DataList
  648. // 数据处理
  649. var dateArr []time.Time
  650. dataMap := make(map[time.Time]*EdbInfoData)
  651. fromDataMap := make(map[time.Time]float64)
  652. //来源指指标数据
  653. for _, v := range dataList {
  654. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  655. dateArr = append(dateArr, v.DataTime)
  656. dataMap[v.DataTime] = v
  657. fromDataMap[v.DataTime] = v.Value
  658. }
  659. if obj.FromFrequency == `日度` {
  660. errMsg = `不能选择日度指标`
  661. err = errors.New(errMsg)
  662. return
  663. }
  664. nowDay := time.Now()
  665. nowDay = time.Date(nowDay.Year(), nowDay.Month(), nowDay.Day(), 0, 0, 0, 0, nowDay.Location())
  666. // 数据计算
  667. dataLen := len(dataList)
  668. lastI := dataLen - 1
  669. for i := 0; i < dataLen; i++ {
  670. //当期
  671. currentItem := dataList[i]
  672. currentDate := currentItem.DataTime
  673. var day int
  674. var preItem *EdbInfoData
  675. var preDate time.Time
  676. // 如果是第一期的数据,那么直接赋值
  677. if i == 0 {
  678. dateDataMap[currentItem.DataTime] = currentItem.Value
  679. continue
  680. }
  681. // 上一期
  682. j := i - 1
  683. preItem = dataList[j]
  684. preDate = preItem.DataTime
  685. day = int(currentDate.Sub(preItem.DataTime).Hours() / float64(24))
  686. //utils.FileLog.Info("preItem.DataTime:" + preItem.DataTime.Format(utils.FormatDate) + ";currentItem.DataTime" + currentItem.DataTime.Format(utils.FormatDate))
  687. fmt.Println("preItem.DataTime:" + preItem.DataTime.Format(utils.FormatDate) + ";currentItem.DataTime:" + currentItem.DataTime.Format(utils.FormatDate))
  688. for k := 1; k < day; k++ {
  689. needDay := preDate.AddDate(0, 0, k)
  690. dateDataMap[needDay] = preItem.Value
  691. }
  692. dateDataMap[currentItem.DataTime] = currentItem.Value
  693. // 如果是最后一期的数据
  694. if i == lastI {
  695. day = int(nowDay.Sub(currentDate).Hours() / float64(24))
  696. //utils.FileLog.Info("preItem.DataTime:" + currentDate.Format(utils.FormatDate) + ";currentItem.DataTime" + nowDay.Format(utils.FormatDate))
  697. for k := 1; k <= day; k++ {
  698. needDay := currentDate.AddDate(0, 0, k)
  699. dateDataMap[needDay] = currentItem.Value
  700. }
  701. }
  702. }
  703. return
  704. }
  705. // DownFrequency
  706. // @Description: 降频计算
  707. // @author: Roc
  708. // @receiver obj
  709. // @datetime2023-11-02 18:51:26
  710. // @return dateDataMap map[time.Time]float64
  711. // @return err error
  712. func (obj BaseCalculate) DownFrequency() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  713. dateDataMap = make(map[time.Time]float64)
  714. dataList := obj.DataList
  715. edbFrequency := obj.Frequency
  716. formula := obj.Formula.(string)
  717. if !CheckFrequency(obj.FromFrequency, obj.Frequency) {
  718. errMsg = "频度异常,不允许低频降频到高频"
  719. err = errors.New(errMsg)
  720. return
  721. }
  722. // 数据处理
  723. var dateArr []time.Time
  724. dataMap := make(map[time.Time]*EdbInfoData)
  725. fromDataMap := make(map[time.Time]float64)
  726. //来源指指标数据
  727. for _, v := range dataList {
  728. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  729. dateArr = append(dateArr, v.DataTime)
  730. dataMap[v.DataTime] = v
  731. fromDataMap[v.DataTime] = v.Value
  732. }
  733. // 数据计算
  734. dataLen := len(dataList)
  735. if dataLen <= 0 {
  736. return
  737. }
  738. startDataTime := dataList[0].DataTime
  739. endDataTime := dataList[dataLen-1].DataTime
  740. var lastValue float64 // 最近的值
  741. var nextEndDate time.Time // 下一个节点的日期
  742. weekDayDataList := make([]float64, 0)
  743. for tmpStartDataTime := startDataTime; !tmpStartDataTime.After(endDataTime); tmpStartDataTime = tmpStartDataTime.AddDate(0, 0, 1) {
  744. // 将当前数据加入到 weekDayDataList
  745. if tmpData, ok := dataMap[tmpStartDataTime]; ok {
  746. tmpValue := decimal.NewFromFloat(tmpData.Value)
  747. tmpValueFloat, _ := tmpValue.Round(4).Float64()
  748. weekDayDataList = append(weekDayDataList, tmpValueFloat)
  749. }
  750. // 如果下个节点的日期不存在,那么就先给赋值(兼容时间区间内只有一组数据的情况)
  751. if nextEndDate.IsZero() {
  752. nextEndDate = utils.GetFrequencyEndDay(tmpStartDataTime, edbFrequency)
  753. }
  754. // 日期处理过滤
  755. switch edbFrequency {
  756. case "周度":
  757. if tmpStartDataTime.Weekday() != 5 {
  758. //不是周五,代表需要进入下一个循环获取数据并计算
  759. continue
  760. } else {
  761. //记录下一个结束节点的日期
  762. nextEndDate = tmpStartDataTime.AddDate(0, 0, 7)
  763. }
  764. case "旬度":
  765. nextDay := tmpStartDataTime.AddDate(0, 0, 1)
  766. if nextDay.Day() != 1 && nextDay.Day() != 11 && nextDay.Day() != 21 {
  767. //不是每月10、20、最后一天,代表需要进入下一个循环获取数据并计算
  768. continue
  769. } else {
  770. //记录下一个结束节点的日期
  771. if nextDay.Day() == 1 || nextDay.Day() == 11 {
  772. //月初或者月末的时候,加10天就好了
  773. nextEndDate = nextDay.AddDate(0, 0, 9)
  774. } else {
  775. tmpNextMonth := nextDay.AddDate(0, 1, 0)
  776. nextEndDate = time.Date(tmpNextMonth.Year(), tmpNextMonth.Month(), 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
  777. }
  778. }
  779. case "月度":
  780. nextDay := tmpStartDataTime.AddDate(0, 0, 1)
  781. if nextDay.Day() != 1 {
  782. //不是每月最后一天,代表需要进入下一个循环获取数据并计算
  783. continue
  784. } else {
  785. //记录下一个结束节点的日期
  786. nextEndDate = nextDay.AddDate(0, 1, -1)
  787. }
  788. case "季度":
  789. nextDay := tmpStartDataTime.AddDate(0, 0, 1)
  790. if (nextDay.Month() == 1 || nextDay.Month() == 4 || nextDay.Month() == 7 || nextDay.Month() == 10) && nextDay.Day() == 1 {
  791. //记录下一个结束节点的日期
  792. nextEndDate = nextDay.AddDate(0, 3, -1)
  793. } else {
  794. //不是3,6,9,12 月份的最后一天,代表需要进入下一个循环获取数据并计算
  795. continue
  796. }
  797. case "年度":
  798. if tmpStartDataTime.Month() == 12 && tmpStartDataTime.Day() == 31 {
  799. //记录下一个结束节点的日期
  800. nextEndDate = tmpStartDataTime.AddDate(1, 0, 0)
  801. } else {
  802. //不是每年的12-31日,代表需要进入下一个循环获取数据并计算
  803. continue
  804. }
  805. default:
  806. err = errors.New("错误的频度:" + edbFrequency)
  807. return
  808. }
  809. // 当前时间段内的数据计算,得出实际值
  810. var currVal float64
  811. lenWeekDayDataList := len(weekDayDataList)
  812. // 如果这个时间区间内没有数据,那么就采用上一个时间区间的值
  813. if len(weekDayDataList) <= 0 {
  814. currVal = lastValue
  815. } else {
  816. if formula == "期末值" {
  817. currVal = weekDayDataList[lenWeekDayDataList-1]
  818. } else {
  819. // 平均值
  820. sumValDeci := decimal.NewFromFloat(0)
  821. for _, v := range weekDayDataList {
  822. tmpValDeci := decimal.NewFromFloat(v)
  823. sumValDeci = sumValDeci.Add(tmpValDeci)
  824. }
  825. lenDeci := decimal.NewFromInt(int64(lenWeekDayDataList))
  826. currVal, _ = sumValDeci.Div(lenDeci).Round(4).Float64()
  827. }
  828. }
  829. // 赋值
  830. dateDataMap[tmpStartDataTime] = currVal
  831. // 一轮结束后,数据清空
  832. weekDayDataList = make([]float64, 0)
  833. }
  834. // 最后已有的日期处理完成后,需要对剩余不在时间段内的数据做处理
  835. if len(weekDayDataList) > 0 {
  836. // 当前时间段内的数据计算,得出实际值
  837. var currVal float64
  838. lenWeekDayDataList := len(weekDayDataList)
  839. // 如果这个时间区间内没有数据,那么就采用上一个时间区间的值
  840. if len(weekDayDataList) < 0 {
  841. currVal = lastValue
  842. } else {
  843. if formula == "期末值" {
  844. currVal = weekDayDataList[lenWeekDayDataList-1]
  845. } else {
  846. // 平均值
  847. sumValDeci := decimal.NewFromFloat(0)
  848. for _, v := range weekDayDataList {
  849. tmpValDeci := decimal.NewFromFloat(v)
  850. sumValDeci = sumValDeci.Add(tmpValDeci)
  851. }
  852. lenDeci := decimal.NewFromInt(int64(lenWeekDayDataList))
  853. currVal, _ = sumValDeci.Div(lenDeci).Round(4).Float64()
  854. }
  855. }
  856. // 赋值
  857. dateDataMap[nextEndDate] = currVal
  858. }
  859. return
  860. }
  861. // TimeShift
  862. // @Description: 时间移位计算
  863. // @author: Roc
  864. // @receiver obj
  865. // @datetime2023-11-03 13:19:07
  866. // @return dateDataMap map[time.Time]float64
  867. // @return err error
  868. func (obj BaseCalculate) TimeShift() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  869. dateDataMap = make(map[time.Time]float64)
  870. dataList := obj.DataList
  871. nVal := fmt.Sprint(obj.Formula)
  872. formulaInt, err := strconv.Atoi(nVal)
  873. if err != nil {
  874. return
  875. }
  876. moveType := obj.MoveType
  877. moveFrequency := obj.MoveFrequency
  878. // 数据处理
  879. var dateArr []time.Time
  880. dataMap := make(map[time.Time]*EdbInfoData)
  881. for _, v := range dataList {
  882. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  883. dateArr = append(dateArr, v.DataTime)
  884. dataMap[v.DataTime] = v
  885. }
  886. var shiftDay int
  887. switch moveFrequency {
  888. case "天":
  889. shiftDay = formulaInt
  890. case "周":
  891. shiftDay = formulaInt * 7
  892. case "月":
  893. shiftDay = formulaInt * 30
  894. case "季":
  895. shiftDay = formulaInt * 90
  896. case "年":
  897. shiftDay = formulaInt * 365
  898. default:
  899. shiftDay = formulaInt
  900. }
  901. if moveType == 2 {
  902. shiftDay = -shiftDay
  903. }
  904. dataLen := len(dataList)
  905. for i := 0; i < dataLen; i++ {
  906. //当期
  907. currentItem := dataList[i]
  908. currentDate := currentItem.DataTime
  909. newDate := currentDate.AddDate(0, 0, shiftDay)
  910. val, _ := decimal.NewFromFloat(currentItem.Value).RoundCeil(4).Float64()
  911. dateDataMap[newDate] = val
  912. }
  913. return
  914. }
  915. // Cjjx
  916. // @Description: 超季节性计算
  917. // @author: Roc
  918. // @receiver obj
  919. // @datetime2023-11-03 13:28:23
  920. // @return dateDataMap map[time.Time]float64
  921. // @return err error
  922. func (obj BaseCalculate) Cjjx() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  923. dateDataMap = make(map[time.Time]float64)
  924. dataList := obj.DataList
  925. nVal := fmt.Sprint(obj.Formula)
  926. formulaInt, err := strconv.Atoi(nVal)
  927. if err != nil {
  928. return
  929. }
  930. calendar := obj.Calendar
  931. // 数据处理
  932. var dateArr []time.Time
  933. dataMap := make(map[time.Time]*EdbInfoData)
  934. for _, v := range dataList {
  935. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  936. dateArr = append(dateArr, v.DataTime)
  937. dataMap[v.DataTime] = v
  938. }
  939. // 通过插值法补全所有数据(包含周末)
  940. handleDataMap := make(map[time.Time]float64)
  941. _, err = HandleDataByLinearRegressionByTime(dataList, handleDataMap)
  942. if err != nil {
  943. return
  944. }
  945. // 每个年份的日期数据需要平移的天数
  946. moveDayMap := make(map[int]int, 0) // 每个年份的春节公历
  947. var lastDataDay time.Time
  948. if len(dataList) > 0 {
  949. lastDataDay = dataList[0].DataTime
  950. }
  951. // 数据计算
  952. for _, currentDate := range dateArr {
  953. //农历的超季节性运算,只计算11月--次年5月,分段计算,与数据区间和N数值有关
  954. if calendar == "农历" && currentDate.Month() > 5 && currentDate.Month() < 11 {
  955. continue
  956. }
  957. currentItem, ok := dataMap[currentDate]
  958. // 找不到数据就退出当前循环,进入下一循环
  959. if !ok {
  960. continue
  961. }
  962. pastValueList := make([]float64, 0) // 过去几期的数据
  963. pastValueList = append(pastValueList, currentItem.Value) //当前日期
  964. for i := 1; i < formulaInt; i++ {
  965. //前几年当天公历的日期
  966. historyPreDate := currentDate.AddDate(-i, 0, 0)
  967. moveDay := 0
  968. if calendar == "农历" {
  969. if tmpMoveDay, ok := moveDayMap[historyPreDate.Year()]; !ok {
  970. moveDay, err = getMoveDay(lastDataDay, historyPreDate)
  971. if err != nil {
  972. return
  973. }
  974. } else {
  975. moveDay = tmpMoveDay
  976. }
  977. // 移动天数到对应农历 的 公历 日期
  978. historyPreDate = historyPreDate.AddDate(0, 0, moveDay)
  979. }
  980. if tmpValue, ok := handleDataMap[historyPreDate]; ok { //上一年同期找到
  981. pastValueList = append(pastValueList, tmpValue)
  982. }
  983. }
  984. if len(pastValueList) == formulaInt {
  985. val, ok := cjjxSub(currentItem.Value, pastValueList)
  986. if !ok {
  987. continue
  988. }
  989. dateDataMap[currentDate] = val
  990. }
  991. }
  992. return
  993. }
  994. // cjjxSub
  995. // @Description: 超季节性计算
  996. // @author: Roc
  997. // @datetime2023-11-03 13:25:49
  998. // @param currValue float64
  999. // @param pastValue []float64
  1000. // @return value float64
  1001. // @return ok bool
  1002. func cjjxSub(currValue float64, pastValue []float64) (value float64, ok bool) {
  1003. num := len(pastValue)
  1004. if num == 0 {
  1005. return
  1006. }
  1007. numDecimal := decimal.NewFromInt(int64(num))
  1008. af := decimal.NewFromFloat(currValue)
  1009. bf := decimal.NewFromFloat(pastValue[0])
  1010. for k := 1; k < num; k++ {
  1011. tmpVal := decimal.NewFromFloat(pastValue[k])
  1012. bf = bf.Add(tmpVal)
  1013. }
  1014. value, _ = af.Sub(bf.Div(numDecimal)).RoundCeil(4).Float64()
  1015. ok = true
  1016. return
  1017. }
  1018. // Annualized
  1019. // @Description: 年化计算
  1020. // @author: Roc
  1021. // @receiver obj
  1022. // @datetime2023-11-03 13:34:32
  1023. // @param to orm.TxOrmer
  1024. // @param edbInfoId int
  1025. // @param source int
  1026. // @param fromEdbInfo *EdbInfo
  1027. // @param edbCode string
  1028. // @return dateDataMap map[time.Time]float64
  1029. // @return err error
  1030. func (obj BaseCalculate) Annualized() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  1031. dateDataMap = make(map[time.Time]float64)
  1032. fromDataList := obj.DataList
  1033. lenFromDataList := len(fromDataList)
  1034. // 如果来源指标没有数据,那么就直接返回得了
  1035. if lenFromDataList <= 0 {
  1036. return
  1037. }
  1038. // 数据处理
  1039. // 插值法数据处理
  1040. handleDataMap := make(map[time.Time]float64)
  1041. _, err = HandleDataByLinearRegressionByTime(fromDataList, handleDataMap)
  1042. if err != nil {
  1043. return
  1044. }
  1045. // 每年的最后一天的数据值
  1046. yearLastValMap := make(map[int]float64)
  1047. startDataTime := fromDataList[0].DataTime
  1048. endDataTime := fromDataList[lenFromDataList-1].DataTime
  1049. for i := startDataTime.Year(); i <= endDataTime.Year(); i++ {
  1050. tmpDate := time.Date(i, 12, 31, startDataTime.Hour(), startDataTime.Minute(), startDataTime.Second(), startDataTime.Nanosecond(), time.Local)
  1051. if tmpVal, ok := handleDataMap[tmpDate]; ok {
  1052. yearLastValMap[i] = tmpVal
  1053. }
  1054. }
  1055. // 数据计算
  1056. // 遍历来源指标数据
  1057. for _, v := range fromDataList {
  1058. currDate := v.DataTime
  1059. perValMap := make(map[time.Time]float64)
  1060. //前3年当日的数据
  1061. for i := 1; i <= 3; i++ {
  1062. tmpDateTime := currDate.AddDate(-i, 0, 0)
  1063. if tmpVal, ok := handleDataMap[tmpDateTime]; ok {
  1064. perValMap[tmpDateTime] = tmpVal
  1065. }
  1066. }
  1067. lenPerValMap := len(perValMap)
  1068. // 如果数据少于2年,那么就不参与计算,结束当前循环,进入下一个循环
  1069. if lenPerValMap < 2 {
  1070. continue
  1071. }
  1072. // N年 当前值占全年比重 的值列表
  1073. divValList := make([]decimal.Decimal, 0)
  1074. for tmpDateTime, tmpVal := range perValMap {
  1075. yearLastVal, ok2 := yearLastValMap[tmpDateTime.Year()]
  1076. // 如果当年最后一天没有数据
  1077. if !ok2 {
  1078. continue
  1079. }
  1080. // 当前值占全年比重
  1081. tmpYearLastVal := decimal.NewFromFloat(yearLastVal)
  1082. if tmpYearLastVal.IsZero() { //如果是0,那么就退出当前循环,进入下一个循环
  1083. continue
  1084. }
  1085. divVal := decimal.NewFromFloat(tmpVal).Div(tmpYearLastVal)
  1086. divValList = append(divValList, divVal)
  1087. }
  1088. lenDivValList := len(divValList)
  1089. // 如果 N年 当前值占全年比重 的值 小于 2个,那么就不参与计算,结束当前循环,进入下一个循环
  1090. if lenDivValList < 2 {
  1091. continue
  1092. }
  1093. divValSum := decimal.NewFromFloat(0)
  1094. for _, divVal := range divValList {
  1095. divValSum = divValSum.Add(divVal)
  1096. }
  1097. // 当前计算出来的结果
  1098. tmpDivVal := divValSum.Div(decimal.NewFromInt(int64(lenDivValList)))
  1099. if tmpDivVal.IsZero() { //如果是0,那么就退出当前循环,进入下一个循环
  1100. continue
  1101. }
  1102. currVal, _ := decimal.NewFromFloat(v.Value).Div(tmpDivVal).Round(4).Float64()
  1103. dateDataMap[currDate] = currVal
  1104. }
  1105. return
  1106. }
  1107. // Ljz
  1108. // @Description: 累计值
  1109. // @author: Roc
  1110. // @receiver obj
  1111. // @datetime2023-11-03 13:49:17
  1112. // @return dateDataMap map[time.Time]float64
  1113. // @return err error
  1114. func (obj BaseCalculate) Ljz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  1115. dateDataMap = make(map[time.Time]float64)
  1116. dataList := obj.DataList
  1117. frequency := obj.Frequency //需要变更的频度
  1118. fromFrequency := obj.FromFrequency //来源的频度
  1119. // 数据处理
  1120. var isWeekData bool // 是否周度数据,如果是周度数据的话,是需要变频的,最后结果还需要除以7
  1121. // 周度数据需要先变成日度的
  1122. if fromFrequency == `周度` {
  1123. isWeekData = true
  1124. }
  1125. fromEdbDataMap := make(map[time.Time]float64)
  1126. if isWeekData {
  1127. dataList, err = HandleDataByLinearRegressionByTime(dataList, fromEdbDataMap)
  1128. if err != nil {
  1129. return
  1130. }
  1131. }
  1132. //日度转周度:日期选周五,计算上周六到本周五的日度值的加总,最新日期为最新值对应的周五。
  1133. //日度转月度:日期选每个月最后一天,计算当月所有日度值的加总,最新日期为最新值对应当月最后一天。
  1134. //日度转季度、年度:方法类似转月度。
  1135. //周度转月度/季度/年度:将周度值转成日度,空值用插值法插值,计算当月/当季/当年所有值的加总,然后除以7。
  1136. //月度转季度/年度: 当季/当年月度值相加。
  1137. dateList := make([]time.Time, 0)
  1138. valueMap := make(map[time.Time]float64)
  1139. switch frequency {
  1140. case "年度":
  1141. yearMap := make(map[int]float64)
  1142. yearList := make([]int, 0)
  1143. for _, item := range dataList {
  1144. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1145. year := item.DataTime.Year()
  1146. yearVal, ok := yearMap[year]
  1147. if ok {
  1148. yearMap[year] = item.Value + yearVal
  1149. } else {
  1150. yearList = append(yearList, year)
  1151. yearMap[year] = item.Value
  1152. }
  1153. }
  1154. for _, v := range yearList {
  1155. currTime := time.Date(v, 12, 31, 0, 0, 0, 0, time.Local)
  1156. dateList = append(dateList, currTime)
  1157. valueMap[currTime] = yearMap[v]
  1158. }
  1159. case "半年度":
  1160. yearMonthMap := make(map[string]float64)
  1161. yearMonthList := make([]string, 0)
  1162. for _, item := range dataList {
  1163. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1164. itemDate := item.DataTime
  1165. year := itemDate.Year()
  1166. var tmpK string
  1167. if itemDate.Month() <= 6 {
  1168. tmpK = fmt.Sprint(year, "06")
  1169. } else {
  1170. tmpK = fmt.Sprint(year, "12")
  1171. }
  1172. yearVal, ok := yearMonthMap[tmpK]
  1173. if ok {
  1174. yearMonthMap[tmpK] = item.Value + yearVal
  1175. } else {
  1176. yearMonthList = append(yearMonthList, tmpK)
  1177. yearMonthMap[tmpK] = item.Value
  1178. }
  1179. }
  1180. for _, v := range yearMonthList {
  1181. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  1182. if tmpErr != nil {
  1183. err = tmpErr
  1184. return
  1185. }
  1186. currTime = currTime.AddDate(0, 1, -1)
  1187. dateList = append(dateList, currTime)
  1188. valueMap[currTime] = yearMonthMap[v]
  1189. }
  1190. case "季度":
  1191. yearMonthMap := make(map[string]float64)
  1192. yearMonthList := make([]string, 0)
  1193. for _, item := range dataList {
  1194. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1195. itemDate := item.DataTime
  1196. year := itemDate.Year()
  1197. var tmpK string
  1198. if itemDate.Month() <= 3 {
  1199. tmpK = fmt.Sprint(year, "03")
  1200. } else if itemDate.Month() <= 6 {
  1201. tmpK = fmt.Sprint(year, "06")
  1202. } else if itemDate.Month() <= 9 {
  1203. tmpK = fmt.Sprint(year, "09")
  1204. } else {
  1205. tmpK = fmt.Sprint(year, "12")
  1206. }
  1207. yearVal, ok := yearMonthMap[tmpK]
  1208. if ok {
  1209. yearMonthMap[tmpK] = item.Value + yearVal
  1210. } else {
  1211. yearMonthList = append(yearMonthList, tmpK)
  1212. yearMonthMap[tmpK] = item.Value
  1213. }
  1214. }
  1215. for _, v := range yearMonthList {
  1216. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  1217. if tmpErr != nil {
  1218. err = tmpErr
  1219. return
  1220. }
  1221. currTime = currTime.AddDate(0, 1, -1)
  1222. dateList = append(dateList, currTime)
  1223. valueMap[currTime] = yearMonthMap[v]
  1224. }
  1225. case "月度":
  1226. yearMonthMap := make(map[string]float64)
  1227. yearMonthList := make([]string, 0)
  1228. for _, item := range dataList {
  1229. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1230. itemDate := item.DataTime
  1231. year := itemDate.Year()
  1232. var tmpK string
  1233. tmpK = fmt.Sprint(year*100 + int(itemDate.Month()))
  1234. yearVal, ok := yearMonthMap[tmpK]
  1235. if ok {
  1236. yearMonthMap[tmpK] = item.Value + yearVal
  1237. } else {
  1238. yearMonthList = append(yearMonthList, tmpK)
  1239. yearMonthMap[tmpK] = item.Value
  1240. }
  1241. }
  1242. for _, v := range yearMonthList {
  1243. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  1244. if tmpErr != nil {
  1245. err = tmpErr
  1246. return
  1247. }
  1248. currTime = currTime.AddDate(0, 1, -1)
  1249. dateList = append(dateList, currTime)
  1250. valueMap[currTime] = yearMonthMap[v]
  1251. }
  1252. case "旬度":
  1253. tmpDateDataMap := make(map[time.Time]float64)
  1254. tmpDateList := make([]time.Time, 0)
  1255. for _, item := range dataList {
  1256. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1257. itemDate := item.DataTime
  1258. dayInt := itemDate.Year()*100 + int(itemDate.Month())
  1259. var currTime time.Time
  1260. if itemDate.Day() <= 10 {
  1261. //本月上旬
  1262. tmpK := fmt.Sprint(dayInt, "10")
  1263. currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
  1264. if err != nil {
  1265. return
  1266. }
  1267. } else if itemDate.Day() <= 20 {
  1268. // 本月中旬
  1269. tmpK := fmt.Sprint(dayInt, "20")
  1270. currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
  1271. if err != nil {
  1272. return
  1273. }
  1274. } else {
  1275. // 本月下旬
  1276. currTime, err = time.ParseInLocation(utils.FormatYearMonthUnSpace, fmt.Sprint(dayInt), time.Local)
  1277. if err != nil {
  1278. return
  1279. }
  1280. currTime = currTime.AddDate(0, 1, -1)
  1281. }
  1282. yearVal, ok := tmpDateDataMap[currTime]
  1283. if ok {
  1284. tmpDateDataMap[currTime] = item.Value + yearVal
  1285. } else {
  1286. tmpDateList = append(tmpDateList, currTime)
  1287. tmpDateDataMap[currTime] = item.Value
  1288. }
  1289. }
  1290. for _, currTime := range tmpDateList {
  1291. dateList = append(dateList, currTime)
  1292. valueMap[currTime] = tmpDateDataMap[currTime]
  1293. }
  1294. case "周度":
  1295. tmpDateDataMap := make(map[time.Time]float64)
  1296. tmpDateList := make([]time.Time, 0)
  1297. for _, item := range dataList {
  1298. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1299. itemDate := item.DataTime
  1300. var currTime time.Time
  1301. // 周六周日,这是下一个周五的数据
  1302. if itemDate.Weekday() == 0 {
  1303. currTime = itemDate.AddDate(0, 0, 5)
  1304. } else if itemDate.Weekday() == 6 {
  1305. currTime = itemDate.AddDate(0, 0, 6)
  1306. } else {
  1307. currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
  1308. }
  1309. yearVal, ok := tmpDateDataMap[currTime]
  1310. if ok {
  1311. tmpDateDataMap[currTime] = item.Value + yearVal
  1312. } else {
  1313. tmpDateList = append(tmpDateList, currTime)
  1314. tmpDateDataMap[currTime] = item.Value
  1315. }
  1316. }
  1317. for _, currTime := range tmpDateList {
  1318. dateList = append(dateList, currTime)
  1319. valueMap[currTime] = tmpDateDataMap[currTime]
  1320. }
  1321. default:
  1322. err = errors.New("错误的频度")
  1323. return
  1324. }
  1325. // 数据计算
  1326. for _, currTime := range dateList {
  1327. tmpVal, ok2 := valueMap[currTime]
  1328. if !ok2 {
  1329. err = errors.New("数据异常,date:" + currTime.Format(utils.FormatDate))
  1330. return
  1331. }
  1332. var saveValue float64
  1333. if isWeekData { //周度指标转的话,最后结果要除以7
  1334. saveValue, _ = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).Float64()
  1335. } else {
  1336. saveValue, _ = decimal.NewFromFloat(tmpVal).Round(4).Float64()
  1337. }
  1338. dateDataMap[currTime] = saveValue
  1339. }
  1340. return
  1341. }
  1342. // LjzNczj
  1343. // @Description:年初至今累计值计算
  1344. // @author: Roc
  1345. // @receiver obj
  1346. // @datetime2023-11-03 13:55:44
  1347. // @return dateDataMap map[time.Time]float64
  1348. // @return err error
  1349. func (obj BaseCalculate) LjzNczj() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  1350. dateDataMap = make(map[time.Time]float64)
  1351. dataList := obj.DataList
  1352. frequency := obj.Frequency //需要变更的频度
  1353. fromFrequency := obj.FromFrequency //来源的频度
  1354. // 数据处理
  1355. var isWeekData bool // 是否周度数据,如果是周度数据的话,是需要变频的,最后结果还需要除以7
  1356. // 周度数据需要先变成日度的
  1357. if fromFrequency == `周度` {
  1358. isWeekData = true
  1359. }
  1360. fromEdbDataMap := make(map[time.Time]float64)
  1361. if isWeekData {
  1362. dataList, err = HandleDataByLinearRegressionByTime(dataList, fromEdbDataMap)
  1363. if err != nil {
  1364. return
  1365. }
  1366. }
  1367. //日度数据年初至今:日期同原日度数据。将每年1月1日(含)到日度数据所在日期含间的日度值,进行加总。
  1368. //周度数据年初至今:日期同原周度数据。将周度值转成日度频率,空值用插值法插值,然后算法同日度年度至今,再除以7
  1369. //月度/季度数据年初至今:日期同原月度/季度数据,将每年1月1日(含)到月度数据所在日期(含)之间的月度/季度值,进行加总
  1370. //以此类推
  1371. dateList := make([]time.Time, 0)
  1372. valueMap := make(map[time.Time]float64)
  1373. yearMap := make(map[int]float64)
  1374. switch frequency {
  1375. case "周度":
  1376. tmpDateDataMap := make(map[time.Time]float64)
  1377. tmpDateList := make([]time.Time, 0)
  1378. for _, item := range dataList {
  1379. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1380. itemDate := item.DataTime
  1381. var currTime time.Time
  1382. // 周六周日,这是下一个周五的数据
  1383. if itemDate.Weekday() == 0 {
  1384. currTime = itemDate.AddDate(0, 0, 5)
  1385. } else if itemDate.Weekday() == 6 {
  1386. currTime = itemDate.AddDate(0, 0, 6)
  1387. } else {
  1388. currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
  1389. }
  1390. year := itemDate.Year()
  1391. yearVal, ok := yearMap[year]
  1392. if ok {
  1393. yearMap[year] = item.Value + yearVal
  1394. } else {
  1395. yearMap[year] = item.Value
  1396. }
  1397. if itemDate.Equal(currTime) {
  1398. tmpDateDataMap[itemDate] = yearMap[year]
  1399. tmpDateList = append(tmpDateList, itemDate)
  1400. }
  1401. }
  1402. for _, currTime := range tmpDateList {
  1403. dateList = append(dateList, currTime)
  1404. valueMap[currTime] = tmpDateDataMap[currTime]
  1405. }
  1406. default:
  1407. for _, item := range dataList {
  1408. item.DataTime = time.Date(item.DataTime.Year(), item.DataTime.Month(), item.DataTime.Day(), 0, 0, 0, 0, item.DataTime.Location())
  1409. itemDate := item.DataTime
  1410. year := itemDate.Year()
  1411. yearVal, ok := yearMap[year]
  1412. if ok {
  1413. yearMap[year] = item.Value + yearVal
  1414. } else {
  1415. yearMap[year] = item.Value
  1416. }
  1417. valueMap[itemDate] = yearMap[year]
  1418. dateList = append(dateList, itemDate)
  1419. }
  1420. }
  1421. // 数据计算
  1422. for _, currTime := range dateList {
  1423. tmpVal, ok2 := valueMap[currTime]
  1424. if !ok2 {
  1425. err = errors.New("数据异常,date:" + currTime.Format(utils.FormatDate))
  1426. return
  1427. }
  1428. var saveValue float64
  1429. if isWeekData { //周度指标转的话,最后结果要除以7
  1430. saveValue, _ = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).Float64()
  1431. } else {
  1432. saveValue, _ = decimal.NewFromFloat(tmpVal).Round(4).Float64()
  1433. }
  1434. dateDataMap[currTime] = saveValue
  1435. }
  1436. return
  1437. }
  1438. // ExponentialSmoothing
  1439. // @Description: 指数修匀计算
  1440. // @author: Roc
  1441. // @receiver obj
  1442. // @datetime2023-11-03 14:07:47
  1443. // @return dateDataMap map[time.Time]float64
  1444. // @return err error
  1445. func (obj BaseCalculate) ExponentialSmoothing() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  1446. dateDataMap = make(map[time.Time]float64)
  1447. dataList := obj.DataList
  1448. formula := obj.Formula.(string) // alpha值
  1449. alpha, _ := strconv.ParseFloat(formula, 64)
  1450. if alpha <= 0 || alpha >= 1 {
  1451. err = fmt.Errorf("alpha值有误: %v", alpha)
  1452. return
  1453. }
  1454. // 获取标准差图表的指标数据
  1455. fromDataList, err := calculateExponentialSmoothingData(dataList, alpha)
  1456. if err != nil {
  1457. return
  1458. }
  1459. // 数据计算
  1460. for _, tmpData := range fromDataList {
  1461. if tmpData.DataTime.IsZero() {
  1462. continue
  1463. }
  1464. // 当前的实际值
  1465. saveValue, _ := decimal.NewFromFloat(tmpData.Value).Round(4).Float64()
  1466. dateDataMap[tmpData.DataTime] = saveValue
  1467. }
  1468. return
  1469. }
  1470. // calculateExponentialSmoothingData
  1471. // @Description: 计算指数修匀
  1472. // @author: Roc
  1473. // @datetime2023-11-03 14:05:41
  1474. // @param dataList []*EdbInfoData 时间基准指标在时间区间内的值
  1475. // @param strAlpha float64
  1476. // @return newDataList []EdbInfoData
  1477. // @return err error
  1478. func calculateExponentialSmoothingData(dataList []*EdbInfoData, alpha float64) (newDataList []EdbInfoData, err error) {
  1479. if alpha <= 0 || alpha >= 1 {
  1480. err = fmt.Errorf("alpha值有误: %v", alpha)
  1481. return
  1482. }
  1483. var preVal float64
  1484. alphaDecimal := decimal.NewFromFloat(alpha)
  1485. subAlpha := decimal.NewFromFloat(1).Sub(alphaDecimal)
  1486. for k, d := range dataList {
  1487. d.DataTime = time.Date(d.DataTime.Year(), d.DataTime.Month(), d.DataTime.Day(), 0, 0, 0, 0, d.DataTime.Location())
  1488. // 首期的值以原始值作为指数修匀的计算值
  1489. if k == 0 {
  1490. newDataList = append(newDataList, EdbInfoData{
  1491. EdbDataId: k,
  1492. DataTime: dataList[k].DataTime,
  1493. Value: d.Value,
  1494. })
  1495. preVal = d.Value
  1496. continue
  1497. }
  1498. // 上一期的值参与计算
  1499. preDecimal := decimal.NewFromFloat(preVal)
  1500. valDecimal := decimal.NewFromFloat(d.Value)
  1501. partA := alphaDecimal.Mul(valDecimal)
  1502. partB := subAlpha.Mul(preDecimal)
  1503. res, _ := (partA.Add(partB)).Float64()
  1504. preVal = res
  1505. newDataList = append(newDataList, EdbInfoData{
  1506. EdbDataId: k,
  1507. DataTime: dataList[k].DataTime,
  1508. Value: res,
  1509. })
  1510. }
  1511. return
  1512. }
  1513. // Rjz
  1514. // @Description: 日均值计算
  1515. // @author: Roc
  1516. // @receiver obj
  1517. // @datetime2023-11-03 14:47:47
  1518. // @return dateDataMap map[time.Time]float64
  1519. // @return err error
  1520. func (obj BaseCalculate) Rjz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
  1521. dateDataMap = make(map[time.Time]float64)
  1522. dataList := obj.DataList
  1523. fromFrequency := obj.FromFrequency
  1524. if fromFrequency == `` {
  1525. errMsg = "原指标的频度不能为空"
  1526. err = errors.New(errMsg)
  1527. return
  1528. }
  1529. if obj.FromFrequency == "日度" {
  1530. errMsg = "日度指标无需进行日均值计算"
  1531. err = errors.New(errMsg)
  1532. return
  1533. }
  1534. // 数据处理
  1535. // 数据降序
  1536. dataList = reverseSliceByDesc(dataList)
  1537. var dateArr []time.Time
  1538. dataMap := make(map[time.Time]*EdbInfoData)
  1539. for _, v := range dataList {
  1540. v.DataTime = time.Date(v.DataTime.Year(), v.DataTime.Month(), v.DataTime.Day(), 0, 0, 0, 0, v.DataTime.Location())
  1541. dateArr = append(dateArr, v.DataTime)
  1542. dataMap[v.DataTime] = v
  1543. }
  1544. // 数据计算
  1545. for _, currentDate := range dateArr {
  1546. //当前日期
  1547. currentItem, ok := dataMap[currentDate]
  1548. if !ok {
  1549. continue
  1550. }
  1551. //根据频度计算需要均分的天数
  1552. days := GetRjzFrequencyDays(currentDate, fromFrequency)
  1553. val, ok := rjzDivV2(currentItem.Value, days)
  1554. if !ok {
  1555. continue
  1556. }
  1557. dateDataMap[currentDate] = val
  1558. }
  1559. return
  1560. }
  1561. // rjzDivV2
  1562. // @Description: 日均值计算
  1563. // @author: Roc
  1564. // @datetime2023-11-03 14:47:36
  1565. // @param a float64
  1566. // @param b int
  1567. // @return val float64
  1568. // @return ok bool
  1569. func rjzDivV2(a float64, b int) (val float64, ok bool) {
  1570. if b == 0 {
  1571. return
  1572. }
  1573. af := decimal.NewFromFloat(a)
  1574. bf := decimal.NewFromFloat(float64(b))
  1575. val, _ = af.Div(bf).Round(4).Float64()
  1576. ok = true
  1577. return
  1578. }
  1579. // reverseSliceByDesc
  1580. // @Description: 倒转切片
  1581. // @author: Roc
  1582. // @datetime2023-11-06 14:08:25
  1583. // @param slice []*EdbInfoData
  1584. // @return []*EdbInfoData
  1585. func reverseSliceByDesc(slice []*EdbInfoData) (newSlice []*EdbInfoData) {
  1586. // 只有一个数据的话,那么就直接返回
  1587. if len(slice) <= 1 {
  1588. newSlice = slice
  1589. return
  1590. }
  1591. if !slice[1].DataTime.After(slice[0].DataTime) {
  1592. newSlice = slice
  1593. return
  1594. }
  1595. newSlice = make([]*EdbInfoData, len(slice))
  1596. for i, j := 0, len(slice)-1; i < len(slice); i, j = i+1, j-1 {
  1597. newSlice[i] = slice[j]
  1598. }
  1599. return
  1600. }
  1601. // EdbInfoSearchDataToData
  1602. // @Description: 将EdbInfoSearchData转成EdbInfoData
  1603. // @author: Roc
  1604. // @datetime2023-11-09 11:11:31
  1605. // @param dataList []*EdbInfoSearchData
  1606. // @return newDateDataList []*EdbInfoData
  1607. // @return err error
  1608. func EdbInfoSearchDataToData(dataList []*EdbInfoSearchData) (newDateDataList []*EdbInfoData, err error) {
  1609. newDateDataList = make([]*EdbInfoData, 0)
  1610. for _, data := range dataList {
  1611. dateTime, tmpErr := time.ParseInLocation(utils.FormatDate, data.DataTime, time.Local)
  1612. if tmpErr != nil {
  1613. err = tmpErr
  1614. return
  1615. }
  1616. newDateDataList = append(newDateDataList, &EdbInfoData{
  1617. EdbDataId: data.EdbDataId,
  1618. DataTime: dateTime,
  1619. Value: data.Value,
  1620. })
  1621. }
  1622. return
  1623. }
  1624. // GetDateDataAndDateList
  1625. // @Description: 通过时间数据map 获取 时间字符串数据map和日期字符串切片
  1626. // @author: Roc
  1627. // @datetime2023-11-09 13:10:38
  1628. // @param dateDataMap map[time.Time]float64
  1629. // @return dateStrDataMap map[string]float64
  1630. // @return dateStrList []string
  1631. func GetDateDataAndDateList(dateDataMap map[time.Time]float64) (dateStrDataMap map[string]float64, dateStrList []string) {
  1632. dateStrDataMap = make(map[string]float64)
  1633. dateStrList = make([]string, 0)
  1634. dateList := make([]time.Time, 0)
  1635. for date, val := range dateDataMap {
  1636. dateStr := date.Format(utils.FormatDate)
  1637. dateStrDataMap[dateStr] = val
  1638. dateList = append(dateList, date)
  1639. }
  1640. // 使用 sort.Slice 对日期切片进行排序
  1641. sort.Slice(dateList, func(i, j int) bool {
  1642. return dateList[i].Before(dateList[j])
  1643. })
  1644. // 将日期排序后返回日期字符串切片
  1645. for _, v := range dateList {
  1646. dateStrList = append(dateStrList, v.Format(utils.FormatDate))
  1647. }
  1648. return
  1649. }