base_from_predict_calculate.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "hongze/hongze_edb_lib/models"
  6. "hongze/hongze_edb_lib/services"
  7. "hongze/hongze_edb_lib/utils"
  8. "strconv"
  9. "strings"
  10. "time"
  11. )
  12. // PredictCalculateController 预测计算指标
  13. type PredictCalculateController struct {
  14. BaseAuthController
  15. }
  16. // Save
  17. // @Title 新增/编辑预测指标运算接口
  18. // @Description 新增预测指标运算接口
  19. // @Success 200 {object} models.EdbInfoCalculateSaveReq
  20. // @router /save [post]
  21. func (this *PredictCalculateController) Save() {
  22. br := new(models.BaseResponse).Init()
  23. defer func() {
  24. this.Data["json"] = br
  25. this.ServeJSON()
  26. }()
  27. var req models.EdbInfoCalculateSaveReq
  28. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  29. if err != nil {
  30. br.Msg = "参数解析异常!"
  31. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  32. return
  33. }
  34. if req.EdbInfoId <= 0 {
  35. addPredictCalculate(br, req)
  36. } else {
  37. editPredictCalculate(br, req)
  38. }
  39. }
  40. // AddPredictCalculate 添加预测指标运算
  41. func addPredictCalculate(br *models.BaseResponse, req models.EdbInfoCalculateSaveReq) {
  42. req.EdbName = strings.Trim(req.EdbName, " ")
  43. if req.EdbName == "" {
  44. br.Msg = "指标名称不能为空"
  45. return
  46. }
  47. if req.Frequency == "" {
  48. br.Msg = "频率不能为空"
  49. return
  50. }
  51. if req.Unit == "" {
  52. br.Msg = "单位不能为空"
  53. return
  54. }
  55. if req.ClassifyId <= 0 {
  56. br.Msg = "请选择分类"
  57. return
  58. }
  59. if len(req.EdbInfoIdArr) <= 0 {
  60. br.Msg = "请选择指标"
  61. return
  62. }
  63. if req.CalculateFormula == "" {
  64. br.Msg = "请填写指标"
  65. return
  66. }
  67. //加入缓存机制,避免创建同一个名称的指标 start
  68. redisKey := fmt.Sprint("predict_edb_info:calculate:batch:save:", utils.DATA_SOURCE_PREDICT_CALCULATE, ":", req.EdbName)
  69. isExist := utils.Rc.IsExist(redisKey)
  70. if isExist {
  71. br.Msg = "指标正在处理,请勿重复提交"
  72. return
  73. } else {
  74. //设置3分钟缓存
  75. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  76. defer func() {
  77. _ = utils.Rc.Delete(redisKey)
  78. }()
  79. }
  80. calculateFormula := req.CalculateFormula
  81. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  82. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  83. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  84. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  85. if strings.Contains(req.CalculateFormula, "%") {
  86. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  87. }
  88. req.CalculateFormula = calculateFormula
  89. //检验公式
  90. var formulaStr string
  91. var edbInfoIdBytes []string
  92. for _, v := range req.EdbInfoIdArr {
  93. formulaStr += v.FromTag + ","
  94. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  95. }
  96. formulaMap := services.CheckFormula(req.CalculateFormula)
  97. for _, v := range formulaMap {
  98. if !strings.Contains(formulaStr, v) {
  99. br.Msg = "公式错误,请重新填写"
  100. return
  101. }
  102. }
  103. // 判断指标名称是否存在
  104. {
  105. var condition string
  106. var pars []interface{}
  107. // 指标类型,0:普通指标,1:预测指标
  108. condition += " AND edb_info_type=? AND edb_name=? "
  109. pars = append(pars, 1, req.EdbName)
  110. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  111. if err != nil {
  112. br.Msg = "判断指标名称是否存在失败"
  113. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  114. return
  115. }
  116. if count > 0 {
  117. br.Msg = "指标名称已存在,请重新填写"
  118. br.ErrMsg = "指标名称已存在,请重新填写"
  119. return
  120. }
  121. }
  122. // 关联指标信息
  123. edbInfoList := make([]*models.EdbInfo, 0)
  124. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  125. for k, v := range req.EdbInfoIdArr {
  126. fromEdbInfo, err := models.GetEdbInfoById(v.EdbInfoId)
  127. if err != nil {
  128. if err.Error() == utils.ErrNoRow() {
  129. br.Msg = "生成计算指标失败"
  130. br.Msg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  131. return
  132. }
  133. br.Msg = "生成计算指标失败"
  134. br.Msg = "获取指标失败:Err:" + err.Error()
  135. return
  136. }
  137. edbInfoList = append(edbInfoList, fromEdbInfo)
  138. //关联关系表
  139. {
  140. calculateMappingItem := &models.EdbInfoCalculateMapping{
  141. CreateTime: time.Now(),
  142. ModifyTime: time.Now(),
  143. Sort: k + 1,
  144. //EdbCode: edbCode,
  145. //EdbInfoId: int(edbInfoId),
  146. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  147. FromEdbCode: fromEdbInfo.EdbCode,
  148. FromEdbName: fromEdbInfo.EdbName,
  149. FromSource: fromEdbInfo.Source,
  150. FromSourceName: fromEdbInfo.SourceName,
  151. FromTag: v.FromTag,
  152. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  153. SourceName: "预测指标运算",
  154. }
  155. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  156. }
  157. }
  158. // 指标入库
  159. randStr := utils.GetRandDigit(4)
  160. edbCode := `C2` + time.Now().Format("060102") + randStr
  161. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  162. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  163. edbInfo := &models.EdbInfo{
  164. EdbInfoType: 1,
  165. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  166. SourceName: "预测指标运算",
  167. EdbCode: edbCode,
  168. EdbName: req.EdbName,
  169. EdbNameSource: req.EdbName,
  170. Frequency: req.Frequency,
  171. Unit: req.Unit,
  172. ClassifyId: req.ClassifyId,
  173. SysUserId: req.AdminId,
  174. SysUserRealName: req.AdminName,
  175. CreateTime: time.Now(),
  176. ModifyTime: time.Now(),
  177. UniqueCode: uniqueCode,
  178. CalculateFormula: req.CalculateFormula,
  179. EdbType: 2,
  180. }
  181. edbInfoId, err := models.AddEdbInfo(edbInfo)
  182. if err != nil {
  183. br.Msg = "生成计算指标失败"
  184. br.Msg = "生成计算指标失败,AddEdbInfo Err:" + err.Error()
  185. return
  186. }
  187. edbInfo.EdbInfoId = int(edbInfoId)
  188. //处理同名指标
  189. {
  190. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  191. if err != nil {
  192. br.Msg = "保存失败"
  193. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  194. return
  195. }
  196. if len(edbNameList) >= 2 {
  197. for _, v := range edbNameList {
  198. edbName := v.EdbName + "(" + v.SourceName + ")"
  199. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  200. if err != nil {
  201. br.Msg = "保存失败"
  202. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  203. return
  204. }
  205. }
  206. }
  207. }
  208. //关联关系表入库
  209. {
  210. if len(calculateMappingList) > 0 {
  211. for k, v := range calculateMappingList {
  212. v.EdbCode = edbInfo.EdbCode
  213. v.EdbInfoId = edbInfo.EdbInfoId
  214. calculateMappingList[k] = v
  215. }
  216. go models.AddEdbInfoCalculateMappingMulti(calculateMappingList)
  217. }
  218. }
  219. // 开始添加预测指标
  220. latestDateStr, latestValue, err := models.AddPredictCalculateData(edbInfoList, int(edbInfoId), edbCode, req.CalculateFormula, edbInfoIdBytes)
  221. if err != nil {
  222. br.Msg = "生成计算指标失败"
  223. br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
  224. return
  225. }
  226. // 更新指标最大最小值
  227. err, errMsg := models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  228. if err != nil {
  229. br.Msg = errMsg
  230. br.ErrMsg = err.Error()
  231. return
  232. }
  233. resp := models.AddEdbInfoResp{
  234. EdbInfoId: int(edbInfoId),
  235. UniqueCode: uniqueCode,
  236. }
  237. br.Ret = 200
  238. br.Success = true
  239. br.Msg = "保存成功"
  240. br.Data = resp
  241. br.IsAddLog = true
  242. }
  243. // editPredictCalculate 编辑预测指标运算
  244. func editPredictCalculate(br *models.BaseResponse, req models.EdbInfoCalculateSaveReq) {
  245. req.EdbName = strings.Trim(req.EdbName, " ")
  246. if req.EdbName == "" {
  247. br.Msg = "指标名称不能为空"
  248. return
  249. }
  250. if req.Frequency == "" {
  251. br.Msg = "频率不能为空"
  252. return
  253. }
  254. if req.Unit == "" {
  255. br.Msg = "单位不能为空"
  256. return
  257. }
  258. if req.ClassifyId <= 0 {
  259. br.Msg = "请选择分类"
  260. return
  261. }
  262. if len(req.EdbInfoIdArr) <= 0 {
  263. br.Msg = "请选择指标"
  264. return
  265. }
  266. if req.CalculateFormula == "" {
  267. br.Msg = "请填写指标"
  268. return
  269. }
  270. //加入缓存机制,避免创建同一个名称的指标 start
  271. redisKey := fmt.Sprint("predict_edb_info:calculate:batch:save:", utils.DATA_SOURCE_PREDICT_CALCULATE, ":", req.EdbName)
  272. isExist := utils.Rc.IsExist(redisKey)
  273. if isExist {
  274. br.Msg = "指标正在处理,请勿重复提交"
  275. return
  276. } else {
  277. //设置3分钟缓存
  278. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  279. defer func() {
  280. _ = utils.Rc.Delete(redisKey)
  281. }()
  282. }
  283. calculateFormula := req.CalculateFormula
  284. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  285. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  286. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  287. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  288. if strings.Contains(req.CalculateFormula, "%") {
  289. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  290. }
  291. req.CalculateFormula = calculateFormula
  292. //检验公式
  293. var formulaStr string
  294. var edbInfoIdBytes []string
  295. for _, v := range req.EdbInfoIdArr {
  296. formulaStr += v.FromTag + ","
  297. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  298. }
  299. formulaMap := services.CheckFormula(req.CalculateFormula)
  300. for _, v := range formulaMap {
  301. if !strings.Contains(formulaStr, v) {
  302. br.Msg = "公式错误,请重新填写"
  303. return
  304. }
  305. }
  306. // 获取计算指标信息
  307. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  308. if err != nil {
  309. br.Msg = "获取指标信息失败"
  310. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  311. return
  312. }
  313. // 判断指标名称是否存在
  314. {
  315. var condition string
  316. var pars []interface{}
  317. // 指标类型,0:普通指标,1:预测指标
  318. condition += " AND edb_info_id !=? AND edb_info_type=? AND edb_name=? "
  319. pars = append(pars, edbInfo.EdbInfoId, edbInfo.EdbInfoType, req.EdbName)
  320. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  321. if err != nil {
  322. br.Msg = "判断指标名称是否存在失败"
  323. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  324. return
  325. }
  326. if count > 0 {
  327. br.Msg = "指标名称已存在,请重新填写"
  328. br.ErrMsg = "指标名称已存在,请重新填写"
  329. return
  330. }
  331. }
  332. // 关联指标信息
  333. edbInfoList := make([]*models.EdbInfo, 0)
  334. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  335. for k, v := range req.EdbInfoIdArr {
  336. fromEdbInfo, err := models.GetEdbInfoById(v.EdbInfoId)
  337. if err != nil {
  338. if err.Error() == utils.ErrNoRow() {
  339. br.Msg = "生成计算指标失败"
  340. br.Msg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  341. return
  342. }
  343. br.Msg = "生成计算指标失败"
  344. br.Msg = "获取指标失败:Err:" + err.Error()
  345. return
  346. }
  347. edbInfoList = append(edbInfoList, fromEdbInfo)
  348. //关联关系表
  349. {
  350. calculateMappingItem := &models.EdbInfoCalculateMapping{
  351. CreateTime: time.Now(),
  352. ModifyTime: time.Now(),
  353. Sort: k + 1,
  354. EdbCode: edbInfo.EdbCode,
  355. EdbInfoId: edbInfo.EdbInfoId,
  356. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  357. FromEdbCode: fromEdbInfo.EdbCode,
  358. FromEdbName: fromEdbInfo.EdbName,
  359. FromSource: fromEdbInfo.Source,
  360. FromSourceName: fromEdbInfo.SourceName,
  361. FromTag: v.FromTag,
  362. Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  363. SourceName: "预测指标运算",
  364. }
  365. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  366. }
  367. }
  368. //处理同名指标
  369. {
  370. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  371. if err != nil {
  372. br.Msg = "保存失败"
  373. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  374. return
  375. }
  376. if len(edbNameList) >= 2 {
  377. for _, v := range edbNameList {
  378. edbName := v.EdbName + "(" + v.SourceName + ")"
  379. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  380. if err != nil {
  381. br.Msg = "保存失败"
  382. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  383. return
  384. }
  385. }
  386. }
  387. }
  388. edbInfo.EdbName = req.EdbName
  389. edbInfo.Frequency = req.Frequency
  390. edbInfo.Unit = req.Unit
  391. edbInfo.ClassifyId = req.ClassifyId
  392. edbInfo.CalculateFormula = req.CalculateFormula
  393. err = models.EditPredictCalculate(edbInfo, calculateMappingList)
  394. if err != nil {
  395. br.Msg = "修改计算指标失败"
  396. br.Msg = "修改计算指标失败,Calculate Err:" + err.Error()
  397. return
  398. }
  399. // 开始添加预测指标
  400. latestDateStr, latestValue, err := models.AddPredictCalculateData(edbInfoList, edbInfo.EdbInfoId, edbInfo.EdbCode, req.CalculateFormula, edbInfoIdBytes)
  401. if err != nil {
  402. br.Msg = "生成计算指标失败"
  403. br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
  404. return
  405. }
  406. // 更新指标最大最小值
  407. err, errMsg := models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  408. if err != nil {
  409. br.Msg = errMsg
  410. br.ErrMsg = err.Error()
  411. return
  412. }
  413. resp := models.AddEdbInfoResp{
  414. EdbInfoId: edbInfo.EdbInfoId,
  415. UniqueCode: edbInfo.UniqueCode,
  416. }
  417. br.Ret = 200
  418. br.Success = true
  419. br.Msg = "保存成功"
  420. br.Data = resp
  421. br.IsAddLog = true
  422. }
  423. // CalculateBatchSave
  424. // @Title 累计值转月-同比值-同差等计算新增
  425. // @Description 累计值转月-同比值-同差等计算新增接口
  426. // @Param request body models.EdbInfoCalculateBatchSaveReq true "type json string"
  427. // @Success Ret=200 返回指标id
  428. // @router /batch/save [post]
  429. func (this *PredictCalculateController) CalculateBatchSave() {
  430. br := new(models.BaseResponse).Init()
  431. defer func() {
  432. this.Data["json"] = br
  433. this.ServeJSON()
  434. }()
  435. var req models.EdbInfoCalculateBatchSaveReq
  436. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  437. if err != nil {
  438. br.Msg = "参数解析异常!"
  439. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  440. return
  441. }
  442. req.EdbName = strings.Trim(req.EdbName, " ")
  443. if req.EdbName == "" {
  444. br.Msg = "指标名称不能为空"
  445. return
  446. }
  447. if req.Frequency == "" {
  448. br.Msg = "频率不能为空"
  449. return
  450. }
  451. if req.Unit == "" {
  452. br.Msg = "单位不能为空"
  453. return
  454. }
  455. if req.ClassifyId <= 0 {
  456. br.Msg = "请选择分类"
  457. return
  458. }
  459. if req.FromEdbInfoId <= 0 {
  460. br.Msg = "请选择指标"
  461. return
  462. }
  463. //加入缓存机制,避免创建同一个名称的指标 start
  464. redisKey := fmt.Sprint("edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
  465. isExist := utils.Rc.IsExist(redisKey)
  466. if isExist {
  467. br.Msg = "指标正在处理,请勿重复提交"
  468. return
  469. } else {
  470. //设置3分钟缓存
  471. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  472. defer func() {
  473. _ = utils.Rc.Delete(redisKey)
  474. }()
  475. }
  476. //加入缓存机制,避免创建同一个名称的指标 end
  477. var condition string
  478. var pars []interface{}
  479. condition += " AND edb_name=? "
  480. pars = append(pars, req.EdbName)
  481. if req.EdbInfoId > 0 {
  482. condition += " AND edb_info_id !=? "
  483. pars = append(pars, req.EdbInfoId)
  484. }
  485. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  486. if err != nil {
  487. br.Msg = "判断指标名称是否存在失败"
  488. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  489. return
  490. }
  491. if count > 0 {
  492. br.Msg = "指标名称已存在,请重新填写"
  493. br.ErrMsg = "指标名称已存在,请重新填写"
  494. return
  495. }
  496. // 来源预测指标信息
  497. fromEdbInfo, err := models.GetEdbInfoById(req.FromEdbInfoId)
  498. if err != nil {
  499. br.Msg = "获取指标信息失败"
  500. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  501. return
  502. }
  503. if fromEdbInfo.EdbInfoType != 1 {
  504. br.Msg = "来源指标不是预测指标"
  505. br.ErrMsg = "来源指标不是预测指标"
  506. return
  507. }
  508. //生成指标编码
  509. randStr := utils.GetRandDigit(4)
  510. edbCode := `C` + time.Now().Format("060102") + randStr
  511. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  512. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  513. adminId := req.AdminId
  514. adminName := req.AdminName
  515. var sourName string
  516. var errMsg string
  517. var edbInfo *models.EdbInfo
  518. var latestDateStr string // 最近实际数据的日期
  519. var latestValue float64 // 最近实际数据的值
  520. if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ {
  521. sourName = "预测同比"
  522. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTbz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  523. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ {
  524. sourName = "预测同差"
  525. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTcz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  526. } else {
  527. br.Msg = "无效计算方式"
  528. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  529. return
  530. }
  531. if err != nil {
  532. br.Msg = "生成" + sourName + "失败"
  533. if errMsg != `` {
  534. br.Msg = errMsg
  535. }
  536. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  537. return
  538. }
  539. if edbInfo == nil {
  540. br.Msg = "生成" + sourName + "失败"
  541. br.ErrMsg = "生成" + sourName + "失败,edbInfo is nil"
  542. return
  543. }
  544. //处理同名指标
  545. {
  546. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  547. if err != nil {
  548. br.Msg = "保存失败"
  549. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  550. return
  551. }
  552. if len(edbNameList) >= 2 {
  553. for _, v := range edbNameList {
  554. edbName := v.EdbName + "(" + v.SourceName + ")"
  555. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  556. if err != nil {
  557. br.Msg = "保存失败"
  558. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  559. return
  560. }
  561. }
  562. }
  563. }
  564. // 更新指标最大最小值
  565. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  566. if err != nil {
  567. br.Msg = errMsg
  568. br.ErrMsg = err.Error()
  569. return
  570. }
  571. resp := models.AddEdbInfoResp{
  572. EdbInfoId: edbInfo.EdbInfoId,
  573. UniqueCode: uniqueCode,
  574. }
  575. br.Ret = 200
  576. br.Success = true
  577. br.Msg = "保存成功"
  578. br.Data = resp
  579. br.IsAddLog = true
  580. }
  581. // Refresh
  582. // @Title 刷新计算指标接口
  583. // @Description 刷新计算指标接口
  584. // @Success 200 {object} models.RefreshEdbInfoReq
  585. // @router /refresh [post]
  586. func (this *PredictCalculateController) Refresh() {
  587. br := new(models.BaseResponse).Init()
  588. var cacheKey string
  589. defer func() {
  590. this.Data["json"] = br
  591. this.ServeJSON()
  592. }()
  593. var req models.RefreshEdbInfoReq
  594. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  595. if err != nil {
  596. br.Msg = "参数解析异常!"
  597. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  598. return
  599. }
  600. if req.EdbCode == "" {
  601. br.Msg = "请输入指标编码!"
  602. br.ErrMsg = "请输入指标编码,指标编码为空"
  603. return
  604. }
  605. if req.EdbInfoId <= 0 {
  606. br.Msg = "请输入指标ID!"
  607. br.ErrMsg = "请输入指标ID"
  608. return
  609. }
  610. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  611. if err != nil {
  612. br.Msg = "指标不存在!"
  613. br.ErrMsg = "指标不存在"
  614. return
  615. }
  616. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  617. if utils.Rc.IsExist(cacheKey) {
  618. br.Ret = 501
  619. br.Success = true
  620. br.Msg = "系统处理中,请稍后重试"
  621. br.IsSendEmail = false
  622. }
  623. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  624. defer func() {
  625. _ = utils.Rc.Delete(cacheKey)
  626. }()
  627. startDate := req.StartDate
  628. var errMsg string
  629. edbInfoId := edbInfo.EdbInfoId
  630. source := edbInfo.Source
  631. var latestDateStr string // 最近实际数据的日期
  632. var latestValue float64 // 最近实际数据的值
  633. switch source {
  634. case utils.DATA_SOURCE_PREDICT_CALCULATE:
  635. //startDate = edbInfo.StartDate
  636. //sTime, err := time.Parse(utils.FormatDate, edbInfo.EndDate)
  637. //if err != nil {
  638. // return
  639. //}
  640. //startDate = sTime.Format(utils.FormatDate)
  641. startDate = ""
  642. var edbInfoIdBytes []string
  643. calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  644. if err != nil {
  645. errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  646. break
  647. }
  648. var formulaStr string
  649. edbInfoList := make([]*models.EdbInfo, 0)
  650. for _, v := range calculateMap {
  651. formulaStr += v.FromTag + ","
  652. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  653. edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  654. edbInfoList = append(edbInfoList, edbInfo)
  655. }
  656. latestDateStr, latestValue, err = models.RefreshAllPredictCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, edbInfoIdBytes)
  657. if err != nil && err.Error() != utils.ErrNoRow() {
  658. errMsg = "RefreshCalculate Err:" + err.Error()
  659. break
  660. }
  661. case utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ: //刷新同比值
  662. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  663. if err != nil {
  664. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  665. break
  666. }
  667. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  668. if err != nil {
  669. errMsg = "GetEdbInfoById Err:" + err.Error()
  670. break
  671. }
  672. startDate = edbInfo.StartDate
  673. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate)
  674. if err != nil && err.Error() != utils.ErrNoRow() {
  675. errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  676. break
  677. }
  678. case utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ: //同差值
  679. calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  680. if err != nil {
  681. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  682. break
  683. }
  684. fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  685. if err != nil {
  686. errMsg = "GetEdbInfoById Err:" + err.Error()
  687. break
  688. }
  689. startDate = edbInfo.StartDate
  690. latestDateStr, latestValue, err = models.RefreshAllPredictCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate)
  691. if err != nil && err.Error() != utils.ErrNoRow() {
  692. errMsg = "RefreshCalculateTcz Err:" + err.Error()
  693. break
  694. }
  695. default:
  696. br.Msg = "来源异常,请联系相关开发!"
  697. br.ErrMsg = "来源异常,请联系相关开发"
  698. return
  699. }
  700. if errMsg != `` {
  701. br.Msg = "刷新指标失败!"
  702. br.ErrMsg = "刷新指标失败,err:" + errMsg
  703. return
  704. }
  705. // 更新指标最大最小值
  706. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  707. if err != nil {
  708. br.Msg = errMsg
  709. br.ErrMsg = err.Error()
  710. return
  711. }
  712. br.Ret = 200
  713. br.Success = true
  714. br.Msg = "获取成功"
  715. }