123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687 |
- package chart
- import (
- "errors"
- "fmt"
- "hongze/hongze_yb/global"
- chartEdbMappingModel "hongze/hongze_yb/models/tables/chart_edb_mapping"
- chartInfoModel "hongze/hongze_yb/models/tables/chart_info"
- company2 "hongze/hongze_yb/models/tables/company"
- edbDataModel "hongze/hongze_yb/models/tables/edb_data"
- edbInfoModel "hongze/hongze_yb/models/tables/edb_info"
- "hongze/hongze_yb/models/tables/my_chart_classify"
- "hongze/hongze_yb/models/tables/yb_chart_daily_visit_log"
- "hongze/hongze_yb/models/tables/yb_chart_visit_log"
- "hongze/hongze_yb/services/alarm_msg"
- edbDataService "hongze/hongze_yb/services/edb_data"
- "hongze/hongze_yb/services/user"
- "hongze/hongze_yb/utils"
- "sort"
- "strconv"
- "strings"
- "time"
- )
- // ModifyChartInfoAndMapping 修改图表信息
- func ModifyChartInfoAndMapping(edbInfoIdStr string, req *chartInfoModel.SaveChartInfoReq, chartType int) (err error) {
- // 开启事务
- tx := global.MYSQL["data"].Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- // 更新图表信息
- if chartType == 1 {
- sql := ` UPDATE chart_info
- SET
- edb_info_ids = ?,
- modify_time = NOW(),
- date_type = ?,
- start_date = ?,
- end_date = ?,
- left_min = ?,
- left_max = ?,
- right_min = ?,
- right_max = ?
- WHERE chart_info_id = ?`
- err = tx.Exec(sql, edbInfoIdStr, req.DateType, req.StartDate, req.EndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.ChartInfoId).Error
- } else {
- sql := ` UPDATE chart_info
- SET
- edb_info_ids = ?,
- modify_time = NOW(),
- calendar = ?,
- season_start_date = ?,
- season_end_date = ?,
- left_min = ?,
- left_max = ?,
- right_min = ?,
- right_max = ?
- WHERE chart_info_id = ?`
- err = tx.Exec(sql, edbInfoIdStr, req.Calendar, req.SeasonStartDate, req.SeasonEndDate, req.LeftMin, req.LeftMax, req.RightMin, req.RightMax, req.ChartInfoId).Error
- }
- if err != nil {
- fmt.Println("UPDATE chart_info Err:", err.Error())
- return
- }
- var edbInfoIdArr []string
- mapList := make([]*chartEdbMappingModel.ChartEdbMapping, 0)
- for _, v := range req.ChartEdbInfoList {
- edbInfoIdArr = append(edbInfoIdArr, strconv.Itoa(v.EdbInfoId))
- var count int
- csql := ` SELECT COUNT(1) AS count FROM chart_edb_mapping WHERE chart_info_id = ? AND edb_info_id = ? `
- err = tx.Raw(csql, req.ChartInfoId, v.EdbInfoId).Scan(&count).Error
- if err != nil {
- fmt.Println("QueryRow Err:", err.Error())
- return err
- }
- if count > 0 {
- // 更新指标
- msql := `UPDATE chart_edb_mapping
- SET
- modify_time = NOW(),
- max_data = ?,
- min_data = ?,
- is_order = ?,
- is_axis = ?,
- edb_info_type = ?,
- lead_value = ?,
- lead_unit = ?,
- chart_style = ?,
- chart_color = ?,
- chart_width = ?
- WHERE chart_info_id =? AND edb_info_id=? `
- err = tx.Exec(msql, v.MaxData, v.MinData, v.IsOrder, v.IsAxis, v.EdbInfoType, v.LeadValue, v.LeadUnit, v.ChartStyle, v.ChartColor, v.ChartWidth, req.ChartInfoId, v.EdbInfoId).Error
- if err != nil {
- fmt.Println("chart_edb_mapping Err:" + err.Error())
- return err
- }
- } else {
- mapItem := new(chartEdbMappingModel.ChartEdbMapping)
- mapItem.ChartInfoId = req.ChartInfoId
- mapItem.EdbInfoId = v.EdbInfoId
- mapItem.CreateTime = time.Now()
- mapItem.ModifyTime = time.Now()
- timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- mapItem.UniqueCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
- mapItem.MaxData = v.MaxData
- mapItem.MinData = v.MinData
- mapItem.IsOrder = 0
- if v.IsOrder {
- mapItem.IsOrder = 1
- }
- mapItem.IsAxis = v.IsAxis
- mapItem.EdbInfoType = v.EdbInfoType
- mapItem.LeadValue = v.LeadValue
- mapItem.LeadUnit = v.LeadUnit
- mapItem.ChartStyle = v.ChartStyle
- mapItem.ChartColor = v.ChartColor
- mapItem.ChartWidth = v.ChartWidth
- mapList = append(mapList, mapItem)
- }
- }
- // 批量新增指标
- if len(mapList) > 0 {
- err = tx.Create(mapList).Error
- if err != nil {
- fmt.Println("AddChartEdbMapping Err:" + err.Error())
- return err
- }
- }
- // 清除原图表指标
- if len(edbInfoIdArr) > 0 {
- edbInfoIdStr := strings.Join(edbInfoIdArr, ",")
- if edbInfoIdStr != "" {
- dsql := `DELETE FROM chart_edb_mapping WHERE chart_info_id=? AND edb_info_id NOT IN(` + edbInfoIdStr + `)`
- err = tx.Exec(dsql, req.ChartInfoId).Error
- if err != nil {
- fmt.Println("delete err:" + err.Error())
- return err
- }
- }
- }
- return
- }
- // GetChartInfoRefreshData 获取需要刷新的指标
- func GetChartInfoRefreshData(chartInfoId int) (baseEdbInfoArr, calculateInfoArr []*edbInfoModel.EdbInfo, err error) {
- sql := ` SELECT b.* FROM chart_edb_mapping AS a
- INNER JOIN edb_info AS b ON a.edb_info_id=b.edb_info_id
- WHERE a.chart_info_id=? `
- edbInfoList := make([]*edbInfoModel.EdbInfo, 0)
- err = global.MYSQL["data"].Raw(sql, chartInfoId).Scan(&edbInfoList).Error
- if err != nil {
- return
- }
- for _, v := range edbInfoList {
- fmt.Println(v.EdbInfoId, v.EdbType)
- if v.EdbType == 1 {
- baseEdbInfoArr = append(baseEdbInfoArr, v)
- } else {
- calculateInfoArr = append(calculateInfoArr, v)
- newBaseEdbInfoArr, newCalculateInfoArr, err := GetChartRefreshEdbInfo(v.EdbInfoId, v.Source, 0)
- if err != nil {
- return baseEdbInfoArr, calculateInfoArr, err
- }
- baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...)
- calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...)
- }
- }
- return
- }
- // GetChartRefreshEdbInfo
- func GetChartRefreshEdbInfo(edbInfoId, source, n int) (baseEdbInfoArr, calculateInfoArr []*edbInfoModel.EdbInfo, err error) {
- calculateList, err := GetEdbInfoCalculateMap(edbInfoId, source)
- if err != nil && err != utils.ErrNoRow {
- return
- }
- n++
- for _, item := range calculateList {
- fmt.Println(item.EdbInfoId)
- if item.EdbType == 1 {
- baseEdbInfoArr = append(baseEdbInfoArr, item)
- } else {
- calculateInfoArr = append(calculateInfoArr, item)
- if n > 10 {
- return
- }
- newBaseEdbInfoArr, newCalculateInfoArr, _ := GetChartRefreshEdbInfo(item.EdbInfoId, item.Source, n)
- baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...)
- calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...)
- }
- }
- return
- }
- func GetEdbInfoCalculateMap(edbInfoId, source int) (list []*edbInfoModel.EdbInfo, err error) {
- sql := ` SELECT b.* FROM edb_info_calculate_mapping AS a
- INNER JOIN edb_info AS b ON a.from_edb_info_id=b.edb_info_id
- WHERE a.edb_info_id=? ORDER BY sort ASC `
- err = global.MYSQL["data"].Exec(sql, edbInfoId).Error
- return
- }
- // SaveChartVisitLog 保存图表访问记录
- func SaveChartVisitLog(userInfo user.UserInfo, chartInfo *chartInfoModel.ChartInfoView, myChartClassifyId int) (err error) {
- userId := int(userInfo.UserID)
- companyId := int(userInfo.CompanyID)
- chartInfoId := chartInfo.ChartInfoId
- // 访问记录
- visitLog := new(yb_chart_visit_log.YbChartVisitLog)
- visitLog.CompanyId = companyId
- companyInfo, err := company2.GetByCompanyId(userInfo.CompanyID)
- if err == nil {
- visitLog.CompanyName = companyInfo.CompanyName
- }
- visitLog.UserId = userId
- visitLog.RealName = userInfo.RealName
- visitLog.Mobile = userInfo.Mobile
- visitLog.Email = userInfo.Email
- visitLog.ChartInfoId = chartInfoId
- visitLog.ChartName = chartInfo.ChartName
- visitLog.CreateTime = time.Now().Local()
- visitLog.Create()
- // 每日访问记录,用户/天/图仅记录一次
- if myChartClassifyId > 0 {
- // 今日是否已访问
- nowTime := time.Now().Local()
- startTime := time.Date(nowTime.Year(), nowTime.Month(), nowTime.Day(), 0, 0, 0, 0, time.Local)
- endTime := time.Date(nowTime.Year(), nowTime.Month(), nowTime.Day(), 23, 59, 59, 0, time.Local)
- _, tmpErr := yb_chart_daily_visit_log.GetTodayVisitLog(userId, chartInfoId, myChartClassifyId, startTime, endTime)
- if tmpErr == nil {
- return
- }
- myChartClassify, tmpErr := my_chart_classify.GetClassifyById(myChartClassifyId)
- if tmpErr != nil {
- return
- }
- dailyLog := new(yb_chart_daily_visit_log.YbChartDailyVisitLog)
- dailyLog.CompanyId = companyId
- if companyInfo != nil {
- dailyLog.CompanyName = companyInfo.CompanyName
- }
- dailyLog.UserId = userId
- dailyLog.RealName = userInfo.RealName
- dailyLog.Mobile = userInfo.Mobile
- dailyLog.Email = userInfo.Email
- dailyLog.ChartInfoId = chartInfoId
- dailyLog.ChartName = chartInfo.ChartName
- dailyLog.MyChartClassifyId = myChartClassifyId
- dailyLog.MyChartClassifyName = myChartClassify.MyChartClassifyName
- dailyLog.CreateTime = time.Now().Local()
- dailyLog.Create()
- }
- return
- }
- // RefreshChart 图表刷新
- func RefreshChart(chartInfoId int) (err error) {
- var errMsg string
- defer func() {
- if err != nil {
- if global.CONFIG.Serve.RunMode == "release" {
- //go services.SendEmail(utils.APPNAME+"【"+global.CONFIG.Serve.RunMode+"】"+"失败提醒", "RefreshChart:"+errMsg, utils.EmailSendToUsers)
- go alarm_msg.SendAlarmMsg("RefreshChart Err:"+errMsg, 3)
- }
- fmt.Println("RefreshChart Err:" + errMsg)
- }
- }()
- // 获取需要刷新的指标
- baseEdbInfoArr, calculateInfoArr, err := GetChartInfoRefreshData(chartInfoId)
- if err != nil {
- errMsg = "获取需要刷新的指标失败:Err:" + err.Error()
- return
- }
- newBaseEdbInfoArr := make([]*edbInfoModel.EdbInfo, 0)
- baseMap := make(map[int]*edbInfoModel.EdbInfo)
- for _, bv := range baseEdbInfoArr {
- // 如果不是普通指标,那么过滤
- if bv.EdbInfoType != 0 {
- continue
- }
- if _, ok := baseMap[bv.EdbInfoId]; !ok {
- newBaseEdbInfoArr = append(newBaseEdbInfoArr, bv)
- }
- baseMap[bv.EdbInfoId] = bv
- }
- newCalculateInfoArr := make([]*edbInfoModel.EdbInfo, 0)
- calculateMap := make(map[int]*edbInfoModel.EdbInfo)
- var calculateArr []int
- for _, bv := range calculateInfoArr {
- if _, ok := calculateMap[bv.EdbInfoId]; !ok {
- newCalculateInfoArr = append(newCalculateInfoArr, bv)
- calculateArr = append(calculateArr, bv.EdbInfoId)
- }
- calculateMap[bv.EdbInfoId] = bv
- }
- sort.Ints(calculateArr)
- // 刷新指标
- var startDate string
- for _, bv := range newBaseEdbInfoArr {
- source := bv.Source
- edbInfoId := bv.EdbInfoId
- edbCode := bv.EdbCode
- startDate = bv.StartDate.Format(utils.FormatDate)
- frequency := bv.Frequency
- if startDate == "0000-00-00" {
- continue
- }
- sTime := bv.EndDate
- var limitDay int
- startDate := ""
- switch frequency {
- case "日度":
- limitDay = utils.DATA_START_REFRESH_LIMIT
- case "周度":
- limitDay = utils.DATA_START_REFRESH_LIMIT * 7
- case "月度":
- limitDay = utils.DATA_START_REFRESH_LIMIT * 30
- case "季度":
- limitDay = utils.DATA_START_REFRESH_LIMIT * 90
- case "年度":
- limitDay = utils.DATA_START_REFRESH_LIMIT * 365
- default:
- limitDay = utils.DATA_START_REFRESH_LIMIT
- }
- startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
- fmt.Println("source:", source)
- respItem, err := RefreshEdbData(edbInfoId, source, edbCode, startDate)
- if err != nil {
- errMsg = errors.New("RefreshEdbData Err:" + err.Error()).Error()
- return err
- }
- if respItem.Ret != 200 {
- errMsg = errors.New(respItem.ErrMsg + ";EdbCode:" + edbCode).Error()
- return err
- }
- //maxAndMinItem, err := edbDataModel.GetEdbInfoMaxAndMinInfo(source, edbCode)
- //if err != nil {
- // return err
- //}
- //if maxAndMinItem != nil {
- // err = edbDataModel.ModifyEdbInfoMaxAndMinInfo(edbInfoId, maxAndMinItem)
- // if err != nil {
- // return err
- // }
- //}
- }
- // 刷新计算指标
- for _, v := range calculateArr {
- edbInfo := calculateMap[v]
- if edbInfo == nil {
- return err
- }
- startDate = edbInfo.StartDate.Format(utils.FormatDate)
- source := edbInfo.Source
- if source == utils.DATA_SOURCE_CALCULATE {
- sTime := edbInfo.EndDate
- startDate = sTime.Format(utils.FormatDate)
- }
- result, err := RefreshEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
- if err != nil {
- fmt.Println(v, "RefreshEdbCalculateData err", time.Now())
- errMsg = "RefreshEdbCalculateData Err:" + err.Error()
- return err
- }
- if result.Ret != 200 {
- fmt.Println(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
- errMsg = fmt.Sprint(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
- return fmt.Errorf("刷新失败")
- }
- }
- return
- }
- // GetChartEdbData 获取图表的指标数据
- func GetChartEdbData(chartInfoId, chartType int, calendar, startDate, endDate string, mappingList []*chartEdbMappingModel.ChartEdbInfoMapping) (edbList []*chartEdbMappingModel.ChartEdbInfoMappingList, sourceArr []string, err error) {
- edbList = make([]*chartEdbMappingModel.ChartEdbInfoMappingList, 0)
- sourceArr = make([]string, 0)
- for _, v := range mappingList {
- //fmt.Println("v:", v.EdbInfoId)
- item := new(chartEdbMappingModel.ChartEdbInfoMappingList)
- item.EdbInfoId = v.EdbInfoId
- item.SourceName = v.SourceName
- item.Source = v.Source
- item.EdbCode = v.EdbCode
- item.EdbName = v.EdbName
- item.EdbNameEn = v.EdbNameEn
- item.Frequency = v.Frequency
- item.FrequencyEn = GetFrequencyEn(v.Frequency)
- if v.Unit != `无` {
- item.Unit = v.Unit
- }
- item.UnitEn = v.UnitEn
- item.StartDate = v.StartDate
- item.EndDate = v.EndDate
- item.ModifyTime = v.ModifyTime
- item.EdbInfoCategoryType = v.EdbInfoCategoryType
- item.PredictChartColor = v.PredictChartColor
- if !utils.InArray(v.Source, utils.SystemSourceList) { //来源于系统的指标,都展示为弘则研究
- if !utils.InArray(v.SourceName, sourceArr) {
- sourceArr = append(sourceArr, v.SourceName)
- }
- }
- if chartInfoId <= 0 {
- item.IsAxis = 1
- item.LeadValue = 0
- item.LeadUnit = ""
- item.ChartEdbMappingId = 0
- item.ChartInfoId = 0
- item.IsOrder = false
- item.EdbInfoType = 1
- item.ChartStyle = ""
- item.ChartColor = ""
- item.ChartWidth = 0
- item.MaxData = v.MaxValue
- item.MinData = v.MinValue
- } else {
- item.IsAxis = v.IsAxis
- item.EdbInfoType = v.EdbInfoType
- item.LeadValue = v.LeadValue
- item.LeadUnit = v.LeadUnit
- item.LeadUnitEn = GetLeadUnitEn(v.LeadUnit)
- item.ChartEdbMappingId = v.ChartEdbMappingId
- item.ChartInfoId = v.ChartInfoId
- item.ChartStyle = v.ChartStyle
- item.ChartColor = v.ChartColor
- item.ChartWidth = v.ChartWidth
- item.IsOrder = v.IsOrder
- item.MaxData = v.MaxData
- item.MinData = v.MinData
- }
- item.LatestValue = v.LatestValue
- item.LatestDate = v.LatestDate
- item.UniqueCode = v.UniqueCode
- var startDateReal string
- var diffSeconds int64
- if chartType == 2 { //季节性图
- startDateReal = startDate
- } else {
- if v.EdbInfoType == 0 && v.LeadUnit != "" && v.LeadValue > 0 { //领先指标
- var startTimeRealTemp time.Time
- startDateParse, _ := time.Parse(utils.FormatDate, startDate)
- switch v.LeadUnit {
- case "天":
- startTimeRealTemp = startDateParse.AddDate(0, 0, -v.LeadValue)
- case "月":
- startTimeRealTemp = startDateParse.AddDate(0, -v.LeadValue, 0)
- case "季":
- startTimeRealTemp = startDateParse.AddDate(0, -3*v.LeadValue, 0)
- case "周":
- startTimeRealTemp = startDateParse.AddDate(0, 0, -7*v.LeadValue)
- case "年":
- startTimeRealTemp = startDateParse.AddDate(-v.LeadValue, 0, 0)
- }
- if startTimeRealTemp.Before(startDateParse) {
- startDateReal = startTimeRealTemp.Format(utils.FormatDate)
- diffSeconds = (int64(startTimeRealTemp.UnixNano()) - int64(startDateParse.UnixNano())) / 1e6
- } else {
- startDateReal = startDate
- diffSeconds = 0
- }
- } else {
- startDateReal = startDate
- }
- }
- //fmt.Println("line 1011 chart:", v.Source, v.EdbInfoId, startDateReal, endDate)
- calendarPreYear := 0
- if calendar == "农历" {
- newStartDateReal, err := time.Parse(utils.FormatDate, startDateReal)
- if err != nil {
- fmt.Println("time.Parse:" + err.Error())
- }
- calendarPreYear = newStartDateReal.Year() - 1
- newStartDateReal = newStartDateReal.AddDate(-1, 0, 0)
- startDateReal = newStartDateReal.Format(utils.FormatDate)
- }
- dataList := make([]*edbDataModel.EdbDataList, 0)
- //fmt.Println("chart:", v.Source, v.EdbInfoId, startDateReal, endDate, ";EdbInfoCategoryType:", v.EdbInfoCategoryType)
- switch v.EdbInfoCategoryType {
- case 0:
- dataList, err = edbDataModel.GetEdbDataList(v.Source, v.EdbInfoId, startDateReal, endDate)
- case 1:
- dataList, _, _, err, _ = GetPredictDataListByPredictEdbInfoId(v.EdbInfoId, startDateReal, endDate, false)
- default:
- err = errors.New(fmt.Sprint("获取失败,指标类型异常", v.EdbInfoCategoryType))
- }
- if err != nil {
- return
- }
- if diffSeconds != 0 && v.EdbInfoType == 0 {
- dataListLen := len(dataList)
- for i := 0; i < dataListLen; i++ {
- dataList[i].DataTimestamp = dataList[i].DataTimestamp - diffSeconds
- }
- }
- if chartType == 2 {
- latestDateStr := v.LatestDate.Format(utils.FormatDate) //实际数据的截止日期
- latestDate := v.LatestDate
- //latestDate, tmpErr := time.Parse(utils.FormatDate, v.LatestDate)
- //if tmpErr != nil {
- // err = errors.New(fmt.Sprint("获取最后实际数据的日期失败,Err:" + tmpErr.Error() + ";LatestDate:" + v.LatestDate))
- // return
- //}
- latestDateYear := latestDate.Year() //实际数据截止年份
- if calendar == "农历" {
- if len(dataList) <= 0 {
- result := new(edbDataModel.EdbDataResult)
- item.DataList = result
- } else {
- result, tmpErr := edbDataService.AddCalculateQuarter(dataList)
- if tmpErr != nil {
- err = errors.New("获取农历数据失败,Err:" + tmpErr.Error())
- return
- }
- // 处理季节图的截止日期
- for k, edbDataItems := range result.List {
- var cuttingDataTimestamp int64
- // 切割的日期时间字符串
- cuttingDataTimeStr := latestDate.AddDate(0, 0, edbDataItems.BetweenDay).Format(utils.FormatDate)
- //如果等于最后的实际日期,那么遍历找到该日期对应的时间戳,并将其赋值为 切割时间戳
- if edbDataItems.Year >= latestDateYear {
- for _, tmpData := range edbDataItems.Items {
- if tmpData.DataTime == cuttingDataTimeStr {
- cuttingDataTimestamp = tmpData.DataTimestamp
- break
- }
- }
- }
- edbDataItems.CuttingDataTimestamp = cuttingDataTimestamp
- result.List[k] = edbDataItems
- }
- if result.List[0].Year != calendarPreYear {
- itemList := make([]*edbDataModel.EdbDataList, 0)
- items := new(edbDataModel.EdbDataItems)
- //items.Year = calendarPreYear
- items.Items = itemList
- newResult := new(edbDataModel.EdbDataResult)
- newResult.List = append(newResult.List, items)
- newResult.List = append(newResult.List, result.List...)
- item.DataList = newResult
- } else {
- item.DataList = result
- }
- }
- } else {
- currentYear := time.Now().Year()
- quarterDataList := make([]*edbDataModel.QuarterData, 0)
- quarterMap := make(map[int][]*edbDataModel.EdbDataList)
- var quarterArr []int
- for _, v := range dataList {
- itemDate, tmpErr := time.Parse(utils.FormatDate, v.DataTime)
- if tmpErr != nil {
- err = errors.New("季度指标日期转换,Err:" + tmpErr.Error() + ";DataTime:" + v.DataTime)
- return
- }
- year := itemDate.Year()
- newItemDate := itemDate.AddDate(currentYear-year, 0, 0)
- timestamp := newItemDate.UnixNano() / 1e6
- v.DataTimestamp = timestamp
- if findVal, ok := quarterMap[year]; !ok {
- quarterArr = append(quarterArr, year)
- findVal = append(findVal, v)
- quarterMap[year] = findVal
- } else {
- findVal = append(findVal, v)
- quarterMap[year] = findVal
- }
- }
- for _, v := range quarterArr {
- itemList := quarterMap[v]
- quarterItem := new(edbDataModel.QuarterData)
- quarterItem.Year = v
- quarterItem.DataList = itemList
- //如果等于最后的实际日期,那么将切割时间戳记录
- if v == latestDateYear {
- var cuttingDataTimestamp int64
- for _, tmpData := range itemList {
- if tmpData.DataTime == latestDateStr {
- cuttingDataTimestamp = tmpData.DataTimestamp
- break
- }
- }
- quarterItem.CuttingDataTimestamp = cuttingDataTimestamp
- } else if v > latestDateYear {
- //如果大于最后的实际日期,那么第一个点就是切割的时间戳
- if len(itemList) > 0 {
- quarterItem.CuttingDataTimestamp = itemList[0].DataTimestamp - 100
- }
- }
- quarterDataList = append(quarterDataList, quarterItem)
- }
- item.DataList = quarterDataList
- }
- } else {
- item.DataList = dataList
- }
- edbList = append(edbList, item)
- }
- return
- }
- // 获取频度的英文版
- func GetFrequencyEn(frequency string) (frequencyEn string) {
- switch frequency {
- case "日度":
- frequencyEn = "day"
- return
- case "周度":
- frequencyEn = "week"
- return
- case "旬度":
- frequencyEn = "ten days"
- return
- case "月度":
- frequencyEn = "month"
- return
- case "季度":
- frequencyEn = "quarter"
- return
- case "年度":
- frequencyEn = "year"
- return
- }
- return
- }
- func GetLeadUnitEn(unit string) (unitEn string) {
- switch unit {
- case "天":
- unitEn = "day"
- return
- case "周":
- unitEn = "week"
- return
- case "月":
- unitEn = "month"
- return
- case "季":
- unitEn = "quarter"
- return
- case "年":
- unitEn = "year"
- return
- }
- return
- }
|