base_from_predict_calculate.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744
  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. // Refresh
  424. // @Title 刷新计算指标接口
  425. // @Description 刷新计算指标接口
  426. // @Success 200 {object} models.RefreshEdbInfoReq
  427. // @router /refresh [post]
  428. func (this *PredictCalculateController) Refresh() {
  429. br := new(models.BaseResponse).Init()
  430. var cacheKey string
  431. defer func() {
  432. utils.Rc.Delete(cacheKey)
  433. this.Data["json"] = br
  434. this.ServeJSON()
  435. }()
  436. var req models.RefreshEdbInfoReq
  437. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  438. if err != nil {
  439. br.Msg = "参数解析异常!"
  440. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  441. return
  442. }
  443. if req.EdbCode == "" {
  444. br.Msg = "请输入指标编码!"
  445. br.ErrMsg = "请输入指标编码,指标编码为空"
  446. return
  447. }
  448. if req.EdbInfoId <= 0 {
  449. br.Msg = "请输入指标ID!"
  450. br.ErrMsg = "请输入指标ID"
  451. return
  452. }
  453. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  454. if err != nil {
  455. br.Msg = "指标不存在!"
  456. br.ErrMsg = "指标不存在"
  457. return
  458. }
  459. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  460. if !utils.Rc.IsExist(cacheKey) {
  461. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  462. startDate := req.StartDate
  463. var errMsg string
  464. endDate := time.Now().Format(utils.FormatDate)
  465. edbInfoId := edbInfo.EdbInfoId
  466. source := edbInfo.Source
  467. switch source {
  468. case utils.DATA_SOURCE_CALCULATE:
  469. //startDate = edbInfo.StartDate
  470. //sTime, err := time.Parse(utils.FormatDate, edbInfo.EndDate)
  471. //if err != nil {
  472. // return
  473. //}
  474. //startDate = sTime.Format(utils.FormatDate)
  475. startDate = ""
  476. var edbInfoIdBytes []string
  477. calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  478. if err != nil {
  479. errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  480. break
  481. }
  482. var formulaStr string
  483. edbInfoList := make([]*models.EdbInfo, 0)
  484. for _, v := range calculateMap {
  485. formulaStr += v.FromTag + ","
  486. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  487. edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  488. edbInfoList = append(edbInfoList, edbInfo)
  489. }
  490. err = models.RefreshAllCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, endDate, edbInfoIdBytes)
  491. if err != nil && err.Error() != utils.ErrNoRow() {
  492. errMsg = "RefreshCalculate Err:" + err.Error()
  493. break
  494. }
  495. case utils.DATA_SOURCE_CALCULATE_LJZZY: //刷新累计值转月值
  496. calculateLjzzy, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  497. if err != nil {
  498. errMsg = "GetEdbInfoCalculateLjzzyDetail Err:" + err.Error()
  499. break
  500. }
  501. fromEdbInfo, err := models.GetEdbInfoById(calculateLjzzy.FromEdbInfoId)
  502. if err != nil {
  503. errMsg = "GetEdbInfoById Err:" + err.Error()
  504. break
  505. }
  506. startDate = edbInfo.StartDate
  507. endDate = time.Now().Format(utils.FormatDate)
  508. err = models.RefreshAllCalculateLjzzy(edbInfoId, source, fromEdbInfo, calculateLjzzy.EdbCode, startDate, endDate)
  509. if err != nil && err.Error() != utils.ErrNoRow() {
  510. errMsg = "RefreshAllCalculateLjzzy Err:" + err.Error()
  511. break
  512. }
  513. case utils.DATA_SOURCE_CALCULATE_TBZ: //刷新同比值
  514. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  515. if err != nil {
  516. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  517. break
  518. }
  519. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  520. if err != nil {
  521. errMsg = "GetEdbInfoById Err:" + err.Error()
  522. break
  523. }
  524. startDate = edbInfo.StartDate
  525. endDate = time.Now().Format(utils.FormatDate)
  526. err = models.RefreshAllCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  527. if err != nil && err.Error() != utils.ErrNoRow() {
  528. errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  529. break
  530. }
  531. case utils.DATA_SOURCE_CALCULATE_TCZ: //同差值
  532. calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  533. if err != nil {
  534. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  535. break
  536. }
  537. fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  538. if err != nil {
  539. errMsg = "GetEdbInfoById Err:" + err.Error()
  540. break
  541. }
  542. startDate = edbInfo.StartDate
  543. endDate = time.Now().Format(utils.FormatDate)
  544. err = models.RefreshAllCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate, endDate)
  545. if err != nil && err.Error() != utils.ErrNoRow() {
  546. errMsg = "RefreshCalculateTcz Err:" + err.Error()
  547. break
  548. }
  549. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS: //N数值移动平均计算
  550. calculateNszydpjjs, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  551. if err != nil {
  552. errMsg = "GetEdbInfoCalculateNszydpjjsDetail Err:" + err.Error()
  553. break
  554. }
  555. fromEdbInfo, err := models.GetEdbInfoById(calculateNszydpjjs.FromEdbInfoId)
  556. if err != nil {
  557. errMsg = "GetEdbInfoById Err:" + err.Error()
  558. break
  559. }
  560. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  561. startDate = edbInfo.StartDate
  562. err = models.RefreshAllCalculateNszydpjjs(edbInfoId, edbInfo.Source, formulaInt, fromEdbInfo, calculateNszydpjjs.EdbCode, startDate)
  563. if err != nil && err.Error() != utils.ErrNoRow() {
  564. errMsg = "RefreshCalculateNszydpjjs Err:" + err.Error()
  565. break
  566. }
  567. case utils.DATA_SOURCE_CALCULATE_HBZ: //刷新环比值
  568. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  569. if err != nil {
  570. errMsg = "GetEdbInfoCalculateHbzDetail Err:" + err.Error()
  571. break
  572. }
  573. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  574. if err != nil {
  575. errMsg = "GetEdbInfoById Err:" + err.Error()
  576. break
  577. }
  578. startDate = edbInfo.StartDate
  579. endDate = time.Now().Format(utils.FormatDate)
  580. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  581. err = models.RefreshAllCalculateHbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  582. if err != nil && err.Error() != utils.ErrNoRow() {
  583. errMsg = "RefreshAllCalculateHbz Err:" + err.Error()
  584. break
  585. }
  586. case utils.DATA_SOURCE_CALCULATE_HCZ: //刷新环差值
  587. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  588. if err != nil {
  589. errMsg = "GetEdbInfoCalculateHczDetail Err:" + err.Error()
  590. break
  591. }
  592. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  593. if err != nil {
  594. errMsg = "GetEdbInfoById Err:" + err.Error()
  595. break
  596. }
  597. startDate = edbInfo.StartDate
  598. endDate = time.Now().Format(utils.FormatDate)
  599. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  600. err = models.RefreshAllCalculateHcz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  601. if err != nil && err.Error() != utils.ErrNoRow() {
  602. errMsg = "RefreshAllCalculateHcz Err:" + err.Error()
  603. break
  604. }
  605. case utils.DATA_SOURCE_CALCULATE_BP: //刷新变频
  606. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  607. if err != nil {
  608. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  609. break
  610. }
  611. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  612. if err != nil {
  613. errMsg = "GetEdbInfoById Err:" + err.Error()
  614. break
  615. }
  616. startDate = edbInfo.StartDate
  617. endDate = time.Now().Format(utils.FormatDate)
  618. err = models.RefreshAllCalculateBp(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  619. if err != nil && err.Error() != utils.ErrNoRow() {
  620. errMsg = "RefreshAllCalculateBp Err:" + err.Error()
  621. break
  622. }
  623. case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT:
  624. calculate, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  625. if err != nil {
  626. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  627. break
  628. }
  629. fromEdbInfo, err := models.GetEdbInfoById(calculate.FromEdbInfoId)
  630. if err != nil {
  631. errMsg = "GetEdbInfoById Err:" + err.Error()
  632. break
  633. }
  634. startDate = edbInfo.StartDate
  635. endDate = time.Now().Format(utils.FormatDate)
  636. formulaInt, _ := strconv.Atoi(calculate.CalculateFormula)
  637. err = models.RefreshAllCalculateTimeShift(edbInfoId, source, formulaInt, calculate.MoveType, fromEdbInfo, calculate.EdbCode, startDate, endDate, calculate.MoveFrequency)
  638. if err != nil && err.Error() != utils.ErrNoRow() {
  639. errMsg = "RefreshAllCalculateTimeShift Err:" + err.Error()
  640. }
  641. case utils.DATA_SOURCE_CALCULATE_ZJPJ: //刷新直接拼接
  642. err = models.RefreshAllCalculateZjpj(edbInfo)
  643. if err != nil && err.Error() != utils.ErrNoRow() {
  644. errMsg = "RefreshAllCalculateZjpj Err:" + err.Error()
  645. break
  646. }
  647. case utils.DATA_SOURCE_CALCULATE_LJZTBPJ: //刷新累计值同比拼接
  648. err = models.RefreshAllCalculateLjztbpj(edbInfo)
  649. if err != nil && err.Error() != utils.ErrNoRow() {
  650. errMsg = "RefreshAllCalculateLjztbpj Err:" + err.Error()
  651. break
  652. }
  653. case utils.DATA_SOURCE_PYTHON: //python代码运算
  654. edbPythonCode, err := models.GetEdbPythonCodeById(edbInfo.EdbInfoId)
  655. if err != nil {
  656. errMsg = "获取python代码失败 Err:" + err.Error()
  657. break
  658. }
  659. edbData, err, errMsg := services.ExecPythonCode(edbInfo.EdbCode, edbPythonCode.PythonCode)
  660. if err != nil {
  661. br.Msg = "获取数据失败"
  662. br.ErrMsg = "python代码获取数据失败,err:" + err.Error()
  663. if errMsg != "" {
  664. br.ErrMsg = errMsg
  665. }
  666. return
  667. }
  668. err = models.RefreshAllPythonEdb(edbInfo, edbData)
  669. if err != nil && err.Error() != utils.ErrNoRow() {
  670. errMsg = "RefreshAllPythonEdb Err:" + err.Error()
  671. break
  672. }
  673. default:
  674. br.Msg = "来源异常,请联系相关开发!"
  675. br.ErrMsg = "来源异常,请联系相关开发"
  676. return
  677. }
  678. if errMsg != `` {
  679. br.Msg = "刷新指标失败!"
  680. br.ErrMsg = "刷新指标失败,err:" + errMsg
  681. return
  682. }
  683. // 更新指标最大最小值
  684. err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  685. if err != nil {
  686. br.Msg = errMsg
  687. br.ErrMsg = err.Error()
  688. return
  689. }
  690. br.Ret = 200
  691. br.Success = true
  692. br.Msg = "获取成功"
  693. } else {
  694. br.Ret = 501
  695. br.Success = true
  696. br.Msg = "系统处理中,请稍后重试"
  697. }
  698. }