base_from_predict_calculate.go 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "hongze/hongze_edb_lib/logic"
  6. "hongze/hongze_edb_lib/models"
  7. "hongze/hongze_edb_lib/services"
  8. "hongze/hongze_edb_lib/utils"
  9. "strconv"
  10. "strings"
  11. "time"
  12. )
  13. // PredictCalculateController 预测计算指标
  14. type PredictCalculateController struct {
  15. BaseAuthController
  16. }
  17. // Save
  18. // @Title 新增/编辑预测指标运算接口
  19. // @Description 新增预测指标运算接口
  20. // @Success 200 {object} models.EdbInfoCalculateSaveReq
  21. // @router /save [post]
  22. func (this *PredictCalculateController) Save() {
  23. br := new(models.BaseResponse).Init()
  24. defer func() {
  25. this.Data["json"] = br
  26. this.ServeJSON()
  27. }()
  28. var req models.EdbInfoCalculateSaveReq
  29. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  30. if err != nil {
  31. br.Msg = "参数解析异常!"
  32. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  33. return
  34. }
  35. if req.EdbInfoId <= 0 {
  36. addPredictCalculate(br, req)
  37. } else {
  38. editPredictCalculate(br, req)
  39. }
  40. }
  41. // AddPredictCalculate 添加预测指标运算
  42. func addPredictCalculate(br *models.BaseResponse, req models.EdbInfoCalculateSaveReq) {
  43. req.EdbName = strings.Trim(req.EdbName, " ")
  44. if req.EdbName == "" {
  45. br.Msg = "指标名称不能为空"
  46. return
  47. }
  48. if req.Frequency == "" {
  49. br.Msg = "频率不能为空"
  50. return
  51. }
  52. if req.Unit == "" {
  53. br.Msg = "单位不能为空"
  54. return
  55. }
  56. if req.ClassifyId <= 0 {
  57. br.Msg = "请选择分类"
  58. return
  59. }
  60. if len(req.EdbInfoIdArr) <= 0 {
  61. br.Msg = "请选择指标"
  62. return
  63. }
  64. if req.CalculateFormula == "" {
  65. br.Msg = "请填写指标"
  66. return
  67. }
  68. //加入缓存机制,避免创建同一个名称的指标 start
  69. redisKey := fmt.Sprint("predict_edb_info:calculate:batch:save:", utils.DATA_SOURCE_PREDICT_CALCULATE, ":", req.EdbName)
  70. isExist := utils.Rc.IsExist(redisKey)
  71. if isExist {
  72. br.Msg = "指标正在处理,请勿重复提交"
  73. return
  74. } else {
  75. //设置3分钟缓存
  76. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  77. defer func() {
  78. _ = utils.Rc.Delete(redisKey)
  79. }()
  80. }
  81. calculateFormula := req.CalculateFormula
  82. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  83. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  84. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  85. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  86. if strings.Contains(req.CalculateFormula, "%") {
  87. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  88. }
  89. req.CalculateFormula = calculateFormula
  90. //检验公式
  91. var formulaStr string
  92. var edbInfoIdBytes []string
  93. for _, v := range req.EdbInfoIdArr {
  94. formulaStr += v.FromTag + ","
  95. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  96. }
  97. formulaMap := services.CheckFormula(req.CalculateFormula)
  98. for _, v := range formulaMap {
  99. if !strings.Contains(formulaStr, v) {
  100. br.Msg = "公式错误,请重新填写"
  101. return
  102. }
  103. }
  104. // 判断指标名称是否存在
  105. {
  106. var condition string
  107. var pars []interface{}
  108. // 指标类型,0:普通指标,1:预测指标
  109. condition += " AND edb_info_type=? AND edb_name=? "
  110. pars = append(pars, 1, req.EdbName)
  111. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  112. if err != nil {
  113. br.Msg = "判断指标名称是否存在失败"
  114. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  115. return
  116. }
  117. if count > 0 {
  118. br.Msg = "指标名称已存在,请重新填写"
  119. br.ErrMsg = "指标名称已存在,请重新填写"
  120. return
  121. }
  122. }
  123. // 关联指标信息
  124. edbInfoList := make([]*models.EdbInfo, 0)
  125. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  126. for k, v := range req.EdbInfoIdArr {
  127. fromEdbInfo, err := models.GetEdbInfoById(v.EdbInfoId)
  128. if err != nil {
  129. if err.Error() == utils.ErrNoRow() {
  130. br.Msg = "生成计算指标失败"
  131. br.Msg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  132. return
  133. }
  134. br.Msg = "生成计算指标失败"
  135. br.Msg = "获取指标失败:Err:" + err.Error()
  136. return
  137. }
  138. edbInfoList = append(edbInfoList, fromEdbInfo)
  139. //关联关系表
  140. {
  141. calculateMappingItem := &models.EdbInfoCalculateMapping{
  142. CreateTime: time.Now(),
  143. ModifyTime: time.Now(),
  144. Sort: k + 1,
  145. //EdbCode: edbCode,
  146. //EdbInfoId: int(edbInfoId),
  147. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  148. FromEdbCode: fromEdbInfo.EdbCode,
  149. FromEdbName: fromEdbInfo.EdbName,
  150. FromSource: fromEdbInfo.Source,
  151. FromSourceName: fromEdbInfo.SourceName,
  152. FromTag: v.FromTag,
  153. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  154. SourceName: "预测指标运算",
  155. }
  156. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  157. }
  158. }
  159. //预先计算,判断公式是否正常
  160. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, calculateFormula, edbInfoIdBytes)
  161. if !ok {
  162. br.Msg = "生成计算指标失败,请使用正确的计算公式"
  163. return
  164. }
  165. // 指标入库
  166. randStr := utils.GetRandDigit(4)
  167. edbCode := `C2` + time.Now().Format("060102") + randStr
  168. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  169. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  170. edbInfo := &models.EdbInfo{
  171. EdbInfoType: 1,
  172. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  173. SourceName: "预测指标运算",
  174. EdbCode: edbCode,
  175. EdbName: req.EdbName,
  176. EdbNameSource: req.EdbName,
  177. Frequency: req.Frequency,
  178. Unit: req.Unit,
  179. ClassifyId: req.ClassifyId,
  180. SysUserId: req.AdminId,
  181. SysUserRealName: req.AdminName,
  182. CreateTime: time.Now(),
  183. ModifyTime: time.Now(),
  184. UniqueCode: uniqueCode,
  185. CalculateFormula: req.CalculateFormula,
  186. EdbType: 2,
  187. }
  188. edbInfoId, err := models.AddEdbInfo(edbInfo)
  189. if err != nil {
  190. br.Msg = "生成计算指标失败"
  191. br.Msg = "生成计算指标失败,AddEdbInfo Err:" + err.Error()
  192. return
  193. }
  194. edbInfo.EdbInfoId = int(edbInfoId)
  195. //处理同名指标
  196. {
  197. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  198. if err != nil {
  199. br.Msg = "保存失败"
  200. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  201. return
  202. }
  203. if len(edbNameList) >= 2 {
  204. for _, v := range edbNameList {
  205. edbName := v.EdbName + "(" + v.SourceName + ")"
  206. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  207. if err != nil {
  208. br.Msg = "保存失败"
  209. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  210. return
  211. }
  212. }
  213. }
  214. }
  215. //关联关系表入库
  216. {
  217. if len(calculateMappingList) > 0 {
  218. for k, v := range calculateMappingList {
  219. v.EdbCode = edbInfo.EdbCode
  220. v.EdbInfoId = edbInfo.EdbInfoId
  221. calculateMappingList[k] = v
  222. }
  223. go models.AddEdbInfoCalculateMappingMulti(calculateMappingList)
  224. }
  225. }
  226. // 开始添加预测指标
  227. latestDateStr, latestValue, err := models.AddPredictCalculateData(edbInfoList, edbInfo, edbCode, req.CalculateFormula, edbInfoIdBytes)
  228. if err != nil {
  229. br.Msg = "生成计算指标失败"
  230. br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
  231. return
  232. }
  233. // 更新指标最大最小值
  234. err, errMsg := models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  235. if err != nil {
  236. br.Msg = errMsg
  237. br.ErrMsg = err.Error()
  238. return
  239. }
  240. resp := models.AddEdbInfoResp{
  241. EdbInfoId: int(edbInfoId),
  242. UniqueCode: uniqueCode,
  243. }
  244. br.Ret = 200
  245. br.Success = true
  246. br.Msg = "保存成功"
  247. br.Data = resp
  248. br.IsAddLog = true
  249. }
  250. // editPredictCalculate 编辑预测指标运算
  251. func editPredictCalculate(br *models.BaseResponse, req models.EdbInfoCalculateSaveReq) {
  252. req.EdbName = strings.Trim(req.EdbName, " ")
  253. if req.EdbName == "" {
  254. br.Msg = "指标名称不能为空"
  255. return
  256. }
  257. if req.Frequency == "" {
  258. br.Msg = "频率不能为空"
  259. return
  260. }
  261. if req.Unit == "" {
  262. br.Msg = "单位不能为空"
  263. return
  264. }
  265. if req.ClassifyId <= 0 {
  266. br.Msg = "请选择分类"
  267. return
  268. }
  269. if len(req.EdbInfoIdArr) <= 0 {
  270. br.Msg = "请选择指标"
  271. return
  272. }
  273. if req.CalculateFormula == "" {
  274. br.Msg = "请填写指标"
  275. return
  276. }
  277. //加入缓存机制,避免创建同一个名称的指标 start
  278. redisKey := fmt.Sprint("predict_edb_info:calculate:batch:save:", utils.DATA_SOURCE_PREDICT_CALCULATE, ":", req.EdbName)
  279. isExist := utils.Rc.IsExist(redisKey)
  280. if isExist {
  281. br.Msg = "指标正在处理,请勿重复提交"
  282. return
  283. } else {
  284. //设置3分钟缓存
  285. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  286. defer func() {
  287. _ = utils.Rc.Delete(redisKey)
  288. }()
  289. }
  290. calculateFormula := req.CalculateFormula
  291. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  292. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  293. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  294. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  295. if strings.Contains(req.CalculateFormula, "%") {
  296. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  297. }
  298. req.CalculateFormula = calculateFormula
  299. //检验公式
  300. var formulaStr string
  301. var edbInfoIdBytes []string
  302. for _, v := range req.EdbInfoIdArr {
  303. formulaStr += v.FromTag + ","
  304. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  305. }
  306. formulaMap := services.CheckFormula(req.CalculateFormula)
  307. for _, v := range formulaMap {
  308. if !strings.Contains(formulaStr, v) {
  309. br.Msg = "公式错误,请重新填写"
  310. return
  311. }
  312. }
  313. // 获取计算指标信息
  314. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  315. if err != nil {
  316. br.Msg = "获取指标信息失败"
  317. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  318. return
  319. }
  320. // 判断指标名称是否存在
  321. {
  322. var condition string
  323. var pars []interface{}
  324. // 指标类型,0:普通指标,1:预测指标
  325. condition += " AND edb_info_id !=? AND edb_info_type=? AND edb_name=? "
  326. pars = append(pars, edbInfo.EdbInfoId, edbInfo.EdbInfoType, req.EdbName)
  327. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  328. if err != nil {
  329. br.Msg = "判断指标名称是否存在失败"
  330. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  331. return
  332. }
  333. if count > 0 {
  334. br.Msg = "指标名称已存在,请重新填写"
  335. br.ErrMsg = "指标名称已存在,请重新填写"
  336. return
  337. }
  338. }
  339. // 关联指标信息
  340. edbInfoList := make([]*models.EdbInfo, 0)
  341. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  342. for k, v := range req.EdbInfoIdArr {
  343. fromEdbInfo, err := models.GetEdbInfoById(v.EdbInfoId)
  344. if err != nil {
  345. if err.Error() == utils.ErrNoRow() {
  346. br.Msg = "生成计算指标失败"
  347. br.Msg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  348. return
  349. }
  350. br.Msg = "生成计算指标失败"
  351. br.Msg = "获取指标失败:Err:" + err.Error()
  352. return
  353. }
  354. edbInfoList = append(edbInfoList, fromEdbInfo)
  355. //关联关系表
  356. {
  357. calculateMappingItem := &models.EdbInfoCalculateMapping{
  358. CreateTime: time.Now(),
  359. ModifyTime: time.Now(),
  360. Sort: k + 1,
  361. EdbCode: edbInfo.EdbCode,
  362. EdbInfoId: edbInfo.EdbInfoId,
  363. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  364. FromEdbCode: fromEdbInfo.EdbCode,
  365. FromEdbName: fromEdbInfo.EdbName,
  366. FromSource: fromEdbInfo.Source,
  367. FromSourceName: fromEdbInfo.SourceName,
  368. FromTag: v.FromTag,
  369. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  370. SourceName: "预测指标运算",
  371. }
  372. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  373. }
  374. }
  375. //预先计算,判断公式是否正常
  376. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, calculateFormula, edbInfoIdBytes)
  377. if !ok {
  378. br.Msg = "生成计算指标失败,请使用正确的计算公式"
  379. return
  380. }
  381. //处理同名指标
  382. {
  383. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  384. if err != nil {
  385. br.Msg = "保存失败"
  386. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  387. return
  388. }
  389. if len(edbNameList) >= 2 {
  390. for _, v := range edbNameList {
  391. edbName := v.EdbName + "(" + v.SourceName + ")"
  392. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  393. if err != nil {
  394. br.Msg = "保存失败"
  395. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  396. return
  397. }
  398. }
  399. }
  400. }
  401. edbInfo.EdbName = req.EdbName
  402. edbInfo.Frequency = req.Frequency
  403. edbInfo.Unit = req.Unit
  404. edbInfo.ClassifyId = req.ClassifyId
  405. edbInfo.CalculateFormula = req.CalculateFormula
  406. err = models.EditPredictCalculate(edbInfo, calculateMappingList)
  407. if err != nil {
  408. br.Msg = "修改计算指标失败"
  409. br.Msg = "修改计算指标失败,Calculate Err:" + err.Error()
  410. return
  411. }
  412. // 开始添加预测指标
  413. latestDateStr, latestValue, err := models.AddPredictCalculateData(edbInfoList, edbInfo, edbInfo.EdbCode, req.CalculateFormula, edbInfoIdBytes)
  414. if err != nil {
  415. br.Msg = "生成计算指标失败"
  416. br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
  417. return
  418. }
  419. // 更新指标最大最小值
  420. err, errMsg := models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  421. if err != nil {
  422. br.Msg = errMsg
  423. br.ErrMsg = err.Error()
  424. return
  425. }
  426. resp := models.AddEdbInfoResp{
  427. EdbInfoId: edbInfo.EdbInfoId,
  428. UniqueCode: edbInfo.UniqueCode,
  429. }
  430. br.Ret = 200
  431. br.Success = true
  432. br.Msg = "保存成功"
  433. br.Data = resp
  434. br.IsAddLog = true
  435. }
  436. // CalculateBatchSave
  437. // @Title 累计值转月-同比值-同差等计算新增
  438. // @Description 累计值转月-同比值-同差等计算新增接口
  439. // @Param request body models.EdbInfoCalculateBatchSaveReq true "type json string"
  440. // @Success Ret=200 返回指标id
  441. // @router /batch/save [post]
  442. func (this *PredictCalculateController) CalculateBatchSave() {
  443. br := new(models.BaseResponse).Init()
  444. defer func() {
  445. this.Data["json"] = br
  446. this.ServeJSON()
  447. }()
  448. var req models.EdbInfoCalculateBatchSaveReq
  449. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  450. if err != nil {
  451. br.Msg = "参数解析异常!"
  452. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  453. return
  454. }
  455. req.EdbName = strings.Trim(req.EdbName, " ")
  456. if req.EdbName == "" {
  457. br.Msg = "指标名称不能为空"
  458. return
  459. }
  460. if req.Frequency == "" {
  461. br.Msg = "频率不能为空"
  462. return
  463. }
  464. if req.Unit == "" {
  465. br.Msg = "单位不能为空"
  466. return
  467. }
  468. if req.ClassifyId <= 0 {
  469. br.Msg = "请选择分类"
  470. return
  471. }
  472. // 基础指标id
  473. fromEdbInfoId := req.FromEdbInfoId
  474. notNeedFromEdbSourceList := []int{utils.DATA_SOURCE_PREDICT_CALCULATE_KSZS, utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC} // 不需要传入来源指标id的 指标类型
  475. if fromEdbInfoId <= 0 && !utils.InArrayByInt(notNeedFromEdbSourceList, req.Source) {
  476. br.Msg = "请选择指标"
  477. return
  478. }
  479. //加入缓存机制,避免创建同一个名称的指标 start
  480. redisKey := fmt.Sprint("edb_lib:predict_calculate:batch:save:", req.Source, ":", req.EdbName)
  481. isExist := utils.Rc.IsExist(redisKey)
  482. if isExist {
  483. br.Msg = "指标正在处理,请勿重复提交"
  484. return
  485. } else {
  486. //设置3分钟缓存
  487. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  488. defer func() {
  489. _ = utils.Rc.Delete(redisKey)
  490. }()
  491. }
  492. //加入缓存机制,避免创建同一个名称的指标 end
  493. var condition string
  494. var pars []interface{}
  495. condition += " AND edb_name=? "
  496. pars = append(pars, req.EdbName)
  497. if req.EdbInfoId > 0 {
  498. condition += " AND edb_info_id !=? "
  499. pars = append(pars, req.EdbInfoId)
  500. }
  501. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  502. if err != nil {
  503. br.Msg = "判断指标名称是否存在失败"
  504. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  505. return
  506. }
  507. if count > 0 {
  508. br.Msg = "指标名称已存在,请重新填写"
  509. br.ErrMsg = "指标名称已存在,请重新填写"
  510. return
  511. }
  512. //生成指标编码
  513. randStr := utils.GetRandDigit(4)
  514. edbCode := `C` + time.Now().Format("060102") + randStr
  515. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  516. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  517. adminId := req.AdminId
  518. adminName := req.AdminName
  519. var sourName string
  520. var errMsg string
  521. var edbInfo *models.EdbInfo
  522. var latestDateStr string // 最近实际数据的日期
  523. var latestValue float64 // 最近实际数据的值
  524. var formulaInt int
  525. var nhccDate models.NhccDate // 拟合残差的日期
  526. switch req.Source {
  527. case utils.DATA_SOURCE_PREDICT_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_PREDICT_CALCULATE_HBZ, utils.DATA_SOURCE_PREDICT_CALCULATE_HCZ, utils.DATA_SOURCE_PREDICT_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_PREDICT_CALCULATE_CJJX:
  528. if req.Formula == "" {
  529. br.Msg = "请填写N值"
  530. return
  531. }
  532. formulaInt, _ = strconv.Atoi(req.Formula)
  533. if formulaInt <= 0 {
  534. br.Msg = "N值输入错误,请重新输入"
  535. return
  536. }
  537. /*case utils.DATA_SOURCE_PREDICT_CALCULATE_ZJPJ:
  538. //直接拼接指标
  539. //校验时间格式
  540. _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
  541. if err != nil {
  542. br.Msg = "拼接日期有误,请重新输入"
  543. return
  544. }*/
  545. case utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC: //拟合残差指标
  546. //指标校验
  547. if len(req.EdbInfoIdArr) != 2 {
  548. br.Msg = "选择的指标异常,请重新选择"
  549. return
  550. }
  551. fromEdbInfoId = req.EdbInfoIdArr[0].EdbInfoId
  552. //校验时间格式
  553. //数据格式:2022-11-01,2022-11-10
  554. timeList := strings.Split(req.Formula, ",")
  555. if len(timeList) != 2 {
  556. br.Msg = "选择时间有误,请重新输入"
  557. return
  558. }
  559. startDate, err := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
  560. if err != nil {
  561. br.Msg = "开始日期有误,请重新输入"
  562. return
  563. }
  564. endDate, err := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
  565. if err != nil {
  566. br.Msg = "结束日期有误,请重新输入"
  567. return
  568. }
  569. if utils.GetTimeSubDay(startDate, endDate) < 2 {
  570. br.Msg = "日期间隔不得少于两天"
  571. return
  572. }
  573. nhccDate.StartDate = startDate
  574. nhccDate.EndDate = endDate
  575. }
  576. if err != nil {
  577. br.Msg = "获取指标信息失败"
  578. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  579. return
  580. }
  581. // 来源预测指标信息
  582. var fromEdbInfo *models.EdbInfo
  583. if fromEdbInfoId > 0 {
  584. fromEdbInfo, err = models.GetEdbInfoById(fromEdbInfoId)
  585. if err != nil {
  586. br.Msg = "获取指标信息失败"
  587. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  588. return
  589. }
  590. if fromEdbInfo.EdbInfoType != 1 {
  591. br.Msg = "来源指标不是预测指标"
  592. br.ErrMsg = "来源指标不是预测指标"
  593. return
  594. }
  595. }
  596. var baseEdbInfoModel models.BasePredictEdbInfoInterface
  597. saveParams := models.BatchSaveCalculateBatchParams{
  598. Req: req,
  599. FromEdbInfo: fromEdbInfo,
  600. EdbCode: edbCode,
  601. UniqueCode: uniqueCode,
  602. SysUserId: req.AdminId,
  603. SysUserRealName: req.AdminName,
  604. }
  605. if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ {
  606. if req.EdbInfoId <= 0 {
  607. if req.Frequency != fromEdbInfo.Frequency {
  608. br.Msg = "当前频度和原指标频度不一致"
  609. return
  610. }
  611. if req.Unit != "无" {
  612. br.Msg = "单位只允许为无,禁止选择其他单位"
  613. return
  614. }
  615. }
  616. sourName = "预测同比"
  617. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTbz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  618. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ {
  619. if req.EdbInfoId <= 0 {
  620. if req.Frequency != fromEdbInfo.Frequency {
  621. br.Msg = "当前频度和原指标频度不一致"
  622. return
  623. }
  624. if req.Unit != fromEdbInfo.Unit {
  625. br.Msg = "当前单位和原指标单位不一致"
  626. return
  627. }
  628. }
  629. sourName = "预测同差"
  630. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTcz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  631. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_NSZYDPJJS {
  632. sourName = "预测N数值移动平均计算"
  633. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateNszydpjjs(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  634. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_BP {
  635. sourName = "预测变频"
  636. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateBp(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  637. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_HBZ {
  638. checkDataList, tmpErr := models.GetPredictEdbDataListAllByStartDate(fromEdbInfo, 0, "")
  639. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  640. br.Msg = "判断环比值是否可计算失败"
  641. br.ErrMsg = "判断环比值是否可计算失败,Err:" + tmpErr.Error()
  642. return
  643. }
  644. for _, v := range checkDataList {
  645. if v.Value <= 0 {
  646. br.Msg = "原始数据中存在0或负数,该指标不能进行环比运算"
  647. br.ErrMsg = "原始数据中出现0和负值时,提示该指标不能进行环比运算"
  648. return
  649. }
  650. }
  651. sourName = "预测环比值"
  652. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateHbz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  653. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_HCZ {
  654. sourName = "预测环差值"
  655. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateHcz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  656. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_LJZZY {
  657. sourName = "预测累计值转月值"
  658. if fromEdbInfo.Frequency != "月度" {
  659. br.Msg = "请选择月度指标"
  660. return
  661. }
  662. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateLjzzy(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  663. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TIME_SHIFT {
  664. sourName = "预测时间移位"
  665. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateTimeShift(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  666. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_CJJX {
  667. sourName = "预测超季节性"
  668. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateCjjx(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  669. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC {
  670. sourName = "预测拟合残差"
  671. secondEdbInfoReq := req.EdbInfoIdArr[1]
  672. secondEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  673. if tmpErr != nil {
  674. br.Msg = "获取因变量的指标信息失败"
  675. br.ErrMsg = "获取因变量的指标信息失败:Err:" + tmpErr.Error()
  676. return
  677. }
  678. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  679. br.Msg = "两个指标不允许为同一个"
  680. br.ErrMsg = "两个指标不允许为同一个"
  681. return
  682. }
  683. edbInfo, latestDateStr, latestValue, err = models.SavePredictCalculateNhcc(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, nhccDate, adminId, adminName)
  684. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_JP {
  685. sourName = utils.DATA_SOURCE_NAME_PREDICT_CALCULATE_JP
  686. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateJp(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  687. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_NH {
  688. sourName = utils.DATA_SOURCE_NAME_PREDICT_CALCULATE_NH
  689. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateNh(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  690. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_KSZS {
  691. sourName = utils.DATA_SOURCE_NAME_PREDICT_CALCULATE_KSZS
  692. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateKszs(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, req.EdbInfoIdArr, edbCode, uniqueCode, adminId, adminName)
  693. } else {
  694. // 获取通用的数据源处理服务
  695. baseEdbInfoModel = models.GetBasePredictEdbInfoModel(req.Source)
  696. // 没有找到的话,那么就直接返回报错吧
  697. if baseEdbInfoModel == nil {
  698. br.Msg = "无效计算方式"
  699. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  700. return
  701. }
  702. sourName = baseEdbInfoModel.GetSourceName()
  703. if req.EdbInfoId <= 0 {
  704. edbInfo, latestDateStr, latestValue, err, errMsg = baseEdbInfoModel.Add(saveParams)
  705. } else {
  706. edbInfo, err = models.GetEdbInfoById(req.EdbInfoId)
  707. if err != nil {
  708. if err.Error() == utils.ErrNoRow() {
  709. br.Msg = "指标已被删除,请刷新页面"
  710. br.ErrMsg = "指标已被删除,请刷新页面:Err:" + err.Error()
  711. return
  712. }
  713. br.Msg = "获取指标信息失败"
  714. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  715. return
  716. }
  717. saveParams.EdbInfo = edbInfo
  718. latestDateStr, latestValue, err, errMsg = baseEdbInfoModel.Edit(saveParams)
  719. }
  720. }
  721. if err != nil {
  722. br.Msg = "生成" + sourName + "失败"
  723. if errMsg != `` {
  724. br.Msg = errMsg
  725. }
  726. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  727. return
  728. }
  729. if edbInfo == nil {
  730. br.Msg = "生成" + sourName + "失败"
  731. br.ErrMsg = "生成" + sourName + "失败,edbInfo is nil"
  732. return
  733. }
  734. //处理同名指标
  735. {
  736. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  737. if err != nil {
  738. br.Msg = "保存失败"
  739. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  740. return
  741. }
  742. if len(edbNameList) >= 2 {
  743. for _, v := range edbNameList {
  744. edbName := v.EdbName + "(" + v.SourceName + ")"
  745. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  746. if err != nil {
  747. br.Msg = "保存失败"
  748. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  749. return
  750. }
  751. }
  752. }
  753. }
  754. // 更新指标最大最小值
  755. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  756. if err != nil {
  757. br.Msg = errMsg
  758. br.ErrMsg = err.Error()
  759. return
  760. }
  761. resp := models.AddEdbInfoResp{
  762. EdbInfoId: edbInfo.EdbInfoId,
  763. UniqueCode: uniqueCode,
  764. }
  765. br.Ret = 200
  766. br.Success = true
  767. br.Msg = "保存成功"
  768. br.Data = resp
  769. br.IsAddLog = true
  770. }
  771. // Refresh
  772. // @Title 刷新计算指标接口
  773. // @Description 刷新计算指标接口
  774. // @Success 200 {object} models.RefreshEdbInfoReq
  775. // @router /refresh [post]
  776. func (this *PredictCalculateController) Refresh() {
  777. br := new(models.BaseResponse).Init()
  778. var cacheKey string
  779. defer func() {
  780. this.Data["json"] = br
  781. this.ServeJSON()
  782. }()
  783. var req models.RefreshEdbInfoReq
  784. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  785. if err != nil {
  786. br.Msg = "参数解析异常!"
  787. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  788. return
  789. }
  790. if req.EdbCode == "" {
  791. br.Msg = "请输入指标编码!"
  792. br.ErrMsg = "请输入指标编码,指标编码为空"
  793. return
  794. }
  795. if req.EdbInfoId <= 0 {
  796. br.Msg = "请输入指标ID!"
  797. br.ErrMsg = "请输入指标ID"
  798. return
  799. }
  800. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  801. if err != nil {
  802. br.Msg = "指标不存在!"
  803. br.ErrMsg = "指标不存在"
  804. return
  805. }
  806. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  807. if utils.Rc.IsExist(cacheKey) {
  808. br.Ret = 501
  809. br.Success = true
  810. br.Msg = "系统处理中,请稍后重试"
  811. br.IsSendEmail = false
  812. return
  813. }
  814. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  815. defer func() {
  816. _ = utils.Rc.Delete(cacheKey)
  817. }()
  818. startDate := req.StartDate
  819. var errMsg string
  820. endDate := time.Now().Format(utils.FormatDate)
  821. edbInfoId := edbInfo.EdbInfoId
  822. source := edbInfo.Source
  823. var latestDateStr string // 最近实际数据的日期
  824. var latestValue float64 // 最近实际数据的值
  825. var baseEdbInfoModel models.BasePredictEdbInfoInterface
  826. refreshParams := models.RefreshParams{
  827. EdbInfo: edbInfo,
  828. }
  829. switch source {
  830. case utils.DATA_SOURCE_PREDICT_CALCULATE:
  831. //startDate = edbInfo.StartDate
  832. //sTime, err := time.ParseInLocation(utils.FormatDate, edbInfo.EndDate, time.Local)
  833. //if err != nil {
  834. // return
  835. //}
  836. //startDate = sTime.Format(utils.FormatDate)
  837. startDate = ""
  838. var edbInfoIdBytes []string
  839. calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  840. if err != nil {
  841. errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  842. break
  843. }
  844. var formulaStr string
  845. edbInfoList := make([]*models.EdbInfo, 0)
  846. for _, v := range calculateMap {
  847. formulaStr += v.FromTag + ","
  848. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  849. edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  850. edbInfoList = append(edbInfoList, edbInfo)
  851. }
  852. latestDateStr, latestValue, err = models.RefreshAllPredictCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, edbInfoIdBytes)
  853. if err != nil && err.Error() != utils.ErrNoRow() {
  854. errMsg = "RefreshCalculate Err:" + err.Error()
  855. break
  856. }
  857. case utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ: //刷新同比值
  858. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  859. if err != nil {
  860. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  861. break
  862. }
  863. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  864. if err != nil {
  865. errMsg = "GetEdbInfoById Err:" + err.Error()
  866. break
  867. }
  868. startDate = edbInfo.StartDate
  869. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate)
  870. if err != nil && err.Error() != utils.ErrNoRow() {
  871. errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  872. break
  873. }
  874. case utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ: //同差值
  875. calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  876. if err != nil {
  877. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  878. break
  879. }
  880. fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  881. if err != nil {
  882. errMsg = "GetEdbInfoById Err:" + err.Error()
  883. break
  884. }
  885. startDate = edbInfo.StartDate
  886. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate)
  887. if err != nil && err.Error() != utils.ErrNoRow() {
  888. errMsg = "RefreshCalculateTcz Err:" + err.Error()
  889. break
  890. }
  891. case utils.DATA_SOURCE_PREDICT_CALCULATE_NSZYDPJJS: //N数值移动平均计算
  892. calculateInfo, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  893. if err != nil {
  894. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  895. break
  896. }
  897. fromEdbInfo, err := models.GetEdbInfoById(calculateInfo.FromEdbInfoId)
  898. if err != nil {
  899. errMsg = "GetEdbInfoById Err:" + err.Error()
  900. break
  901. }
  902. var formulaInt int
  903. formulaInt, _ = strconv.Atoi(calculateInfo.CalculateFormula)
  904. if formulaInt <= 0 {
  905. errMsg = "N值输入错误,请重新输入"
  906. break
  907. }
  908. startDate = edbInfo.StartDate
  909. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateNszydpjjs(edbInfoId, source, formulaInt, fromEdbInfo, calculateInfo.EdbCode, startDate)
  910. if err != nil && err.Error() != utils.ErrNoRow() {
  911. errMsg = "RefreshAllPredictCalculateNszydpjjs Err:" + err.Error()
  912. break
  913. }
  914. case utils.DATA_SOURCE_PREDICT_CALCULATE_BP: //刷新变频
  915. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  916. if err != nil {
  917. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  918. break
  919. }
  920. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  921. if err != nil {
  922. errMsg = "GetEdbInfoById Err:" + err.Error()
  923. break
  924. }
  925. endDate = time.Now().Format(utils.FormatDate)
  926. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateBp(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  927. if err != nil && err.Error() != utils.ErrNoRow() {
  928. errMsg = "RefreshAllPredictCalculateBp Err:" + err.Error()
  929. break
  930. }
  931. case utils.DATA_SOURCE_PREDICT_CALCULATE_HBZ: //刷新环比值
  932. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  933. if err != nil {
  934. errMsg = "GetEdbInfoCalculateHbzDetail Err:" + err.Error()
  935. break
  936. }
  937. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  938. if err != nil {
  939. errMsg = "GetEdbInfoById Err:" + err.Error()
  940. break
  941. }
  942. startDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  943. endDate = time.Now().Format(utils.FormatDate)
  944. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  945. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateHbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  946. if err != nil && err.Error() != utils.ErrNoRow() {
  947. errMsg = "RefreshAllPredictCalculateHbz Err:" + err.Error()
  948. break
  949. }
  950. case utils.DATA_SOURCE_PREDICT_CALCULATE_HCZ: //刷新环差值
  951. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  952. if err != nil {
  953. errMsg = "GetEdbInfoCalculateHczDetail Err:" + err.Error()
  954. break
  955. }
  956. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  957. if err != nil {
  958. errMsg = "GetEdbInfoById Err:" + err.Error()
  959. break
  960. }
  961. startDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  962. endDate = time.Now().Format(utils.FormatDate)
  963. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  964. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateHcz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  965. if err != nil && err.Error() != utils.ErrNoRow() {
  966. errMsg = "RefreshAllPredictCalculateHcz Err:" + err.Error()
  967. break
  968. }
  969. case utils.DATA_SOURCE_PREDICT_CALCULATE_LJZZY: //刷新累计值转月值
  970. calculateLjzzy, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  971. if err != nil {
  972. errMsg = "GetEdbInfoCalculateLjzzyDetail Err:" + err.Error()
  973. break
  974. }
  975. fromEdbInfo, err := models.GetEdbInfoById(calculateLjzzy.FromEdbInfoId)
  976. if err != nil {
  977. errMsg = "GetEdbInfoById Err:" + err.Error()
  978. break
  979. }
  980. endDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  981. startDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  982. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateLjzzy(edbInfoId, source, fromEdbInfo, calculateLjzzy.EdbCode, startDate, endDate)
  983. if err != nil && err.Error() != utils.ErrNoRow() {
  984. errMsg = "RefreshAllPredictCalculateLjzzy Err:" + err.Error()
  985. break
  986. }
  987. case utils.DATA_SOURCE_PREDICT_CALCULATE_TIME_SHIFT:
  988. calculate, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  989. if err != nil {
  990. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  991. break
  992. }
  993. fromEdbInfo, err := models.GetEdbInfoById(calculate.FromEdbInfoId)
  994. if err != nil {
  995. errMsg = "GetEdbInfoById Err:" + err.Error()
  996. break
  997. }
  998. //startDate = edbInfo.StartDate
  999. startDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  1000. endDate = time.Now().Format(utils.FormatDate)
  1001. formulaInt, _ := strconv.Atoi(calculate.CalculateFormula)
  1002. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTimeShift(edbInfoId, source, formulaInt, calculate.MoveType, fromEdbInfo, calculate.EdbCode, startDate, endDate, calculate.MoveFrequency)
  1003. if err != nil && err.Error() != utils.ErrNoRow() {
  1004. errMsg = "RefreshAllCalculateTimeShift Err:" + err.Error()
  1005. }
  1006. case utils.DATA_SOURCE_PREDICT_CALCULATE_LJZTBPJ: //刷新累计值同比拼接
  1007. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateLjztbpj(edbInfo)
  1008. if err != nil && err.Error() != utils.ErrNoRow() {
  1009. errMsg = "RefreshAllPredictCalculateLjztbpj Err:" + err.Error()
  1010. break
  1011. }
  1012. case utils.DATA_SOURCE_PREDICT_CALCULATE_ZJPJ: //刷新直接拼接
  1013. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateZjpj(edbInfo)
  1014. if err != nil && err.Error() != utils.ErrNoRow() {
  1015. errMsg = "RefreshAllPredictCalculateZjpj Err:" + err.Error()
  1016. break
  1017. }
  1018. case utils.DATA_SOURCE_PREDICT_CALCULATE_CJJX: //超季节性
  1019. calculateCjjx, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1020. if err != nil {
  1021. errMsg = "GetEdbInfoCalculateNszydpjjsDetail Err:" + err.Error()
  1022. break
  1023. }
  1024. fromEdbInfo, err := models.GetEdbInfoById(calculateCjjx.FromEdbInfoId)
  1025. if err != nil {
  1026. errMsg = "GetEdbInfoById Err:" + err.Error()
  1027. break
  1028. }
  1029. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  1030. startDate = `` //只要填写日期,就会出现问题,还是把日期给去掉吧
  1031. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateCjjx(edbInfoId, edbInfo.Source, fromEdbInfo, calculateCjjx.EdbCode, startDate, "", edbInfo.Calendar, formulaInt)
  1032. if err != nil && err.Error() != utils.ErrNoRow() {
  1033. errMsg = "RefreshAllPredictCalculateCjjx Err:" + err.Error()
  1034. break
  1035. }
  1036. case utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC: //nhcc
  1037. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateNhcc(edbInfo)
  1038. if err != nil && err.Error() != utils.ErrNoRow() {
  1039. errMsg = "RefreshAllPredictCalculateNhcc Err:" + err.Error()
  1040. break
  1041. }
  1042. case utils.DATA_SOURCE_PREDICT_CALCULATE_JP: //降频
  1043. calculateInfo, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1044. if err != nil {
  1045. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  1046. break
  1047. }
  1048. fromEdbInfo, err := models.GetEdbInfoById(calculateInfo.FromEdbInfoId)
  1049. if err != nil {
  1050. errMsg = "GetEdbInfoById Err:" + err.Error()
  1051. break
  1052. }
  1053. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateJp(edbInfoId, source, fromEdbInfo, calculateInfo.EdbCode, edbInfo.Frequency, edbInfo.CalculateFormula)
  1054. if err != nil && err.Error() != utils.ErrNoRow() {
  1055. errMsg = "RefreshAllPredictCalculateJp Err:" + err.Error()
  1056. break
  1057. }
  1058. case utils.DATA_SOURCE_PREDICT_CALCULATE_NH: //年化
  1059. calculateInfo, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1060. if err != nil {
  1061. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  1062. break
  1063. }
  1064. fromEdbInfo, err := models.GetEdbInfoById(calculateInfo.FromEdbInfoId)
  1065. if err != nil {
  1066. errMsg = "GetEdbInfoById Err:" + err.Error()
  1067. break
  1068. }
  1069. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateNh(edbInfoId, source, fromEdbInfo, calculateInfo.EdbCode, edbInfo.Frequency, edbInfo.CalculateFormula)
  1070. if err != nil && err.Error() != utils.ErrNoRow() {
  1071. errMsg = "RefreshAllPredictCalculateNh Err:" + err.Error()
  1072. break
  1073. }
  1074. case utils.DATA_SOURCE_PREDICT_CALCULATE_KSZS: //扩散指数
  1075. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateKszs(edbInfo)
  1076. if err != nil && err.Error() != utils.ErrNoRow() {
  1077. errMsg = "RefreshAllPredictCalculateKszs Err:" + err.Error()
  1078. break
  1079. }
  1080. default:
  1081. // 获取通用的数据源处理服务
  1082. baseEdbInfoModel = models.GetBasePredictEdbInfoModel(source)
  1083. // 没有找到的话,那么就直接返回报错吧
  1084. if baseEdbInfoModel == nil {
  1085. br.Msg = "来源异常,请联系相关开发!"
  1086. br.ErrMsg = "来源异常,请联系相关开发"
  1087. return
  1088. }
  1089. latestDateStr, latestValue, err, errMsg = baseEdbInfoModel.Refresh(refreshParams)
  1090. }
  1091. if errMsg != `` {
  1092. br.Msg = "刷新指标失败!"
  1093. br.ErrMsg = "刷新指标失败,err:" + errMsg
  1094. return
  1095. }
  1096. // 更新指标最大最小值
  1097. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  1098. if err != nil {
  1099. br.Msg = errMsg
  1100. br.ErrMsg = err.Error()
  1101. return
  1102. }
  1103. // 更新ES
  1104. go logic.UpdateEs(edbInfo.EdbInfoId)
  1105. br.Ret = 200
  1106. br.Success = true
  1107. br.Msg = "获取成功"
  1108. }