base_from_predict_calculate.go 26 KB

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