base_from_predict_calculate.go 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407
  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. var formulaInt int
  464. if req.Source == utils.DATA_SOURCE_CALCULATE_NSZYDPJJS ||
  465. req.Source == utils.DATA_SOURCE_CALCULATE_HBZ ||
  466. req.Source == utils.DATA_SOURCE_CALCULATE_HCZ ||
  467. req.Source == utils.DATA_SOURCE_CALCULATE_TIME_SHIFT {
  468. if req.Formula == "" {
  469. br.Msg = "请填写N值"
  470. return
  471. }
  472. formulaInt, _ = strconv.Atoi(req.Formula)
  473. if formulaInt <= 0 {
  474. br.Msg = "N值输入错误,请重新输入"
  475. return
  476. }
  477. } else if req.Source == utils.DATA_SOURCE_CALCULATE_ZJPJ {
  478. //直接拼接指标
  479. //校验时间格式
  480. _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
  481. if err != nil {
  482. br.Msg = "拼接日期有误,请重新输入"
  483. return
  484. }
  485. }
  486. // 获取指标详情
  487. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  488. if err != nil {
  489. br.Msg = "指标不存在!"
  490. br.ErrMsg = "指标不存在"
  491. return
  492. }
  493. //加入缓存机制,避免创建同一个名称的指标 start
  494. redisKey := fmt.Sprint("edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
  495. isExist := utils.Rc.IsExist(redisKey)
  496. if isExist {
  497. br.Msg = "指标正在处理,请勿重复提交"
  498. return
  499. } else {
  500. //设置3分钟缓存
  501. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  502. defer func() {
  503. utils.Rc.Delete(redisKey)
  504. }()
  505. }
  506. //加入缓存机制,避免创建同一个名称的指标 end
  507. var condition string
  508. var pars []interface{}
  509. condition += " AND edb_name=? "
  510. pars = append(pars, req.EdbName)
  511. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  512. if err != nil {
  513. br.Msg = "判断指标名称是否存在失败"
  514. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  515. return
  516. }
  517. if count > 0 {
  518. br.Msg = "指标名称已存在,请重新填写"
  519. br.ErrMsg = "指标名称已存在,请重新填写"
  520. return
  521. }
  522. fromEdbInfo, err := models.GetEdbInfoById(req.FromEdbInfoId)
  523. if err != nil {
  524. br.Msg = "获取指标信息失败"
  525. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  526. return
  527. }
  528. //生成指标编码
  529. randStr := utils.GetRandDigit(4)
  530. edbCode := `C` + time.Now().Format("060102") + randStr
  531. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  532. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  533. adminId := req.AdminId
  534. adminName := req.AdminName
  535. var sourName string
  536. var edbInfoId int
  537. if req.Source == utils.DATA_SOURCE_CALCULATE_LJZZY {
  538. sourName = "累计值转月值"
  539. if fromEdbInfo.Frequency != "月度" {
  540. br.Msg = "请选择月度指标"
  541. return
  542. }
  543. edbInfoId, err = models.AddCalculateLjzzy(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  544. } else if req.Source == utils.DATA_SOURCE_CALCULATE_TBZ {
  545. sourName = "同比值"
  546. edbInfoId, err = models.AddCalculateTbz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  547. } else if req.Source == utils.DATA_SOURCE_CALCULATE_TCZ {
  548. sourName = "同差值"
  549. edbInfoId, err = models.AddCalculateTcz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  550. } else if req.Source == utils.DATA_SOURCE_CALCULATE_NSZYDPJJS {
  551. sourName = "N数值移动平均计算"
  552. edbInfoId, err = models.AddCalculateNszydpjjs(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  553. } else if req.Source == utils.DATA_SOURCE_CALCULATE_HBZ { //环比值
  554. var condition string
  555. var pars []interface{}
  556. condition += " AND edb_info_id =? "
  557. pars = append(pars, req.FromEdbInfoId)
  558. condition += " AND value <=0 "
  559. checkCount, err := models.GetEdbDataCount(condition, pars, fromEdbInfo.Source)
  560. if err != nil && err.Error() != utils.ErrNoRow() {
  561. br.Msg = "判断环比值是否可计算失败"
  562. br.ErrMsg = "判断环比值是否可计算失败,Err:" + err.Error()
  563. return
  564. }
  565. if checkCount > 0 {
  566. br.Msg = "原始数据中存在0或负数,该指标不能进行环比运算"
  567. br.ErrMsg = "原始数据中出现0和负值时,提示该指标不能进行环比运算"
  568. return
  569. }
  570. sourName = "环比值"
  571. edbInfoId, err = models.AddCalculateHbz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  572. } else if req.Source == utils.DATA_SOURCE_CALCULATE_HCZ {
  573. sourName = "环差值"
  574. edbInfoId, err = models.AddCalculateHcz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName, formulaInt)
  575. } else if req.Source == utils.DATA_SOURCE_CALCULATE_BP {
  576. if fromEdbInfo.Frequency == "日度" {
  577. br.Msg = "日度指标,无法进行变频操作"
  578. br.ErrMsg = "日度指标,无法进行变频操作:edbcode:" + fromEdbInfo.EdbCode
  579. return
  580. }
  581. sourName = "变频"
  582. edbInfoId, err = models.AddCalculateBp(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  583. } else if req.Source == utils.DATA_SOURCE_CALCULATE_TIME_SHIFT { //时间移位
  584. sourName = "时间移位"
  585. edbInfoId, err = models.AddCalculateTimeShift(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  586. } else if req.Source == utils.DATA_SOURCE_CALCULATE_ZJPJ { //直接拼接
  587. sourName = "直接拼接"
  588. if len(req.EdbInfoIdArr) != 1 {
  589. br.Msg = "请传入拼接日期之后的指标"
  590. br.ErrMsg = "请传入拼接日期之后的指标"
  591. return
  592. }
  593. secondEdbInfoReq := req.EdbInfoIdArr[0]
  594. secondEdbInfo, err := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  595. if err != nil {
  596. br.Msg = "获取拼接日期之后的指标信息失败"
  597. br.ErrMsg = "获取拼接日期之后的指标信息失败:Err:" + err.Error()
  598. return
  599. }
  600. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  601. br.Msg = "两个指标不允许为同一个"
  602. br.ErrMsg = "两个指标不允许为同一个"
  603. return
  604. }
  605. edbInfoId, err = models.AddCalculateZjpj(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, adminId, adminName)
  606. } else if req.Source == utils.DATA_SOURCE_CALCULATE_LJZTBPJ { //累计值同比拼接
  607. sourName = "累计值同比拼接"
  608. if fromEdbInfo.Frequency != "月度" {
  609. br.Msg = "待拼接指标只能筛选月度指标"
  610. br.ErrMsg = "待拼接指标只能筛选月度指标"
  611. return
  612. }
  613. if len(req.EdbInfoIdArr) != 1 {
  614. br.Msg = "请传入同比值指标"
  615. br.ErrMsg = "请传入同比值指标"
  616. return
  617. }
  618. secondEdbInfoReq := req.EdbInfoIdArr[0]
  619. tbzEdbInfo, err := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  620. if err != nil {
  621. br.Msg = "获取同比值指标信息失败"
  622. br.ErrMsg = "获取同比值指标信息失败:Err:" + err.Error()
  623. return
  624. }
  625. if tbzEdbInfo.Source != utils.DATA_SOURCE_CALCULATE_TBZ {
  626. br.Msg = "指标必须是传入同比值指标类型"
  627. br.ErrMsg = "指标必须是传入同比值指标类型"
  628. return
  629. }
  630. if tbzEdbInfo.Frequency != "月度" {
  631. br.Msg = "同比值指标只能筛选月度指标"
  632. br.ErrMsg = "同比值指标只能筛选月度指标"
  633. return
  634. }
  635. if fromEdbInfo.EdbInfoId == tbzEdbInfo.EdbInfoId {
  636. br.Msg = "两个指标不允许为同一个"
  637. br.ErrMsg = "两个指标不允许为同一个"
  638. return
  639. }
  640. edbInfoId, err = models.AddCalculateLjztbpj(&req, fromEdbInfo, tbzEdbInfo, edbCode, uniqueCode, adminId, adminName)
  641. } else {
  642. br.Msg = "无效计算方式"
  643. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  644. return
  645. }
  646. if err != nil {
  647. br.Msg = "生成" + sourName + "失败"
  648. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  649. return
  650. }
  651. if edbInfoId <= 0 {
  652. br.Msg = "生成" + sourName + "失败"
  653. br.ErrMsg = "生成" + sourName + "失败,指标ID错误:" + strconv.Itoa(edbInfoId)
  654. return
  655. }
  656. //处理同名指标
  657. {
  658. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  659. if err != nil {
  660. br.Msg = "保存失败"
  661. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  662. return
  663. }
  664. if len(edbNameList) >= 2 {
  665. for _, v := range edbNameList {
  666. edbName := v.EdbName + "(" + v.SourceName + ")"
  667. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  668. if err != nil {
  669. br.Msg = "保存失败"
  670. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  671. return
  672. }
  673. }
  674. }
  675. }
  676. // 更新指标最大最小值
  677. err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  678. if err != nil {
  679. br.Msg = errMsg
  680. br.ErrMsg = err.Error()
  681. return
  682. }
  683. resp := models.AddEdbInfoResp{
  684. EdbInfoId: edbInfoId,
  685. UniqueCode: uniqueCode,
  686. }
  687. br.Ret = 200
  688. br.Success = true
  689. br.Msg = "保存成功"
  690. br.Data = resp
  691. br.IsAddLog = true
  692. }
  693. func AddCalculateBatchSave(br *models.BaseResponse, req models.EdbInfoCalculateBatchSaveReq) {
  694. req.EdbName = strings.Trim(req.EdbName, " ")
  695. if req.EdbName == "" {
  696. br.Msg = "指标名称不能为空"
  697. return
  698. }
  699. if req.Frequency == "" {
  700. br.Msg = "频率不能为空"
  701. return
  702. }
  703. if req.Unit == "" {
  704. br.Msg = "单位不能为空"
  705. return
  706. }
  707. if req.ClassifyId <= 0 {
  708. br.Msg = "请选择分类"
  709. return
  710. }
  711. if req.FromEdbInfoId <= 0 {
  712. br.Msg = "请选择指标"
  713. return
  714. }
  715. // 获取指标详情
  716. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  717. if err != nil {
  718. br.Msg = "指标不存在!"
  719. br.ErrMsg = "指标不存在"
  720. return
  721. }
  722. //加入缓存机制,避免创建同一个名称的指标 start
  723. redisKey := fmt.Sprint("edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
  724. isExist := utils.Rc.IsExist(redisKey)
  725. if isExist {
  726. br.Msg = "指标正在处理,请勿重复提交"
  727. return
  728. } else {
  729. //设置3分钟缓存
  730. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  731. defer func() {
  732. utils.Rc.Delete(redisKey)
  733. }()
  734. }
  735. //加入缓存机制,避免创建同一个名称的指标 end
  736. var condition string
  737. var pars []interface{}
  738. condition += " AND edb_name=? "
  739. pars = append(pars, req.EdbName)
  740. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  741. if err != nil {
  742. br.Msg = "判断指标名称是否存在失败"
  743. br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  744. return
  745. }
  746. if count > 0 {
  747. br.Msg = "指标名称已存在,请重新填写"
  748. br.ErrMsg = "指标名称已存在,请重新填写"
  749. return
  750. }
  751. fromEdbInfo, err := models.GetEdbInfoById(req.FromEdbInfoId)
  752. if err != nil {
  753. br.Msg = "获取指标信息失败"
  754. br.ErrMsg = "获取指标信息失败:Err:" + err.Error()
  755. return
  756. }
  757. //生成指标编码
  758. randStr := utils.GetRandDigit(4)
  759. edbCode := `C` + time.Now().Format("060102") + randStr
  760. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  761. uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  762. adminId := req.AdminId
  763. adminName := req.AdminName
  764. var sourName string
  765. var edbInfoId int
  766. if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ {
  767. sourName = "预测同比"
  768. edbInfoId, err = models.AddPredictCalculateTbz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  769. } else if req.Source == utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ {
  770. sourName = "预测同差"
  771. edbInfoId, err = models.AddPredictCalculateTcz(&req, fromEdbInfo, edbCode, uniqueCode, adminId, adminName)
  772. } else {
  773. br.Msg = "无效计算方式"
  774. br.ErrMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  775. return
  776. }
  777. if err != nil {
  778. br.Msg = "生成" + sourName + "失败"
  779. br.Msg = "生成" + sourName + "失败 Err:" + err.Error()
  780. return
  781. }
  782. if edbInfoId <= 0 {
  783. br.Msg = "生成" + sourName + "失败"
  784. br.ErrMsg = "生成" + sourName + "失败,指标ID错误:" + strconv.Itoa(edbInfoId)
  785. return
  786. }
  787. //处理同名指标
  788. {
  789. edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  790. if err != nil {
  791. br.Msg = "保存失败"
  792. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  793. return
  794. }
  795. if len(edbNameList) >= 2 {
  796. for _, v := range edbNameList {
  797. edbName := v.EdbName + "(" + v.SourceName + ")"
  798. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  799. if err != nil {
  800. br.Msg = "保存失败"
  801. br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  802. return
  803. }
  804. }
  805. }
  806. }
  807. // 更新指标最大最小值
  808. err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  809. if err != nil {
  810. br.Msg = errMsg
  811. br.ErrMsg = err.Error()
  812. return
  813. }
  814. resp := models.AddEdbInfoResp{
  815. EdbInfoId: edbInfoId,
  816. UniqueCode: uniqueCode,
  817. }
  818. br.Ret = 200
  819. br.Success = true
  820. br.Msg = "保存成功"
  821. br.Data = resp
  822. br.IsAddLog = true
  823. }
  824. // AddPredictCalculate 添加预测指标运算
  825. //func addPredictCalculate(br *models.BaseResponse, req models.EdbInfoCalculateSaveReq) {
  826. //
  827. // req.EdbName = strings.Trim(req.EdbName, " ")
  828. // if req.EdbName == "" {
  829. // br.Msg = "指标名称不能为空"
  830. // return
  831. // }
  832. //
  833. // if req.Frequency == "" {
  834. // br.Msg = "频率不能为空"
  835. // return
  836. // }
  837. //
  838. // if req.Unit == "" {
  839. // br.Msg = "单位不能为空"
  840. // return
  841. // }
  842. //
  843. // if req.ClassifyId <= 0 {
  844. // br.Msg = "请选择分类"
  845. // return
  846. // }
  847. //
  848. // if len(req.EdbInfoIdArr) <= 0 {
  849. // br.Msg = "请选择指标"
  850. // return
  851. // }
  852. //
  853. // if req.CalculateFormula == "" {
  854. // br.Msg = "请填写指标"
  855. // return
  856. // }
  857. //
  858. // //加入缓存机制,避免创建同一个名称的指标 start
  859. // redisKey := fmt.Sprint("predict_edb_info:calculate:batch:save:", utils.DATA_SOURCE_PREDICT_CALCULATE, ":", req.EdbName)
  860. // isExist := utils.Rc.IsExist(redisKey)
  861. // if isExist {
  862. // br.Msg = "指标正在处理,请勿重复提交"
  863. // return
  864. // } else {
  865. // //设置3分钟缓存
  866. // utils.Rc.SetNX(redisKey, 1, time.Second*300)
  867. // defer func() {
  868. // utils.Rc.Delete(redisKey)
  869. // }()
  870. // }
  871. //
  872. // calculateFormula := req.CalculateFormula
  873. // calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  874. // calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  875. // calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  876. // calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  877. // if strings.Contains(req.CalculateFormula, "%") {
  878. // calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  879. // }
  880. // req.CalculateFormula = calculateFormula
  881. //
  882. // //检验公式
  883. // var formulaStr string
  884. // var edbInfoIdBytes []string
  885. // for _, v := range req.EdbInfoIdArr {
  886. // formulaStr += v.FromTag + ","
  887. // edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  888. // }
  889. // formulaMap := services.CheckFormula(req.CalculateFormula)
  890. // for _, v := range formulaMap {
  891. // if !strings.Contains(formulaStr, v) {
  892. // br.Msg = "公式错误,请重新填写"
  893. // return
  894. // }
  895. // }
  896. //
  897. // // 判断指标名称是否存在
  898. // {
  899. // var condition string
  900. // var pars []interface{}
  901. // // 指标类型,0:普通指标,1:预测指标
  902. // condition += " AND edb_info_type=? AND edb_name=? "
  903. // pars = append(pars, 1, req.EdbName)
  904. //
  905. // count, err := models.GetEdbInfoCountByCondition(condition, pars)
  906. // if err != nil {
  907. // br.Msg = "判断指标名称是否存在失败"
  908. // br.ErrMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  909. // return
  910. // }
  911. // if count > 0 {
  912. // br.Msg = "指标名称已存在,请重新填写"
  913. // br.ErrMsg = "指标名称已存在,请重新填写"
  914. // return
  915. // }
  916. // }
  917. //
  918. // // 关联指标信息
  919. // edbInfoList := make([]*models.EdbInfo, 0)
  920. // calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  921. // for k, v := range req.EdbInfoIdArr {
  922. // fromEdbInfo, err := models.GetEdbInfoById(v.EdbInfoId)
  923. // if err != nil {
  924. // if err.Error() == utils.ErrNoRow() {
  925. // br.Msg = "生成计算指标失败"
  926. // br.Msg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  927. // return
  928. // }
  929. // br.Msg = "生成计算指标失败"
  930. // br.Msg = "获取指标失败:Err:" + err.Error()
  931. // return
  932. // }
  933. // edbInfoList = append(edbInfoList, fromEdbInfo)
  934. //
  935. // //关联关系表
  936. // {
  937. // calculateMappingItem := &models.EdbInfoCalculateMapping{
  938. // CreateTime: time.Now(),
  939. // ModifyTime: time.Now(),
  940. // Sort: k + 1,
  941. // //EdbCode: edbCode,
  942. // //EdbInfoId: int(edbInfoId),
  943. // FromEdbInfoId: fromEdbInfo.EdbInfoId,
  944. // FromEdbCode: fromEdbInfo.EdbCode,
  945. // FromEdbName: fromEdbInfo.EdbName,
  946. // FromSource: fromEdbInfo.Source,
  947. // FromSourceName: fromEdbInfo.SourceName,
  948. // FromTag: v.FromTag,
  949. // Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  950. // SourceName: "预测指标运算",
  951. // }
  952. // calculateMappingList = append(calculateMappingList, calculateMappingItem)
  953. // }
  954. // }
  955. //
  956. // // 指标入库
  957. // randStr := utils.GetRandDigit(4)
  958. // edbCode := `C2` + time.Now().Format("060102") + randStr
  959. // timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  960. // uniqueCode := utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  961. // edbInfo := &models.EdbInfo{
  962. // EdbInfoType: 1,
  963. // Source: utils.DATA_SOURCE_PREDICT_CALCULATE,
  964. // SourceName: "预测指标运算",
  965. // EdbCode: edbCode,
  966. // EdbName: req.EdbName,
  967. // EdbNameSource: req.EdbName,
  968. // Frequency: req.Frequency,
  969. // Unit: req.Unit,
  970. // ClassifyId: req.ClassifyId,
  971. // SysUserId: req.AdminId,
  972. // SysUserRealName: req.AdminName,
  973. // CreateTime: time.Now(),
  974. // ModifyTime: time.Now(),
  975. // UniqueCode: uniqueCode,
  976. // CalculateFormula: req.CalculateFormula,
  977. // EdbType: 2,
  978. // }
  979. // edbInfoId, err := models.AddEdbInfo(edbInfo)
  980. // if err != nil {
  981. // br.Msg = "生成计算指标失败"
  982. // br.Msg = "生成计算指标失败,AddEdbInfo Err:" + err.Error()
  983. // return
  984. // }
  985. // edbInfo.EdbInfoId = int(edbInfoId)
  986. //
  987. // //处理同名指标
  988. // {
  989. // edbNameList, err := models.GetEdbInfoByName(req.EdbName)
  990. // if err != nil {
  991. // br.Msg = "保存失败"
  992. // br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  993. // return
  994. // }
  995. // if len(edbNameList) >= 2 {
  996. // for _, v := range edbNameList {
  997. // edbName := v.EdbName + "(" + v.SourceName + ")"
  998. // err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  999. // if err != nil {
  1000. // br.Msg = "保存失败"
  1001. // br.ErrMsg = "修改指标名称失败,Err:" + err.Error()
  1002. // return
  1003. // }
  1004. // }
  1005. // }
  1006. // }
  1007. //
  1008. // //关联关系表入库
  1009. // {
  1010. // if len(calculateMappingList) > 0 {
  1011. // for k, v := range calculateMappingList {
  1012. // v.EdbCode = edbInfo.EdbCode
  1013. // v.EdbInfoId = edbInfo.EdbInfoId
  1014. // calculateMappingList[k] = v
  1015. // }
  1016. // go models.AddEdbInfoCalculateMappingMulti(calculateMappingList)
  1017. // }
  1018. // }
  1019. //
  1020. // // 开始添加预测指标
  1021. // latestDateStr, latestValue, err := models.AddPredictCalculateData(edbInfoList, int(edbInfoId), edbCode, req.CalculateFormula, edbInfoIdBytes)
  1022. // if err != nil {
  1023. // br.Msg = "生成计算指标失败"
  1024. // br.Msg = "生成计算指标失败,Calculate Err:" + err.Error()
  1025. // return
  1026. // }
  1027. //
  1028. // // 更新指标最大最小值
  1029. // err, errMsg := models.UnifiedModifyPredictEdbInfoMaxAndMinInfo(edbInfo, latestDateStr, latestValue)
  1030. // if err != nil {
  1031. // br.Msg = errMsg
  1032. // br.ErrMsg = err.Error()
  1033. // return
  1034. // }
  1035. // resp := models.AddEdbInfoResp{
  1036. // EdbInfoId: int(edbInfoId),
  1037. // UniqueCode: uniqueCode,
  1038. // }
  1039. // br.Ret = 200
  1040. // br.Success = true
  1041. // br.Msg = "保存成功"
  1042. // br.Data = resp
  1043. // br.IsAddLog = true
  1044. //}
  1045. // Refresh
  1046. // @Title 刷新计算指标接口
  1047. // @Description 刷新计算指标接口
  1048. // @Success 200 {object} models.RefreshEdbInfoReq
  1049. // @router /refresh [post]
  1050. func (this *PredictCalculateController) Refresh() {
  1051. br := new(models.BaseResponse).Init()
  1052. var cacheKey string
  1053. defer func() {
  1054. utils.Rc.Delete(cacheKey)
  1055. this.Data["json"] = br
  1056. this.ServeJSON()
  1057. }()
  1058. var req models.RefreshEdbInfoReq
  1059. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1060. if err != nil {
  1061. br.Msg = "参数解析异常!"
  1062. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1063. return
  1064. }
  1065. if req.EdbCode == "" {
  1066. br.Msg = "请输入指标编码!"
  1067. br.ErrMsg = "请输入指标编码,指标编码为空"
  1068. return
  1069. }
  1070. if req.EdbInfoId <= 0 {
  1071. br.Msg = "请输入指标ID!"
  1072. br.ErrMsg = "请输入指标ID"
  1073. return
  1074. }
  1075. edbInfo, err := models.GetEdbInfoById(req.EdbInfoId)
  1076. if err != nil {
  1077. br.Msg = "指标不存在!"
  1078. br.ErrMsg = "指标不存在"
  1079. return
  1080. }
  1081. cacheKey = utils.CACHE_EDB_DATA_REFRESH + strconv.Itoa(edbInfo.Source) + "_" + req.EdbCode
  1082. if !utils.Rc.IsExist(cacheKey) {
  1083. utils.Rc.SetNX(cacheKey, 1, 1*time.Minute)
  1084. startDate := req.StartDate
  1085. var errMsg string
  1086. endDate := time.Now().Format(utils.FormatDate)
  1087. edbInfoId := edbInfo.EdbInfoId
  1088. source := edbInfo.Source
  1089. switch source {
  1090. case utils.DATA_SOURCE_CALCULATE:
  1091. //startDate = edbInfo.StartDate
  1092. //sTime, err := time.Parse(utils.FormatDate, edbInfo.EndDate)
  1093. //if err != nil {
  1094. // return
  1095. //}
  1096. //startDate = sTime.Format(utils.FormatDate)
  1097. startDate = ""
  1098. var edbInfoIdBytes []string
  1099. calculateMap, err := models.GetEdbInfoCalculateDetailList(edbInfo.EdbInfoId)
  1100. if err != nil {
  1101. errMsg = "GetEdbInfoCalculateDetail Err:" + err.Error()
  1102. break
  1103. }
  1104. var formulaStr string
  1105. edbInfoList := make([]*models.EdbInfo, 0)
  1106. for _, v := range calculateMap {
  1107. formulaStr += v.FromTag + ","
  1108. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  1109. edbInfo, _ := models.GetEdbInfoById(v.FromEdbInfoId)
  1110. edbInfoList = append(edbInfoList, edbInfo)
  1111. }
  1112. err = models.RefreshAllCalculate(edbInfoList, edbInfo.EdbInfoId, source, edbInfo.EdbCode, edbInfo.CalculateFormula, startDate, endDate, edbInfoIdBytes)
  1113. if err != nil && err.Error() != utils.ErrNoRow() {
  1114. errMsg = "RefreshCalculate Err:" + err.Error()
  1115. break
  1116. }
  1117. case utils.DATA_SOURCE_CALCULATE_LJZZY: //刷新累计值转月值
  1118. calculateLjzzy, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1119. if err != nil {
  1120. errMsg = "GetEdbInfoCalculateLjzzyDetail Err:" + err.Error()
  1121. break
  1122. }
  1123. fromEdbInfo, err := models.GetEdbInfoById(calculateLjzzy.FromEdbInfoId)
  1124. if err != nil {
  1125. errMsg = "GetEdbInfoById Err:" + err.Error()
  1126. break
  1127. }
  1128. startDate = edbInfo.StartDate
  1129. endDate = time.Now().Format(utils.FormatDate)
  1130. err = models.RefreshAllCalculateLjzzy(edbInfoId, source, fromEdbInfo, calculateLjzzy.EdbCode, startDate, endDate)
  1131. if err != nil && err.Error() != utils.ErrNoRow() {
  1132. errMsg = "RefreshAllCalculateLjzzy Err:" + err.Error()
  1133. break
  1134. }
  1135. case utils.DATA_SOURCE_CALCULATE_TBZ: //刷新同比值
  1136. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1137. if err != nil {
  1138. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  1139. break
  1140. }
  1141. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  1142. if err != nil {
  1143. errMsg = "GetEdbInfoById Err:" + err.Error()
  1144. break
  1145. }
  1146. startDate = edbInfo.StartDate
  1147. endDate = time.Now().Format(utils.FormatDate)
  1148. err = models.RefreshAllCalculateTbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  1149. if err != nil && err.Error() != utils.ErrNoRow() {
  1150. errMsg = "RefreshAllCalculateTbz Err:" + err.Error()
  1151. break
  1152. }
  1153. case utils.DATA_SOURCE_CALCULATE_TCZ: //同差值
  1154. calculateTcz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1155. if err != nil {
  1156. errMsg = "GetEdbInfoCalculateTczDetail Err:" + err.Error()
  1157. break
  1158. }
  1159. fromEdbInfo, err := models.GetEdbInfoById(calculateTcz.FromEdbInfoId)
  1160. if err != nil {
  1161. errMsg = "GetEdbInfoById Err:" + err.Error()
  1162. break
  1163. }
  1164. startDate = edbInfo.StartDate
  1165. endDate = time.Now().Format(utils.FormatDate)
  1166. err = models.RefreshAllCalculateTcz(edbInfoId, source, fromEdbInfo, calculateTcz.EdbCode, startDate, endDate)
  1167. if err != nil && err.Error() != utils.ErrNoRow() {
  1168. errMsg = "RefreshCalculateTcz Err:" + err.Error()
  1169. break
  1170. }
  1171. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS: //N数值移动平均计算
  1172. calculateNszydpjjs, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1173. if err != nil {
  1174. errMsg = "GetEdbInfoCalculateNszydpjjsDetail Err:" + err.Error()
  1175. break
  1176. }
  1177. fromEdbInfo, err := models.GetEdbInfoById(calculateNszydpjjs.FromEdbInfoId)
  1178. if err != nil {
  1179. errMsg = "GetEdbInfoById Err:" + err.Error()
  1180. break
  1181. }
  1182. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  1183. startDate = edbInfo.StartDate
  1184. err = models.RefreshAllCalculateNszydpjjs(edbInfoId, edbInfo.Source, formulaInt, fromEdbInfo, calculateNszydpjjs.EdbCode, startDate)
  1185. if err != nil && err.Error() != utils.ErrNoRow() {
  1186. errMsg = "RefreshCalculateNszydpjjs Err:" + err.Error()
  1187. break
  1188. }
  1189. case utils.DATA_SOURCE_CALCULATE_HBZ: //刷新环比值
  1190. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1191. if err != nil {
  1192. errMsg = "GetEdbInfoCalculateHbzDetail Err:" + err.Error()
  1193. break
  1194. }
  1195. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  1196. if err != nil {
  1197. errMsg = "GetEdbInfoById Err:" + err.Error()
  1198. break
  1199. }
  1200. startDate = edbInfo.StartDate
  1201. endDate = time.Now().Format(utils.FormatDate)
  1202. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  1203. err = models.RefreshAllCalculateHbz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  1204. if err != nil && err.Error() != utils.ErrNoRow() {
  1205. errMsg = "RefreshAllCalculateHbz Err:" + err.Error()
  1206. break
  1207. }
  1208. case utils.DATA_SOURCE_CALCULATE_HCZ: //刷新环差值
  1209. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1210. if err != nil {
  1211. errMsg = "GetEdbInfoCalculateHczDetail Err:" + err.Error()
  1212. break
  1213. }
  1214. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  1215. if err != nil {
  1216. errMsg = "GetEdbInfoById Err:" + err.Error()
  1217. break
  1218. }
  1219. startDate = edbInfo.StartDate
  1220. endDate = time.Now().Format(utils.FormatDate)
  1221. formulaInt, _ := strconv.Atoi(edbInfo.CalculateFormula)
  1222. err = models.RefreshAllCalculateHcz(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate, formulaInt)
  1223. if err != nil && err.Error() != utils.ErrNoRow() {
  1224. errMsg = "RefreshAllCalculateHcz Err:" + err.Error()
  1225. break
  1226. }
  1227. case utils.DATA_SOURCE_CALCULATE_BP: //刷新变频
  1228. calculateTbz, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1229. if err != nil {
  1230. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  1231. break
  1232. }
  1233. fromEdbInfo, err := models.GetEdbInfoById(calculateTbz.FromEdbInfoId)
  1234. if err != nil {
  1235. errMsg = "GetEdbInfoById Err:" + err.Error()
  1236. break
  1237. }
  1238. startDate = edbInfo.StartDate
  1239. endDate = time.Now().Format(utils.FormatDate)
  1240. err = models.RefreshAllCalculateBp(edbInfoId, source, fromEdbInfo, calculateTbz.EdbCode, startDate, endDate)
  1241. if err != nil && err.Error() != utils.ErrNoRow() {
  1242. errMsg = "RefreshAllCalculateBp Err:" + err.Error()
  1243. break
  1244. }
  1245. case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT:
  1246. calculate, err := models.GetEdbInfoCalculateMappingDetail(edbInfoId)
  1247. if err != nil {
  1248. errMsg = "GetEdbInfoCalculateTbzDetail Err:" + err.Error()
  1249. break
  1250. }
  1251. fromEdbInfo, err := models.GetEdbInfoById(calculate.FromEdbInfoId)
  1252. if err != nil {
  1253. errMsg = "GetEdbInfoById Err:" + err.Error()
  1254. break
  1255. }
  1256. startDate = edbInfo.StartDate
  1257. endDate = time.Now().Format(utils.FormatDate)
  1258. formulaInt, _ := strconv.Atoi(calculate.CalculateFormula)
  1259. err = models.RefreshAllCalculateTimeShift(edbInfoId, source, formulaInt, calculate.MoveType, fromEdbInfo, calculate.EdbCode, startDate, endDate, calculate.MoveFrequency)
  1260. if err != nil && err.Error() != utils.ErrNoRow() {
  1261. errMsg = "RefreshAllCalculateTimeShift Err:" + err.Error()
  1262. }
  1263. case utils.DATA_SOURCE_CALCULATE_ZJPJ: //刷新直接拼接
  1264. err = models.RefreshAllCalculateZjpj(edbInfo)
  1265. if err != nil && err.Error() != utils.ErrNoRow() {
  1266. errMsg = "RefreshAllCalculateZjpj Err:" + err.Error()
  1267. break
  1268. }
  1269. case utils.DATA_SOURCE_CALCULATE_LJZTBPJ: //刷新累计值同比拼接
  1270. err = models.RefreshAllCalculateLjztbpj(edbInfo)
  1271. if err != nil && err.Error() != utils.ErrNoRow() {
  1272. errMsg = "RefreshAllCalculateLjztbpj Err:" + err.Error()
  1273. break
  1274. }
  1275. case utils.DATA_SOURCE_PYTHON: //python代码运算
  1276. edbPythonCode, err := models.GetEdbPythonCodeById(edbInfo.EdbInfoId)
  1277. if err != nil {
  1278. errMsg = "获取python代码失败 Err:" + err.Error()
  1279. break
  1280. }
  1281. edbData, err, errMsg := services.ExecPythonCode(edbInfo.EdbCode, edbPythonCode.PythonCode)
  1282. if err != nil {
  1283. br.Msg = "获取数据失败"
  1284. br.ErrMsg = "python代码获取数据失败,err:" + err.Error()
  1285. if errMsg != "" {
  1286. br.ErrMsg = errMsg
  1287. }
  1288. return
  1289. }
  1290. err = models.RefreshAllPythonEdb(edbInfo, edbData)
  1291. if err != nil && err.Error() != utils.ErrNoRow() {
  1292. errMsg = "RefreshAllPythonEdb Err:" + err.Error()
  1293. break
  1294. }
  1295. default:
  1296. br.Msg = "来源异常,请联系相关开发!"
  1297. br.ErrMsg = "来源异常,请联系相关开发"
  1298. return
  1299. }
  1300. if errMsg != `` {
  1301. br.Msg = "刷新指标失败!"
  1302. br.ErrMsg = "刷新指标失败,err:" + errMsg
  1303. return
  1304. }
  1305. // 更新指标最大最小值
  1306. err, errMsg := models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  1307. if err != nil {
  1308. br.Msg = errMsg
  1309. br.ErrMsg = err.Error()
  1310. return
  1311. }
  1312. br.Ret = 200
  1313. br.Success = true
  1314. br.Msg = "获取成功"
  1315. } else {
  1316. br.Ret = 501
  1317. br.Success = true
  1318. br.Msg = "系统处理中,请稍后重试"
  1319. }
  1320. }