base_from_predict_calculate.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801
  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. if req.FromEdbInfoId <= 0 {
  473. br.Msg = "请选择指标"
  474. return
  475. }
  476. //加入缓存机制,避免创建同一个名称的指标 start
  477. redisKey := fmt.Sprint("edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
  478. isExist := utils.Rc.IsExist(redisKey)
  479. if isExist {
  480. br.Msg = "指标正在处理,请勿重复提交"
  481. return
  482. } else {
  483. //设置3分钟缓存
  484. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  485. defer func() {
  486. _ = utils.Rc.Delete(redisKey)
  487. }()
  488. }
  489. //加入缓存机制,避免创建同一个名称的指标 end
  490. var condition string
  491. var pars []interface{}
  492. condition += " AND edb_name=? "
  493. pars = append(pars, req.EdbName)
  494. if req.EdbInfoId > 0 {
  495. condition += " AND edb_info_id !=? "
  496. pars = append(pars, req.EdbInfoId)
  497. }
  498. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  499. if err != nil {
  500. br.Msg = "判断指标名称是否存在失败"
  501. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  502. return
  503. }
  504. if count > 0 {
  505. br.Msg = "指标名称已存在,请重新填写"
  506. br.ErrMsg = "指标名称已存在,请重新填写"
  507. return
  508. }
  509. // 来源预测指标信息
  510. fromEdbInfo, err := models.GetEdbInfoById(req.FromEdbInfoId)
  511. if err != nil {
  512. br.Msg = "获取指标信息失败"
  513. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  514. return
  515. }
  516. if fromEdbInfo.EdbInfoType != 1 {
  517. br.Msg = "来源指标不是预测指标"
  518. br.ErrMsg = "来源指标不是预测指标"
  519. return
  520. }
  521. //生成指标编码
  522. randStr := utils.GetRandDigit(4)
  523. edbCode := `C` + time.Now().Format("060102") + randStr
  524. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  525. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  526. adminId := req.AdminId
  527. adminName := req.AdminName
  528. var sourName string
  529. var errMsg string
  530. var edbInfo *models.EdbInfo
  531. var latestDateStr string // 最近实际数据的日期
  532. var latestValue float64 // 最近实际数据的值
  533. if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ {
  534. sourName = "预测同比"
  535. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTbz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  536. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ {
  537. sourName = "预测同差"
  538. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTcz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  539. } else {
  540. br.Msg = "无效计算方式"
  541. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  542. return
  543. }
  544. if err != nil {
  545. br.Msg = "生成" + sourName + "失败"
  546. if errMsg != `` {
  547. br.Msg = errMsg
  548. }
  549. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  550. return
  551. }
  552. if edbInfo == nil {
  553. br.Msg = "生成" + sourName + "失败"
  554. br.ErrMsg = "生成" + sourName + "失败,edbInfo is nil"
  555. return
  556. }
  557. //处理同名指标
  558. {
  559. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  560. if err != nil {
  561. br.Msg = "保存失败"
  562. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  563. return
  564. }
  565. if len(edbNameList) >= 2 {
  566. for _, v := range edbNameList {
  567. edbName := v.EdbName + "(" + v.SourceName + ")"
  568. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  569. if err != nil {
  570. br.Msg = "保存失败"
  571. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  572. return
  573. }
  574. }
  575. }
  576. }
  577. // 更新指标最大最小值
  578. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  579. if err != nil {
  580. br.Msg = errMsg
  581. br.ErrMsg = err.Error()
  582. return
  583. }
  584. resp := models.AddEdbInfoResp{
  585. EdbInfoId: edbInfo.EdbInfoId,
  586. UniqueCode: uniqueCode,
  587. }
  588. br.Ret = 200
  589. br.Success = true
  590. br.Msg = "保存成功"
  591. br.Data = resp
  592. br.IsAddLog = true
  593. }
  594. // Refresh
  595. // @Title 刷新计算指标接口
  596. // @Description 刷新计算指标接口
  597. // @Success 200 {object} models.RefreshEdbInfoReq
  598. // @router /refresh [post]
  599. func (this *PredictCalculateController) Refresh() {
  600. br := new(models.BaseResponse).Init()
  601. var cacheKey string
  602. defer func() {
  603. this.Data["json"] = br
  604. this.ServeJSON()
  605. }()
  606. var req models.RefreshEdbInfoReq
  607. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  608. if err != nil {
  609. br.Msg = "参数解析异常!"
  610. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  611. return
  612. }
  613. if req.EdbCode == "" {
  614. br.Msg = "请输入指标编码!"
  615. br.ErrMsg = "请输入指标编码,指标编码为空"
  616. return
  617. }
  618. if req.EdbInfoId <= 0 {
  619. br.Msg = "请输入指标ID!"
  620. br.ErrMsg = "请输入指标ID"
  621. return
  622. }
  623. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  624. if err != nil {
  625. br.Msg = "指标不存在!"
  626. br.ErrMsg = "指标不存在"
  627. return
  628. }
  629. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  630. if utils.Rc.IsExist(cacheKey) {
  631. br.Ret = 501
  632. br.Success = true
  633. br.Msg = "系统处理中,请稍后重试"
  634. br.IsSendEmail = false
  635. return
  636. }
  637. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  638. defer func() {
  639. _ = utils.Rc.Delete(cacheKey)
  640. }()
  641. startDate := req.StartDate
  642. var errMsg string
  643. edbInfoId := edbInfo.EdbInfoId
  644. source := edbInfo.Source
  645. var latestDateStr string // 最近实际数据的日期
  646. var latestValue float64 // 最近实际数据的值
  647. switch source {
  648. case utils.DATA_SOURCE_PREDICT_CALCULATE:
  649. //startDate = edbInfo.StartDate
  650. //sTime, err := time.Parse(utils.FormatDate, edbInfo.EndDate)
  651. //if err != nil {
  652. // return
  653. //}
  654. //startDate = sTime.Format(utils.FormatDate)
  655. startDate = ""
  656. var edbInfoIdBytes []string
  657. calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  658. if err != nil {
  659. errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  660. break
  661. }
  662. var formulaStr string
  663. edbInfoList := make([]*models.EdbInfo, 0)
  664. for _, v := range calculateMap {
  665. formulaStr += v.FromTag + ","
  666. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  667. edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  668. edbInfoList = append(edbInfoList, edbInfo)
  669. }
  670. latestDateStr, latestValue, err = models.RefreshAllPredictCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, edbInfoIdBytes)
  671. if err != nil && err.Error() != utils.ErrNoRow() {
  672. errMsg = "RefreshCalculate Err:" + err.Error()
  673. break
  674. }
  675. case utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ: //刷新同比值
  676. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  677. if err != nil {
  678. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  679. break
  680. }
  681. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  682. if err != nil {
  683. errMsg = "GetEdbInfoById Err:" + err.Error()
  684. break
  685. }
  686. startDate = edbInfo.StartDate
  687. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate)
  688. if err != nil && err.Error() != utils.ErrNoRow() {
  689. errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  690. break
  691. }
  692. case utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ: //同差值
  693. calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  694. if err != nil {
  695. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  696. break
  697. }
  698. fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  699. if err != nil {
  700. errMsg = "GetEdbInfoById Err:" + err.Error()
  701. break
  702. }
  703. startDate = edbInfo.StartDate
  704. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate)
  705. if err != nil && err.Error() != utils.ErrNoRow() {
  706. errMsg = "RefreshCalculateTcz Err:" + err.Error()
  707. break
  708. }
  709. default:
  710. br.Msg = "来源异常,请联系相关开发!"
  711. br.ErrMsg = "来源异常,请联系相关开发"
  712. return
  713. }
  714. if errMsg != `` {
  715. br.Msg = "刷新指标失败!"
  716. br.ErrMsg = "刷新指标失败,err:" + errMsg
  717. return
  718. }
  719. // 更新指标最大最小值
  720. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  721. if err != nil {
  722. br.Msg = errMsg
  723. br.ErrMsg = err.Error()
  724. return
  725. }
  726. // 更新ES
  727. go logic.UpdateEs(edbInfo.EdbInfoId)
  728. br.Ret = 200
  729. br.Success = true
  730. br.Msg = "获取成功"
  731. }