1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117 |
- package excel
- import (
- "errors"
- "eta/eta_chart_lib/models"
- "eta/eta_chart_lib/models/data_manage"
- "eta/eta_chart_lib/models/request"
- "eta/eta_chart_lib/services/data"
- "eta/eta_chart_lib/utils"
- "fmt"
- "sort"
- "time"
- )
- func GetFirstEdbDataList(edbInfo *data_manage.EdbInfo, num int, manualDateList []string) (resultDataList []request.ManualDataReq, err error) {
- var dataList []*models.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = models.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- return
- }
-
- lenData := len(dataList)
- if lenData <= 0 {
- return
- }
- tmpManualDateNum := 0
- lenManualDate := len(manualDateList)
- if lenManualDate > 0 {
- sortDateList := manualDateList
- baseDateList := utils.StrArr{}
- baseDateList = append(baseDateList, sortDateList...)
- sort.Sort(baseDateList)
- sortDateList = append([]string{}, baseDateList...)
- lastData := dataList[lenData-1]
- lastDataDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
-
- for _, tmpDateStr := range sortDateList {
- tmpDate, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDateStr, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if tmpDate.After(lastDataDate) {
- tmpManualDateNum++
- continue
- }
- break
- }
- }
-
- num = num - tmpManualDateNum
- if num > lenData {
- num = lenData
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- for i := 1; i <= num; i++ {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
- dataType := 1
-
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: dataList[lenData-i].DataTime,
- ShowValue: fmt.Sprint(dataList[lenData-i].Value),
- Value: fmt.Sprint(dataList[lenData-i].Value),
- DataTimeType: 1,
- })
- }
- return
- }
- func GetOtherEdbDataList(edbInfo *data_manage.EdbInfo, dateList []string) (resultDataList []request.ManualDataReq, err error) {
- lenDate := len(dateList)
- if lenDate <= 0 {
- return
- }
- sortDateList := dateList
- baseDateList := utils.StrArr{}
- baseDateList = append(baseDateList, sortDateList...)
- sort.Sort(baseDateList)
- sortDateList = append([]string{}, baseDateList...)
- endDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[0], time.Local)
- if err != nil {
- return
- }
- firstDateTime, err := time.ParseInLocation(utils.FormatDate, sortDateList[lenDate-1], time.Local)
- if err != nil {
- return
- }
- var dataList []*models.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = models.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, ``)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, ``, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- return
- }
-
- baseDataList := make([]*models.EdbDataList, 0)
- var lastData *models.EdbDataList
- var isInsert bool
- for _, data := range dataList {
- tmpDate := data.DataTime
- tmpDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if tmpDateTime.Before(firstDateTime) {
- lastData = data
- continue
- }
-
- if !isInsert && lastData != nil {
- baseDataList = append(baseDataList, lastData)
- }
- if tmpDateTime.After(endDateTime) {
- baseDataList = append(baseDataList, data)
- break
- }
- baseDataList = append(baseDataList, data)
- isInsert = true
- }
-
- realValMap := make(map[string]string)
- for _, v := range baseDataList {
- realValMap[v.DataTime] = v.DataTime
- }
-
- handleDataMap := make(map[string]float64)
- err = data.HandleDataByLinearRegression(baseDataList, handleDataMap)
- if err != nil {
- return
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
-
- for _, date := range sortDateList {
- dataType := 1
- if _, ok := realValMap[date]; !ok {
- dataType = 2
- } else {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, date, time.Local)
-
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- }
- var value, showValue string
- if tmpVal, ok := handleDataMap[date]; ok {
- value = fmt.Sprint(tmpVal)
- showValue = value
- } else {
- dataType = 3
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: date,
- ShowValue: showValue,
- Value: value,
- })
- }
- return
- }
- func GetFirstHistoryEdbDataList(edbInfo *data_manage.EdbInfo, num int, endDate string) (resultDataList []request.ManualDataReq, err error) {
- endDateTime, err := time.ParseInLocation(utils.FormatDate, endDate, time.Local)
- if err != nil {
- return
- }
- var dataList []*models.EdbDataList
- switch edbInfo.EdbInfoType {
- case 0:
- dataList, err = models.GetEdbDataList(edbInfo.Source, edbInfo.SubSource, edbInfo.EdbInfoId, ``, endDate)
- case 1:
- _, dataList, _, _, err, _ = data.GetPredictDataListByPredictEdbInfoId(edbInfo.EdbInfoId, ``, endDate, true)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", edbInfo.EdbInfoType))
- return
- }
-
- lenData := len(dataList)
- if lenData <= 0 {
- return
- }
- lastData := dataList[lenData-1]
- lastDataDateTime, err := time.ParseInLocation(utils.FormatDate, lastData.DataTime, time.Local)
- if err != nil {
- return
- }
- if endDateTime.Equal(lastDataDateTime) || lastDataDateTime.After(endDateTime) {
- dataList = dataList[:lenData-1]
- lenData = len(dataList)
- }
- if num > lenData {
- num = lenData
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, edbInfo.LatestDate, time.Local)
- for i := 1; i <= num; i++ {
- dataTime, _ := time.ParseInLocation(utils.FormatDate, dataList[lenData-i].DataTime, time.Local)
- dataType := 1
-
- if edbInfo.EdbInfoType == 1 && dataTime.After(latestDateTime) {
- dataType = 5
- }
- resultDataList = append(resultDataList, request.ManualDataReq{
- DataType: dataType,
- DataTime: dataList[lenData-i].DataTime,
- ShowValue: fmt.Sprint(dataList[lenData-i].Value),
- Value: fmt.Sprint(dataList[lenData-i].Value),
- })
- }
- return
- }
|