base_from_calculate.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609
  1. package services
  2. import (
  3. "eta/eta_index_lib/logic"
  4. "eta/eta_index_lib/models"
  5. "eta/eta_index_lib/utils"
  6. "fmt"
  7. "strconv"
  8. "strings"
  9. "time"
  10. )
  11. // EdbCalculateBatchSave 批量指标运算
  12. func EdbCalculateBatchSave(req models.EdbInfoCalculateBatchSaveReq) (edbInfo *models.EdbInfo, uniqueCode string, err error, errMsg string) {
  13. // 校验参数
  14. req.EdbName = strings.Trim(req.EdbName, " ")
  15. if req.EdbName == "" {
  16. err = fmt.Errorf("指标名称不能为空")
  17. return
  18. }
  19. if req.Frequency == "" {
  20. err = fmt.Errorf("频率不能为空")
  21. return
  22. }
  23. if req.Unit == "" {
  24. err = fmt.Errorf("单位不能为空")
  25. return
  26. }
  27. if req.ClassifyId <= 0 {
  28. err = fmt.Errorf("请选择分类")
  29. return
  30. }
  31. // 基础指标id
  32. fromEdbInfoId := req.FromEdbInfoId
  33. var formulaInt int
  34. var nhccDate models.NhccDate // 拟合残差的日期
  35. switch req.Source {
  36. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_CALCULATE_HBZ, utils.DATA_SOURCE_CALCULATE_HCZ, utils.DATA_SOURCE_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_CALCULATE_CJJX:
  37. if req.Formula == "" {
  38. err = fmt.Errorf("请填写N值")
  39. return
  40. }
  41. formulaInt, _ = strconv.Atoi(req.Formula)
  42. if formulaInt <= 0 {
  43. err = fmt.Errorf(" N值输入错误,请重新输入")
  44. return
  45. }
  46. case utils.DATA_SOURCE_CALCULATE_ZJPJ:
  47. //直接拼接指标
  48. //校验时间格式
  49. _, err = time.ParseInLocation(utils.FormatDate, req.Formula, time.Local)
  50. if err != nil {
  51. err = fmt.Errorf("拼接日期有误,请重新输入")
  52. return
  53. }
  54. case utils.DATA_SOURCE_CALCULATE_NHCC: //拟合残差指标
  55. //指标校验
  56. if len(req.EdbInfoIdArr) != 2 {
  57. err = fmt.Errorf("选择的指标异常,请重新选择")
  58. return
  59. }
  60. fromEdbInfoId = req.EdbInfoIdArr[0].EdbInfoId
  61. //校验时间格式
  62. //数据格式:2022-11-01,2022-11-10
  63. timeList := strings.Split(req.Formula, ",")
  64. if len(timeList) != 2 {
  65. err = fmt.Errorf("选择时间有误,请重新输入")
  66. return
  67. }
  68. startDate, e := time.ParseInLocation(utils.FormatDate, timeList[0], time.Local)
  69. if e != nil {
  70. err = fmt.Errorf("开始日期有误,请重新输入")
  71. return
  72. }
  73. endDate, e := time.ParseInLocation(utils.FormatDate, timeList[1], time.Local)
  74. if e != nil {
  75. err = fmt.Errorf("结束日期有误,请重新输入")
  76. return
  77. }
  78. if utils.GetTimeSubDay(startDate, endDate) < 2 {
  79. err = fmt.Errorf("日期间隔不得少于两天")
  80. return
  81. }
  82. nhccDate.StartDate = startDate
  83. nhccDate.EndDate = endDate
  84. case utils.DATA_SOURCE_CALCULATE_JP:
  85. if req.Formula != "期末值" && req.Formula != "平均值" {
  86. err = fmt.Errorf("数据取值类型错误:" + req.Formula)
  87. return
  88. }
  89. case utils.DATA_SOURCE_CALCULATE_ZSXY: // 指数修匀
  90. if req.Formula == "" {
  91. err = fmt.Errorf("请填写alpha值")
  92. return
  93. }
  94. a, e := strconv.ParseFloat(req.Formula, 64)
  95. if e != nil {
  96. err = fmt.Errorf("alpha值输入错误, 请重新输入")
  97. return
  98. }
  99. if a <= 0 || a >= 1 {
  100. err = fmt.Errorf("alpha值输入错误, 请重新输入")
  101. return
  102. }
  103. }
  104. notNeedFromEdbSourceList := []int{utils.DATA_SOURCE_CALCULATE_KSZS, utils.DATA_SOURCE_CALCULATE_CORRELATION, utils.DATA_SOURCE_CALCULATE_ZDYFX, utils.DATA_SOURCE_CALCULATE_SUM, utils.DATA_SOURCE_CALCULATE_AVG} // 不需要传入来源指标id的 指标类型
  105. if fromEdbInfoId <= 0 && !utils.InArrayByInt(notNeedFromEdbSourceList, req.Source) {
  106. err = fmt.Errorf("请选择指标")
  107. return
  108. }
  109. //加入缓存机制,避免创建同一个名称的指标 start
  110. redisKey := fmt.Sprint("edb_lib:edb_info:calculate:batch:save:", req.Source, ":", req.EdbName)
  111. isExist := utils.Rc.IsExist(redisKey)
  112. if isExist {
  113. err = fmt.Errorf("指标正在处理,请勿重复提交")
  114. return
  115. } else {
  116. //设置3分钟缓存
  117. utils.Rc.SetNX(redisKey, 1, time.Second*300)
  118. defer func() {
  119. _ = utils.Rc.Delete(redisKey)
  120. }()
  121. }
  122. //加入缓存机制,避免创建同一个名称的指标 end
  123. var condition string
  124. var pars []interface{}
  125. condition += " AND edb_name=? "
  126. pars = append(pars, req.EdbName)
  127. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  128. if err != nil {
  129. err = fmt.Errorf("判断指标名称是否存在失败")
  130. errMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  131. return
  132. }
  133. if count > 0 {
  134. err = fmt.Errorf("指标名称已存在,请重新填写")
  135. errMsg = "指标名称已存在,请重新填写"
  136. return
  137. }
  138. // 来源指标
  139. var fromEdbInfo *models.EdbInfo
  140. if fromEdbInfoId > 0 {
  141. fromEdbInfo, err = models.GetEdbInfoById(fromEdbInfoId)
  142. if err != nil {
  143. errMsg = "获取指标信息失败:Err:" + err.Error()
  144. err = fmt.Errorf("获取指标信息失败")
  145. return
  146. }
  147. }
  148. //生成指标编码
  149. randStr := utils.GetRandDigit(4)
  150. edbCode := `C` + time.Now().Format("060102") + randStr
  151. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  152. uniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  153. sysUserId := req.AdminId
  154. sysUserName := req.AdminName
  155. var sourName string
  156. var edbInfoId int
  157. var baseEdbInfoModel models.BaseEdbInfoInterface
  158. addParam := models.AddCalculateBatchParams{
  159. Req: &req,
  160. FromEdbInfo: fromEdbInfo,
  161. EdbCode: edbCode,
  162. UniqueCode: uniqueCode,
  163. SysUserId: sysUserId,
  164. SysUserRealName: sysUserName,
  165. }
  166. switch req.Source {
  167. case utils.DATA_SOURCE_CALCULATE:
  168. sourName = "计算指标"
  169. req2 := models.EdbInfoCalculateSaveReq{
  170. AdminId: sysUserId,
  171. AdminName: sysUserName,
  172. EdbName: req.EdbName,
  173. Frequency: req.Frequency,
  174. Unit: req.Unit,
  175. ClassifyId: req.ClassifyId,
  176. CalculateFormula: req.CalculateFormula,
  177. //EdbInfoIdArr: req.EdbInfoIdArr,
  178. EmptyType: req.EmptyType,
  179. MaxEmptyType: req.MaxEmptyType,
  180. Extra: req.Extra,
  181. }
  182. // 放入A指标
  183. t := models.EdbInfoFromTag{
  184. EdbInfoId: req.FromEdbInfoId,
  185. FromTag: "A",
  186. }
  187. req2.EdbInfoIdArr = append(req2.EdbInfoIdArr, t)
  188. // 放入B指标
  189. if len(req.EdbInfoIdArr) > 0 {
  190. t = models.EdbInfoFromTag{
  191. EdbInfoId: req.EdbInfoIdArr[0].EdbInfoId,
  192. FromTag: "B",
  193. }
  194. req2.EdbInfoIdArr = append(req2.EdbInfoIdArr, t)
  195. }
  196. edbInfo, _, err, errMsg = EdbCalculateAdd(req2)
  197. case utils.DATA_SOURCE_CALCULATE_LJZZY:
  198. sourName = "累计值转月值"
  199. if fromEdbInfo.Frequency != "月度" {
  200. err = fmt.Errorf("请选择月度指标")
  201. return
  202. }
  203. edbInfo, err = models.AddCalculateLjzzy(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  204. case utils.DATA_SOURCE_CALCULATE_TBZ:
  205. if req.Frequency != fromEdbInfo.Frequency {
  206. err = fmt.Errorf("当前频度和原指标频度不一致")
  207. return
  208. }
  209. if req.Unit != "无" {
  210. err = fmt.Errorf("单位只允许为无,禁止选择其他单位")
  211. return
  212. }
  213. sourName = "同比值"
  214. edbInfo, err = models.AddCalculateTbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  215. case utils.DATA_SOURCE_CALCULATE_TCZ:
  216. if req.Frequency != fromEdbInfo.Frequency {
  217. err = fmt.Errorf("当前频度和原指标频度不一致")
  218. return
  219. }
  220. if req.Unit != fromEdbInfo.Unit {
  221. err = fmt.Errorf("当前单位和原指标单位不一致")
  222. return
  223. }
  224. sourName = "同差值"
  225. edbInfo, err = models.AddCalculateTcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  226. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS:
  227. sourName = "N数值移动平均计算"
  228. edbInfo, err = models.AddCalculateNszydpjjs(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  229. case utils.DATA_SOURCE_CALCULATE_HBZ:
  230. var condition string
  231. var pars []interface{}
  232. condition += " AND edb_info_id =? "
  233. pars = append(pars, fromEdbInfoId)
  234. condition += " AND value <=0 "
  235. checkCount, tmpErr := models.GetEdbDataCount(condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource)
  236. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  237. err = fmt.Errorf("判断环比值是否可计算失败")
  238. errMsg = "判断环比值是否可计算失败,Err:" + tmpErr.Error()
  239. return
  240. }
  241. if checkCount > 0 {
  242. err = fmt.Errorf("原始数据中存在0或负数,该指标不能进行环比运算")
  243. errMsg = "原始数据中出现0和负值时,提示该指标不能进行环比运算"
  244. return
  245. }
  246. sourName = "环比值"
  247. edbInfo, err = models.AddCalculateHbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  248. case utils.DATA_SOURCE_CALCULATE_HCZ:
  249. sourName = "环差值"
  250. edbInfo, err = models.AddCalculateHcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  251. case utils.DATA_SOURCE_CALCULATE_BP:
  252. sourName = utils.DATA_SOURCE_NAME_CALCULATE_BP
  253. edbInfo, err = models.AddCalculateBp(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  254. case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT:
  255. sourName = "时间移位"
  256. edbInfo, err = models.AddCalculateTimeShift(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  257. case utils.DATA_SOURCE_CALCULATE_ZJPJ:
  258. sourName = "直接拼接"
  259. if len(req.EdbInfoIdArr) != 1 {
  260. err = fmt.Errorf("请传入拼接日期之后的指标")
  261. errMsg = "请传入拼接日期之后的指标"
  262. return
  263. }
  264. secondEdbInfoReq := req.EdbInfoIdArr[0]
  265. secondEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  266. if tmpErr != nil {
  267. err = fmt.Errorf("获取拼接日期之后的指标信息失败")
  268. errMsg = "获取拼接日期之后的指标信息失败:Err:" + tmpErr.Error()
  269. return
  270. }
  271. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  272. err = fmt.Errorf("两个指标不允许为同一个")
  273. errMsg = "两个指标不允许为同一个"
  274. return
  275. }
  276. edbInfo, err = models.AddCalculateZjpj(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  277. case utils.DATA_SOURCE_CALCULATE_LJZTBPJ:
  278. sourName = "累计值同比拼接"
  279. if fromEdbInfo.Frequency != "月度" {
  280. err = fmt.Errorf("待拼接指标只能筛选月度指标")
  281. errMsg = "待拼接指标只能筛选月度指标"
  282. return
  283. }
  284. if len(req.EdbInfoIdArr) != 1 {
  285. err = fmt.Errorf("请传入同比值指标")
  286. errMsg = "请传入同比值指标"
  287. return
  288. }
  289. secondEdbInfoReq := req.EdbInfoIdArr[0]
  290. tbzEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  291. if tmpErr != nil {
  292. err = fmt.Errorf("获取同比值指标信息失败")
  293. errMsg = "获取同比值指标信息失败:Err:" + tmpErr.Error()
  294. return
  295. }
  296. if tbzEdbInfo.Source != utils.DATA_SOURCE_CALCULATE_TBZ {
  297. err = fmt.Errorf("指标必须是传入同比值指标类型")
  298. errMsg = "指标必须是传入同比值指标类型"
  299. return
  300. }
  301. if tbzEdbInfo.Frequency != "月度" {
  302. err = fmt.Errorf("同比值指标只能筛选月度指标")
  303. errMsg = "同比值指标只能筛选月度指标"
  304. return
  305. }
  306. if fromEdbInfo.EdbInfoId == tbzEdbInfo.EdbInfoId {
  307. err = fmt.Errorf("两个指标不允许为同一个")
  308. errMsg = "两个指标不允许为同一个"
  309. return
  310. }
  311. edbInfo, err = models.AddCalculateLjztbpj(&req, fromEdbInfo, tbzEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  312. case utils.DATA_SOURCE_CALCULATE_CJJX:
  313. sourName = "超季节性"
  314. edbInfo, err = models.AddCalculateCjjx(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  315. case utils.DATA_SOURCE_CALCULATE_NHCC:
  316. sourName = "拟合残差"
  317. secondEdbInfoReq := req.EdbInfoIdArr[1]
  318. secondEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  319. if tmpErr != nil {
  320. err = fmt.Errorf("获取因变量的指标信息失败")
  321. errMsg = "获取因变量的指标信息失败:Err:" + tmpErr.Error()
  322. return
  323. }
  324. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  325. err = fmt.Errorf("两个指标不允许为同一个")
  326. errMsg = "两个指标不允许为同一个"
  327. return
  328. }
  329. edbInfo, err, errMsg = models.AddCalculateNhcc(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, nhccDate, sysUserId, sysUserName)
  330. case utils.DATA_SOURCE_CALCULATE_JP:
  331. if !models.CheckFrequency(fromEdbInfo.Frequency, req.Frequency) {
  332. err = fmt.Errorf("频度异常,不允许低频降频到高频")
  333. return
  334. }
  335. sourName = utils.DATA_SOURCE_NAME_CALCULATE_JP
  336. edbInfo, err = models.AddCalculateJp(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  337. case utils.DATA_SOURCE_CALCULATE_NH:
  338. sourName = utils.DATA_SOURCE_NAME_CALCULATE_NH
  339. edbInfo, err = models.AddCalculateNh(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  340. case utils.DATA_SOURCE_CALCULATE_KSZS:
  341. //关联的指标信息
  342. if len(req.EdbInfoIdArr) < 2 {
  343. err = fmt.Errorf("指标数量不能小于2个,请重新选择")
  344. errMsg = "指标数量不能小于2个,请重新选择"
  345. //br.IsSendEmail = false
  346. return
  347. }
  348. sourName = utils.DATA_SOURCE_NAME_CALCULATE_KSZS
  349. edbInfo, err = models.AddCalculateKszs(&req, edbCode, uniqueCode, sysUserId, sysUserName)
  350. case utils.DATA_SOURCE_CALCULATE_CORRELATION:
  351. //关联的指标信息
  352. if len(req.EdbInfoIdArr) < 2 {
  353. err = fmt.Errorf("指标数量不能小于2个,请重新选择")
  354. errMsg = "指标数量不能小于2个,请重新选择"
  355. //br.IsSendEmail = false
  356. return
  357. }
  358. sourName = utils.DATA_SOURCE_NAME_CALCULATE_CORRELATION
  359. edbInfo, err, errMsg = models.AddCalculateCorrelation(&req, edbCode, uniqueCode, sysUserId, sysUserName)
  360. case utils.DATA_SOURCE_CALCULATE_RJZ:
  361. if req.Frequency == "日度" {
  362. err = fmt.Errorf("日度指标无需进行日均值计算")
  363. return
  364. }
  365. if req.Frequency != fromEdbInfo.Frequency {
  366. err = fmt.Errorf("当前频度和原指标频度不一致")
  367. return
  368. }
  369. if req.Unit != fromEdbInfo.Unit {
  370. err = fmt.Errorf("单位只允许为和原指标频度保持一致,禁止选择其他单位")
  371. return
  372. }
  373. sourName = "日均值"
  374. edbInfo, err = models.AddCalculateRjz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  375. default:
  376. // 获取通用的数据源处理服务
  377. baseEdbInfoModel = models.GetBaseEdbInfoModel(req.Source)
  378. // 没有找到的话,那么就直接返回报错吧
  379. if baseEdbInfoModel == nil {
  380. err = fmt.Errorf("无效计算方式")
  381. errMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  382. return
  383. }
  384. sourName = baseEdbInfoModel.GetSourceName()
  385. edbInfo, err, errMsg = baseEdbInfoModel.Add(addParam)
  386. }
  387. if err != nil {
  388. errMsg = "生成" + sourName + "失败 Err:" + err.Error() + errMsg
  389. err = fmt.Errorf("生成" + sourName + "失败")
  390. return
  391. }
  392. if edbInfo == nil {
  393. err = fmt.Errorf("生成" + sourName + "失败")
  394. errMsg = "生成" + sourName + "失败,指标ID错误:" + strconv.Itoa(edbInfoId)
  395. return
  396. }
  397. edbInfoId = edbInfo.EdbInfoId
  398. //处理同名指标
  399. {
  400. edbNameList, tErr := models.GetEdbInfoByName(req.EdbName)
  401. if tErr != nil {
  402. err = fmt.Errorf("保存失败")
  403. errMsg = "获取指标信息失败,Err:" + tErr.Error()
  404. return
  405. }
  406. if len(edbNameList) >= 2 {
  407. for _, v := range edbNameList {
  408. edbName := v.EdbName + "(" + v.SourceName + ")"
  409. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  410. if err != nil {
  411. errMsg = "修改指标名称失败,Err:" + err.Error()
  412. err = fmt.Errorf("保存失败")
  413. return
  414. }
  415. }
  416. }
  417. }
  418. // 更新指标最大最小值
  419. err, errMsg = models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  420. if err != nil {
  421. return
  422. }
  423. // 更新ES
  424. go logic.UpdateEs(edbInfo.EdbInfoId)
  425. return
  426. }
  427. // EdbCalculateAdd 新增指标运算
  428. func EdbCalculateAdd(req models.EdbInfoCalculateSaveReq) (edbInfo *models.EdbInfo, uniqueCode string, err error, errMsg string) {
  429. calculateFormula := req.CalculateFormula
  430. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  431. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  432. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  433. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  434. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  435. req.CalculateFormula = calculateFormula
  436. //判断是否重复指标
  437. edbInfoMap := make(map[int]string)
  438. //移除研究员选择指标中的未使用的指标
  439. {
  440. //转大写的计算公式
  441. upperCalculateFormulaStr := strings.ToUpper(req.CalculateFormula)
  442. //用到的指标
  443. newEdbInfoIdArr := make([]models.EdbInfoFromTag, 0)
  444. for _, tmpEdbInfo := range req.EdbInfoIdArr {
  445. _, ok := edbInfoMap[tmpEdbInfo.EdbInfoId]
  446. if ok {
  447. err = fmt.Errorf("选择指标失败,请勿选择重复指标!")
  448. return
  449. }
  450. edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo.FromTag
  451. upperFromTag := strings.ToUpper(tmpEdbInfo.FromTag)
  452. if strings.Contains(upperCalculateFormulaStr, upperFromTag) {
  453. newEdbInfoIdArr = append(newEdbInfoIdArr, tmpEdbInfo)
  454. }
  455. }
  456. req.EdbInfoIdArr = newEdbInfoIdArr
  457. }
  458. var condition string
  459. var pars []interface{}
  460. condition += " AND edb_name=? "
  461. pars = append(pars, req.EdbName)
  462. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  463. if err != nil {
  464. errMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  465. err = fmt.Errorf("判断指标名称是否存在失败")
  466. return
  467. }
  468. if count > 0 {
  469. err = fmt.Errorf("指标名称已存在,请重新填写")
  470. errMsg = "指标名称已存在,请重新填写"
  471. //br.IsSendEmail = false
  472. return
  473. }
  474. //检验公式
  475. var formulaStr string
  476. var edbInfoIdBytes []string
  477. for _, v := range req.EdbInfoIdArr {
  478. formulaStr += v.FromTag + ","
  479. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  480. }
  481. formulaSlice, err := utils.CheckFormulaJson(req.CalculateFormula)
  482. if err != nil {
  483. err = fmt.Errorf("公式格式错误,请重新填写")
  484. return
  485. }
  486. for _, formula := range formulaSlice {
  487. formulaMap, e := utils.CheckFormula(formula)
  488. if e != nil {
  489. err = fmt.Errorf("公式错误,请重新填写")
  490. return
  491. }
  492. for _, v := range formulaMap {
  493. if !strings.Contains(formulaStr, v) {
  494. err = fmt.Errorf("公式错误,请重新填写")
  495. return
  496. }
  497. }
  498. }
  499. //关联的指标信息
  500. edbInfoList := make([]*models.EdbInfo, 0)
  501. //关联的指标数据表
  502. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  503. for k, v := range req.EdbInfoIdArr {
  504. fromEdbInfo, e := models.GetEdbInfoById(v.EdbInfoId)
  505. if e != nil {
  506. if e.Error() == utils.ErrNoRow() {
  507. err = fmt.Errorf("生成计算指标失败")
  508. errMsg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  509. return
  510. }
  511. err = fmt.Errorf("生成计算指标失败")
  512. errMsg = "获取指标失败:Err:" + e.Error()
  513. return
  514. }
  515. edbInfoList = append(edbInfoList, fromEdbInfo)
  516. //关联关系表
  517. {
  518. calculateMappingItem := &models.EdbInfoCalculateMapping{
  519. EdbInfoCalculateMappingId: 0,
  520. EdbInfoId: 0,
  521. Source: utils.DATA_SOURCE_CALCULATE,
  522. SourceName: "指标运算",
  523. EdbCode: "",
  524. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  525. FromEdbCode: fromEdbInfo.EdbCode,
  526. FromEdbName: fromEdbInfo.EdbName,
  527. FromSource: fromEdbInfo.Source,
  528. FromSourceName: fromEdbInfo.SourceName,
  529. FromTag: v.FromTag,
  530. Sort: k + 1,
  531. CreateTime: time.Now(),
  532. ModifyTime: time.Now(),
  533. }
  534. //calculateMappingItem.EdbCode = edbCode
  535. //calculateMappingItem.EdbInfoId = int(edbInfoId)
  536. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  537. }
  538. }
  539. for _, v := range formulaSlice {
  540. formulaMap, e := utils.CheckFormula(v)
  541. if e != nil {
  542. err = fmt.Errorf("公式错误,请重新填写")
  543. return
  544. }
  545. //预先计算,判断公式是否正常
  546. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, v, edbInfoIdBytes)
  547. if !ok {
  548. err = fmt.Errorf("生成计算指标失败,请使用正确的计算公式")
  549. return
  550. }
  551. }
  552. randStr := utils.GetRandDigit(4)
  553. edbCode := `C` + time.Now().Format("060102") + randStr
  554. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  555. uniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  556. // 新增计算指标
  557. edbInfo, err = models.AddCalculateInfo(req, calculateMappingList, edbInfoList, edbCode, uniqueCode, edbInfoIdBytes)
  558. if err != nil {
  559. errMsg = "生成计算指标失败,AddEdbInfo Err:" + err.Error()
  560. err = fmt.Errorf("生成计算指标失败")
  561. return
  562. }
  563. return
  564. }