123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992 |
- package logic
- import (
- "errors"
- "eta/eta_index_lib/models"
- "eta/eta_index_lib/utils"
- "fmt"
- "strconv"
- "strings"
- "time"
- )
- // AddPredictEdbInfo 新增预测指标
- func AddPredictEdbInfo(sourceEdbInfoId, classifyId int, edbName, dataDateType string, ruleList []models.RuleConfig, minValue, maxValue float64, sysUserId int, sysUserName string) (edbInfo *models.EdbInfo, err error, errMsg string) {
- var sourceEdbInfo *models.EdbInfo
- // 来源指标信息校验
- {
- sourceEdbInfo, err = models.GetEdbInfoById(sourceEdbInfoId)
- if err != nil && err.Error() != utils.ErrNoRow() {
- errMsg = "新增失败"
- err = errors.New("获取来源指标失败,Err:" + err.Error())
- return
- }
- if sourceEdbInfo == nil {
- errMsg = "找不到该来源指标"
- err = errors.New(errMsg)
- return
- }
- //必须是普通的指标
- if sourceEdbInfo.EdbInfoType != 0 {
- errMsg = "来源指标异常,不是普通的指标"
- err = errors.New(errMsg)
- return
- }
- //if !utils.InArrayByStr([]string{"日度", "周度", "月度", "年度"}, sourceEdbInfo.Frequency) {
- // errMsg = "预测指标只支持选择日度、周度、月度、年度的指标"
- // err = errors.New(errMsg)
- // return
- //}
- }
- var classifyInfo *models.EdbClassify
- // 来源分类信息校验
- {
- classifyInfo, err = models.GetEdbClassifyById(classifyId)
- if err != nil && err.Error() != utils.ErrNoRow() {
- errMsg = "新增失败"
- err = errors.New("获取预测指标分类失败,Err:" + err.Error())
- return
- }
- if classifyInfo == nil {
- errMsg = "找不到该预测指标分类"
- err = errors.New(errMsg)
- return
- }
- //必须是预测指标分类
- if classifyInfo.ClassifyType != 1 {
- errMsg = "预测指标分类异常,不是预测指标分类"
- err = errors.New(errMsg)
- return
- }
- }
- edbName = strings.Trim(edbName, " ")
- edbCode := sourceEdbInfo.EdbCode + "_" + time.Now().Format(utils.FormatShortDateTimeUnSpace)
- //判断指标名称是否存在
- var condition string
- var pars []interface{}
- condition += " AND edb_info_type=? "
- pars = append(pars, 1)
- condition += " AND edb_name=? "
- pars = append(pars, edbName)
- count, err := models.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- errMsg = "判断指标名称是否存在失败"
- err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
- return
- }
- if count > 0 {
- errMsg = "指标名称已存在,请重新填写"
- err = errors.New(errMsg)
- return
- }
- timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- if dataDateType == `` {
- dataDateType = `自然日`
- }
- edbInfo = &models.EdbInfo{
- //EdbInfoId: 0,
- EdbInfoType: 1,
- SourceName: "预测指标",
- Source: utils.DATA_SOURCE_PREDICT,
- EdbCode: edbCode,
- EdbName: edbName,
- EdbNameSource: edbName,
- Frequency: sourceEdbInfo.Frequency,
- Unit: sourceEdbInfo.Unit,
- StartDate: sourceEdbInfo.StartDate,
- ClassifyId: classifyId,
- SysUserId: sysUserId,
- SysUserRealName: sysUserName,
- UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- MinValue: minValue,
- MaxValue: maxValue,
- CalculateFormula: sourceEdbInfo.CalculateFormula,
- EdbType: 1,
- //Sort: sourceEdbInfo.,
- LatestDate: sourceEdbInfo.LatestDate,
- LatestValue: sourceEdbInfo.LatestValue,
- MoveType: sourceEdbInfo.MoveType,
- MoveFrequency: sourceEdbInfo.MoveFrequency,
- NoUpdate: sourceEdbInfo.NoUpdate,
- ServerUrl: "",
- DataDateType: dataDateType,
- }
- // 关联关系表
- calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
- fromEdbMap := make(map[int]int)
- // 源指标关联关系表
- calculateMappingItem := &models.EdbInfoCalculateMapping{
- //EdbInfoCalculateMappingId: 0,
- //EdbInfoId: 0,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: sourceEdbInfo.EdbInfoId,
- FromEdbCode: sourceEdbInfo.EdbCode,
- FromEdbName: sourceEdbInfo.EdbName,
- FromSource: sourceEdbInfo.Source,
- FromSourceName: sourceEdbInfo.SourceName,
- //FromTag: "",
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- fromEdbMap[sourceEdbInfoId] = sourceEdbInfoId
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- // 动态环差 计算列表
- calculateRuleMap := make(map[int]models.CalculateRule, 0)
- // 预测指标配置
- predictEdbConfList := make([]*models.PredictEdbConf, 0)
- for ruleIndex, v := range ruleList {
- // 预测指标配置
- ruleEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.EndDate, time.Local)
- if tmpErr != nil {
- errMsg = "规则配置的截止日期异常,请重新填写"
- err = errors.New(errMsg)
- return
- }
- //1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值,9:动态环差,10:根据 给定终值后插值 规则获取预测数据,11:根据 季节性 规则获取预测数据,12:根据 移动平均同比 规则获取预测数据
- // 环比、环差、动态环差、季节性、移动平均同比不支持年度
- if sourceEdbInfo.Frequency == "年度" && utils.InArrayByInt([]int{5, 6, 11, 12}, v.RuleType) {
- errMsg = "环比、环差、动态环差、季节性、移动平均同比不支持年度指标"
- err = errors.New(errMsg)
- return
- }
- switch v.RuleType {
- case 8: //N期段线性外推值
- valInt, tmpErr := strconv.Atoi(v.Value)
- if tmpErr != nil {
- errMsg = "N期段线性外推值的N值异常"
- err = errors.New(errMsg)
- return
- }
- if valInt <= 1 {
- errMsg = "N期段线性外推值的N值必须大于1"
- err = errors.New(errMsg)
- return
- }
- case 9: //9:动态环差
- if v.Value == "" {
- errMsg = "请填写计算规则"
- err = errors.New(errMsg)
- return
- }
- formula := v.Value
- formula = strings.Replace(formula, "(", "(", -1)
- formula = strings.Replace(formula, ")", ")", -1)
- formula = strings.Replace(formula, ",", ",", -1)
- formula = strings.Replace(formula, "。", ".", -1)
- formula = strings.Replace(formula, "%", "*0.01", -1)
- v.Value = formula
- //检验公式
- var formulaStr string
- var edbInfoIdBytes []string
- for _, tmpEdbInfoId := range v.EdbInfoIdArr {
- formulaStr += tmpEdbInfoId.FromTag + ","
- edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
- }
- formulaSlice, tErr := utils.CheckFormulaJson(formula)
- if tErr != nil {
- errMsg = "公式格式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- for _, fm := range formulaSlice {
- formulaMap, e := utils.CheckFormula(fm)
- if e != nil {
- err = fmt.Errorf("公式错误,请重新填写")
- return
- }
- for _, f := range formulaMap {
- if !strings.Contains(formulaStr, f) {
- errMsg = "公式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- }
- }
- //关联的指标信息
- edbInfoList := make([]*models.EdbInfo, 0)
- // 动态环差规则 关系表
- trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
- for k, tmpEdbInfoId := range v.EdbInfoIdArr {
- fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- if err.Error() == utils.ErrNoRow() {
- errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
- err = errors.New(errMsg)
- return
- }
- errMsg = "获取指标失败:Err:" + err.Error()
- err = errors.New(errMsg)
- return
- }
- edbInfoList = append(edbInfoList, fromEdbInfo)
- //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
- {
- if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
- fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
- calculateMappingItem := &models.EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: 0,
- Source: utils.DATA_SOURCE_CALCULATE,
- SourceName: "指标运算",
- EdbCode: "",
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- //FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- }
- }
- // 动态环差规则 关系表
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
- //PredictEdbConfCalculateMappingId: 0,
- EdbInfoId: 0,
- ConfigId: 0,
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
- }
- for _, f := range formulaSlice {
- formulaMap, e := utils.CheckFormula(f)
- if e != nil {
- err = fmt.Errorf("公式错误,请重新填写")
- return
- }
- //预先计算,判断公式是否正常
- ok, _ := models.CheckFormula2(edbInfoList, formulaMap, f, edbInfoIdBytes)
- if !ok {
- errMsg = "生成计算指标失败,请使用正确的计算公式"
- err = errors.New(errMsg)
- return
- }
- }
- calculateRuleMap[ruleIndex] = models.CalculateRule{
- TrendsCalculateMappingList: trendsMappingList,
- EdbInfoList: edbInfoList,
- EdbInfoIdBytes: edbInfoIdBytes,
- Formula: formula,
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- case 14: //14:根据 一元线性拟合 规则获取预测数据
- if v.Value == "" {
- errMsg = "请填写一元线性拟合规则"
- err = errors.New(errMsg)
- return
- }
- //关联的指标信息
- edbInfoList := make([]*models.EdbInfo, 0)
- // 动态环差规则 关系表
- trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
- for k, tmpEdbInfoId := range v.EdbInfoIdArr {
- fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- if err.Error() == utils.ErrNoRow() {
- errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
- err = errors.New(errMsg)
- return
- }
- errMsg = "获取指标失败:Err:" + err.Error()
- err = errors.New(errMsg)
- return
- }
- edbInfoList = append(edbInfoList, fromEdbInfo)
- //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
- {
- if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
- fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
- tmpCalculateMappingItem := &models.EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: 0,
- Source: utils.DATA_SOURCE_CALCULATE,
- SourceName: "指标运算",
- EdbCode: "",
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- //FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, tmpCalculateMappingItem)
- }
- }
- // 动态环差规则 关系表
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
- //PredictEdbConfCalculateMappingId: 0,
- EdbInfoId: 0,
- ConfigId: 0,
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
- }
- calculateRuleMap[ruleIndex] = models.CalculateRule{
- TrendsCalculateMappingList: trendsMappingList,
- EdbInfoList: edbInfoList,
- //EdbInfoIdBytes: edbInfoIdBytes,
- //Formula: formula,
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- }
- tmpPredictEdbConf := &models.PredictEdbConf{
- PredictEdbInfoId: 0,
- SourceEdbInfoId: sourceEdbInfoId,
- RuleType: v.RuleType,
- //FixedValue: v.Value,
- Value: v.Value,
- EndDate: ruleEndDate,
- ModifyTime: time.Now(),
- CreateTime: time.Now(),
- }
- edbInfo.EndDate = v.EndDate
- predictEdbConfList = append(predictEdbConfList, tmpPredictEdbConf)
- }
- err, errMsg = models.AddPredictEdb(edbInfo, calculateMappingList, predictEdbConfList, calculateRuleMap)
- return
- }
- // EditPredictEdbInfo 编辑预测指标
- func EditPredictEdbInfo(edbInfoId, classifyId int, edbName, dataDateType string, ruleList []models.RuleConfig, minValue, maxValue float64) (edbInfo *models.EdbInfo, err error, errMsg string) {
- // 指标信息校验
- {
- edbInfo, err = models.GetEdbInfoById(edbInfoId)
- if err != nil && err.Error() != utils.ErrNoRow() {
- errMsg = "修改失败"
- err = errors.New("获取预测指标失败,Err:" + err.Error())
- return
- }
- if edbInfo == nil {
- errMsg = "找不到该预测指标"
- err = errors.New(errMsg)
- return
- }
- //必须是普通的指标
- if edbInfo.EdbInfoType != 1 {
- errMsg = "指标异常,不是预测指标"
- err = errors.New(errMsg)
- return
- }
- }
- var predictEdbConf *models.PredictEdbConf
- // 指标配置信息校验
- {
- // 查找该预测指标配置
- predictEdbConfList, tmpErr := models.GetPredictEdbConfListById(edbInfo.EdbInfoId)
- if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
- errMsg = "修改失败"
- err = errors.New("获取预测指标配置信息失败,Err:" + tmpErr.Error())
- return
- }
- if len(predictEdbConfList) == 0 {
- errMsg = "找不到该预测指标配置"
- err = errors.New(errMsg)
- return
- }
- predictEdbConf = predictEdbConfList[0]
- }
- //判断指标名称是否存在
- var condition string
- var pars []interface{}
- condition += " AND edb_info_id<>? "
- pars = append(pars, edbInfoId)
- condition += " AND edb_info_type=? "
- pars = append(pars, 1)
- condition += " AND edb_name=? "
- pars = append(pars, edbName)
- count, err := models.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- errMsg = "判断指标名称是否存在失败"
- err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
- return
- }
- if count > 0 {
- errMsg = "指标名称已存在,请重新填写"
- err = errors.New(errMsg)
- return
- }
- if dataDateType == `` {
- dataDateType = `自然日`
- }
- edbInfo.EdbName = edbName
- edbInfo.EdbNameSource = edbName
- edbInfo.ClassifyId = classifyId
- edbInfo.MinValue = minValue
- edbInfo.MaxValue = maxValue
- edbInfo.DataDateType = dataDateType
- edbInfo.ModifyTime = time.Now()
- updateEdbInfoCol := []string{"EdbName", "EdbNameSource", "ClassifyId", "EndDate", "MinValue", "MaxValue", "DataDateType", "ModifyTime"}
- var sourceEdbInfo *models.EdbInfo
- // 来源指标信息校验
- {
- sourceEdbInfo, err = models.GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
- if err != nil && err.Error() != utils.ErrNoRow() {
- errMsg = "新增失败"
- err = errors.New("获取来源指标失败,Err:" + err.Error())
- return
- }
- if sourceEdbInfo == nil {
- errMsg = "找不到该来源指标"
- err = errors.New(errMsg)
- return
- }
- //必须是普通的指标
- if sourceEdbInfo.EdbInfoType != 0 {
- errMsg = "来源指标异常,不是普通的指标"
- err = errors.New(errMsg)
- return
- }
- //if !utils.InArrayByStr([]string{"日度", "周度", "月度", "年度"}, sourceEdbInfo.Frequency) {
- // errMsg = "预测指标只支持选择日度、周度、月度、年度的指标"
- // err = errors.New(errMsg)
- // return
- //}
- }
- // 预测指标配置
- // 关联关系表
- calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
- fromEdbMap := make(map[int]int)
- // 源指标关联关系表
- calculateMappingItem := &models.EdbInfoCalculateMapping{
- //EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfoId,
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: sourceEdbInfo.EdbInfoId,
- FromEdbCode: sourceEdbInfo.EdbCode,
- FromEdbName: sourceEdbInfo.EdbName,
- FromSource: sourceEdbInfo.Source,
- FromSourceName: sourceEdbInfo.SourceName,
- //FromTag: "",
- Sort: 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- fromEdbMap[sourceEdbInfo.EdbInfoId] = sourceEdbInfo.EdbInfoId
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- // 动态环差 计算列表
- calculateRuleMap := make(map[int]models.CalculateRule, 0)
- // 预测指标配置
- predictEdbConfList := make([]*models.PredictEdbConf, 0)
- for ruleIndex, v := range ruleList {
- // 预测指标配置
- ruleEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.EndDate, time.Local)
- if tmpErr != nil {
- errMsg = "规则配置的截止日期异常,请重新填写"
- err = errors.New(errMsg)
- return
- }
- //1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值,9:动态环差,10:根据 给定终值后插值 规则获取预测数据,11:根据 季节性 规则获取预测数据,12:根据 移动平均同比 规则获取预测数据
- // 环比、环差、动态环差、季节性、移动平均同比不支持年度
- if sourceEdbInfo.Frequency == "年度" && utils.InArrayByInt([]int{5, 6, 11, 12}, v.RuleType) {
- errMsg = "环比、环差、动态环差、季节性、移动平均同比不支持年度指标"
- err = errors.New(errMsg)
- return
- }
- switch v.RuleType {
- case 8: //N期段线性外推值
- valInt, tmpErr := strconv.Atoi(v.Value)
- if tmpErr != nil {
- errMsg = "N期段线性外推值的N值异常"
- err = errors.New(errMsg)
- return
- }
- if valInt <= 1 {
- errMsg = "N期段线性外推值的N值必须大于1"
- err = errors.New(errMsg)
- return
- }
- case 9: //9:动态环差
- if v.Value == "" {
- errMsg = "请填写计算规则"
- err = errors.New(errMsg)
- return
- }
- formula := v.Value
- formula = strings.Replace(formula, "(", "(", -1)
- formula = strings.Replace(formula, ")", ")", -1)
- formula = strings.Replace(formula, ",", ",", -1)
- formula = strings.Replace(formula, "。", ".", -1)
- formula = strings.Replace(formula, "%", "*0.01", -1)
- v.Value = formula
- //检验公式
- var formulaStr string
- var edbInfoIdBytes []string
- for _, tmpEdbInfoId := range v.EdbInfoIdArr {
- formulaStr += tmpEdbInfoId.FromTag + ","
- edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
- }
- formulaSlice, tErr := utils.CheckFormulaJson(formula)
- if tErr != nil {
- errMsg = "公式格式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- for _, fm := range formulaSlice {
- formulaMap, e := utils.CheckFormula(fm)
- if e != nil {
- err = fmt.Errorf("公式错误,请重新填写")
- return
- }
- for _, f := range formulaMap {
- if !strings.Contains(formulaStr, f) {
- errMsg = "公式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- }
- }
- //关联的指标信息
- edbInfoList := make([]*models.EdbInfo, 0)
- // 动态环差规则 关系表
- trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
- for k, tmpEdbInfoId := range v.EdbInfoIdArr {
- fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- if err.Error() == utils.ErrNoRow() {
- errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
- err = errors.New(errMsg)
- return
- }
- errMsg = "获取指标失败:Err:" + err.Error()
- err = errors.New(errMsg)
- return
- }
- edbInfoList = append(edbInfoList, fromEdbInfo)
- //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
- {
- if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
- fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
- calculateMappingItem := &models.EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: edbInfoId,
- Source: utils.DATA_SOURCE_CALCULATE,
- SourceName: "指标运算",
- EdbCode: "",
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- //FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- }
- }
- // 动态环差规则 关系表
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
- //PredictEdbConfCalculateMappingId: 0,
- EdbInfoId: edbInfoId,
- ConfigId: 0,
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
- }
- for _, f := range formulaSlice {
- formulaMap, e := utils.CheckFormula(f)
- if e != nil {
- err = fmt.Errorf("公式错误,请重新填写")
- return
- }
- //预先计算,判断公式是否正常
- ok, _ := models.CheckFormula2(edbInfoList, formulaMap, f, edbInfoIdBytes)
- if !ok {
- errMsg = "生成计算指标失败,请使用正确的计算公式"
- err = errors.New(errMsg)
- return
- }
- }
- calculateRuleMap[ruleIndex] = models.CalculateRule{
- TrendsCalculateMappingList: trendsMappingList,
- EdbInfoList: edbInfoList,
- EdbInfoIdBytes: edbInfoIdBytes,
- Formula: formula,
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- case 14: //14:根据 一元线性拟合 规则获取预测数据
- if v.Value == "" {
- errMsg = "请填写一元线性拟合规则"
- err = errors.New(errMsg)
- return
- }
- //关联的指标信息
- edbInfoList := make([]*models.EdbInfo, 0)
- // 动态环差规则 关系表
- trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
- for k, tmpEdbInfoId := range v.EdbInfoIdArr {
- fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
- if tmpErr != nil {
- err = tmpErr
- if err.Error() == utils.ErrNoRow() {
- errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
- err = errors.New(errMsg)
- return
- }
- errMsg = "获取指标失败:Err:" + err.Error()
- err = errors.New(errMsg)
- return
- }
- edbInfoList = append(edbInfoList, fromEdbInfo)
- //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
- {
- if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
- fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
- tmpCalculateMappingItem := &models.EdbInfoCalculateMapping{
- EdbInfoCalculateMappingId: 0,
- EdbInfoId: 0,
- Source: utils.DATA_SOURCE_CALCULATE,
- SourceName: "指标运算",
- EdbCode: "",
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- //FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, tmpCalculateMappingItem)
- }
- }
- // 动态环差规则 关系表
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
- //PredictEdbConfCalculateMappingId: 0,
- EdbInfoId: 0,
- ConfigId: 0,
- FromEdbInfoId: fromEdbInfo.EdbInfoId,
- FromEdbCode: fromEdbInfo.EdbCode,
- FromEdbName: fromEdbInfo.EdbName,
- FromSource: fromEdbInfo.Source,
- FromSourceName: fromEdbInfo.SourceName,
- FromTag: tmpEdbInfoId.FromTag,
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
- }
- calculateRuleMap[ruleIndex] = models.CalculateRule{
- TrendsCalculateMappingList: trendsMappingList,
- EdbInfoList: edbInfoList,
- //EdbInfoIdBytes: edbInfoIdBytes,
- //Formula: formula,
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- }
- tmpPredictEdbConf := &models.PredictEdbConf{
- PredictEdbInfoId: edbInfoId,
- SourceEdbInfoId: sourceEdbInfo.EdbInfoId,
- RuleType: v.RuleType,
- //FixedValue: v.Value,
- Value: v.Value,
- EndDate: ruleEndDate,
- ModifyTime: time.Now(),
- CreateTime: time.Now(),
- }
- edbInfo.EndDate = v.EndDate
- predictEdbConfList = append(predictEdbConfList, tmpPredictEdbConf)
- }
- err, errMsg = models.EditPredictEdb(edbInfo, updateEdbInfoCol, calculateMappingList, predictEdbConfList, calculateRuleMap)
- return
- }
- // RefreshPredictEdbInfo 更新基础预测指标规则中的动态数据
- func RefreshPredictEdbInfo(edbInfoId int) (edbInfo *models.EdbInfo, err error, errMsg string) {
- // 指标信息校验
- {
- edbInfo, err = models.GetEdbInfoById(edbInfoId)
- if err != nil && err.Error() != utils.ErrNoRow() {
- errMsg = "刷新失败"
- err = errors.New("获取预测指标失败,Err:" + err.Error())
- return
- }
- if edbInfo == nil {
- errMsg = "找不到该预测指标"
- err = nil
- return
- }
- //必须是普通的指标
- if edbInfo.EdbInfoType != 1 {
- errMsg = "指标异常,不是预测指标"
- return
- }
- }
- // 配置 与 指标的 关联关系表
- list, err := models.GetPredictEdbConfCalculateMappingListByEdbInfoId(edbInfoId)
- if err != nil {
- return
- }
- // 没有关联指标,不需要刷新
- if len(list) <= 0 {
- return
- }
- // 配置关联的指标信息
- predictEdbConfCalculateMappingListMap := make(map[int][]*models.PredictEdbConfCalculateMapping)
- configIdList := make([]int, 0) //关联配置id
- edbInfoIdList := make([]int, 0) //关联指标配置id
- edbInfoIdMap := make(map[int]int, 0) //关联指标配置map
- for _, v := range list {
- configList, ok := predictEdbConfCalculateMappingListMap[v.ConfigId]
- if !ok {
- configList = make([]*models.PredictEdbConfCalculateMapping, 0)
- configIdList = append(configIdList, v.ConfigId)
- }
- if _, ok := edbInfoIdMap[v.FromEdbInfoId]; !ok {
- edbInfoIdList = append(edbInfoIdList, v.FromEdbInfoId)
- }
- configList = append(configList, v)
- predictEdbConfCalculateMappingListMap[v.ConfigId] = configList
- }
- predictEdbConfList, err := models.GetPredictEdbConfListByConfigIdList(configIdList)
- if err != nil {
- errMsg = "刷新失败"
- err = errors.New("获取预测指标配置信息失败,Err:" + err.Error())
- return
- }
- if len(predictEdbConfList) == 0 {
- errMsg = "找不到该预测指标配置"
- err = nil
- return
- }
- // 指标信息
- edbInfoList, err := models.GetEdbInfoByIdList(edbInfoIdList)
- if err != nil {
- err = errors.New("获取关联指标失败,Err:" + err.Error())
- return
- }
- // 指标信息map
- edbInfoListMap := make(map[int]*models.EdbInfo)
- for _, v := range edbInfoList {
- edbInfoListMap[v.EdbInfoId] = v
- }
- predictEdbConfAndDataList := make([]*models.PredictEdbConfAndData, 0)
- // 刷新所有的规则
- for _, v := range predictEdbConfList {
- // 每次规则计算的时候,产生的临时数据
- resultDataList := make([]*models.EdbInfoSearchData, 0)
- switch v.RuleType {
- case 9: //动态环差值
- if v.Value == "" {
- errMsg = "请填写计算规则"
- return
- }
- // todo 动态环差的空值类型处理
- formula := v.Value
- // 动态环差规则 关系表
- trendsMappingList := predictEdbConfCalculateMappingListMap[v.ConfigId]
- // 关联标签
- edbInfoIdArr := make([]models.EdbInfoFromTag, 0)
- //关联的指标信息
- edbInfoList := make([]*models.EdbInfo, 0)
- for _, trendsMapping := range trendsMappingList {
- tmpEdbInfo, ok := edbInfoListMap[trendsMapping.FromEdbInfoId]
- if ok {
- edbInfoList = append(edbInfoList, tmpEdbInfo)
- }
- // 关联标签
- edbInfoIdArr = append(edbInfoIdArr, models.EdbInfoFromTag{
- EdbInfoId: trendsMapping.FromEdbInfoId,
- FromTag: trendsMapping.FromTag,
- })
- }
- //检验公式
- var formulaStr string
- var edbInfoIdBytes []string
- for _, tmpEdbInfoId := range edbInfoIdArr {
- formulaStr += tmpEdbInfoId.FromTag + ","
- edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
- }
- formulaSlice, tErr := utils.CheckFormulaJson(formula)
- if tErr != nil {
- errMsg = "公式格式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- for _, fm := range formulaSlice {
- formulaMap, e := utils.CheckFormula(fm)
- if e != nil {
- err = fmt.Errorf("公式错误,请重新填写")
- return
- }
- for _, f := range formulaMap {
- if !strings.Contains(formulaStr, f) {
- errMsg = "公式错误,请重新填写"
- err = errors.New(errMsg)
- return
- }
- }
- //预先计算,判断公式是否正常
- ok, _ := models.CheckFormula2(edbInfoList, formulaMap, fm, edbInfoIdBytes)
- if !ok {
- errMsg = "生成计算指标失败,请使用正确的计算公式"
- return
- }
- }
- rule := models.CalculateRule{
- EdbInfoId: v.PredictEdbInfoId,
- ConfigId: v.ConfigId,
- TrendsCalculateMappingList: trendsMappingList,
- EdbInfoList: edbInfoList,
- EdbInfoIdBytes: edbInfoIdBytes,
- Formula: formula,
- RuleType: v.RuleType,
- EndDate: v.EndDate.Format(utils.FormatDate),
- EdbInfoIdArr: edbInfoIdArr,
- }
- resultDataList, err = models.RefreshCalculateByRuleBy9(rule)
- if err != nil {
- return
- }
- case 14: //14:根据 一元线性拟合 规则获取预测数据
- if v.Value == "" {
- errMsg = "一元线性拟合规则信息未配置"
- return
- }
- err, errMsg = models.RefreshCalculateByRuleByLineNh(*edbInfo, predictEdbConfAndDataList, *v)
- if err != nil {
- return
- }
- }
- // 规则配置(含数据)
- tmpPredictEdbConfAndData := &models.PredictEdbConfAndData{
- ConfigId: 0,
- PredictEdbInfoId: 0,
- SourceEdbInfoId: v.SourceEdbInfoId,
- RuleType: v.RuleType,
- FixedValue: v.FixedValue,
- Value: v.Value,
- EndDate: v.EndDate,
- ModifyTime: v.ModifyTime,
- CreateTime: v.CreateTime,
- DataList: resultDataList,
- }
- predictEdbConfAndDataList = append(predictEdbConfAndDataList, tmpPredictEdbConfAndData)
- }
- return
- }
|