123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905 |
- package logic
- import (
- "encoding/json"
- "errors"
- "eta/eta_index_lib/models"
- "eta/eta_index_lib/models/future_good"
- "eta/eta_index_lib/utils"
- "fmt"
- "github.com/dengsgo/math-engine/engine"
- "github.com/shopspring/decimal"
- "sort"
- "strconv"
- "strings"
- "time"
- )
- type ChartInfoDateReq struct {
- Type int `description:"配置类型"`
- Date string `description:"固定日期"`
- Value int `description:"N天的值"`
- Color string `description:"颜色"`
- Name string `description:"别名"`
- }
- type SectionScatterReq struct {
- XName string `description:"x轴名称"`
- XNameEn string `description:"x轴名称(英文)"`
- XUnitName string `description:"x轴单位名称"`
- XUnitNameEn string `description:"x轴单位名称(英文)"`
- YName string `description:"y轴名称"`
- YNameEn string `description:"y轴名称(英文)"`
- YUnitName string `description:"y轴单位名称"`
- YUnitNameEn string `description:"y轴单位名称(英文)"`
- XMinValue string `description:"X轴的最小值"`
- XMaxValue string `description:"X轴的最大值"`
- YMinValue string `description:"Y轴的最小值"`
- YMaxValue string `description:"Y轴的最大值"`
-
- SeriesList []SectionScatterSeriesItemReq `description:"系列数据"`
- }
- type SectionScatterSeriesItemReq struct {
- Name string `description:"系列名"`
- NameEn string `description:"系列名(英文名)"`
- Color string `description:"颜色"`
- EdbInfoList []SectionScatterEdbItemReq
- ShowTrendLine bool `description:"是否展示趋势线"`
- ShowFitEquation bool `description:"是否展示方程式"`
- ShowRSquare bool `description:"是否展示R平方"`
- }
- type SectionScatterEdbItemReq struct {
- XEdbInfoId int `description:"X轴的指标ID"`
- YEdbInfoId int `description:"Y轴的指标ID"`
- Name string `description:"别名"`
- NameEn string `description:"英文别名"`
- XDateType int `description:"X轴的日期配置类型"`
- XDate string `description:"X轴的日期固定日期"`
- XDateValue int `description:"X轴的日期N天的值"`
- YDateType int `description:"Y轴的日期配置类型"`
- YDate string `description:"Y轴的日期固定日期"`
- YDateValue int `description:"Y轴的日期N天的值"`
- IsShow bool `description:"是否展示"`
- }
- type XData struct {
- Name string `description:"别名"`
- NameEn string `description:"英文别名"`
- }
- type YData struct {
- Date string `description:"数据日期"`
- ConfigDate time.Time `description:"配置的日期" json:"-"`
- Color string `description:"数据颜色"`
- Name string `description:"别名"`
- NameEn string `description:"英文别名"`
- Value []float64 `description:"每个指标的值"`
- NoDataEdbList []int `description:"没有数据的指标列表"`
- XEdbInfoIdList []int `description:"对应X轴的指标id列表"`
- NameList []string `description:"每个值对应的名称"`
- EnNameList []string `description:"每个值对应的英文名称"`
- EdbValMap map[int]float64 `description:"指标与值的对应" json:"-"`
- M []int `description:"对应开始日期的间隔值" json:"-"`
- }
- type BarChartInfoEdbItemReq struct {
- EdbInfoId int `description:"指标ID"`
- Name string `description:"别名"`
- NameEn string `description:"英文别名"`
- Source int `description:"1:ETA图库;2:商品价格"`
- }
- type ChartInfoReq struct {
- FutureGoodEdbInfoIdList []models.EdbInfoFromTag `description:"指标信息"`
- CalculateFormula string `description:"计算公式"`
- BaseEdbInfoId int `description:"基础的指标id"`
- DateList []ChartInfoDateReq `description:"日期配置"`
- ProfitNameEn string `description:"利润英文名称"`
- }
- func RefreshByChartId(chartInfoId int) (err error, errMsg string) {
-
- chartInfo, err := models.GetChartInfoById(chartInfoId)
- if err != nil {
- if err.Error() == utils.ErrNoRow() {
- err = nil
- }
- return
- }
- if chartInfo.Source != utils.CHART_SOURCE_FUTURE_GOOD_PROFIT {
- return
- }
- var extraConf ChartInfoReq
- err = json.Unmarshal([]byte(chartInfo.ExtraConfig), &extraConf)
- if err != nil {
- errMsg = "商品利润曲线图配置异常"
- return
- }
-
- chartInfoFutureGoodProfit := new(future_good.ChartInfoFutureGoodProfit)
- if err = chartInfoFutureGoodProfit.GetItemById(chartInfoId); err != nil {
- errMsg = "获取失败"
- if err.Error() == utils.ErrNoRow() {
- err = nil
- }
- return
- }
- baseEdbInfo, err := models.GetEdbInfoById(extraConf.BaseEdbInfoId)
- if err != nil {
- errMsg = "获取失败"
- return
- }
-
- futureGoodEdbInfoMap := make(map[int]*future_good.FutureGoodEdbInfo)
- zlFutureGoodEdbInfoList := make([]*future_good.FutureGoodEdbInfo, 0)
- for _, v := range extraConf.FutureGoodEdbInfoIdList {
- if _, ok := futureGoodEdbInfoMap[v.EdbInfoId]; ok {
- continue
- }
- zlFutureGoodEdbInfo, tmpErr := future_good.GetFutureGoodEdbInfo(v.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- errMsg = "获取失败"
- return
- }
- futureGoodEdbInfoMap[v.EdbInfoId] = zlFutureGoodEdbInfo
- zlFutureGoodEdbInfoList = append(zlFutureGoodEdbInfoList, zlFutureGoodEdbInfo)
- }
- xDataList, yDataList, err := GetProfitChartEdbData(baseEdbInfo, zlFutureGoodEdbInfoList, extraConf.DateList, extraConf.CalculateFormula, extraConf.FutureGoodEdbInfoIdList)
- xDataListByte, err := json.Marshal(xDataList)
- if err != nil {
- errMsg = "保存失败"
- err = errors.New("X轴数据转换失败,ERR:" + err.Error())
- return
- }
- yDataListByte, err := json.Marshal(yDataList)
- if err != nil {
- errMsg = "保存失败"
- err = errors.New("Y轴数据转换失败,ERR:" + err.Error())
- return
- }
- extraUpdateCol := make([]string, 0)
- chartInfoFutureGoodProfit.XValue = string(xDataListByte)
- chartInfoFutureGoodProfit.YValue = string(yDataListByte)
- chartInfoFutureGoodProfit.ProfitName = zlFutureGoodEdbInfoList[0].FutureGoodEdbName + "盘面利润"
- chartInfoFutureGoodProfit.ModifyTime = time.Now()
- extraUpdateCol = []string{"XValue", "YValue", "ProfitName", "ModifyTime"}
- err = chartInfoFutureGoodProfit.Update(extraUpdateCol)
- return
- }
- func GetProfitChartEdbData(baseEdbInfo *models.EdbInfo, zlFutureGoodEdbInfoList []*future_good.FutureGoodEdbInfo, chartInfoDateList []ChartInfoDateReq, formulaStr string, edbInfoFromTagList []models.EdbInfoFromTag) (xDataList []XData, yDataList []YData, err error) {
- if baseEdbInfo == nil {
- err = errors.New("ETA指标未选取")
- return
- }
- if len(zlFutureGoodEdbInfoList) <= 0 {
- err = errors.New("商品指标未选取")
- return
- }
-
- tagEdbIdMap := make(map[string]int)
-
- futureGoodEdbInfoIdMap := make(map[int]int)
- {
- tmpTagEdbIdMap := make(map[string]int)
- for _, v := range edbInfoFromTagList {
- tmpTagEdbIdMap[v.FromTag] = v.EdbInfoId
- }
- formulaMap := CheckFormula(formulaStr)
- for _, tag := range formulaMap {
- tagEdbIdMap[tag] = tmpTagEdbIdMap[tag]
- futureGoodEdbInfoIdMap[tmpTagEdbIdMap[tag]] = tmpTagEdbIdMap[tag]
- }
- }
-
- baseDataList := make([]*models.EdbDataList, 0)
- baseDataList, err = models.GetEdbDataList(baseEdbInfo.Source, baseEdbInfo.SubSource, baseEdbInfo.EdbInfoId, "", "")
- if err != nil {
- return
- }
- latestDate := zlFutureGoodEdbInfoList[0].EndDate
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, latestDate, time.Local)
- earliestDateTime := latestDateTime
- for _, barChartInfoDate := range chartInfoDateList {
- var findDateTime time.Time
- switch barChartInfoDate.Type {
- case 1:
- findDateTime = latestDateTime
- case 2:
- findDateTime = latestDateTime.AddDate(0, 0, -barChartInfoDate.Value)
- case 3:
-
- tmpFindDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- findDateTime = tmpFindDateTime
- default:
- err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
- return
- }
- if findDateTime.IsZero() {
- err = errors.New("错误的日期")
- return
- }
- if findDateTime.Before(earliestDateTime) {
- earliestDateTime = findDateTime
- }
- }
- monthNum := (latestDateTime.Year()-earliestDateTime.Year())*12 + int(latestDateTime.Month()-earliestDateTime.Month())
-
- futureGoodEdbInfoDateMap := make(map[int]map[string]*future_good.FutureGoodEdbInfo)
- futureGoodDataListMap := make(map[int][]*models.EdbDataList, 0)
-
- specialFutureGoodEdbInfoMap := make(map[int]map[int]*future_good.FutureGoodEdbInfo, 0)
- isAllChina := true
- for _, v := range zlFutureGoodEdbInfoList {
- if v.RegionType != "国内" {
- isAllChina = false
- break
- }
- }
- nMap := make(map[string]string)
- var maxN int
- for _, v := range zlFutureGoodEdbInfoList {
-
- if _, ok := futureGoodEdbInfoIdMap[v.FutureGoodEdbInfoId]; !ok {
- continue
- }
-
- tmpFutureGoodEdbInfoList, tmpErr := future_good.GetChildFutureGoodEdbInfoListByParentId(v.FutureGoodEdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- childFutureGoodEdbInfoMap, tmpMaxN, tmpErr := getProfitFutureGoodEdbInfoList(earliestDateTime, v, tmpFutureGoodEdbInfoList, isAllChina, monthNum)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- if maxN < tmpMaxN {
- maxN = tmpMaxN
- }
- futureGoodEdbInfoDateMap[v.FutureGoodEdbInfoId] = childFutureGoodEdbInfoMap
- if v.FutureGoodEdbType == 2 {
- specialFutureGoodEdbInfoMap[v.FutureGoodEdbInfoId] = make(map[int]*future_good.FutureGoodEdbInfo)
- }
-
- for date, childFutureGoodEdbInfo := range childFutureGoodEdbInfoMap {
- nMap[date] = date
- dataList := make([]*models.EdbDataList, 0)
- tmpDataList, tmpErr := future_good.GetFutureGoodEdbDataListByDate(childFutureGoodEdbInfo.FutureGoodEdbInfoId, "", "")
- if tmpErr != nil {
- return
- }
- for _, tmpData := range tmpDataList {
- dataList = append(dataList, &models.EdbDataList{
- EdbDataId: tmpData.FutureGoodEdbDataId,
- EdbInfoId: tmpData.FutureGoodEdbInfoId,
- DataTime: tmpData.DataTime.Format(utils.FormatDate),
- DataTimestamp: tmpData.DataTimestamp,
- Value: tmpData.Close,
- })
- }
- futureGoodDataListMap[childFutureGoodEdbInfo.FutureGoodEdbInfoId] = dataList
- if childFutureGoodEdbInfo.FutureGoodEdbType == 2 {
- specialFutureGoodEdbInfoMap[v.FutureGoodEdbInfoId][childFutureGoodEdbInfo.Month] = childFutureGoodEdbInfo
- }
- }
- }
-
- dateList := make([]string, 0)
- for _, n := range nMap {
- dateList = append(dateList, n)
- }
- sort.Slice(dateList, func(i, j int) bool {
- return dateList[i] < dateList[j]
- })
- _, yDataList, err = ProfitChartChartData(baseDataList, futureGoodEdbInfoDateMap, futureGoodDataListMap, chartInfoDateList, baseEdbInfo.EndDate, specialFutureGoodEdbInfoMap, formulaStr, tagEdbIdMap, dateList, maxN)
- if err != nil {
- return
- }
- tmpXDataList, newYDataList, err := handleProfitResultData(baseEdbInfo, yDataList, earliestDateTime)
- if err != nil {
- return
- }
- xDataList = []XData{
- {
- Name: "现货利润",
- NameEn: "Spot Price",
- },
- }
- xDataList = append(xDataList, tmpXDataList...)
- yDataList = newYDataList
- return
- }
- func ProfitChartChartData(baseDataList []*models.EdbDataList, futureGoodEdbInfoMap map[int]map[string]*future_good.FutureGoodEdbInfo, futureGoodEdbDataListMap map[int][]*models.EdbDataList, chartInfoDateList []ChartInfoDateReq, latestDate string, specialFutureGoodEdbInfoMap map[int]map[int]*future_good.FutureGoodEdbInfo, formulaStr string, tagEdbIdMap map[string]int, dateList []string, maxN int) (edbIdList []int, yDataList []YData, err error) {
-
-
-
- baseEdbDateData := make(map[string]float64)
- for _, edbData := range baseDataList {
- baseEdbDateData[edbData.DataTime] = edbData.Value
- }
-
- edbDataMap := make(map[int]map[string]float64)
- for edbInfoId, edbDataList := range futureGoodEdbDataListMap {
- edbDateData := make(map[string]float64)
- for _, edbData := range edbDataList {
- edbDateData[edbData.DataTime] = edbData.Value
- }
- edbDataMap[edbInfoId] = edbDateData
- }
- latestDateTime, _ := time.ParseInLocation(utils.FormatDate, latestDate, time.Local)
- yDataList = make([]YData, 0)
-
- formulaStr = strings.ToUpper(formulaStr)
- for _, barChartInfoDate := range chartInfoDateList {
- yDataMap := make(map[int]float64)
- var maxDate time.Time
- var findDateTime time.Time
- switch barChartInfoDate.Type {
- case 1:
- findDateTime = latestDateTime
- case 2:
- findDateTime = latestDateTime.AddDate(0, 0, -barChartInfoDate.Value)
- case 3:
-
- tmpFindDateTime, tmpErr := time.ParseInLocation(utils.FormatDate, barChartInfoDate.Date, time.Local)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- findDateTime = tmpFindDateTime
- default:
- err = errors.New(fmt.Sprint("日期类型异常,Type:", barChartInfoDate.Type))
- return
- }
- if findDateTime.IsZero() {
- err = errors.New("错误的日期")
- return
- }
- findDataList := make([]float64, 0)
- noDataIdList := make([]int, 0)
- xEdbInfoIdList := make([]int, 0)
-
- realDateTime, findDataValue, isFind, tmpErr := GetNeedDateData(findDateTime, baseDataList, baseEdbDateData, edbDataMap)
- if tmpErr != nil {
- err = tmpErr
- return
- }
- findDataList = append(findDataList, findDataValue)
- yDataMap[0] = findDataValue
- if isFind {
- maxDate = realDateTime
- }
- xEdbInfoIdList = append(xEdbInfoIdList, 0)
- mList := make([]int, 0)
-
-
- for _, date := range dateList {
- currDate, _ := time.ParseInLocation(utils.FormatYearMonthDate, date, time.Local)
-
-
-
-
-
-
-
-
- tmpN := (currDate.Year()-realDateTime.Year())*12 + int(currDate.Month()-realDateTime.Month())
- if tmpN <= 0 {
- continue
- }
-
- if tmpN >= maxN {
- break
- }
- zlAndChildEdbId := make(map[int]int)
- childFutureGoodEdbInfoIdList := make([]int, 0)
- for zlFutureGoodEdbInfoId, futureGoodEdbInfoList := range futureGoodEdbInfoMap {
-
- if childFutureGoodEdbInfoIdMap, ok := specialFutureGoodEdbInfoMap[zlFutureGoodEdbInfoId]; ok {
- if childFutureGoodEdbInfo, ok2 := childFutureGoodEdbInfoIdMap[tmpN]; ok2 {
- childFutureGoodEdbInfoIdList = append(childFutureGoodEdbInfoIdList, childFutureGoodEdbInfo.FutureGoodEdbInfoId)
- zlAndChildEdbId[zlFutureGoodEdbInfoId] = childFutureGoodEdbInfo.FutureGoodEdbInfoId
- }
- } else {
- if childFutureGoodEdbInfo, ok2 := futureGoodEdbInfoList[date]; ok2 {
- childFutureGoodEdbInfoIdList = append(childFutureGoodEdbInfoIdList, childFutureGoodEdbInfo.FutureGoodEdbInfoId)
- zlAndChildEdbId[zlFutureGoodEdbInfoId] = childFutureGoodEdbInfo.FutureGoodEdbInfoId
- }
- }
- }
-
-
-
-
- calculateMap := make(map[int]float64)
- for _, childFutureGoodEdbInfoId := range childFutureGoodEdbInfoIdList {
- tmpFindDataValue, tmpIsFind := edbDataMap[childFutureGoodEdbInfoId][realDateTime.Format(utils.FormatDate)]
- if tmpIsFind && tmpFindDataValue != 0 {
- calculateMap[childFutureGoodEdbInfoId] = tmpFindDataValue
- }
- }
-
-
-
-
- newTagEdbIdMap := make(map[string]int)
- for tag, zlEdbId := range tagEdbIdMap {
- newTagEdbIdMap[tag] = zlAndChildEdbId[zlEdbId]
- }
-
- formulaFormStr := ReplaceFormula(newTagEdbIdMap, calculateMap, formulaStr)
-
- if formulaFormStr == `` {
-
-
- continue
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- calVal, err := engine.ParseAndExec(formulaFormStr)
-
- if err != nil {
-
- if utils.IsDivideZero(err) {
-
- continue
- }
- err = errors.New("计算失败:获取计算值失败 Err:" + err.Error() + ";formulaStr:" + formulaFormStr)
- fmt.Println(err)
- return nil, nil, err
- }
-
-
-
- nanCheck := fmt.Sprintf("%0.f", calVal)
-
- if nanCheck == "NaN" || nanCheck == "+Inf" || nanCheck == "-Inf" {
- continue
- }
- calVal, _ = decimal.NewFromFloat(calVal).Round(4).Float64()
- yDataMap[tmpN] = calVal
- xEdbInfoIdList = append(xEdbInfoIdList, tmpN)
- findDataList = append(findDataList, calVal)
- }
- yName := barChartInfoDate.Name
- yNameEn := barChartInfoDate.Name
- if yName == `` {
- if barChartInfoDate.Type == 2 {
- yName = strconv.Itoa(barChartInfoDate.Value) + "天前"
- if barChartInfoDate.Value == 1 {
- yNameEn = strconv.Itoa(barChartInfoDate.Value) + "day ago"
- } else {
- yNameEn = strconv.Itoa(barChartInfoDate.Value) + " days ago"
- }
- } else {
- yName = maxDate.Format(utils.FormatDate)
- yNameEn = maxDate.Format(utils.FormatDate)
- }
- }
- yDate := "0000-00-00"
- if !maxDate.IsZero() {
- yDate = maxDate.Format(utils.FormatDate)
- }
- yDataList = append(yDataList, YData{
- Date: yDate,
- ConfigDate: realDateTime,
- Value: findDataList,
- NoDataEdbList: noDataIdList,
- XEdbInfoIdList: xEdbInfoIdList,
- Color: barChartInfoDate.Color,
- Name: yName,
- NameEn: yNameEn,
- EdbValMap: yDataMap,
- M: mList,
- })
- }
- return
- }
- func getProfitFutureGoodEdbInfoList(earliestDateTime time.Time, zlFutureGoodEdbInfo *future_good.FutureGoodEdbInfo, tmpFutureGoodEdbInfoList []*future_good.FutureGoodEdbInfo, isAllChina bool, monthNum int) (futureGoodEdbInfoDateMap map[string]*future_good.FutureGoodEdbInfo, newMaxN int, err error) {
- maxN := 36
- futureGoodEdbInfoList := make([]*future_good.FutureGoodEdbInfo, 0)
- futureGoodEdbInfoDateMap = make(map[string]*future_good.FutureGoodEdbInfo)
- earliestDateTime = time.Date(earliestDateTime.Year(), earliestDateTime.Month(), 1, 0, 0, 0, 0, time.Local)
- if zlFutureGoodEdbInfo.RegionType == "国内" {
- startMonth := int(earliestDateTime.Month())
- if startMonth == 1 {
- futureGoodEdbInfoList = tmpFutureGoodEdbInfoList
- } else {
-
- index := startMonth - 1
- futureGoodEdbInfoList = tmpFutureGoodEdbInfoList[index:]
- futureGoodEdbInfoList = append(futureGoodEdbInfoList, tmpFutureGoodEdbInfoList[:index]...)
- }
- lenFutureGoodEdbInfoList := len(futureGoodEdbInfoList)
-
-
-
-
-
-
- maxN = lenFutureGoodEdbInfoList + monthNum
- for i := 1; i < maxN; i++ {
- k := i % lenFutureGoodEdbInfoList
- futureGoodEdbInfoDateMap[earliestDateTime.AddDate(0, i, 0).Format(utils.FormatYearMonthDate)] = futureGoodEdbInfoList[k]
- if i > newMaxN {
- newMaxN = i
- }
- }
-
- newMaxN = 12
- return
- }
- for _, v := range tmpFutureGoodEdbInfoList {
-
- if v.FutureGoodEdbType == 2 {
- if v.Month <= newMaxN {
- futureGoodEdbInfoDateMap[earliestDateTime.AddDate(0, v.Month, 0).Format(utils.FormatYearMonthDate)] = v
- if v.Month > newMaxN {
- newMaxN = v.Month
- }
- }
- continue
- }
- if v.Year < earliestDateTime.Year() {
- continue
- }
-
- if v.Year == earliestDateTime.Year() && v.Month <= int(earliestDateTime.Month()) {
- continue
- }
- subYear := v.Year - earliestDateTime.Year()
- subMonth := v.Month - int(earliestDateTime.Month())
-
- tmpN := subYear*12 + subMonth
- if tmpN < newMaxN {
- tmpDateTime := time.Date(v.Year, time.Month(v.Month), 0, 0, 0, 0, 0, time.Local)
- futureGoodEdbInfoDateMap[tmpDateTime.Format(utils.FormatYearMonthDate)] = v
- if tmpN > newMaxN {
- newMaxN = tmpN
- }
- continue
- }
- }
- return
- }
- func handleProfitResultData(baseEdbInfo *models.EdbInfo, yDataList []YData, earliestDateTime time.Time) (xDataList []XData, newYDataList []YData, err error) {
- xDataList = make([]XData, 0)
- newYDataList = yDataList
- nMap := make(map[int]int)
- for _, v := range yDataList {
- for _, n := range v.XEdbInfoIdList {
- nMap[n] = n
- }
- }
-
- nList := make([]int, 0)
- for _, n := range nMap {
- nList = append(nList, n)
- }
- sort.Slice(nList, func(i, j int) bool {
- return nList[i] < nList[j]
- })
- for _, n := range nList {
- if n == 0 {
- continue
- }
- xDataList = append(xDataList, XData{
- Name: fmt.Sprint("M+", n),
- NameEn: fmt.Sprint("M+", n),
- })
- }
- for yIndex, yData := range yDataList {
- newYDataList[yIndex].XEdbInfoIdList = []int{}
- newYDataList[yIndex].Value = []float64{}
- tmpNList := nList
- xEdbInfoIdList := yData.XEdbInfoIdList
- valIndex := 0
- needNum := 0
- for _, n := range tmpNList {
- if len(xEdbInfoIdList) > 0 {
- currN := xEdbInfoIdList[0]
-
- if n == currN {
- if needNum > 0 {
- currVal := yData.Value[valIndex]
- preVal := yData.Value[valIndex-1]
- hcValDeci := decimal.NewFromFloat(currVal).Sub(decimal.NewFromFloat(preVal)).Div(decimal.NewFromInt(int64(needNum + 1)))
- for tmpNum := 0; tmpNum < needNum; tmpNum++ {
- newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, 0)
-
- tmpVal, _ := decimal.NewFromFloat(preVal).Add(hcValDeci.Mul(decimal.NewFromInt(int64(tmpNum + 1)))).Round(4).Float64()
- newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, tmpVal)
- }
- }
- newYDataList[yIndex].XEdbInfoIdList = append(newYDataList[yIndex].XEdbInfoIdList, currN+1)
- newYDataList[yIndex].Value = append(newYDataList[yIndex].Value, yData.Value[valIndex])
- valIndex++
- needNum = 0
- if len(xEdbInfoIdList) > 0 {
- xEdbInfoIdList = xEdbInfoIdList[1:]
- }
- } else {
- needNum++
- }
- }
- }
- }
- maxI := 0
- for _, yData := range newYDataList {
- lenEdb := len(yData.XEdbInfoIdList)
- for i := 0; i < lenEdb; i++ {
- if yData.XEdbInfoIdList[i] != 0 && !utils.InArrayByInt(yData.NoDataEdbList, yData.XEdbInfoIdList[i]) {
- if maxI < i {
- maxI = i
- }
- }
- }
- }
- earliestDateTime = time.Date(earliestDateTime.Year(), earliestDateTime.Month(), 1, 0, 0, 0, 0, time.Local)
- xDataList = xDataList[0:maxI]
- for yIndex, yData := range newYDataList {
- if len(yData.XEdbInfoIdList) > maxI+1 {
- newYDataList[yIndex].XEdbInfoIdList = yData.XEdbInfoIdList[0 : maxI+1]
- }
- if len(yData.Value) > maxI+1 {
- newYDataList[yIndex].Value = yData.Value[0 : maxI+1]
- }
- currDate, _ := time.ParseInLocation(utils.FormatDate, yData.Date, time.Local)
- nameList := make([]string, 0)
- enNameList := make([]string, 0)
- for _, n := range newYDataList[yIndex].XEdbInfoIdList {
- if n == 1 {
- nameList = append(nameList, baseEdbInfo.EdbName)
- enNameList = append(enNameList, baseEdbInfo.EdbNameEn)
- continue
- }
- if n <= 0 {
- nameList = append(nameList, `无合约`)
- enNameList = append(enNameList, `no contract`)
- } else {
- var date string
- currDateTime := currDate.AddDate(0, n-1, 0)
- month := int(currDateTime.Month())
- date = fmt.Sprintf("%d%d", currDateTime.Year(), month)
- if month < 10 {
- date = fmt.Sprintf("%d0%d", currDateTime.Year(), month)
- }
- nameList = append(nameList, date)
- enNameList = append(enNameList, date)
- }
- }
- newYDataList[yIndex].NameList = nameList
- newYDataList[yIndex].EnNameList = enNameList
- }
- return
- }
- func CheckFormula(formula string) map[string]string {
- mathFormula := []string{"MAX", "MIN", "ABS", "ACOS", "ASIN", "CEIL", "MOD", "POW", "ROUND", "SIGN", "SIN", "TAN", "LOG10", "LOG2", "LOG", "LN"}
- str := strings.ToUpper(formula)
- for _, v := range mathFormula {
- str = strings.Replace(str, v, "", -1)
- }
- str = strings.Replace(str, "(", "", -1)
- str = strings.Replace(str, ")", "", -1)
- byteMap := make(map[string]string)
- for i := 0; i < len(str); i++ {
- byteInt := str[i]
- if byteInt >= 65 && byteInt <= 90 {
- byteStr := string(byteInt)
- if _, ok := byteMap[byteStr]; !ok {
- byteMap[byteStr] = byteStr
- }
- }
- }
- return byteMap
- }
- func ReplaceFormula(tagEdbIdMap map[string]int, valArr map[int]float64, formulaStr string) string {
- funMap := GetFormulaMap()
- for k, v := range funMap {
- formulaStr = strings.Replace(formulaStr, k, v, -1)
- }
- replaceCount := 0
- for tag, edbInfoId := range tagEdbIdMap {
- if val, valOk := valArr[edbInfoId]; valOk {
- dvStr := fmt.Sprintf("%v", val)
- formulaStr = strings.Replace(formulaStr, tag, dvStr, -1)
- replaceCount++
- }
- }
- for k, v := range funMap {
- formulaStr = strings.Replace(formulaStr, v, k, -1)
- }
- if replaceCount == len(tagEdbIdMap) {
- return formulaStr
- } else {
- return ""
- }
- }
- func GetFormulaMap() map[string]string {
- funMap := make(map[string]string)
- funMap["MAX"] = "[@@]"
- funMap["MIN"] = "[@!]"
- funMap["ABS"] = "[@#]"
- funMap["CEIL"] = "[@$]"
- funMap["COS"] = "[@%]"
- funMap["FLOOR"] = "[@^]"
- funMap["MOD"] = "[@&]"
- funMap["POW"] = "[@*]"
- funMap["ROUND"] = "[@(]"
- return funMap
- }
- func GetNeedDateData(needDateTime time.Time, dataList []*models.EdbDataList, edbDataMap map[string]float64, allEdbDataMap map[int]map[string]float64) (findDateTime time.Time, findDataValue float64, isFind bool, err error) {
-
- if len(dataList) <= 0 {
-
- return
- }
-
- minDateTime, err := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
- if err != nil {
- return
- }
-
- maxCount := 1
- for tmpDateTime := needDateTime; tmpDateTime.After(minDateTime) || tmpDateTime.Equal(minDateTime); tmpDateTime = tmpDateTime.AddDate(0, 0, -1) {
- tmpDate := tmpDateTime.Format(utils.FormatDate)
- tmpValue, ok := edbDataMap[tmpDate]
- if !ok {
- continue
- }
-
- count := 0
- for _, currEdbDataMap := range allEdbDataMap {
- _, tmpIsFind := currEdbDataMap[tmpDate]
- if tmpIsFind {
- count++
- if count >= maxCount {
- continue
- }
- }
- }
-
- if count < maxCount {
- continue
- }
-
-
- if tmpValue == 0 {
- return
- }
- findDateTime, _ = time.ParseInLocation(utils.FormatDate, tmpDate, time.Local)
- findDataValue = tmpValue
- isFind = true
- return
- }
- return
- }
|