base_from_predict_calculate.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920
  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. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  482. if err != nil {
  483. br.Msg = "判断指标名称是否存在失败"
  484. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  485. return
  486. }
  487. if count > 0 {
  488. br.Msg = "指标名称已存在,请重新填写"
  489. br.ErrMsg = "指标名称已存在,请重新填写"
  490. return
  491. }
  492. // 来源预测指标信息
  493. fromEdbInfo, err := models.GetEdbInfoById(req.FromEdbInfoId)
  494. if err != nil {
  495. br.Msg = "获取指标信息失败"
  496. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  497. return
  498. }
  499. if fromEdbInfo.EdbInfoType != 1 {
  500. br.Msg = "来源指标不是预测指标"
  501. br.ErrMsg = "来源指标不是预测指标"
  502. return
  503. }
  504. //生成指标编码
  505. randStr := utils.GetRandDigit(4)
  506. edbCode := `C` + time.Now().Format("060102") + randStr
  507. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  508. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  509. adminId := req.AdminId
  510. adminName := req.AdminName
  511. var sourName string
  512. var errMsg string
  513. var edbInfo *models.EdbInfo
  514. var latestDateStr string // 最近实际数据的日期
  515. var latestValue float64 // 最近实际数据的值
  516. if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ {
  517. sourName = "预测同比"
  518. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTbz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  519. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ {
  520. sourName = "预测同差"
  521. edbInfo, latestDateStr, latestValue, err, errMsg = models.SavePredictCalculateTcz(req.EdbInfoId, req.ClassifyId, req.EdbName, req.Frequency, req.Unit, req.Formula, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  522. } else {
  523. br.Msg = "无效计算方式"
  524. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  525. return
  526. }
  527. if err != nil {
  528. br.Msg = "生成" + sourName + "失败"
  529. if errMsg != `` {
  530. br.Msg = errMsg
  531. }
  532. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  533. return
  534. }
  535. if edbInfo == nil {
  536. br.Msg = "生成" + sourName + "失败"
  537. br.ErrMsg = "生成" + sourName + "失败,edbInfo is nil"
  538. return
  539. }
  540. //处理同名指标
  541. {
  542. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  543. if err != nil {
  544. br.Msg = "保存失败"
  545. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  546. return
  547. }
  548. if len(edbNameList) >= 2 {
  549. for _, v := range edbNameList {
  550. edbName := v.EdbName + "(" + v.SourceName + ")"
  551. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  552. if err != nil {
  553. br.Msg = "保存失败"
  554. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  555. return
  556. }
  557. }
  558. }
  559. }
  560. // 更新指标最大最小值
  561. err, errMsg = models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  562. if err != nil {
  563. br.Msg = errMsg
  564. br.ErrMsg = err.Error()
  565. return
  566. }
  567. resp := models.AddEdbInfoResp{
  568. EdbInfoId: edbInfo.EdbInfoId,
  569. UniqueCode: uniqueCode,
  570. }
  571. br.Ret = 200
  572. br.Success = true
  573. br.Msg = "保存成功"
  574. br.Data = resp
  575. br.IsAddLog = true
  576. }
  577. // Refresh
  578. // @Title 刷新计算指标接口
  579. // @Description 刷新计算指标接口
  580. // @Success 200 {object} models.RefreshEdbInfoReq
  581. // @router /refresh [post]
  582. //func (this *PredictCalculateController) Refresh() {
  583. // br := new(models.BaseResponse).Init()
  584. // var cacheKey string
  585. // defer func() {
  586. // this.Data["json"] = br
  587. // this.ServeJSON()
  588. // }()
  589. // var req models.RefreshEdbInfoReq
  590. // err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  591. // if err != nil {
  592. // br.Msg = "参数解析异常!"
  593. // br.ErrMsg = "参数解析失败,Err:" + err.Error()
  594. // return
  595. // }
  596. // if req.EdbCode == "" {
  597. // br.Msg = "请输入指标编码!"
  598. // br.ErrMsg = "请输入指标编码,指标编码为空"
  599. // return
  600. // }
  601. // if req.EdbInfoId <= 0 {
  602. // br.Msg = "请输入指标ID!"
  603. // br.ErrMsg = "请输入指标ID"
  604. // return
  605. // }
  606. // edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  607. // if err != nil {
  608. // br.Msg = "指标不存在!"
  609. // br.ErrMsg = "指标不存在"
  610. // return
  611. // }
  612. //
  613. // cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  614. // if !utils.Rc.IsExist(cacheKey) {
  615. // utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  616. // defer func() {
  617. // _ = utils.Rc.Delete(cacheKey)
  618. //
  619. // }()
  620. // startDate := req.StartDate
  621. // var errMsg string
  622. // endDate := time.Now().Format(utils.FormatDate)
  623. // edbInfoId := edbInfo.EdbInfoId
  624. // source := edbInfo.Source
  625. //
  626. // switch source {
  627. // case utils.DATA_SOURCE_CALCULATE:
  628. // //startDate = edbInfo.StartDate
  629. // //sTime, err := time.Parse(utils.FormatDate, edbInfo.EndDate)
  630. // //if err != nil {
  631. // // return
  632. // //}
  633. // //startDate = sTime.Format(utils.FormatDate)
  634. // startDate = ""
  635. // var edbInfoIdBytes []string
  636. // calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  637. // if err != nil {
  638. // errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  639. // break
  640. // }
  641. // var formulaStr string
  642. // edbInfoList := make([]*models.EdbInfo, 0)
  643. //
  644. // for _, v := range calculateMap {
  645. // formulaStr += v.FromTag + ","
  646. // edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  647. // edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  648. // edbInfoList = append(edbInfoList, edbInfo)
  649. // }
  650. // err = models.RefreshAllCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, endDate, edbInfoIdBytes)
  651. // if err != nil && err.Error() != utils.ErrNoRow() {
  652. // errMsg = "RefreshCalculate Err:" + err.Error()
  653. // break
  654. // }
  655. // case utils.DATA_SOURCE_CALCULATE_LJZZY: //刷新累计值转月值
  656. // calculateLjzzy, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  657. // if err != nil {
  658. // errMsg = "GetEdbInfoCalculateLjzzyDetail Err:" + err.Error()
  659. // break
  660. // }
  661. // fromEdbInfo, err := models.GetEdbInfoById(calculateLjzzy.FromEdbInfoId)
  662. // if err != nil {
  663. // errMsg = "GetEdbInfoById Err:" + err.Error()
  664. // break
  665. // }
  666. // startDate = edbInfo.StartDate
  667. // endDate = time.Now().Format(utils.FormatDate)
  668. // err = models.RefreshAllCalculateLjzzy(edbInfoId, source, fromEdbInfo, calculateLjzzy.EdbCode, startDate, endDate)
  669. // if err != nil && err.Error() != utils.ErrNoRow() {
  670. // errMsg = "RefreshAllCalculateLjzzy Err:" + err.Error()
  671. // break
  672. // }
  673. // case utils.DATA_SOURCE_CALCULATE_TBZ: //刷新同比值
  674. // calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  675. // if err != nil {
  676. // errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  677. // break
  678. // }
  679. // fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  680. // if err != nil {
  681. // errMsg = "GetEdbInfoById Err:" + err.Error()
  682. // break
  683. // }
  684. // startDate = edbInfo.StartDate
  685. // endDate = time.Now().Format(utils.FormatDate)
  686. // err = models.RefreshAllCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  687. // if err != nil && err.Error() != utils.ErrNoRow() {
  688. // errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  689. // break
  690. // }
  691. // case utils.DATA_SOURCE_CALCULATE_TCZ: //同差值
  692. // calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  693. // if err != nil {
  694. // errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  695. // break
  696. // }
  697. // fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  698. // if err != nil {
  699. // errMsg = "GetEdbInfoById Err:" + err.Error()
  700. // break
  701. // }
  702. // startDate = edbInfo.StartDate
  703. // endDate = time.Now().Format(utils.FormatDate)
  704. // err = models.RefreshAllCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate, endDate)
  705. // if err != nil && err.Error() != utils.ErrNoRow() {
  706. // errMsg = "RefreshCalculateTcz Err:" + err.Error()
  707. // break
  708. // }
  709. // case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS: //N数值移动平均计算
  710. // calculateNszydpjjs, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  711. // if err != nil {
  712. // errMsg = "GetEdbInfoCalculateNszydpjjsDetail Err:" + err.Error()
  713. // break
  714. // }
  715. // fromEdbInfo, err := models.GetEdbInfoById(calculateNszydpjjs.FromEdbInfoId)
  716. // if err != nil {
  717. // errMsg = "GetEdbInfoById Err:" + err.Error()
  718. // break
  719. // }
  720. // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  721. // startDate = edbInfo.StartDate
  722. // err = models.RefreshAllCalculateNszydpjjs(edbInfoId, edbInfo.Source, formulaInt, fromEdbInfo, calculateNszydpjjs.EdbCode, startDate)
  723. // if err != nil && err.Error() != utils.ErrNoRow() {
  724. // errMsg = "RefreshCalculateNszydpjjs Err:" + err.Error()
  725. // break
  726. // }
  727. // case utils.DATA_SOURCE_CALCULATE_HBZ: //刷新环比值
  728. // calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  729. // if err != nil {
  730. // errMsg = "GetEdbInfoCalculateHbzDetail Err:" + err.Error()
  731. // break
  732. // }
  733. // fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  734. // if err != nil {
  735. // errMsg = "GetEdbInfoById Err:" + err.Error()
  736. // break
  737. // }
  738. // startDate = edbInfo.StartDate
  739. // endDate = time.Now().Format(utils.FormatDate)
  740. // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  741. // err = models.RefreshAllCalculateHbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  742. // if err != nil && err.Error() != utils.ErrNoRow() {
  743. // errMsg = "RefreshAllCalculateHbz Err:" + err.Error()
  744. // break
  745. // }
  746. // case utils.DATA_SOURCE_CALCULATE_HCZ: //刷新环差值
  747. // calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  748. // if err != nil {
  749. // errMsg = "GetEdbInfoCalculateHczDetail Err:" + err.Error()
  750. // break
  751. // }
  752. // fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  753. // if err != nil {
  754. // errMsg = "GetEdbInfoById Err:" + err.Error()
  755. // break
  756. // }
  757. // startDate = edbInfo.StartDate
  758. // endDate = time.Now().Format(utils.FormatDate)
  759. // formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  760. // err = models.RefreshAllCalculateHcz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  761. // if err != nil && err.Error() != utils.ErrNoRow() {
  762. // errMsg = "RefreshAllCalculateHcz Err:" + err.Error()
  763. // break
  764. // }
  765. // case utils.DATA_SOURCE_CALCULATE_BP: //刷新变频
  766. // calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  767. // if err != nil {
  768. // errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  769. // break
  770. // }
  771. // fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  772. // if err != nil {
  773. // errMsg = "GetEdbInfoById Err:" + err.Error()
  774. // break
  775. // }
  776. // startDate = edbInfo.StartDate
  777. // endDate = time.Now().Format(utils.FormatDate)
  778. // err = models.RefreshAllCalculateBp(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  779. // if err != nil && err.Error() != utils.ErrNoRow() {
  780. // errMsg = "RefreshAllCalculateBp Err:" + err.Error()
  781. // break
  782. // }
  783. // case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT:
  784. // calculate, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  785. // if err != nil {
  786. // errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  787. // break
  788. // }
  789. // fromEdbInfo, err := models.GetEdbInfoById(calculate.FromEdbInfoId)
  790. // if err != nil {
  791. // errMsg = "GetEdbInfoById Err:" + err.Error()
  792. // break
  793. // }
  794. // startDate = edbInfo.StartDate
  795. // endDate = time.Now().Format(utils.FormatDate)
  796. // formulaInt, _ := strconv.Atoi(calculate.CalculateFormula)
  797. // err = models.RefreshAllCalculateTimeShift(edbInfoId, source, formulaInt, calculate.MoveType, fromEdbInfo, calculate.EdbCode, startDate, endDate, calculate.MoveFrequency)
  798. // if err != nil && err.Error() != utils.ErrNoRow() {
  799. // errMsg = "RefreshAllCalculateTimeShift Err:" + err.Error()
  800. // }
  801. // case utils.DATA_SOURCE_CALCULATE_ZJPJ: //刷新直接拼接
  802. // err = models.RefreshAllCalculateZjpj(edbInfo)
  803. // if err != nil && err.Error() != utils.ErrNoRow() {
  804. // errMsg = "RefreshAllCalculateZjpj Err:" + err.Error()
  805. // break
  806. // }
  807. // case utils.DATA_SOURCE_CALCULATE_LJZTBPJ: //刷新累计值同比拼接
  808. // err = models.RefreshAllCalculateLjztbpj(edbInfo)
  809. // if err != nil && err.Error() != utils.ErrNoRow() {
  810. // errMsg = "RefreshAllCalculateLjztbpj Err:" + err.Error()
  811. // break
  812. // }
  813. // case utils.DATA_SOURCE_PYTHON: //python代码运算
  814. // edbPythonCode, err := models.GetEdbPythonCodeById(edbInfo.EdbInfoId)
  815. // if err != nil {
  816. // errMsg = "获取python代码失败 Err:" + err.Error()
  817. // break
  818. // }
  819. // edbData, err, errMsg := services.ExecPythonCode(edbInfo.EdbCode, edbPythonCode.PythonCode)
  820. // if err != nil {
  821. // br.Msg = "获取数据失败"
  822. // br.ErrMsg = "python代码获取数据失败,err:" + err.Error()
  823. // if errMsg != "" {
  824. // br.ErrMsg = errMsg
  825. // }
  826. // return
  827. // }
  828. // err = models.RefreshAllPythonEdb(edbInfo, edbData)
  829. // if err != nil && err.Error() != utils.ErrNoRow() {
  830. // errMsg = "RefreshAllPythonEdb Err:" + err.Error()
  831. // break
  832. // }
  833. //
  834. // default:
  835. // br.Msg = "来源异常,请联系相关开发!"
  836. // br.ErrMsg = "来源异常,请联系相关开发"
  837. // return
  838. // }
  839. // if errMsg != `` {
  840. // br.Msg = "刷新指标失败!"
  841. // br.ErrMsg = "刷新指标失败,err:" + errMsg
  842. // return
  843. // }
  844. //
  845. // // 更新指标最大最小值
  846. // err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  847. // if err != nil {
  848. // br.Msg = errMsg
  849. // br.ErrMsg = err.Error()
  850. // return
  851. // }
  852. // br.Ret = 200
  853. // br.Success = true
  854. // br.Msg = "获取成功"
  855. // } else {
  856. // br.Ret = 501
  857. // br.Success = true
  858. // br.Msg = "系统处理中,请稍后重试"
  859. // }
  860. //}