12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079 |
- package logic
- import (
- "errors"
- "eta_gn/eta_index_lib/models"
- "eta_gn/eta_index_lib/utils"
- "fmt"
- "strconv"
- "strings"
- "time"
- )
- func AddPredictEdbInfo(sourceEdbInfoId, classifyId int, edbName, dataDateType string, ruleList []models.RuleConfig, minValue, maxValue float64, sysUserId int, sysUserName, lang 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
- }
-
-
-
-
-
- }
- 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)
-
- existEdbName, err := CheckExistByEdbNameAndEdbInfoId(0, 0, edbName, lang)
- if err != nil {
- errMsg = "判断指标名称是否存在失败"
- err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
- return
- }
- if existEdbName {
- errMsg = "指标名称已存在,请重新填写"
- err = errors.New(errMsg)
- return
- }
- timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- if dataDateType == `` {
- dataDateType = `自然日`
- }
- edbInfo = &models.EdbInfo{
-
- 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,
-
- LatestDate: sourceEdbInfo.LatestDate,
- LatestValue: sourceEdbInfo.LatestValue,
- MoveType: sourceEdbInfo.MoveType,
- MoveFrequency: sourceEdbInfo.MoveFrequency,
- NoUpdate: sourceEdbInfo.NoUpdate,
- ServerUrl: "",
- EdbNameEn: edbName,
- UnitEn: sourceEdbInfo.UnitEn,
- DataDateType: dataDateType,
- Sort: models.GetAddEdbMaxSortByClassifyId(classifyId, utils.PREDICT_EDB_INFO_TYPE),
- }
-
- calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
- fromEdbMap := make(map[int]int)
-
- calculateMappingItem := &models.EdbInfoCalculateMapping{
-
-
- Source: edbInfo.Source,
- SourceName: edbInfo.SourceName,
- EdbCode: edbInfo.EdbCode,
- FromEdbInfoId: sourceEdbInfo.EdbInfoId,
- FromEdbCode: sourceEdbInfo.EdbCode,
- FromEdbName: sourceEdbInfo.EdbName,
- FromSource: sourceEdbInfo.Source,
- FromSourceName: sourceEdbInfo.SourceName,
-
- 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
- }
-
-
- if sourceEdbInfo.Frequency == "年度" && utils.InArrayByInt([]int{5, 6, 11, 12}, v.RuleType) {
- errMsg = "环比、环差、动态环差、季节性、移动平均同比不支持年度指标"
- err = errors.New(errMsg)
- return
- }
- switch v.RuleType {
- case 8:
- 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:
- 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,
-
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- }
- }
-
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
-
- 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:
- 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,
-
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, tmpCalculateMappingItem)
- }
- }
-
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
-
- 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,
-
-
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- }
- tmpPredictEdbConf := &models.PredictEdbConf{
- PredictEdbInfoId: 0,
- SourceEdbInfoId: sourceEdbInfoId,
- RuleType: v.RuleType,
-
- 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
- }
- func EditPredictEdbInfo(edbInfoId, classifyId int, edbName, dataDateType string, ruleList []models.RuleConfig, minValue, maxValue float64, lang string) (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]
- }
-
- existEdbName, err := CheckExistByEdbNameAndEdbInfoId(1, edbInfoId, edbName, lang)
- if err != nil {
- errMsg = "判断指标名称是否存在失败"
- err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
- return
- }
- if existEdbName {
- errMsg = "指标名称已存在,请重新填写"
- err = errors.New(errMsg)
- return
- }
- if dataDateType == `` {
- dataDateType = `自然日`
- }
- switch lang {
- case utils.EnLangVersion:
- edbInfo.EdbNameEn = edbName
- default:
- 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", "EdbNameEn", "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
- }
-
-
-
-
-
- }
-
-
- calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
- fromEdbMap := make(map[int]int)
-
- calculateMappingItem := &models.EdbInfoCalculateMapping{
-
- 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,
-
- 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
- }
-
-
- if sourceEdbInfo.Frequency == "年度" && utils.InArrayByInt([]int{5, 6, 11, 12}, v.RuleType) {
- errMsg = "环比、环差、动态环差、季节性、移动平均同比不支持年度指标"
- err = errors.New(errMsg)
- return
- }
- switch v.RuleType {
- case 8:
- 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:
- 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,
-
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, calculateMappingItem)
- }
- }
-
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
-
- 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:
- 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,
-
- Sort: k + 1,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- }
- calculateMappingList = append(calculateMappingList, tmpCalculateMappingItem)
- }
- }
-
- tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
-
- 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,
-
-
- RuleType: v.RuleType,
- EndDate: v.EndDate,
- EdbInfoIdArr: v.EdbInfoIdArr,
- }
- }
- tmpPredictEdbConf := &models.PredictEdbConf{
- PredictEdbInfoId: edbInfoId,
- SourceEdbInfoId: sourceEdbInfo.EdbInfoId,
- RuleType: v.RuleType,
-
- 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
- }
- 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)
- edbInfoIdList := make([]int, 0)
- edbInfoIdMap := make(map[int]int, 0)
- 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
- }
-
- 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
- }
-
- 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:
- 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
- }
- func checkExistByEdbName(edbInfoType int, edbName, lang string) (has bool, err error) {
- var condition string
- var pars []interface{}
- condition += " AND edb_info_type=? "
- pars = append(pars, 0)
- switch lang {
- case utils.EnLangVersion:
- condition += " AND edb_name_en = ? "
- default:
- condition += " AND edb_name=? "
- }
- pars = append(pars, edbName)
- count, err := models.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- return
- }
- if count > 0 {
- has = true
- return
- }
- return
- }
- func checkExistByEdbNameAndEdbInfoId(edbInfoType, edbInfoId int, edbName, lang string) (has bool, err error) {
- var condition string
- var pars []interface{}
- condition += " AND edb_info_type=? "
- pars = append(pars, edbInfoType)
- condition += " AND edb_info_id<>? "
- pars = append(pars, edbInfoId)
- switch lang {
- case utils.EnLangVersion:
- condition += " AND edb_name_en = ? "
- default:
- condition += " AND edb_name=? "
- }
- pars = append(pars, edbName)
- count, err := models.GetEdbInfoCountByCondition(condition, pars)
- if err != nil {
- return
- }
- if count > 0 {
- has = true
- return
- }
- return
- }
- func CheckExistByEdbNameAndEdbInfoId(edbInfoType, edbInfoId int, edbName, lang string) (has bool, err error) {
-
- if edbInfoId == 0 {
- return checkExistByEdbName(edbInfoType, edbName, lang)
- }
-
- return checkExistByEdbNameAndEdbInfoId(edbInfoType, edbInfoId, edbName, lang)
- }
|