base_from_calculate.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610
  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) (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 edbInfo *models.EdbInfo
  158. var baseEdbInfoModel models.BaseEdbInfoInterface
  159. addParam := models.AddCalculateBatchParams{
  160. Req: &req,
  161. FromEdbInfo: fromEdbInfo,
  162. EdbCode: edbCode,
  163. UniqueCode: uniqueCode,
  164. SysUserId: sysUserId,
  165. SysUserRealName: sysUserName,
  166. }
  167. switch req.Source {
  168. case utils.DATA_SOURCE_CALCULATE:
  169. sourName = "计算指标"
  170. req2 := models.EdbInfoCalculateSaveReq{
  171. AdminId: sysUserId,
  172. AdminName: sysUserName,
  173. EdbName: req.EdbName,
  174. Frequency: req.Frequency,
  175. Unit: req.Unit,
  176. ClassifyId: req.ClassifyId,
  177. CalculateFormula: req.CalculateFormula,
  178. //EdbInfoIdArr: req.EdbInfoIdArr,
  179. EmptyType: req.EmptyType,
  180. MaxEmptyType: req.MaxEmptyType,
  181. Extra: req.Extra,
  182. }
  183. // 放入A指标
  184. t := models.EdbInfoFromTag{
  185. EdbInfoId: req.FromEdbInfoId,
  186. FromTag: "A",
  187. }
  188. req2.EdbInfoIdArr = append(req2.EdbInfoIdArr, t)
  189. // 放入B指标
  190. if len(req.EdbInfoIdArr) > 0 {
  191. t = models.EdbInfoFromTag{
  192. EdbInfoId: req.EdbInfoIdArr[0].EdbInfoId,
  193. FromTag: "B",
  194. }
  195. req2.EdbInfoIdArr = append(req2.EdbInfoIdArr, t)
  196. }
  197. edbInfo, _, err, errMsg = EdbCalculateAdd(req2)
  198. case utils.DATA_SOURCE_CALCULATE_LJZZY:
  199. sourName = "累计值转月值"
  200. if fromEdbInfo.Frequency != "月度" {
  201. err = fmt.Errorf("请选择月度指标")
  202. return
  203. }
  204. edbInfo, err = models.AddCalculateLjzzy(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  205. case utils.DATA_SOURCE_CALCULATE_TBZ:
  206. if req.Frequency != fromEdbInfo.Frequency {
  207. err = fmt.Errorf("当前频度和原指标频度不一致")
  208. return
  209. }
  210. if req.Unit != "无" {
  211. err = fmt.Errorf("单位只允许为无,禁止选择其他单位")
  212. return
  213. }
  214. sourName = "同比值"
  215. edbInfo, err = models.AddCalculateTbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  216. case utils.DATA_SOURCE_CALCULATE_TCZ:
  217. if req.Frequency != fromEdbInfo.Frequency {
  218. err = fmt.Errorf("当前频度和原指标频度不一致")
  219. return
  220. }
  221. if req.Unit != fromEdbInfo.Unit {
  222. err = fmt.Errorf("当前单位和原指标单位不一致")
  223. return
  224. }
  225. sourName = "同差值"
  226. edbInfo, err = models.AddCalculateTcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  227. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS:
  228. sourName = "N数值移动平均计算"
  229. edbInfo, err = models.AddCalculateNszydpjjs(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  230. case utils.DATA_SOURCE_CALCULATE_HBZ:
  231. var condition string
  232. var pars []interface{}
  233. condition += " AND edb_info_id =? "
  234. pars = append(pars, fromEdbInfoId)
  235. condition += " AND value <=0 "
  236. checkCount, tmpErr := models.GetEdbDataCount(condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource)
  237. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  238. err = fmt.Errorf("判断环比值是否可计算失败")
  239. errMsg = "判断环比值是否可计算失败,Err:" + tmpErr.Error()
  240. return
  241. }
  242. if checkCount > 0 {
  243. err = fmt.Errorf("原始数据中存在0或负数,该指标不能进行环比运算")
  244. errMsg = "原始数据中出现0和负值时,提示该指标不能进行环比运算"
  245. return
  246. }
  247. sourName = "环比值"
  248. edbInfo, err = models.AddCalculateHbz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  249. case utils.DATA_SOURCE_CALCULATE_HCZ:
  250. sourName = "环差值"
  251. edbInfo, err = models.AddCalculateHcz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  252. case utils.DATA_SOURCE_CALCULATE_BP:
  253. sourName = utils.DATA_SOURCE_NAME_CALCULATE_BP
  254. edbInfo, err = models.AddCalculateBp(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  255. case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT:
  256. sourName = "时间移位"
  257. edbInfo, err = models.AddCalculateTimeShift(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  258. case utils.DATA_SOURCE_CALCULATE_ZJPJ:
  259. sourName = "直接拼接"
  260. if len(req.EdbInfoIdArr) != 1 {
  261. err = fmt.Errorf("请传入拼接日期之后的指标")
  262. errMsg = "请传入拼接日期之后的指标"
  263. return
  264. }
  265. secondEdbInfoReq := req.EdbInfoIdArr[0]
  266. secondEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  267. if tmpErr != nil {
  268. err = fmt.Errorf("获取拼接日期之后的指标信息失败")
  269. errMsg = "获取拼接日期之后的指标信息失败:Err:" + tmpErr.Error()
  270. return
  271. }
  272. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  273. err = fmt.Errorf("两个指标不允许为同一个")
  274. errMsg = "两个指标不允许为同一个"
  275. return
  276. }
  277. edbInfo, err = models.AddCalculateZjpj(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  278. case utils.DATA_SOURCE_CALCULATE_LJZTBPJ:
  279. sourName = "累计值同比拼接"
  280. if fromEdbInfo.Frequency != "月度" {
  281. err = fmt.Errorf("待拼接指标只能筛选月度指标")
  282. errMsg = "待拼接指标只能筛选月度指标"
  283. return
  284. }
  285. if len(req.EdbInfoIdArr) != 1 {
  286. err = fmt.Errorf("请传入同比值指标")
  287. errMsg = "请传入同比值指标"
  288. return
  289. }
  290. secondEdbInfoReq := req.EdbInfoIdArr[0]
  291. tbzEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  292. if tmpErr != nil {
  293. err = fmt.Errorf("获取同比值指标信息失败")
  294. errMsg = "获取同比值指标信息失败:Err:" + tmpErr.Error()
  295. return
  296. }
  297. if tbzEdbInfo.Source != utils.DATA_SOURCE_CALCULATE_TBZ {
  298. err = fmt.Errorf("指标必须是传入同比值指标类型")
  299. errMsg = "指标必须是传入同比值指标类型"
  300. return
  301. }
  302. if tbzEdbInfo.Frequency != "月度" {
  303. err = fmt.Errorf("同比值指标只能筛选月度指标")
  304. errMsg = "同比值指标只能筛选月度指标"
  305. return
  306. }
  307. if fromEdbInfo.EdbInfoId == tbzEdbInfo.EdbInfoId {
  308. err = fmt.Errorf("两个指标不允许为同一个")
  309. errMsg = "两个指标不允许为同一个"
  310. return
  311. }
  312. edbInfo, err = models.AddCalculateLjztbpj(&req, fromEdbInfo, tbzEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  313. case utils.DATA_SOURCE_CALCULATE_CJJX:
  314. sourName = "超季节性"
  315. edbInfo, err = models.AddCalculateCjjx(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName, formulaInt)
  316. case utils.DATA_SOURCE_CALCULATE_NHCC:
  317. sourName = "拟合残差"
  318. secondEdbInfoReq := req.EdbInfoIdArr[1]
  319. secondEdbInfo, tmpErr := models.GetEdbInfoById(secondEdbInfoReq.EdbInfoId)
  320. if tmpErr != nil {
  321. err = fmt.Errorf("获取因变量的指标信息失败")
  322. errMsg = "获取因变量的指标信息失败:Err:" + tmpErr.Error()
  323. return
  324. }
  325. if fromEdbInfo.EdbInfoId == secondEdbInfo.EdbInfoId {
  326. err = fmt.Errorf("两个指标不允许为同一个")
  327. errMsg = "两个指标不允许为同一个"
  328. return
  329. }
  330. edbInfo, err, errMsg = models.AddCalculateNhcc(&req, fromEdbInfo, secondEdbInfo, edbCode, uniqueCode, nhccDate, sysUserId, sysUserName)
  331. case utils.DATA_SOURCE_CALCULATE_JP:
  332. if !models.CheckFrequency(fromEdbInfo.Frequency, req.Frequency) {
  333. err = fmt.Errorf("频度异常,不允许低频降频到高频")
  334. return
  335. }
  336. sourName = utils.DATA_SOURCE_NAME_CALCULATE_JP
  337. edbInfo, err = models.AddCalculateJp(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  338. case utils.DATA_SOURCE_CALCULATE_NH:
  339. sourName = utils.DATA_SOURCE_NAME_CALCULATE_NH
  340. edbInfo, err = models.AddCalculateNh(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  341. case utils.DATA_SOURCE_CALCULATE_KSZS:
  342. //关联的指标信息
  343. if len(req.EdbInfoIdArr) < 2 {
  344. err = fmt.Errorf("指标数量不能小于2个,请重新选择")
  345. errMsg = "指标数量不能小于2个,请重新选择"
  346. //br.IsSendEmail = false
  347. return
  348. }
  349. sourName = utils.DATA_SOURCE_NAME_CALCULATE_KSZS
  350. edbInfo, err = models.AddCalculateKszs(&req, edbCode, uniqueCode, sysUserId, sysUserName)
  351. case utils.DATA_SOURCE_CALCULATE_CORRELATION:
  352. //关联的指标信息
  353. if len(req.EdbInfoIdArr) < 2 {
  354. err = fmt.Errorf("指标数量不能小于2个,请重新选择")
  355. errMsg = "指标数量不能小于2个,请重新选择"
  356. //br.IsSendEmail = false
  357. return
  358. }
  359. sourName = utils.DATA_SOURCE_NAME_CALCULATE_CORRELATION
  360. edbInfo, err, errMsg = models.AddCalculateCorrelation(&req, edbCode, uniqueCode, sysUserId, sysUserName)
  361. case utils.DATA_SOURCE_CALCULATE_RJZ:
  362. if req.Frequency == "日度" {
  363. err = fmt.Errorf("日度指标无需进行日均值计算")
  364. return
  365. }
  366. if req.Frequency != fromEdbInfo.Frequency {
  367. err = fmt.Errorf("当前频度和原指标频度不一致")
  368. return
  369. }
  370. if req.Unit != fromEdbInfo.Unit {
  371. err = fmt.Errorf("单位只允许为和原指标频度保持一致,禁止选择其他单位")
  372. return
  373. }
  374. sourName = "日均值"
  375. edbInfo, err = models.AddCalculateRjz(&req, fromEdbInfo, edbCode, uniqueCode, sysUserId, sysUserName)
  376. default:
  377. // 获取通用的数据源处理服务
  378. baseEdbInfoModel = models.GetBaseEdbInfoModel(req.Source)
  379. // 没有找到的话,那么就直接返回报错吧
  380. if baseEdbInfoModel == nil {
  381. err = fmt.Errorf("无效计算方式")
  382. errMsg = "无效计算方式,source:" + strconv.Itoa(req.Source)
  383. return
  384. }
  385. sourName = baseEdbInfoModel.GetSourceName()
  386. edbInfo, err, errMsg = baseEdbInfoModel.Add(addParam)
  387. }
  388. if err != nil {
  389. errMsg = "生成" + sourName + "失败 Err:" + err.Error() + errMsg
  390. err = fmt.Errorf("生成" + sourName + "失败")
  391. return
  392. }
  393. if edbInfo == nil {
  394. err = fmt.Errorf("生成" + sourName + "失败")
  395. errMsg = "生成" + sourName + "失败,指标ID错误:" + strconv.Itoa(edbInfoId)
  396. return
  397. }
  398. edbInfoId = edbInfo.EdbInfoId
  399. //处理同名指标
  400. {
  401. edbNameList, tErr := models.GetEdbInfoByName(req.EdbName)
  402. if tErr != nil {
  403. err = fmt.Errorf("保存失败")
  404. errMsg = "获取指标信息失败,Err:" + tErr.Error()
  405. return
  406. }
  407. if len(edbNameList) >= 2 {
  408. for _, v := range edbNameList {
  409. edbName := v.EdbName + "(" + v.SourceName + ")"
  410. err = models.ModifyEdbInfoNameSource(edbName, v.EdbInfoId)
  411. if err != nil {
  412. errMsg = "修改指标名称失败,Err:" + err.Error()
  413. err = fmt.Errorf("保存失败")
  414. return
  415. }
  416. }
  417. }
  418. }
  419. // 更新指标最大最小值
  420. err, errMsg = models.UnifiedModifyEdbInfoMaxAndMinInfo(edbInfo)
  421. if err != nil {
  422. return
  423. }
  424. // 更新ES
  425. go logic.UpdateEs(edbInfo.EdbInfoId)
  426. return
  427. }
  428. // EdbCalculateAdd 新增指标运算
  429. func EdbCalculateAdd(req models.EdbInfoCalculateSaveReq) (edbInfo *models.EdbInfo, uniqueCode string, err error, errMsg string) {
  430. calculateFormula := req.CalculateFormula
  431. calculateFormula = strings.Replace(calculateFormula, "(", "(", -1)
  432. calculateFormula = strings.Replace(calculateFormula, ")", ")", -1)
  433. calculateFormula = strings.Replace(calculateFormula, ",", ",", -1)
  434. calculateFormula = strings.Replace(calculateFormula, "。", ".", -1)
  435. calculateFormula = strings.Replace(calculateFormula, "%", "*0.01", -1)
  436. req.CalculateFormula = calculateFormula
  437. //判断是否重复指标
  438. edbInfoMap := make(map[int]string)
  439. //移除研究员选择指标中的未使用的指标
  440. {
  441. //转大写的计算公式
  442. upperCalculateFormulaStr := strings.ToUpper(req.CalculateFormula)
  443. //用到的指标
  444. newEdbInfoIdArr := make([]models.EdbInfoFromTag, 0)
  445. for _, tmpEdbInfo := range req.EdbInfoIdArr {
  446. _, ok := edbInfoMap[tmpEdbInfo.EdbInfoId]
  447. if ok {
  448. err = fmt.Errorf("选择指标失败,请勿选择重复指标!")
  449. return
  450. }
  451. edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo.FromTag
  452. upperFromTag := strings.ToUpper(tmpEdbInfo.FromTag)
  453. if strings.Contains(upperCalculateFormulaStr, upperFromTag) {
  454. newEdbInfoIdArr = append(newEdbInfoIdArr, tmpEdbInfo)
  455. }
  456. }
  457. req.EdbInfoIdArr = newEdbInfoIdArr
  458. }
  459. var condition string
  460. var pars []interface{}
  461. condition += " AND edb_name=? "
  462. pars = append(pars, req.EdbName)
  463. count, err := models.GetEdbInfoCountByCondition(condition, pars)
  464. if err != nil {
  465. errMsg = "判断指标名称是否存在失败,Err:" + err.Error()
  466. err = fmt.Errorf("判断指标名称是否存在失败")
  467. return
  468. }
  469. if count > 0 {
  470. err = fmt.Errorf("指标名称已存在,请重新填写")
  471. errMsg = "指标名称已存在,请重新填写"
  472. //br.IsSendEmail = false
  473. return
  474. }
  475. //检验公式
  476. var formulaStr string
  477. var edbInfoIdBytes []string
  478. for _, v := range req.EdbInfoIdArr {
  479. formulaStr += v.FromTag + ","
  480. edbInfoIdBytes = append(edbInfoIdBytes, v.FromTag)
  481. }
  482. formulaSlice, err := utils.CheckFormulaJson(req.CalculateFormula)
  483. if err != nil {
  484. err = fmt.Errorf("公式格式错误,请重新填写")
  485. return
  486. }
  487. for _, formula := range formulaSlice {
  488. formulaMap, e := utils.CheckFormula(formula)
  489. if e != nil {
  490. err = fmt.Errorf("公式错误,请重新填写")
  491. return
  492. }
  493. for _, v := range formulaMap {
  494. if !strings.Contains(formulaStr, v) {
  495. err = fmt.Errorf("公式错误,请重新填写")
  496. return
  497. }
  498. }
  499. }
  500. //关联的指标信息
  501. edbInfoList := make([]*models.EdbInfo, 0)
  502. //关联的指标数据表
  503. calculateMappingList := make([]*models.EdbInfoCalculateMapping, 0)
  504. for k, v := range req.EdbInfoIdArr {
  505. fromEdbInfo, e := models.GetEdbInfoById(v.EdbInfoId)
  506. if e != nil {
  507. if e.Error() == utils.ErrNoRow() {
  508. err = fmt.Errorf("生成计算指标失败")
  509. errMsg = "指标 " + strconv.Itoa(v.EdbInfoId) + " 不存在"
  510. return
  511. }
  512. err = fmt.Errorf("生成计算指标失败")
  513. errMsg = "获取指标失败:Err:" + e.Error()
  514. return
  515. }
  516. edbInfoList = append(edbInfoList, fromEdbInfo)
  517. //关联关系表
  518. {
  519. calculateMappingItem := &models.EdbInfoCalculateMapping{
  520. EdbInfoCalculateMappingId: 0,
  521. EdbInfoId: 0,
  522. Source: utils.DATA_SOURCE_CALCULATE,
  523. SourceName: "指标运算",
  524. EdbCode: "",
  525. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  526. FromEdbCode: fromEdbInfo.EdbCode,
  527. FromEdbName: fromEdbInfo.EdbName,
  528. FromSource: fromEdbInfo.Source,
  529. FromSourceName: fromEdbInfo.SourceName,
  530. FromTag: v.FromTag,
  531. Sort: k + 1,
  532. CreateTime: time.Now(),
  533. ModifyTime: time.Now(),
  534. }
  535. //calculateMappingItem.EdbCode = edbCode
  536. //calculateMappingItem.EdbInfoId = int(edbInfoId)
  537. calculateMappingList = append(calculateMappingList, calculateMappingItem)
  538. }
  539. }
  540. for _, v := range formulaSlice {
  541. formulaMap, e := utils.CheckFormula(v)
  542. if e != nil {
  543. err = fmt.Errorf("公式错误,请重新填写")
  544. return
  545. }
  546. //预先计算,判断公式是否正常
  547. ok, _ := models.CheckFormula2(edbInfoList, formulaMap, v, edbInfoIdBytes)
  548. if !ok {
  549. err = fmt.Errorf("生成计算指标失败,请使用正确的计算公式")
  550. return
  551. }
  552. }
  553. randStr := utils.GetRandDigit(4)
  554. edbCode := `C` + time.Now().Format("060102") + randStr
  555. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  556. uniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  557. // 新增计算指标
  558. edbInfo, err = models.AddCalculateInfo(req, calculateMappingList, edbInfoList, edbCode, uniqueCode, edbInfoIdBytes)
  559. if err != nil {
  560. errMsg = "生成计算指标失败,AddEdbInfo Err:" + err.Error()
  561. err = fmt.Errorf("生成计算指标失败")
  562. return
  563. }
  564. return
  565. }