12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814 |
- package models
- import (
- "errors"
- "eta/eta_index_lib/utils"
- "fmt"
- "github.com/shopspring/decimal"
- "sort"
- "strconv"
- "strings"
- "time"
- )
- type EdbInfoData struct {
- EdbDataId int `description:"数据ID"`
- DataTime time.Time `description:"数据日期"`
- Value float64 `description:"数据"`
- }
- type BaseCalculateBySearchData struct {
- DataList []*EdbInfoSearchData
- Frequency string `description:"需要转换的频度"`
- Formula interface{}
- Calendar string `description:"公历/农历"`
- MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
- MoveFrequency string `description:"移动频度"`
- FromFrequency string `description:"来源的频度"`
- Source int `description:"1:累计值转月;2:累计值转季;3:同比值;4:同差值;5:N数值移动平均数计算;6:环比值;7:环差值;8:升频;9:降频;10:时间移位;11:超季节性;12:年化;13:累计值;14:累计值年初至今;15:指数修匀;16:日均值"`
- }
- type BaseCalculate struct {
- DataList []*EdbInfoData
- Frequency string `description:"需要转换的频度"`
- Formula interface{}
- Calendar string `description:"公历/农历"`
- MoveType int `description:"移动方式:1:领先(默认),2:滞后"`
- MoveFrequency string `description:"移动频度"`
- FromFrequency string `description:"来源的频度"`
- Source int `description:"1:累计值转月;2:累计值转季;3:同比值;4:同差值;5:N数值移动平均数计算;6:环比值;7:环差值;8:升频;9:降频;10:时间移位;11:超季节性;12:年化;13:累计值;14:累计值年初至今;15:指数修匀;16:日均值"`
- }
- type BaseCalculateResp struct {
- DataMap map[string]float64
- DateList []string
- }
- func (obj BaseCalculate) Ljzzy() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
-
- yearMap := make(map[int]map[int]*EdbInfoData)
- dataLen := len(dataList)
- for i := 0; i < dataLen; i++ {
- item := dataList[i]
-
- itemDate := item.DataTime
- year := itemDate.Year()
- month := int(itemDate.Month())
- if monthMap, yok := yearMap[year]; yok {
- monthMap[month] = item
- yearMap[year] = monthMap
- } else {
- monthMap = make(map[int]*EdbInfoData)
- monthMap[month] = item
- yearMap[year] = monthMap
- }
- }
-
- for yk, yv := range yearMap {
- _, oneMonthOk := yv[1]
- _, twoMonthOk := yv[2]
- if !oneMonthOk && !twoMonthOk {
- continue
- }
- for i := 1; i <= 12; i++ {
-
- dataCurrentItem := yv[i]
- if dataCurrentItem == nil {
- continue
- }
- var date time.Time
- var val float64
-
- if i == 1 || i == 2 {
- if _, mok := yv[1]; mok {
- if i == 1 {
- date = dataCurrentItem.DataTime
- val, _ = decimal.NewFromFloat(dataCurrentItem.Value).Float64()
- }
- if i == 2 {
- dataOneItem := yv[1]
- date = dataCurrentItem.DataTime
- twoMonth := decimal.NewFromFloat(dataCurrentItem.Value)
- oneMonth := decimal.NewFromFloat(dataOneItem.Value)
- val, _ = twoMonth.Sub(oneMonth).Float64()
- }
- } else {
- dataTwoItem := yv[2]
- if i == 1 {
- dateStr := strconv.Itoa(yk) + "-01-31"
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
- if err != nil {
- return
- }
- }
- if i == 2 {
-
- {
- dateStr := strconv.Itoa(yk) + "-01-31"
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
- if err != nil {
- return
- }
- dateDataMap[date] = val
- }
-
-
- date = dataCurrentItem.DataTime
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- }
- }
- } else {
- dataPreItem := yv[i-1]
- if dataCurrentItem != nil && dataPreItem != nil {
- date = dataCurrentItem.DataTime
-
- a := decimal.NewFromFloat(dataCurrentItem.Value)
- b := decimal.NewFromFloat(dataPreItem.Value)
- val, _ = a.Sub(b).Float64()
- }
- }
-
- if date.IsZero() {
- continue
- }
- dateDataMap[date] = val
- }
- }
- return
- }
- func (obj BaseCalculate) Ljzzj() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
-
- yearMap := make(map[int]map[int]*EdbInfoData)
- dataLen := len(dataList)
- for i := 0; i < dataLen; i++ {
- item := dataList[i]
-
- itemDate := item.DataTime
- year := itemDate.Year()
- quarter := utils.MonthQuarterMap[int(itemDate.Month())]
- if quarterMap, yok := yearMap[year]; yok {
- quarterMap[quarter] = item
- yearMap[year] = quarterMap
- } else {
- quarterMap = make(map[int]*EdbInfoData)
- quarterMap[quarter] = item
- yearMap[year] = quarterMap
- }
- }
-
- for yk, yv := range yearMap {
- _, oneQuarterOk := yv[1]
- _, twoQuarterOk := yv[2]
- if !oneQuarterOk && !twoQuarterOk {
- continue
- }
- for i := 1; i <= 4; i++ {
-
- dataCurrentItem := yv[i]
- if dataCurrentItem == nil {
- continue
- }
- var date time.Time
- var val float64
-
- if i == 1 || i == 2 {
- if _, mok := yv[1]; mok {
- if i == 1 {
- date = dataCurrentItem.DataTime
- val, _ = decimal.NewFromFloat(dataCurrentItem.Value).Float64()
- }
- if i == 2 {
- dataOneItem := yv[1]
- date = dataCurrentItem.DataTime
- twoQuarter := decimal.NewFromFloat(dataCurrentItem.Value)
- oneQuarter := decimal.NewFromFloat(dataOneItem.Value)
- val, _ = twoQuarter.Sub(oneQuarter).Float64()
- }
- } else {
- dataTwoItem := yv[2]
- if i == 1 {
- dateStr := strconv.Itoa(yk) + "-03-31"
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
- if err != nil {
- return
- }
- }
- if i == 2 {
-
- {
- dateStr := strconv.Itoa(yk) + "-03-31"
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- date, err = time.ParseInLocation(utils.FormatDate, dateStr, time.Local)
- if err != nil {
- return
- }
- dateDataMap[date] = val
- }
-
- date = dataCurrentItem.DataTime
- a := decimal.NewFromFloat(dataTwoItem.Value)
- b := decimal.NewFromFloat(2.0)
- val, _ = a.Div(b).Float64()
- }
- }
- } else {
- dataPreItem := yv[i-1]
- if dataCurrentItem != nil && dataPreItem != nil {
- date = dataCurrentItem.DataTime
-
- a := decimal.NewFromFloat(dataCurrentItem.Value)
- b := decimal.NewFromFloat(dataPreItem.Value)
- val, _ = a.Sub(b).Float64()
- }
- }
-
- if date.IsZero() {
- continue
- }
- dateDataMap[date] = val
- }
- }
- return
- }
- func (obj BaseCalculate) Tbz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- frequency := obj.FromFrequency
-
-
- dataList = reverseSliceByDesc(dataList)
- var dateArr []time.Time
- dataMap := make(map[string]*EdbInfoData)
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime.Format(utils.FormatDate)] = v
- }
-
- for _, currentDate := range dateArr {
-
- currentDateStr := currentDate.Format(utils.FormatDate)
- currentItem, ok := dataMap[currentDateStr]
- if !ok {
- continue
- }
-
- var findItem *EdbInfoData
-
- preDate := currentDate.AddDate(-1, 0, 0)
- preDateStr := preDate.Format(utils.FormatDate)
- if findItem, ok = dataMap[preDateStr]; !ok {
- if frequency == "月度" {
- for i := 0; i <= 35; i++ {
- nextDateDay := preDate.AddDate(0, 0, i)
- nextDateDayStr := nextDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[nextDateDayStr]; ok {
- break
- } else {
- preDateDay := preDate.AddDate(0, 0, -i)
- preDateDayStr := preDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[preDateDayStr]; ok {
- break
- }
- }
- }
- } else if frequency == "季度" || frequency == "年度" {
- if findItem, ok = dataMap[preDateStr]; ok {
- break
- }
- } else {
- nextDateDay := preDate.AddDate(0, 0, 1)
- nextDateDayStr := nextDateDay.Format(utils.FormatDate)
- preDateDay := preDate.AddDate(0, 0, -1)
- preDateDayStr := preDateDay.Format(utils.FormatDate)
- for i := 0; i < 35; i++ {
- if i >= 1 {
- nextDateDay = nextDateDay.AddDate(0, 0, i)
- nextDateDayStr = nextDateDay.Format(utils.FormatDate)
- }
- if findItem, ok = dataMap[nextDateDayStr]; ok {
- break
- } else {
- if i >= 1 {
- preDateDay = preDateDay.AddDate(0, 0, -i)
- preDateDayStr = preDateDay.Format(utils.FormatDate)
- }
- if findItem, ok = dataMap[preDateDayStr]; ok {
- break
- }
- }
- }
- }
- }
-
- if findItem == nil {
- continue
- }
- dateDataMap[currentDate] = tbzDiv(currentItem.Value, findItem.Value)
- }
- return
- }
- func tbzDiv(a, b float64) float64 {
- var val float64
- if b != 0 {
- af := decimal.NewFromFloat(a)
- bf := decimal.NewFromFloat(b)
- val, _ = af.Div(bf).Sub(decimal.NewFromFloat(1)).RoundCeil(4).Float64()
- }
- return val
- }
- func (obj BaseCalculate) Tcz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- frequency := obj.FromFrequency
-
-
- dataList = reverseSliceByDesc(dataList)
- var dateArr []time.Time
- dataMap := make(map[string]*EdbInfoData)
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime.Format(utils.FormatDate)] = v
- }
-
- for _, currentDate := range dateArr {
-
- currentItem, ok := dataMap[currentDate.Format(utils.FormatDate)]
- if !ok {
- continue
- }
-
- var findItem *EdbInfoData
-
- preDate := currentDate.AddDate(-1, 0, 0)
- preDateStr := preDate.Format(utils.FormatDate)
- if findItem, ok = dataMap[preDateStr]; !ok {
-
- if frequency == "月度" {
- for i := 0; i <= 35; i++ {
- nextDateDay := preDate.AddDate(0, 0, i)
- nextDateDayStr := nextDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[nextDateDayStr]; ok {
- break
- } else {
- preDateDay := preDate.AddDate(0, 0, -i)
- preDateDayStr := preDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[preDateDayStr]; ok {
- break
- }
- }
- }
- } else if frequency == "季度" || frequency == "年度" {
- if findItem, ok = dataMap[preDateStr]; ok {
- break
- }
- } else {
- for i := 0; i < 35; i++ {
- nextDateDay := preDate.AddDate(0, 0, i)
- nextDateDayStr := nextDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[nextDateDayStr]; ok {
- break
- } else {
- preDateDay := preDate.AddDate(0, 0, -i)
- preDateDayStr := preDateDay.Format(utils.FormatDate)
- if findItem, ok = dataMap[preDateDayStr]; ok {
- break
- }
- }
- }
- }
- }
-
- if findItem == nil {
- continue
- }
- dateDataMap[currentDate] = tczSub(currentItem.Value, findItem.Value)
- }
- return
- }
- func tczSub(a, b float64) float64 {
- af := decimal.NewFromFloat(a)
- bf := decimal.NewFromFloat(b)
- val, _ := af.Sub(bf).RoundCeil(4).Float64()
- return val
- }
- func (obj BaseCalculate) Nszydpjjs() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- fromDataList := obj.DataList
- nVal := fmt.Sprint(obj.Formula)
- formulaInt, err := strconv.Atoi(nVal)
- if err != nil {
- return
- }
-
-
- fromDataList = reverseSliceByDesc(fromDataList)
- var fromDateArr []time.Time
- fromDataMap := make(map[time.Time]*EdbInfoData)
- for _, v := range fromDataList {
- fromDateArr = append(fromDateArr, v.DataTime)
- fromDataMap[v.DataTime] = v
- }
- arrLen := len(fromDateArr)
- for ak, currentDate := range fromDateArr {
-
- var valArr []float64
- if findItem, ok := fromDataMap[currentDate]; ok {
- valArr = append(valArr, findItem.Value)
- } else {
- continue
- }
- if ak+1 != arrLen {
-
- for i := 1; i < formulaInt; i++ {
- arrIndex := ak + i
- if arrIndex >= arrLen {
- break
- }
- arrVal := fromDateArr[arrIndex]
- if findItem, ok := fromDataMap[arrVal]; ok {
- valArr = append(valArr, findItem.Value)
- } else {
- continue
- }
- }
- }
- valArrLen := len(valArr)
-
- totalVal := decimal.NewFromFloat(0.00)
- for _, v := range valArr {
- newDecimal := decimal.NewFromFloat(v)
- totalVal = totalVal.Add(newDecimal)
- }
- af := totalVal
- bf := decimal.NewFromFloat(float64(valArrLen))
- val, _ := af.Div(bf).RoundCeil(4).Float64()
- dateDataMap[currentDate] = val
- }
- return
- }
- func (obj BaseCalculate) Hbz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- nVal := fmt.Sprint(obj.Formula)
- formulaInt, err := strconv.Atoi(nVal)
- if err != nil {
- return
- }
-
-
- dataList = reverseSliceByDesc(dataList)
- var dateArr []time.Time
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- }
-
- dataLen := len(dataList)
-
- for i := 0; i < dataLen; i++ {
- j := i + formulaInt
- if j < dataLen {
-
- currentItem := dataList[i]
- preItem := dataList[j]
-
- if currentItem == nil || preItem == nil {
- continue
- }
-
- val, ok := hbzDiv(currentItem.Value, preItem.Value)
-
- if !ok {
- continue
- }
- dateDataMap[currentItem.DataTime] = val
- }
- }
- return
- }
- func hbzDiv(current, pre float64) (val float64, ok bool) {
- if pre == 0 {
- return
- }
- currentVal := decimal.NewFromFloat(current)
- preVal := decimal.NewFromFloat(pre)
- val, _ = currentVal.Sub(preVal).Div(preVal).RoundCeil(4).Float64()
-
- ok = true
- return
- }
- func (obj BaseCalculate) Hcz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- nVal := fmt.Sprint(obj.Formula)
- formulaInt, err := strconv.Atoi(nVal)
- if err != nil {
- return
- }
-
-
- dataList = reverseSliceByDesc(dataList)
- var dateArr []time.Time
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- }
- dataLen := len(dataList)
- fmt.Println("dataLen:", dataLen)
- for i := 0; i < dataLen; i++ {
- j := i + formulaInt
- if j < dataLen {
-
- currentItem := dataList[i]
- preItem := dataList[j]
-
- if currentItem == nil || preItem == nil {
- continue
- }
-
- val := hczDiv(currentItem.Value, preItem.Value)
- dateDataMap[currentItem.DataTime] = val
- }
- }
- return
- }
- func hczDiv(current, pre float64) float64 {
- currentVal := decimal.NewFromFloat(current)
- preVal := decimal.NewFromFloat(pre)
- val, _ := currentVal.Sub(preVal).RoundCeil(4).Float64()
-
- return val
- }
- func (obj BaseCalculate) UpFrequency() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
-
- var dateArr []time.Time
- dataMap := make(map[time.Time]*EdbInfoData)
- fromDataMap := make(map[time.Time]float64)
-
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- fromDataMap[v.DataTime] = v.Value
- }
- if obj.FromFrequency == `日度` {
- errMsg = `不能选择日度指标`
- err = errors.New(errMsg)
- return
- }
-
- dataLen := len(dataList)
- for i := 0; i < dataLen; i++ {
-
- currentItem := dataList[i]
- currentDate := currentItem.DataTime
- var day int
- var preItem *EdbInfoData
- var preDate time.Time
- if i == 0 {
- day = int(time.Now().Sub(currentDate).Hours() / float64(24))
- preDate = time.Now()
- } else {
- j := i - 1
- if j < dataLen {
- preItem = dataList[j]
- day = int(preItem.DataTime.Sub(currentDate).Hours() / float64(24))
- utils.FileLog.Info("preItem.DataTime:" + preItem.DataTime.Format(utils.FormatDate) + ";currentItem.DataTime" + currentItem.DataTime.Format(utils.FormatDate))
- }
- }
- for k := 0; k <= day; k++ {
- needDay := preDate.AddDate(0, 0, -k)
- dateDataMap[needDay] = currentItem.Value
- }
- dateDataMap[currentItem.DataTime] = currentItem.Value
- }
- return
- }
- func (obj BaseCalculate) DownFrequency() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- edbFrequency := obj.Frequency
- formula := obj.Formula.(string)
- if !CheckFrequency(obj.FromFrequency, obj.Frequency) {
- errMsg = "频度异常,不允许低频降频到高频"
- err = errors.New(errMsg)
- return
- }
-
- var dateArr []time.Time
- dataMap := make(map[time.Time]*EdbInfoData)
- fromDataMap := make(map[time.Time]float64)
-
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- fromDataMap[v.DataTime] = v.Value
- }
-
- dataLen := len(dataList)
- if dataLen <= 0 {
- return
- }
- startDataTime := dataList[0].DataTime
- endDataTime := dataList[dataLen-1].DataTime
- var lastValue float64
- var nextEndDate time.Time
- weekDayDataList := make([]float64, 0)
- for tmpStartDataTime := startDataTime; !tmpStartDataTime.After(endDataTime); tmpStartDataTime = tmpStartDataTime.AddDate(0, 0, 1) {
-
- if tmpData, ok := dataMap[tmpStartDataTime]; ok {
- tmpValue := decimal.NewFromFloat(tmpData.Value)
- tmpValueFloat, _ := tmpValue.Round(4).Float64()
- weekDayDataList = append(weekDayDataList, tmpValueFloat)
- }
-
- if nextEndDate.IsZero() {
- nextEndDate = utils.GetFrequencyEndDay(tmpStartDataTime, edbFrequency)
- }
-
- switch edbFrequency {
- case "周度":
- if tmpStartDataTime.Weekday() != 5 {
-
- continue
- } else {
-
- nextEndDate = tmpStartDataTime.AddDate(0, 0, 7)
- }
- case "旬度":
- nextDay := tmpStartDataTime.AddDate(0, 0, 1)
- if nextDay.Day() != 1 && nextDay.Day() != 11 && nextDay.Day() != 21 {
-
- continue
- } else {
-
- if nextDay.Day() == 1 || nextDay.Day() == 11 {
-
- nextEndDate = nextDay.AddDate(0, 0, 9)
- } else {
- tmpNextMonth := nextDay.AddDate(0, 1, 0)
- nextEndDate = time.Date(tmpNextMonth.Year(), tmpNextMonth.Month(), 1, 0, 0, 0, 0, time.Local).AddDate(0, 0, -1)
- }
- }
- case "月度":
- nextDay := tmpStartDataTime.AddDate(0, 0, 1)
- if nextDay.Day() != 1 {
-
- continue
- } else {
-
- nextEndDate = nextDay.AddDate(0, 1, -1)
- }
- case "季度":
- nextDay := tmpStartDataTime.AddDate(0, 0, 1)
- if (nextDay.Month() == 1 || nextDay.Month() == 4 || nextDay.Month() == 7 || nextDay.Month() == 10) && nextDay.Day() == 1 {
-
- nextEndDate = nextDay.AddDate(0, 3, -1)
- } else {
-
- continue
- }
- case "年度":
- if tmpStartDataTime.Month() == 12 && tmpStartDataTime.Day() == 31 {
-
- nextEndDate = tmpStartDataTime.AddDate(1, 0, 0)
- } else {
-
- continue
- }
- default:
- err = errors.New("错误的频度:" + edbFrequency)
- return
- }
-
- var currVal float64
- lenWeekDayDataList := len(weekDayDataList)
-
- if len(weekDayDataList) <= 0 {
- currVal = lastValue
- } else {
- if formula == "期末值" {
- currVal = weekDayDataList[lenWeekDayDataList-1]
- } else {
-
- sumValDeci := decimal.NewFromFloat(0)
- for _, v := range weekDayDataList {
- tmpValDeci := decimal.NewFromFloat(v)
- sumValDeci = sumValDeci.Add(tmpValDeci)
- }
- lenDeci := decimal.NewFromInt(int64(lenWeekDayDataList))
- currVal, _ = sumValDeci.Div(lenDeci).Round(4).Float64()
- }
- }
-
- dateDataMap[tmpStartDataTime] = currVal
-
- weekDayDataList = make([]float64, 0)
- }
-
- if len(weekDayDataList) > 0 {
-
- var currVal float64
- lenWeekDayDataList := len(weekDayDataList)
-
- if len(weekDayDataList) < 0 {
- currVal = lastValue
- } else {
- if formula == "期末值" {
- currVal = weekDayDataList[lenWeekDayDataList-1]
- } else {
-
- sumValDeci := decimal.NewFromFloat(0)
- for _, v := range weekDayDataList {
- tmpValDeci := decimal.NewFromFloat(v)
- sumValDeci = sumValDeci.Add(tmpValDeci)
- }
- lenDeci := decimal.NewFromInt(int64(lenWeekDayDataList))
- currVal, _ = sumValDeci.Div(lenDeci).Round(4).Float64()
- }
- }
-
- dateDataMap[nextEndDate] = currVal
- }
- return
- }
- func (obj BaseCalculate) TimeShift() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- nVal := fmt.Sprint(obj.Formula)
- formulaInt, err := strconv.Atoi(nVal)
- if err != nil {
- return
- }
- moveType := obj.MoveType
- moveFrequency := obj.MoveFrequency
-
- var dateArr []time.Time
- dataMap := make(map[time.Time]*EdbInfoData)
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- }
- var shiftDay int
- switch moveFrequency {
- case "天":
- shiftDay = formulaInt
- case "周":
- shiftDay = formulaInt * 7
- case "月":
- shiftDay = formulaInt * 30
- case "季":
- shiftDay = formulaInt * 90
- case "年":
- shiftDay = formulaInt * 365
- default:
- shiftDay = formulaInt
- }
- if moveType == 2 {
- shiftDay = -shiftDay
- }
- dataLen := len(dataList)
- for i := 0; i < dataLen; i++ {
-
- currentItem := dataList[i]
- currentDate := currentItem.DataTime
- newDate := currentDate.AddDate(0, 0, shiftDay)
- val, _ := decimal.NewFromFloat(currentItem.Value).RoundCeil(4).Float64()
- dateDataMap[newDate] = val
- }
- return
- }
- func (obj BaseCalculate) Cjjx() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- nVal := fmt.Sprint(obj.Formula)
- formulaInt, err := strconv.Atoi(nVal)
- if err != nil {
- return
- }
- calendar := obj.Calendar
-
- var dateArr []time.Time
- dataMap := make(map[time.Time]*EdbInfoData)
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- }
-
- handleDataMap := make(map[time.Time]float64)
- _, err = HandleDataByLinearRegressionByTime(dataList, handleDataMap)
- if err != nil {
- return
- }
-
- moveDayMap := make(map[int]int, 0)
- var lastDataDay time.Time
- if len(dataList) > 0 {
- lastDataDay = dataList[0].DataTime
- }
-
- for _, currentDate := range dateArr {
-
- if strings.Contains(currentDate.Format(utils.FormatDate), "02-29") {
- continue
- }
-
- if calendar == "农历" && currentDate.Month() > 5 && currentDate.Month() < 11 {
- continue
- }
- currentItem, ok := dataMap[currentDate]
-
- if !ok {
- continue
- }
- pastValueList := make([]float64, 0)
- pastValueList = append(pastValueList, currentItem.Value)
- for i := 1; i < formulaInt; i++ {
-
- historyPreDate := currentDate.AddDate(-i, 0, 0)
- moveDay := 0
- if calendar == "农历" {
- if tmpMoveDay, ok := moveDayMap[historyPreDate.Year()]; !ok {
- moveDay, err = getMoveDay(lastDataDay, historyPreDate)
- if err != nil {
- return
- }
- } else {
- moveDay = tmpMoveDay
- }
-
- historyPreDate = historyPreDate.AddDate(0, 0, moveDay)
- }
- if tmpValue, ok := handleDataMap[historyPreDate]; ok {
- pastValueList = append(pastValueList, tmpValue)
- }
- }
- if len(pastValueList) == formulaInt {
- val, ok := cjjxSub(currentItem.Value, pastValueList)
- if !ok {
- continue
- }
- dateDataMap[currentDate] = val
- }
- }
- return
- }
- func cjjxSub(currValue float64, pastValue []float64) (value float64, ok bool) {
- num := len(pastValue)
- if num == 0 {
- return
- }
- numDecimal := decimal.NewFromInt(int64(num))
- af := decimal.NewFromFloat(currValue)
- bf := decimal.NewFromFloat(pastValue[0])
- for k := 1; k < num; k++ {
- tmpVal := decimal.NewFromFloat(pastValue[k])
- bf = bf.Add(tmpVal)
- }
- value, _ = af.Sub(bf.Div(numDecimal)).RoundCeil(4).Float64()
- ok = true
- return
- }
- func (obj BaseCalculate) Annualized() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- fromDataList := obj.DataList
- lenFromDataList := len(fromDataList)
-
- if lenFromDataList <= 0 {
- return
- }
-
-
- handleDataMap := make(map[time.Time]float64)
- _, err = HandleDataByLinearRegressionByTime(fromDataList, handleDataMap)
- if err != nil {
- return
- }
-
- yearLastValMap := make(map[int]float64)
- startDataTime := fromDataList[0].DataTime
- endDataTime := fromDataList[lenFromDataList-1].DataTime
- for i := startDataTime.Year(); i <= endDataTime.Year(); i++ {
- tmpDate := time.Date(i, 12, 31, startDataTime.Hour(), startDataTime.Minute(), startDataTime.Second(), startDataTime.Nanosecond(), time.Local)
- if tmpVal, ok := handleDataMap[tmpDate]; ok {
- yearLastValMap[i] = tmpVal
- }
- }
-
-
- for _, v := range fromDataList {
- currDate := v.DataTime
- perValMap := make(map[time.Time]float64)
-
- for i := 1; i <= 3; i++ {
- tmpDateTime := currDate.AddDate(-i, 0, 0)
- if tmpVal, ok := handleDataMap[tmpDateTime]; ok {
- perValMap[tmpDateTime] = tmpVal
- }
- }
- lenPerValMap := len(perValMap)
-
- if lenPerValMap < 2 {
- continue
- }
-
- divValList := make([]decimal.Decimal, 0)
- for tmpDateTime, tmpVal := range perValMap {
- yearLastVal, ok2 := yearLastValMap[tmpDateTime.Year()]
-
- if !ok2 {
- continue
- }
-
- tmpYearLastVal := decimal.NewFromFloat(yearLastVal)
- if tmpYearLastVal.IsZero() {
- continue
- }
- divVal := decimal.NewFromFloat(tmpVal).Div(tmpYearLastVal)
- divValList = append(divValList, divVal)
- }
- lenDivValList := len(divValList)
-
- if lenDivValList < 2 {
- continue
- }
- divValSum := decimal.NewFromFloat(0)
- for _, divVal := range divValList {
- divValSum = divValSum.Add(divVal)
- }
-
- tmpDivVal := divValSum.Div(decimal.NewFromInt(int64(lenDivValList)))
- if tmpDivVal.IsZero() {
- continue
- }
- currVal, _ := decimal.NewFromFloat(v.Value).Div(tmpDivVal).Round(4).Float64()
- dateDataMap[currDate] = currVal
- }
- return
- }
- func (obj BaseCalculate) Ljz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- frequency := obj.Frequency
- fromFrequency := obj.FromFrequency
-
- var isWeekData bool
-
- if fromFrequency == `周度` {
- isWeekData = true
- }
- fromEdbDataMap := make(map[time.Time]float64)
- if isWeekData {
- dataList, err = HandleDataByLinearRegressionByTime(dataList, fromEdbDataMap)
- if err != nil {
- return
- }
- }
-
-
-
-
-
- dateList := make([]time.Time, 0)
- valueMap := make(map[time.Time]float64)
- switch frequency {
- case "年度":
- yearMap := make(map[int]float64)
- yearList := make([]int, 0)
- for _, item := range dataList {
- year := item.DataTime.Year()
- yearVal, ok := yearMap[year]
- if ok {
- yearMap[year] = item.Value + yearVal
- } else {
- yearList = append(yearList, year)
- yearMap[year] = item.Value
- }
- }
- for _, v := range yearList {
- currTime := time.Date(v, 12, 31, 0, 0, 0, 0, time.Local)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMap[v]
- }
- case "半年度":
- yearMonthMap := make(map[string]float64)
- yearMonthList := make([]string, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- year := itemDate.Year()
- var tmpK string
- if itemDate.Month() <= 6 {
- tmpK = fmt.Sprint(year, "06")
- } else {
- tmpK = fmt.Sprint(year, "12")
- }
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- }
- case "季度":
- yearMonthMap := make(map[string]float64)
- yearMonthList := make([]string, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- year := itemDate.Year()
- var tmpK string
- if itemDate.Month() <= 3 {
- tmpK = fmt.Sprint(year, "03")
- } else if itemDate.Month() <= 6 {
- tmpK = fmt.Sprint(year, "06")
- } else if itemDate.Month() <= 9 {
- tmpK = fmt.Sprint(year, "09")
- } else {
- tmpK = fmt.Sprint(year, "12")
- }
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- }
- case "月度":
- yearMonthMap := make(map[string]float64)
- yearMonthList := make([]string, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- year := itemDate.Year()
- var tmpK string
- tmpK = fmt.Sprint(year*100 + int(itemDate.Month()))
- yearVal, ok := yearMonthMap[tmpK]
- if ok {
- yearMonthMap[tmpK] = item.Value + yearVal
- } else {
- yearMonthList = append(yearMonthList, tmpK)
- yearMonthMap[tmpK] = item.Value
- }
- }
- for _, v := range yearMonthList {
- currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- dateList = append(dateList, currTime)
- valueMap[currTime] = yearMonthMap[v]
- }
- case "旬度":
- tmpDateDataMap := make(map[time.Time]float64)
- tmpDateList := make([]time.Time, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- dayInt := itemDate.Year()*100 + int(itemDate.Month())
- var currTime time.Time
- if itemDate.Day() <= 10 {
-
- tmpK := fmt.Sprint(dayInt, "10")
- currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
- if err != nil {
- return
- }
- } else if itemDate.Day() <= 20 {
-
- tmpK := fmt.Sprint(dayInt, "20")
- currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
- if err != nil {
- return
- }
- } else {
-
- currTime, err = time.ParseInLocation(utils.FormatYearMonthUnSpace, fmt.Sprint(dayInt), time.Local)
- if err != nil {
- return
- }
- currTime = currTime.AddDate(0, 1, -1)
- }
- yearVal, ok := tmpDateDataMap[currTime]
- if ok {
- tmpDateDataMap[currTime] = item.Value + yearVal
- } else {
- tmpDateList = append(tmpDateList, currTime)
- tmpDateDataMap[currTime] = item.Value
- }
- }
- for _, currTime := range tmpDateList {
- dateList = append(dateList, currTime)
- valueMap[currTime] = tmpDateDataMap[currTime]
- }
- case "周度":
- tmpDateDataMap := make(map[time.Time]float64)
- tmpDateList := make([]time.Time, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- var currTime time.Time
-
- if itemDate.Weekday() == 0 {
- currTime = itemDate.AddDate(0, 0, 5)
- } else if itemDate.Weekday() == 6 {
- currTime = itemDate.AddDate(0, 0, 6)
- } else {
- currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
- }
- yearVal, ok := tmpDateDataMap[currTime]
- if ok {
- tmpDateDataMap[currTime] = item.Value + yearVal
- } else {
- tmpDateList = append(tmpDateList, currTime)
- tmpDateDataMap[currTime] = item.Value
- }
- }
- for _, currTime := range tmpDateList {
- dateList = append(dateList, currTime)
- valueMap[currTime] = tmpDateDataMap[currTime]
- }
- default:
- err = errors.New("错误的频度")
- return
- }
-
- for _, currTime := range dateList {
- tmpVal, ok2 := valueMap[currTime]
- if !ok2 {
- err = errors.New("数据异常,date:" + currTime.Format(utils.FormatDate))
- return
- }
- var saveValue float64
- if isWeekData {
- saveValue, _ = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).Float64()
- } else {
- saveValue, _ = decimal.NewFromFloat(tmpVal).Round(4).Float64()
- }
- dateDataMap[currTime] = saveValue
- }
- return
- }
- func (obj BaseCalculate) LjzNczj() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- frequency := obj.Frequency
- fromFrequency := obj.FromFrequency
-
- var isWeekData bool
-
- if fromFrequency == `周度` {
- isWeekData = true
- }
- fromEdbDataMap := make(map[time.Time]float64)
- if isWeekData {
- dataList, err = HandleDataByLinearRegressionByTime(dataList, fromEdbDataMap)
- if err != nil {
- return
- }
- }
-
-
-
-
- dateList := make([]time.Time, 0)
- valueMap := make(map[time.Time]float64)
- yearMap := make(map[int]float64)
- switch frequency {
- case "周度":
- tmpDateDataMap := make(map[time.Time]float64)
- tmpDateList := make([]time.Time, 0)
- for _, item := range dataList {
- itemDate := item.DataTime
- var currTime time.Time
-
- if itemDate.Weekday() == 0 {
- currTime = itemDate.AddDate(0, 0, 5)
- } else if itemDate.Weekday() == 6 {
- currTime = itemDate.AddDate(0, 0, 6)
- } else {
- currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
- }
- year := itemDate.Year()
- yearVal, ok := yearMap[year]
- if ok {
- yearMap[year] = item.Value + yearVal
- } else {
- yearMap[year] = item.Value
- }
- if itemDate.Equal(currTime) {
- tmpDateDataMap[itemDate] = yearMap[year]
- tmpDateList = append(tmpDateList, itemDate)
- }
- }
- for _, currTime := range tmpDateList {
- dateList = append(dateList, currTime)
- valueMap[currTime] = tmpDateDataMap[currTime]
- }
- default:
- for _, item := range dataList {
- itemDate := item.DataTime
- year := itemDate.Year()
- yearVal, ok := yearMap[year]
- if ok {
- yearMap[year] = item.Value + yearVal
- } else {
- yearMap[year] = item.Value
- }
- valueMap[itemDate] = yearMap[year]
- dateList = append(dateList, itemDate)
- }
- }
-
- for _, currTime := range dateList {
- tmpVal, ok2 := valueMap[currTime]
- if !ok2 {
- err = errors.New("数据异常,date:" + currTime.Format(utils.FormatDate))
- return
- }
- var saveValue float64
- if isWeekData {
- saveValue, _ = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).Float64()
- } else {
- saveValue, _ = decimal.NewFromFloat(tmpVal).Round(4).Float64()
- }
- dateDataMap[currTime] = saveValue
- }
- return
- }
- func (obj BaseCalculate) ExponentialSmoothing() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- formula := obj.Formula.(string)
- alpha, _ := strconv.ParseFloat(formula, 64)
- if alpha <= 0 || alpha >= 1 {
- err = fmt.Errorf("alpha值有误: %v", alpha)
- return
- }
-
- fromDataList, err := calculateExponentialSmoothingData(dataList, alpha)
- if err != nil {
- return
- }
-
- for _, tmpData := range fromDataList {
- if tmpData.DataTime.IsZero() {
- continue
- }
-
- saveValue, _ := decimal.NewFromFloat(tmpData.Value).Round(4).Float64()
- dateDataMap[tmpData.DataTime] = saveValue
- }
- return
- }
- func calculateExponentialSmoothingData(dataList []*EdbInfoData, alpha float64) (newDataList []EdbInfoData, err error) {
- if alpha <= 0 || alpha >= 1 {
- err = fmt.Errorf("alpha值有误: %v", alpha)
- return
- }
- var preVal float64
- alphaDecimal := decimal.NewFromFloat(alpha)
- subAlpha := decimal.NewFromFloat(1).Sub(alphaDecimal)
- for k, d := range dataList {
-
- if k == 0 {
- newDataList = append(newDataList, EdbInfoData{
- EdbDataId: k,
- DataTime: dataList[k].DataTime,
- Value: d.Value,
- })
- preVal = d.Value
- continue
- }
-
- preDecimal := decimal.NewFromFloat(preVal)
- valDecimal := decimal.NewFromFloat(d.Value)
- partA := alphaDecimal.Mul(valDecimal)
- partB := subAlpha.Mul(preDecimal)
- res, _ := (partA.Add(partB)).Float64()
- preVal = res
- newDataList = append(newDataList, EdbInfoData{
- EdbDataId: k,
- DataTime: dataList[k].DataTime,
- Value: res,
- })
- }
- return
- }
- func (obj BaseCalculate) Rjz() (dateDataMap map[time.Time]float64, err error, errMsg string) {
- dateDataMap = make(map[time.Time]float64)
- dataList := obj.DataList
- fromFrequency := obj.FromFrequency
- if fromFrequency == `` {
- errMsg = "原指标的频度不能为空"
- err = errors.New(errMsg)
- return
- }
- if obj.FromFrequency == "日度" {
- errMsg = "日度指标无需进行日均值计算"
- err = errors.New(errMsg)
- return
- }
-
-
- dataList = reverseSliceByDesc(dataList)
- var dateArr []time.Time
- dataMap := make(map[time.Time]*EdbInfoData)
- for _, v := range dataList {
- dateArr = append(dateArr, v.DataTime)
- dataMap[v.DataTime] = v
- }
-
- for _, currentDate := range dateArr {
-
- currentItem, ok := dataMap[currentDate]
- if !ok {
- continue
- }
-
- days := GetRjzFrequencyDays(currentDate, fromFrequency)
- val, ok := rjzDivV2(currentItem.Value, days)
- if !ok {
- continue
- }
- dateDataMap[currentDate] = val
- }
- return
- }
- func rjzDivV2(a float64, b int) (val float64, ok bool) {
- if b == 0 {
- return
- }
- af := decimal.NewFromFloat(a)
- bf := decimal.NewFromFloat(float64(b))
- val, _ = af.Div(bf).Round(4).Float64()
- ok = true
- return
- }
- func reverseSliceByDesc(slice []*EdbInfoData) (newSlice []*EdbInfoData) {
-
- if len(slice) <= 1 {
- newSlice = slice
- return
- }
- if !slice[1].DataTime.After(slice[0].DataTime) {
- newSlice = slice
- return
- }
- newSlice = make([]*EdbInfoData, len(slice))
- for i, j := 0, len(slice)-1; i < len(slice); i, j = i+1, j-1 {
- newSlice[i] = slice[j]
- }
- return
- }
- func EdbInfoSearchDataToData(dataList []*EdbInfoSearchData) (newDateDataList []*EdbInfoData, err error) {
- newDateDataList = make([]*EdbInfoData, 0)
- for _, data := range dataList {
- dateTime, tmpErr := time.ParseInLocation(utils.FormatDate, data.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- newDateDataList = append(newDateDataList, &EdbInfoData{
- EdbDataId: data.EdbDataId,
- DataTime: dateTime,
- Value: data.Value,
- })
- }
- return
- }
- func GetDateDataAndDateList(dateDataMap map[time.Time]float64) (dateStrDataMap map[string]float64, dateStrList []string) {
- dateStrDataMap = make(map[string]float64)
- dateStrList = make([]string, 0)
- dateList := make([]time.Time, 0)
- for date, val := range dateDataMap {
- dateStr := date.Format(utils.FormatDate)
- dateStrDataMap[dateStr] = val
- dateList = append(dateList, date)
- }
-
- sort.Slice(dateList, func(i, j int) bool {
- return dateList[i].Before(dateList[j])
- })
-
- for _, v := range dateList {
- dateStrList = append(dateStrList, v.Format(utils.FormatDate))
- }
- return
- }
|