predict_edb.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725
  1. package logic
  2. import (
  3. "errors"
  4. "hongze/hongze_edb_lib/models"
  5. "hongze/hongze_edb_lib/services"
  6. "hongze/hongze_edb_lib/utils"
  7. "strconv"
  8. "strings"
  9. "time"
  10. )
  11. // AddPredictEdbInfo 新增预测指标
  12. func AddPredictEdbInfo(sourceEdbInfoId, classifyId int, edbName string, ruleList []models.RuleConfig, sysUserId int, sysUserName string) (edbInfo *models.EdbInfo, err error, errMsg string) {
  13. var sourceEdbInfo *models.EdbInfo
  14. // 来源指标信息校验
  15. {
  16. sourceEdbInfo, err = models.GetEdbInfoById(sourceEdbInfoId)
  17. if err != nil && err.Error() != utils.ErrNoRow() {
  18. errMsg = "新增失败"
  19. err = errors.New("获取来源指标失败,Err:" + err.Error())
  20. return
  21. }
  22. if sourceEdbInfo == nil {
  23. errMsg = "找不到该来源指标"
  24. err = errors.New(errMsg)
  25. return
  26. }
  27. //必须是普通的指标
  28. if sourceEdbInfo.EdbInfoType != 0 {
  29. errMsg = "来源指标异常,不是普通的指标"
  30. err = errors.New(errMsg)
  31. return
  32. }
  33. if !utils.InArrayByStr([]string{"日度", "周度", "月度"}, sourceEdbInfo.Frequency) {
  34. errMsg = "预测指标只支持选择日度、周度、月度的指标"
  35. err = errors.New(errMsg)
  36. return
  37. }
  38. }
  39. var classifyInfo *models.EdbClassify
  40. // 来源分类信息校验
  41. {
  42. classifyInfo, err = models.GetEdbClassifyById(classifyId)
  43. if err != nil && err.Error() != utils.ErrNoRow() {
  44. errMsg = "新增失败"
  45. err = errors.New("获取预测指标分类失败,Err:" + err.Error())
  46. return
  47. }
  48. if classifyInfo == nil {
  49. errMsg = "找不到该预测指标分类"
  50. err = errors.New(errMsg)
  51. return
  52. }
  53. //必须是预测指标分类
  54. if classifyInfo.ClassifyType != 1 {
  55. errMsg = "预测指标分类异常,不是预测指标分类"
  56. err = errors.New(errMsg)
  57. return
  58. }
  59. }
  60. edbName = strings.Trim(edbName, " ")
  61. edbCode := sourceEdbInfo.EdbCode + "_" + time.Now().Format(utils.FormatShortDateTimeUnSpace)
  62. //判断指标名称是否存在
  63. var condition string
  64. var pars []interface{}
  65. condition += " AND edb_info_type=? "
  66. pars = append(pars, 1)
  67. condition += " AND edb_name=? "
  68. pars = append(pars, edbName)
  69. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  70. if err != nil {
  71. errMsg = "判断指标名称是否存在失败"
  72. err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
  73. return
  74. }
  75. if count > 0 {
  76. errMsg = "指标名称已存在,请重新填写"
  77. err = errors.New(errMsg)
  78. return
  79. }
  80. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  81. edbInfo = &models.EdbInfo{
  82. //EdbInfoId: 0,
  83. EdbInfoType: 1,
  84. SourceName: "预测指标",
  85. Source: utils.DATA_SOURCE_PREDICT,
  86. EdbCode: edbCode,
  87. EdbName: edbName,
  88. EdbNameSource: edbName,
  89. Frequency: sourceEdbInfo.Frequency,
  90. Unit: sourceEdbInfo.Unit,
  91. StartDate: sourceEdbInfo.StartDate,
  92. ClassifyId: classifyId,
  93. SysUserId: sysUserId,
  94. SysUserRealName: sysUserName,
  95. UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
  96. CreateTime: time.Now(),
  97. ModifyTime: time.Now(),
  98. MinValue: sourceEdbInfo.MinValue,
  99. MaxValue: sourceEdbInfo.MaxValue,
  100. CalculateFormula: sourceEdbInfo.CalculateFormula,
  101. EdbType: 1,
  102. //Sort: sourceEdbInfo.,
  103. LatestDate: sourceEdbInfo.LatestDate,
  104. LatestValue: sourceEdbInfo.LatestValue,
  105. MoveType: sourceEdbInfo.MoveType,
  106. MoveFrequency: sourceEdbInfo.MoveFrequency,
  107. NoUpdate: sourceEdbInfo.NoUpdate,
  108. ServerUrl: "",
  109. }
  110. // 关联关系表
  111. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  112. fromEdbMap := make(map[int]int)
  113. // 源指标关联关系表
  114. calculateMappingItem := &models.EdbInfoCalculateMapping{
  115. //EdbInfoCalculateMappingId: 0,
  116. //EdbInfoId: 0,
  117. Source: edbInfo.Source,
  118. SourceName: edbInfo.SourceName,
  119. EdbCode: edbInfo.EdbCode,
  120. FromEdbInfoId: sourceEdbInfo.EdbInfoId,
  121. FromEdbCode: sourceEdbInfo.EdbCode,
  122. FromEdbName: sourceEdbInfo.EdbName,
  123. FromSource: sourceEdbInfo.Source,
  124. FromSourceName: sourceEdbInfo.SourceName,
  125. //FromTag: "",
  126. Sort: 1,
  127. CreateTime: time.Now(),
  128. ModifyTime: time.Now(),
  129. }
  130. fromEdbMap[sourceEdbInfoId] = sourceEdbInfoId
  131. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  132. // 动态环差 计算列表
  133. calculateRuleList := make([]models.CalculateRule, 0)
  134. // 预测指标配置
  135. predictEdbConfList := make([]*models.PredictEdbConf, 0)
  136. for _, v := range ruleList {
  137. // 预测指标配置
  138. ruleEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.EndDate, time.Local)
  139. if tmpErr != nil {
  140. errMsg = "规则配置的截止日期异常,请重新填写"
  141. err = errors.New(errMsg)
  142. return
  143. }
  144. switch v.RuleType {
  145. case 8: //N期段线性外推值
  146. valInt, tmpErr := strconv.Atoi(v.Value)
  147. if tmpErr != nil {
  148. errMsg = "N期段线性外推值的N值异常"
  149. err = errors.New(errMsg)
  150. return
  151. }
  152. if valInt <= 1 {
  153. errMsg = "N期段线性外推值的N值必须大于1"
  154. err = errors.New(errMsg)
  155. return
  156. }
  157. case 9: //9:动态环差
  158. if v.Value == "" {
  159. errMsg = "请填写计算规则"
  160. err = errors.New(errMsg)
  161. return
  162. }
  163. formula := v.Value
  164. formula = strings.Replace(formula, "(", "(", -1)
  165. formula = strings.Replace(formula, ")", ")", -1)
  166. formula = strings.Replace(formula, ",", ",", -1)
  167. formula = strings.Replace(formula, "。", ".", -1)
  168. formula = strings.Replace(formula, "%", "*0.01", -1)
  169. v.Value = formula
  170. //检验公式
  171. var formulaStr string
  172. var edbInfoIdBytes []string
  173. for _, tmpEdbInfoId := range v.EdbInfoIdArr {
  174. formulaStr += tmpEdbInfoId.FromTag + ","
  175. edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
  176. }
  177. formulaMap := services.CheckFormula(formula)
  178. for _, formula := range formulaMap {
  179. if !strings.Contains(formulaStr, formula) {
  180. errMsg = "公式错误,请重新填写"
  181. err = errors.New(errMsg)
  182. return
  183. }
  184. }
  185. //关联的指标信息
  186. edbInfoList := make([]*models.EdbInfo, 0)
  187. // 动态环差规则 关系表
  188. trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
  189. for k, tmpEdbInfoId := range v.EdbInfoIdArr {
  190. fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
  191. if tmpErr != nil {
  192. err = tmpErr
  193. if err.Error() == utils.ErrNoRow() {
  194. errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
  195. err = errors.New(errMsg)
  196. return
  197. }
  198. errMsg = "获取指标失败:Err:" + err.Error()
  199. err = errors.New(errMsg)
  200. return
  201. }
  202. edbInfoList = append(edbInfoList, fromEdbInfo)
  203. //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
  204. {
  205. if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
  206. fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
  207. calculateMappingItem := &models.EdbInfoCalculateMapping{
  208. EdbInfoCalculateMappingId: 0,
  209. EdbInfoId: 0,
  210. Source: utils.DATA_SOURCE_CALCULATE,
  211. SourceName: "指标运算",
  212. EdbCode: "",
  213. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  214. FromEdbCode: fromEdbInfo.EdbCode,
  215. FromEdbName: fromEdbInfo.EdbName,
  216. FromSource: fromEdbInfo.Source,
  217. FromSourceName: fromEdbInfo.SourceName,
  218. //FromTag: tmpEdbInfoId.FromTag,
  219. Sort: k + 1,
  220. CreateTime: time.Now(),
  221. ModifyTime: time.Now(),
  222. }
  223. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  224. }
  225. }
  226. // 动态环差规则 关系表
  227. tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
  228. //PredictEdbConfCalculateMappingId: 0,
  229. EdbInfoId: 0,
  230. ConfigId: 0,
  231. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  232. FromEdbCode: fromEdbInfo.EdbCode,
  233. FromEdbName: fromEdbInfo.EdbName,
  234. FromSource: fromEdbInfo.Source,
  235. FromSourceName: fromEdbInfo.SourceName,
  236. FromTag: tmpEdbInfoId.FromTag,
  237. Sort: k + 1,
  238. CreateTime: time.Now(),
  239. ModifyTime: time.Now(),
  240. }
  241. trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
  242. }
  243. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, formula, edbInfoIdBytes)
  244. if !ok {
  245. errMsg = "生成计算指标失败,请使用正确的计算公式"
  246. err = errors.New(errMsg)
  247. return
  248. }
  249. calculateRuleList = append(calculateRuleList, models.CalculateRule{
  250. TrendsCalculateMappingList: trendsMappingList,
  251. EdbInfoList: edbInfoList,
  252. EdbInfoIdBytes: edbInfoIdBytes,
  253. Formula: formula,
  254. RuleType: v.RuleType,
  255. EndDate: v.EndDate,
  256. EdbInfoIdArr: v.EdbInfoIdArr,
  257. })
  258. }
  259. tmpPredictEdbConf := &models.PredictEdbConf{
  260. PredictEdbInfoId: 0,
  261. SourceEdbInfoId: sourceEdbInfoId,
  262. RuleType: v.RuleType,
  263. //FixedValue: v.Value,
  264. Value: v.Value,
  265. EndDate: ruleEndDate,
  266. ModifyTime: time.Now(),
  267. CreateTime: time.Now(),
  268. }
  269. edbInfo.EndDate = v.EndDate
  270. predictEdbConfList = append(predictEdbConfList, tmpPredictEdbConf)
  271. }
  272. err = models.AddPredictEdb(edbInfo, calculateMappingList, predictEdbConfList, calculateRuleList)
  273. return
  274. }
  275. // EditPredictEdbInfo 编辑预测指标
  276. func EditPredictEdbInfo(edbInfoId, classifyId int, edbName string, ruleList []models.RuleConfig) (edbInfo *models.EdbInfo, err error, errMsg string) {
  277. // 指标信息校验
  278. {
  279. edbInfo, err = models.GetEdbInfoById(edbInfoId)
  280. if err != nil && err.Error() != utils.ErrNoRow() {
  281. errMsg = "修改失败"
  282. err = errors.New("获取预测指标失败,Err:" + err.Error())
  283. return
  284. }
  285. if edbInfo == nil {
  286. errMsg = "找不到该预测指标"
  287. err = errors.New(errMsg)
  288. return
  289. }
  290. //必须是普通的指标
  291. if edbInfo.EdbInfoType != 1 {
  292. errMsg = "指标异常,不是预测指标"
  293. err = errors.New(errMsg)
  294. return
  295. }
  296. }
  297. var predictEdbConf *models.PredictEdbConf
  298. // 指标配置信息校验
  299. {
  300. // 查找该预测指标配置
  301. predictEdbConfList, tmpErr := models.GetPredictEdbConfListById(edbInfo.EdbInfoId)
  302. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  303. errMsg = "修改失败"
  304. err = errors.New("获取预测指标配置信息失败,Err:" + tmpErr.Error())
  305. return
  306. }
  307. if len(predictEdbConfList) == 0 {
  308. errMsg = "找不到该预测指标配置"
  309. err = errors.New(errMsg)
  310. return
  311. }
  312. predictEdbConf = predictEdbConfList[0]
  313. }
  314. //判断指标名称是否存在
  315. var condition string
  316. var pars []interface{}
  317. condition += " AND edb_info_id<>? "
  318. pars = append(pars, edbInfoId)
  319. condition += " AND edb_info_type=? "
  320. pars = append(pars, 1)
  321. condition += " AND edb_name=? "
  322. pars = append(pars, edbName)
  323. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  324. if err != nil {
  325. errMsg = "判断指标名称是否存在失败"
  326. err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
  327. return
  328. }
  329. if count > 0 {
  330. errMsg = "指标名称已存在,请重新填写"
  331. err = errors.New(errMsg)
  332. return
  333. }
  334. edbInfo.EdbName = edbName
  335. edbInfo.EdbNameSource = edbName
  336. edbInfo.ClassifyId = classifyId
  337. edbInfo.ModifyTime = time.Now()
  338. updateEdbInfoCol := []string{"EdbName", "EdbNameSource", "ClassifyId", "EndDate", "ModifyTime"}
  339. var sourceEdbInfo *models.EdbInfo
  340. // 来源指标信息校验
  341. {
  342. sourceEdbInfo, err = models.GetEdbInfoById(predictEdbConf.SourceEdbInfoId)
  343. if err != nil && err.Error() != utils.ErrNoRow() {
  344. errMsg = "新增失败"
  345. err = errors.New("获取来源指标失败,Err:" + err.Error())
  346. return
  347. }
  348. if sourceEdbInfo == nil {
  349. errMsg = "找不到该来源指标"
  350. err = errors.New(errMsg)
  351. return
  352. }
  353. //必须是普通的指标
  354. if sourceEdbInfo.EdbInfoType != 0 {
  355. errMsg = "来源指标异常,不是普通的指标"
  356. err = errors.New(errMsg)
  357. return
  358. }
  359. if !utils.InArrayByStr([]string{"日度", "周度", "月度"}, sourceEdbInfo.Frequency) {
  360. errMsg = "预测指标只支持选择日度、周度、月度的指标"
  361. err = errors.New(errMsg)
  362. return
  363. }
  364. }
  365. // 预测指标配置
  366. // 关联关系表
  367. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  368. fromEdbMap := make(map[int]int)
  369. // 源指标关联关系表
  370. calculateMappingItem := &models.EdbInfoCalculateMapping{
  371. //EdbInfoCalculateMappingId: 0,
  372. EdbInfoId: edbInfoId,
  373. Source: edbInfo.Source,
  374. SourceName: edbInfo.SourceName,
  375. EdbCode: edbInfo.EdbCode,
  376. FromEdbInfoId: sourceEdbInfo.EdbInfoId,
  377. FromEdbCode: sourceEdbInfo.EdbCode,
  378. FromEdbName: sourceEdbInfo.EdbName,
  379. FromSource: sourceEdbInfo.Source,
  380. FromSourceName: sourceEdbInfo.SourceName,
  381. //FromTag: "",
  382. Sort: 1,
  383. CreateTime: time.Now(),
  384. ModifyTime: time.Now(),
  385. }
  386. fromEdbMap[sourceEdbInfo.EdbInfoId] = sourceEdbInfo.EdbInfoId
  387. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  388. // 动态环差 计算列表
  389. calculateRuleList := make([]models.CalculateRule, 0)
  390. // 预测指标配置
  391. predictEdbConfList := make([]*models.PredictEdbConf, 0)
  392. for _, v := range ruleList {
  393. // 预测指标配置
  394. ruleEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, v.EndDate, time.Local)
  395. if tmpErr != nil {
  396. errMsg = "规则配置的截止日期异常,请重新填写"
  397. err = errors.New(errMsg)
  398. return
  399. }
  400. switch v.RuleType {
  401. case 8: //N期段线性外推值
  402. valInt, tmpErr := strconv.Atoi(v.Value)
  403. if tmpErr != nil {
  404. errMsg = "N期段线性外推值的N值异常"
  405. err = errors.New(errMsg)
  406. return
  407. }
  408. if valInt <= 1 {
  409. errMsg = "N期段线性外推值的N值必须大于1"
  410. err = errors.New(errMsg)
  411. return
  412. }
  413. case 9: //9:动态环差
  414. if v.Value == "" {
  415. errMsg = "请填写计算规则"
  416. err = errors.New(errMsg)
  417. return
  418. }
  419. formula := v.Value
  420. formula = strings.Replace(formula, "(", "(", -1)
  421. formula = strings.Replace(formula, ")", ")", -1)
  422. formula = strings.Replace(formula, ",", ",", -1)
  423. formula = strings.Replace(formula, "。", ".", -1)
  424. formula = strings.Replace(formula, "%", "*0.01", -1)
  425. v.Value = formula
  426. //检验公式
  427. var formulaStr string
  428. var edbInfoIdBytes []string
  429. for _, tmpEdbInfoId := range v.EdbInfoIdArr {
  430. formulaStr += tmpEdbInfoId.FromTag + ","
  431. edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
  432. }
  433. formulaMap := services.CheckFormula(formula)
  434. for _, formula := range formulaMap {
  435. if !strings.Contains(formulaStr, formula) {
  436. errMsg = "公式错误,请重新填写"
  437. err = errors.New(errMsg)
  438. return
  439. }
  440. }
  441. //关联的指标信息
  442. edbInfoList := make([]*models.EdbInfo, 0)
  443. // 动态环差规则 关系表
  444. trendsMappingList := make([]*models.PredictEdbConfCalculateMapping, 0)
  445. for k, tmpEdbInfoId := range v.EdbInfoIdArr {
  446. fromEdbInfo, tmpErr := models.GetEdbInfoById(tmpEdbInfoId.EdbInfoId)
  447. if tmpErr != nil {
  448. err = tmpErr
  449. if err.Error() == utils.ErrNoRow() {
  450. errMsg = "指标 " + strconv.Itoa(tmpEdbInfoId.EdbInfoId) + " 不存在"
  451. err = errors.New(errMsg)
  452. return
  453. }
  454. errMsg = "获取指标失败:Err:" + err.Error()
  455. err = errors.New(errMsg)
  456. return
  457. }
  458. edbInfoList = append(edbInfoList, fromEdbInfo)
  459. //总的 预测指标与所有相关联指标的关系表(不仅仅该条规则)
  460. {
  461. if _, ok := fromEdbMap[tmpEdbInfoId.EdbInfoId]; !ok {
  462. fromEdbMap[tmpEdbInfoId.EdbInfoId] = tmpEdbInfoId.EdbInfoId
  463. calculateMappingItem := &models.EdbInfoCalculateMapping{
  464. EdbInfoCalculateMappingId: 0,
  465. EdbInfoId: edbInfoId,
  466. Source: utils.DATA_SOURCE_CALCULATE,
  467. SourceName: "指标运算",
  468. EdbCode: "",
  469. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  470. FromEdbCode: fromEdbInfo.EdbCode,
  471. FromEdbName: fromEdbInfo.EdbName,
  472. FromSource: fromEdbInfo.Source,
  473. FromSourceName: fromEdbInfo.SourceName,
  474. //FromTag: tmpEdbInfoId.FromTag,
  475. Sort: k + 1,
  476. CreateTime: time.Now(),
  477. ModifyTime: time.Now(),
  478. }
  479. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  480. }
  481. }
  482. // 动态环差规则 关系表
  483. tmpPredictEdbConfCalculateMapping := &models.PredictEdbConfCalculateMapping{
  484. //PredictEdbConfCalculateMappingId: 0,
  485. EdbInfoId: edbInfoId,
  486. ConfigId: 0,
  487. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  488. FromEdbCode: fromEdbInfo.EdbCode,
  489. FromEdbName: fromEdbInfo.EdbName,
  490. FromSource: fromEdbInfo.Source,
  491. FromSourceName: fromEdbInfo.SourceName,
  492. FromTag: tmpEdbInfoId.FromTag,
  493. Sort: k + 1,
  494. CreateTime: time.Now(),
  495. ModifyTime: time.Now(),
  496. }
  497. trendsMappingList = append(trendsMappingList, tmpPredictEdbConfCalculateMapping)
  498. }
  499. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, formula, edbInfoIdBytes)
  500. if !ok {
  501. errMsg = "生成计算指标失败,请使用正确的计算公式"
  502. err = errors.New(errMsg)
  503. return
  504. }
  505. calculateRuleList = append(calculateRuleList, models.CalculateRule{
  506. TrendsCalculateMappingList: trendsMappingList,
  507. EdbInfoList: edbInfoList,
  508. EdbInfoIdBytes: edbInfoIdBytes,
  509. Formula: formula,
  510. RuleType: v.RuleType,
  511. EndDate: v.EndDate,
  512. EdbInfoIdArr: v.EdbInfoIdArr,
  513. })
  514. }
  515. tmpPredictEdbConf := &models.PredictEdbConf{
  516. PredictEdbInfoId: edbInfoId,
  517. SourceEdbInfoId: sourceEdbInfo.EdbInfoId,
  518. RuleType: v.RuleType,
  519. //FixedValue: v.Value,
  520. Value: v.Value,
  521. EndDate: ruleEndDate,
  522. ModifyTime: time.Now(),
  523. CreateTime: time.Now(),
  524. }
  525. edbInfo.EndDate = v.EndDate
  526. predictEdbConfList = append(predictEdbConfList, tmpPredictEdbConf)
  527. }
  528. err = models.EditPredictEdb(edbInfo, updateEdbInfoCol, calculateMappingList, predictEdbConfList, calculateRuleList)
  529. return
  530. }
  531. // RefreshPredictEdbInfo 更新基础预测指标规则中的动态数据
  532. func RefreshPredictEdbInfo(edbInfoId int) (edbInfo *models.EdbInfo, err error, errMsg string) {
  533. // 指标信息校验
  534. {
  535. edbInfo, err = models.GetEdbInfoById(edbInfoId)
  536. if err != nil && err.Error() != utils.ErrNoRow() {
  537. errMsg = "刷新失败"
  538. err = errors.New("获取预测指标失败,Err:" + err.Error())
  539. return
  540. }
  541. if edbInfo == nil {
  542. errMsg = "找不到该预测指标"
  543. err = nil
  544. return
  545. }
  546. //必须是普通的指标
  547. if edbInfo.EdbInfoType != 1 {
  548. errMsg = "指标异常,不是预测指标"
  549. return
  550. }
  551. }
  552. // 配置 与 指标的 关联关系表
  553. list, err := models.GetPredictEdbConfCalculateMappingListByEdbInfoId(edbInfoId)
  554. if err != nil {
  555. return
  556. }
  557. // 没有关联指标,不需要刷新
  558. if len(list) <= 0 {
  559. return
  560. }
  561. // 配置关联的指标信息
  562. predictEdbConfCalculateMappingListMap := make(map[int][]*models.PredictEdbConfCalculateMapping)
  563. configIdList := make([]int, 0) //关联配置id
  564. edbInfoIdList := make([]int, 0) //关联指标配置id
  565. edbInfoIdMap := make(map[int]int, 0) //关联指标配置map
  566. for _, v := range list {
  567. configList, ok := predictEdbConfCalculateMappingListMap[v.ConfigId]
  568. if !ok {
  569. configList = make([]*models.PredictEdbConfCalculateMapping, 0)
  570. configIdList = append(configIdList, v.ConfigId)
  571. }
  572. if _, ok := edbInfoIdMap[v.FromEdbInfoId]; !ok {
  573. edbInfoIdList = append(edbInfoIdList, v.FromEdbInfoId)
  574. }
  575. configList = append(configList, v)
  576. predictEdbConfCalculateMappingListMap[v.ConfigId] = configList
  577. }
  578. predictEdbConfList, err := models.GetPredictEdbConfListByConfigIdList(configIdList)
  579. if err != nil {
  580. errMsg = "刷新失败"
  581. err = errors.New("获取预测指标配置信息失败,Err:" + err.Error())
  582. return
  583. }
  584. if len(predictEdbConfList) == 0 {
  585. errMsg = "找不到该预测指标配置"
  586. err = nil
  587. return
  588. }
  589. // 指标信息
  590. edbInfoList, err := models.GetEdbInfoByIdList(edbInfoIdList)
  591. if err != nil {
  592. err = errors.New("获取关联指标失败,Err:" + err.Error())
  593. return
  594. }
  595. // 指标信息map
  596. edbInfoListMap := make(map[int]*models.EdbInfo)
  597. for _, v := range edbInfoList {
  598. edbInfoListMap[v.EdbInfoId] = v
  599. }
  600. // 刷新所有的规则
  601. for _, v := range predictEdbConfList {
  602. switch v.RuleType {
  603. case 9:
  604. if v.Value == "" {
  605. errMsg = "请填写计算规则"
  606. return
  607. }
  608. formula := v.Value
  609. // 动态环差规则 关系表
  610. trendsMappingList := predictEdbConfCalculateMappingListMap[v.ConfigId]
  611. // 关联标签
  612. edbInfoIdArr := make([]models.EdbInfoFromTag, 0)
  613. //关联的指标信息
  614. edbInfoList := make([]*models.EdbInfo, 0)
  615. for _, trendsMapping := range trendsMappingList {
  616. tmpEdbInfo, ok := edbInfoListMap[trendsMapping.FromEdbInfoId]
  617. if ok {
  618. edbInfoList = append(edbInfoList, tmpEdbInfo)
  619. }
  620. // 关联标签
  621. edbInfoIdArr = append(edbInfoIdArr, models.EdbInfoFromTag{
  622. EdbInfoId: trendsMapping.FromEdbInfoId,
  623. FromTag: trendsMapping.FromTag,
  624. })
  625. }
  626. //检验公式
  627. var formulaStr string
  628. var edbInfoIdBytes []string
  629. for _, tmpEdbInfoId := range edbInfoIdArr {
  630. formulaStr += tmpEdbInfoId.FromTag + ","
  631. edbInfoIdBytes = append(edbInfoIdBytes, tmpEdbInfoId.FromTag)
  632. }
  633. formulaMap := services.CheckFormula(formula)
  634. for _, formula := range formulaMap {
  635. if !strings.Contains(formulaStr, formula) {
  636. errMsg = "公式错误,请重新填写"
  637. return
  638. }
  639. }
  640. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, formula, edbInfoIdBytes)
  641. if !ok {
  642. errMsg = "生成计算指标失败,请使用正确的计算公式"
  643. return
  644. }
  645. rule := models.CalculateRule{
  646. EdbInfoId: v.PredictEdbInfoId,
  647. ConfigId: v.ConfigId,
  648. TrendsCalculateMappingList: trendsMappingList,
  649. EdbInfoList: edbInfoList,
  650. EdbInfoIdBytes: edbInfoIdBytes,
  651. Formula: formula,
  652. RuleType: v.RuleType,
  653. EndDate: v.EndDate.Format(utils.FormatDate),
  654. EdbInfoIdArr: edbInfoIdArr,
  655. }
  656. err = models.RefreshCalculateByRuleBy9(rule)
  657. if err != nil {
  658. return
  659. }
  660. }
  661. }
  662. return
  663. }