edb_info.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708
  1. package services
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_forum_admin/models"
  6. "eta/eta_forum_admin/models/mgodb"
  7. "eta/eta_forum_admin/utils"
  8. "fmt"
  9. "strconv"
  10. "strings"
  11. "time"
  12. )
  13. // BatchAddOrUpdateEdbInfo 添加批量添加指标到指标库
  14. func BatchAddOrUpdateEdbInfo(edbList []*models.EdbInfo, edbMapping []*models.EdbInfoCalculateMapping, sysUserId int, sysUserRealName string) (newList []*models.EdbInfo, err error, errMsg string, isSendEmail bool) {
  15. isSendEmail = true
  16. edbCodeMap := make(map[string]*models.EdbInfo)
  17. for _, v := range edbList {
  18. tmp, e, msg, _ := AddOrUpdateEdbInfo(v, sysUserId, sysUserRealName)
  19. if e != nil {
  20. err = e
  21. errMsg = msg
  22. return
  23. }
  24. newList = append(newList, tmp)
  25. edbCodeMap[tmp.EdbCode] = tmp
  26. }
  27. //批量添加指标的mapping信息
  28. calculateMappingItemList := make([]*models.EdbInfoCalculateMapping, 0)
  29. for _, v := range edbMapping {
  30. edbInfo, ok := edbCodeMap[v.EdbCode]
  31. if !ok {
  32. continue
  33. }
  34. fromEdbInfo, ok := edbCodeMap[v.FromEdbCode]
  35. if !ok {
  36. continue
  37. }
  38. calculateMappingItem := new(models.EdbInfoCalculateMapping)
  39. calculateMappingItem.CreateTime = time.Now()
  40. calculateMappingItem.ModifyTime = time.Now()
  41. calculateMappingItem.Sort = v.Sort
  42. calculateMappingItem.EdbCode = edbInfo.EdbCode
  43. calculateMappingItem.EdbInfoId = edbInfo.EdbInfoId
  44. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  45. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  46. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  47. calculateMappingItem.FromSource = fromEdbInfo.Source
  48. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  49. calculateMappingItem.FromTag = v.FromTag
  50. calculateMappingItem.Source = edbInfo.Source
  51. calculateMappingItem.SourceName = edbInfo.SourceName
  52. calculateMappingItem.FromSubSource = edbInfo.SubSource
  53. calculateMappingItemList = append(calculateMappingItemList, calculateMappingItem)
  54. }
  55. err = models.AddEdbInfoCalculateMappingMulti(calculateMappingItemList)
  56. if err != nil {
  57. errMsg = "指标映射关系添加失败"
  58. return
  59. }
  60. return
  61. }
  62. // AddOrUpdateEdbInfo 添加指标到指标库
  63. func AddOrUpdateEdbInfo(edbItem *models.EdbInfo, sysUserId int, sysUserRealName string) (edbInfo *models.EdbInfo, err error, errMsg string, isSendEmail bool) {
  64. isSendEmail = true
  65. //判断指标是否存在
  66. edbInfo, err = models.GetEdbInfoByEdbCode(edbItem.Source, edbItem.EdbCode)
  67. if err != nil {
  68. if err.Error() == utils.ErrNoRow() {
  69. err = nil
  70. } else {
  71. errMsg = "判断指标名称是否存在失败"
  72. err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
  73. return
  74. }
  75. }
  76. var edbInfoId int64
  77. // 指标已存在
  78. if edbInfo != nil && edbInfo.EdbInfoId > 0 {
  79. edbInfoId = int64(edbInfo.EdbInfoId)
  80. // 更新指标信息
  81. edbInfo.Source = edbItem.Source
  82. edbInfo.SourceName = edbItem.SourceName
  83. edbInfo.EdbType = edbItem.EdbType
  84. edbInfo.EdbCode = edbItem.EdbCode
  85. edbInfo.EdbName = edbItem.EdbName
  86. edbInfo.EdbNameSource = edbItem.EdbNameSource
  87. edbInfo.Frequency = edbItem.Frequency
  88. edbInfo.Unit = edbItem.Unit
  89. edbInfo.StartDate = edbItem.StartDate
  90. edbInfo.EndDate = edbItem.EndDate
  91. edbInfo.ClassifyId = edbItem.ClassifyId
  92. edbInfo.SysUserId = sysUserId
  93. edbInfo.SysUserRealName = sysUserRealName
  94. edbInfo.ModifyTime = time.Now()
  95. edbInfo.ServerUrl = edbItem.ServerUrl
  96. edbInfo.Sort = edbItem.Sort
  97. edbInfo.DataDateType = edbItem.DataDateType
  98. edbInfo.TerminalCode = edbItem.TerminalCode
  99. edbInfo.SourceIndexName = edbItem.SourceIndexName
  100. edbInfo.MaxValue = edbItem.MaxValue
  101. edbInfo.MinValue = edbItem.MinValue
  102. edbInfo.EdbType = edbItem.EdbType
  103. err = edbInfo.Update([]string{})
  104. if err != nil {
  105. errMsg = "保存失败"
  106. err = errors.New("保存失败,Err:" + err.Error())
  107. return
  108. }
  109. //同时删除指标的映射信息
  110. err = models.DeleteEdbInfoMapping(edbInfoId)
  111. if err != nil {
  112. errMsg = "删除指标映射信息失败"
  113. err = errors.New("删除指标映射信息失败,Err:" + err.Error())
  114. return
  115. }
  116. } else {
  117. edbInfo = new(models.EdbInfo)
  118. edbInfo.Source = edbItem.Source
  119. edbInfo.SourceName = edbItem.SourceName
  120. edbInfo.EdbType = edbItem.EdbType
  121. edbInfo.EdbCode = edbItem.EdbCode
  122. edbInfo.EdbName = edbItem.EdbName
  123. edbInfo.EdbNameSource = edbItem.EdbNameSource
  124. edbInfo.Frequency = edbItem.Frequency
  125. edbInfo.Unit = edbItem.Unit
  126. edbInfo.StartDate = edbItem.StartDate
  127. edbInfo.EndDate = edbItem.EndDate
  128. edbInfo.ClassifyId = edbItem.ClassifyId
  129. edbInfo.SysUserId = sysUserId
  130. edbInfo.SysUserRealName = sysUserRealName
  131. edbInfo.CreateTime = time.Now()
  132. edbInfo.ModifyTime = time.Now()
  133. edbInfo.ServerUrl = edbItem.ServerUrl
  134. edbInfo.Sort = edbItem.Sort
  135. edbInfo.DataDateType = edbItem.DataDateType
  136. edbInfo.TerminalCode = edbItem.TerminalCode
  137. edbInfo.SourceIndexName = edbItem.SourceIndexName
  138. timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
  139. edbInfo.UniqueCode = utils.MD5(utils.DATA_PREFIX + "_" + timestamp)
  140. edbInfo.MaxValue = edbItem.MaxValue
  141. edbInfo.MinValue = edbItem.MinValue
  142. edbInfo.EdbType = edbItem.EdbType
  143. edbInfoId, err = models.AddEdbInfo(edbInfo)
  144. if err != nil {
  145. errMsg = "保存失败"
  146. err = errors.New("保存失败,Err:" + err.Error())
  147. return
  148. }
  149. edbInfo.EdbInfoId = int(edbInfoId)
  150. }
  151. err = mgodb.ModifyEdbInfoDataStatus(edbInfoId, edbInfo.EdbCode)
  152. if err != nil {
  153. errMsg = "保存失败"
  154. err = errors.New("更新指标数据失败,Err:" + err.Error())
  155. return
  156. }
  157. return
  158. }
  159. // 批量删除指标,
  160. func BatchDeleteEdbInfo(edbInfoIds []int) (err error, errMsg string) {
  161. for _, v := range edbInfoIds {
  162. //查询单个指标是否允许删除
  163. err, errMsg = DeleteEdbInfo(v)
  164. if err != nil {
  165. errMsg = "删除指标失败"
  166. return
  167. }
  168. }
  169. return
  170. }
  171. // 删除单个指标
  172. func DeleteEdbInfo(edbInfoId int) (err error, errMsg string) {
  173. //判断指标是否存在
  174. _, err = models.GetEdbInfoById(edbInfoId)
  175. if err != nil {
  176. if err.Error() == utils.ErrNoRow() {
  177. err = nil
  178. return
  179. } else {
  180. errMsg = "判断指标名称是否存在失败"
  181. err = errors.New("判断指标名称是否存在失败,Err:" + err.Error())
  182. return
  183. }
  184. }
  185. //判断指标是否用于作图,如果用于作图,则不可删除
  186. chartCount, tmpErr := models.GetChartEdbMappingCount(edbInfoId)
  187. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  188. errMsg = "删除失败"
  189. err = errors.New("判断指标是否被用于作图失败,Err:" + tmpErr.Error())
  190. return
  191. }
  192. if chartCount > 0 {
  193. errMsg = "当前指标已用作画图,不可删除"
  194. return
  195. }
  196. //判断指标是否用作其他指标的计算
  197. calculateCount, tmpErr := models.GetEdbInfoCalculateMappingCount(edbInfoId)
  198. if tmpErr != nil {
  199. errMsg = "删除失败"
  200. err = errors.New("判断指标是否被用于计算失败,GetEdbInfoCalculateCount Err:" + tmpErr.Error())
  201. return
  202. }
  203. if calculateCount > 0 {
  204. errMsg = "当前指标已用作,指标运算,不可删除"
  205. return
  206. }
  207. // 删除指标,删除映射关系,删除指标数据
  208. err = models.DeleteEdbInfo(edbInfoId)
  209. if err != nil {
  210. errMsg = "删除失败"
  211. err = errors.New("删除指标失败,Err:" + err.Error())
  212. return
  213. }
  214. err = mgodb.DeleteEdbInfoDataByEdbInfoId(edbInfoId)
  215. if err != nil {
  216. errMsg = "删除失败"
  217. err = errors.New("删除指标数据失败,Err:" + err.Error())
  218. return
  219. }
  220. return
  221. }
  222. // TraceEdbInfoByEdbInfoId 指标追溯
  223. func TraceEdbInfoByEdbInfoId(edbInfoId, sysUserId int) (traceEdbInfo models.TraceEdbInfoResp, err error) {
  224. edbInfo, err := models.GetEdbInfoById(edbInfoId)
  225. if err != nil {
  226. return
  227. }
  228. edbInfoRuleMap := make(map[int]string, 0)
  229. edbMappingMap := make(map[int][]*models.EdbInfoCalculateMappingInfo)
  230. //edbInfoRuleMap[edbInfoId] = getEdbRuleTitle(edbInfo)
  231. traceEdbInfo = models.TraceEdbInfoResp{
  232. //EdbInfoId: edbInfo.EdbInfoId,
  233. EdbInfoId: edbInfoId,
  234. EdbInfoType: edbInfo.EdbInfoType,
  235. EdbName: edbInfo.EdbName,
  236. EdbType: edbInfo.EdbType,
  237. //Source: edbInfo.Source,
  238. UniqueCode: edbInfo.UniqueCode,
  239. ClassifyId: edbInfo.ClassifyId,
  240. IsStop: edbInfo.NoUpdate,
  241. EdbInfo: edbInfo,
  242. }
  243. findIdMap := make(map[int]int)
  244. findIdMap[edbInfoId] = edbInfoId
  245. existMap := make(map[int]models.TraceEdbInfoResp)
  246. traceEdbInfo.Child, err = traceEdbInfoByEdbInfoId(edbInfoId, traceEdbInfo, edbInfoRuleMap, findIdMap, existMap, edbMappingMap)
  247. edbInfoIdList := make([]int, 0)
  248. for _, v := range findIdMap {
  249. edbInfoIdList = append(edbInfoIdList, v)
  250. }
  251. classifyIdList := make([]int, 0)
  252. edbInfoList, err := models.GetEdbInfoByIdList(edbInfoIdList)
  253. if err != nil {
  254. return
  255. }
  256. edbInfoMap := make(map[int]*models.EdbInfo)
  257. for _, tmpEdbInfo := range edbInfoList {
  258. edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo
  259. classifyIdList = append(classifyIdList, tmpEdbInfo.ClassifyId)
  260. }
  261. traceEdbInfo, err = handleTraceEdbInfo(traceEdbInfo, 0, edbInfoMap, edbMappingMap)
  262. // 权限校验
  263. return
  264. }
  265. // TraceEdbInfoByEdbInfoIdList 指标追溯
  266. func TraceEdbInfoByEdbInfoIdList(edbInfoIdList []int) (traceEdbInfoList []models.TraceEdbInfoResp, err error) {
  267. traceEdbInfoList = make([]models.TraceEdbInfoResp, 0)
  268. edbInfoList, err := models.GetEdbInfoByIdList(edbInfoIdList)
  269. if err != nil {
  270. return
  271. }
  272. edbInfoRuleMap := make(map[int]string, 0)
  273. edbMappingMap := make(map[int][]*models.EdbInfoCalculateMappingInfo)
  274. findIdMap := make(map[int]int)
  275. existMap := make(map[int]models.TraceEdbInfoResp)
  276. for _, edbInfo := range edbInfoList {
  277. findIdMap[edbInfo.EdbInfoId] = edbInfo.EdbInfoId
  278. //edbInfoRuleMap[edbInfoId] = getEdbRuleTitle(edbInfo)
  279. traceEdbInfo := models.TraceEdbInfoResp{
  280. //EdbInfoId: edbInfo.EdbInfoId,
  281. EdbInfoId: edbInfo.EdbInfoId,
  282. EdbInfoType: edbInfo.EdbInfoType,
  283. EdbName: edbInfo.EdbName,
  284. EdbType: edbInfo.EdbType,
  285. //Source: edbInfo.Source,
  286. UniqueCode: edbInfo.UniqueCode,
  287. ClassifyId: edbInfo.ClassifyId,
  288. IsStop: edbInfo.NoUpdate,
  289. EdbInfo: edbInfo,
  290. }
  291. traceEdbInfo.Child, err = traceEdbInfoByEdbInfoId(edbInfo.EdbInfoId, traceEdbInfo, edbInfoRuleMap, findIdMap, existMap, edbMappingMap)
  292. traceEdbInfoList = append(traceEdbInfoList, traceEdbInfo)
  293. }
  294. //findEdbInfoIdList := make([]int, 0)
  295. //for _, v := range findIdMap {
  296. // findEdbInfoIdList = append(findEdbInfoIdList, v)
  297. //}
  298. //findEdbInfoList, err := models.GetEdbInfoByIdList(findEdbInfoIdList)
  299. //if err != nil {
  300. // return
  301. //}
  302. //edbInfoMap := make(map[int]*models.EdbInfo)
  303. //for _, tmpEdbInfo := range findEdbInfoList {
  304. // edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo
  305. //}
  306. //for k, traceEdbInfo := range traceEdbInfoList {
  307. // traceEdbInfoList[k], err = handleTraceEdbInfo(traceEdbInfo, 0, edbInfoMap, edbMappingMap)
  308. //}
  309. return
  310. }
  311. // traceEdbInfoByEdbInfoId 指标追溯
  312. func traceEdbInfoByEdbInfoId(edbInfoId int, traceEdbInfo models.TraceEdbInfoResp, edbInfoRuleMap map[int]string, findIdMap map[int]int, existMap map[int]models.TraceEdbInfoResp, edbMappingMap map[int][]*models.EdbInfoCalculateMappingInfo) (child []models.TraceEdbInfoResp, err error) {
  313. traceEdbInfo, ok := existMap[edbInfoId]
  314. if ok {
  315. return
  316. }
  317. child = make([]models.TraceEdbInfoResp, 0)
  318. edbInfoMappingList, e := models.GetEdbInfoCalculateMappingListByEdbInfoId(edbInfoId)
  319. if e != nil {
  320. err = fmt.Errorf("GetEdbInfoCalculateMappingListByEdbInfoId err: %s", e.Error())
  321. return
  322. }
  323. // 指标信息map
  324. edbInfoMap := make(map[int]*models.EdbInfo)
  325. if len(edbInfoMappingList) > 0 {
  326. fromEdbInfoIdList := make([]int, 0)
  327. for _, v := range edbInfoMappingList {
  328. fromEdbInfoIdList = append(fromEdbInfoIdList, v.FromEdbInfoId)
  329. }
  330. edbInfoList, tmpErr := models.GetEdbInfoByIdList(fromEdbInfoIdList)
  331. if tmpErr != nil {
  332. err = fmt.Errorf("traceEdbInfoByEdbInfoId GetEdbInfoByIdList err: %s", tmpErr.Error())
  333. return
  334. }
  335. for _, v := range edbInfoList {
  336. edbInfoMap[v.EdbInfoId] = v
  337. }
  338. }
  339. edbMappingMap[edbInfoId] = edbInfoMappingList
  340. for _, v := range edbInfoMappingList {
  341. tmpEdbInfoId := v.FromEdbInfoId
  342. tmpTraceEdbInfo := models.TraceEdbInfoResp{
  343. EdbInfoId: tmpEdbInfoId,
  344. EdbInfoType: v.FromEdbInfoType,
  345. EdbType: v.FromEdbType,
  346. UniqueCode: v.FromUniqueCode,
  347. ClassifyId: v.FromClassifyId,
  348. IsStop: v.NoUpdate,
  349. EdbInfo: edbInfoMap[v.FromEdbInfoId],
  350. }
  351. // 计算指标/预测指标继续溯源
  352. if edbInfoId != v.FromEdbInfoId && (v.FromEdbType == 2 || v.FromEdbInfoType == 1) {
  353. // 查过了就不查了
  354. if _, ok2 := findIdMap[tmpEdbInfoId]; !ok2 {
  355. tmpTraceEdbInfo.Child, e = traceEdbInfoByEdbInfoId(tmpEdbInfoId, tmpTraceEdbInfo, edbInfoRuleMap, findIdMap, existMap, edbMappingMap)
  356. if e != nil {
  357. err = fmt.Errorf("traceEdbInfoByEdbInfoId err: %s", e.Error())
  358. return
  359. }
  360. }
  361. }
  362. child = append(child, tmpTraceEdbInfo)
  363. findIdMap[tmpEdbInfoId] = tmpEdbInfoId
  364. }
  365. existMap[edbInfoId] = traceEdbInfo
  366. return
  367. }
  368. func handleTraceEdbInfo(traceEdbInfoResp models.TraceEdbInfoResp, parentEdbInfoId int, edbInfoMap map[int]*models.EdbInfo, edbMappingMap map[int][]*models.EdbInfoCalculateMappingInfo) (newTraceEdbInfoResp models.TraceEdbInfoResp, err error) {
  369. edbInfo, ok := edbInfoMap[traceEdbInfoResp.EdbInfoId]
  370. if !ok {
  371. err = errors.New("指标异常")
  372. return
  373. }
  374. var parentEdbInfo *models.EdbInfo
  375. if parentEdbInfoId > 0 {
  376. parentEdbInfo, ok = edbInfoMap[parentEdbInfoId]
  377. if !ok {
  378. err = errors.New("指标异常")
  379. return
  380. }
  381. }
  382. //traceEdbInfoResp.EdbName = edbInfo.EdbName
  383. traceEdbInfoResp.EdbName, traceEdbInfoResp.RuleTitle = getEdbRuleTitle(edbInfo, parentEdbInfo, traceEdbInfoResp.Child, edbInfoMap, edbMappingMap)
  384. if traceEdbInfoResp.Child != nil && len(traceEdbInfoResp.Child) > 0 {
  385. for k, v := range traceEdbInfoResp.Child {
  386. traceEdbInfoResp.Child[k], err = handleTraceEdbInfo(v, traceEdbInfoResp.EdbInfoId, edbInfoMap, edbMappingMap)
  387. if err != nil {
  388. return
  389. }
  390. }
  391. }
  392. newTraceEdbInfoResp = traceEdbInfoResp
  393. return
  394. }
  395. // 获取频度的英文版
  396. func GetFrequencyEn(frequency string) (frequencyEn string) {
  397. switch frequency {
  398. case "日度":
  399. frequencyEn = "day"
  400. return
  401. case "周度":
  402. frequencyEn = "week"
  403. return
  404. case "旬度":
  405. frequencyEn = "ten days"
  406. return
  407. case "月度":
  408. frequencyEn = "month"
  409. return
  410. case "季度":
  411. frequencyEn = "quarter"
  412. return
  413. case "年度":
  414. frequencyEn = "year"
  415. return
  416. }
  417. return
  418. }
  419. func GetLeadUnitEn(unit string) (unitEn string) {
  420. switch unit {
  421. case "天":
  422. unitEn = "day"
  423. return
  424. case "周":
  425. unitEn = "week"
  426. return
  427. case "月":
  428. unitEn = "month"
  429. return
  430. case "季":
  431. unitEn = "quarter"
  432. return
  433. case "年":
  434. unitEn = "year"
  435. return
  436. }
  437. return
  438. }
  439. func GetRefreshEdbInfoFromBase(edbInfoId, source int) (baseEdbInfoArr, calculateInfoArr []*models.EdbInfo, err error) {
  440. calculateList, err := models.GetEdbInfoCalculateMap(edbInfoId, source)
  441. if err != nil && err.Error() != utils.ErrNoRow() {
  442. return
  443. }
  444. for _, item := range calculateList {
  445. if item.EdbInfoId == edbInfoId { // 如果查出来关联的指标就是自己的话,那么就过滤
  446. continue
  447. }
  448. if item.EdbType == 1 {
  449. baseEdbInfoArr = append(baseEdbInfoArr, item)
  450. } else {
  451. calculateInfoArr = append(calculateInfoArr, item)
  452. newBaseEdbInfoArr, newCalculateInfoArr, _ := GetRefreshEdbInfoFromBase(item.EdbInfoId, item.Source)
  453. baseEdbInfoArr = append(baseEdbInfoArr, newBaseEdbInfoArr...)
  454. calculateInfoArr = append(calculateInfoArr, newCalculateInfoArr...)
  455. }
  456. }
  457. return
  458. }
  459. // getEdbRule 获取规则名称
  460. func getEdbRuleTitle(edbInfo, parentEdbInfo *models.EdbInfo, childList []models.TraceEdbInfoResp, edbInfoMap map[int]*models.EdbInfo, edbMappingMap map[int][]*models.EdbInfoCalculateMappingInfo) (edbName, ruleTitle string) {
  461. edbName = edbInfo.EdbName
  462. ruleTitle = `来源于` + edbInfo.SourceName
  463. if parentEdbInfo != nil {
  464. edbMappingList, ok := edbMappingMap[parentEdbInfo.EdbInfoId]
  465. if !ok {
  466. edbMappingList = []*models.EdbInfoCalculateMappingInfo{}
  467. }
  468. // 指标名称
  469. switch parentEdbInfo.Source {
  470. case utils.DATA_SOURCE_CALCULATE, utils.DATA_SOURCE_PREDICT_CALCULATE:
  471. for _, v := range edbMappingList {
  472. if v.FromEdbInfoId == edbInfo.EdbInfoId {
  473. edbName = fmt.Sprintf("%s(%s)", edbInfo.EdbName, v.FromTag)
  474. }
  475. }
  476. case utils.DATA_SOURCE_CALCULATE_ZJPJ, utils.DATA_SOURCE_PREDICT_CALCULATE_ZJPJ, utils.DATA_SOURCE_CALCULATE_LJZTBPJ, utils.DATA_SOURCE_PREDICT_CALCULATE_LJZTBPJ: // 直接拼接 ,累计值同比拼接
  477. for _, v := range edbMappingList {
  478. if v.FromEdbInfoId == edbInfo.EdbInfoId {
  479. tmpName := ``
  480. if v.FromTag == `A` {
  481. tmpName = `拼接日期前`
  482. } else if v.FromTag == `B` {
  483. tmpName = `拼接日期后`
  484. }
  485. edbName = fmt.Sprintf("%s(%s)", edbInfo.EdbName, tmpName)
  486. }
  487. }
  488. case utils.DATA_SOURCE_CALCULATE_NHCC, utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC: //计算指标(拟合残差)
  489. for _, v := range edbMappingList {
  490. //(需对上游指标+自变量,领先10天/因变量)
  491. if v.FromEdbInfoId == edbInfo.EdbInfoId {
  492. tmpName := ``
  493. if v.FromTag == `A` {
  494. tmpName = fmt.Sprintf(`自变量,领先%d天`, v.MoveValue)
  495. } else if v.FromTag == `B` {
  496. tmpName = `因变量`
  497. }
  498. edbName = fmt.Sprintf("%s(%s)", edbInfo.EdbName, tmpName)
  499. }
  500. }
  501. case utils.DATA_SOURCE_CALCULATE_CORRELATION: // 滚动相关性
  502. for _, v := range edbMappingList {
  503. if v.FromEdbInfoId == edbInfo.EdbInfoId {
  504. edbName = fmt.Sprintf("%s(%s)", edbInfo.EdbName, v.FromTag)
  505. }
  506. }
  507. }
  508. }
  509. if edbInfo.EdbType == 1 {
  510. // 基础指标的话,直接返回来源名称
  511. //ruleTitle = `来源于`+edbInfo.SourceName
  512. return
  513. }
  514. // 规则
  515. switch edbInfo.Source {
  516. case utils.DATA_SOURCE_CALCULATE, utils.DATA_SOURCE_PREDICT_CALCULATE:
  517. ruleTitle = "=" + edbInfo.CalculateFormula
  518. case utils.DATA_SOURCE_CALCULATE_LJZZY, utils.DATA_SOURCE_PREDICT_CALCULATE_LJZZY:
  519. ruleTitle = `累计转月值计算`
  520. case utils.DATA_SOURCE_CALCULATE_TBZ, utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ:
  521. ruleTitle = `同比值计算`
  522. case utils.DATA_SOURCE_CALCULATE_TCZ, utils.DATA_SOURCE_PREDICT_CALCULATE_TCZ:
  523. ruleTitle = `同差值计算`
  524. case utils.DATA_SOURCE_CALCULATE_NSZYDPJJS, utils.DATA_SOURCE_PREDICT_CALCULATE_NSZYDPJJS:
  525. ruleTitle = fmt.Sprintf("N数值移动均值计算(N=%s)", edbInfo.CalculateFormula)
  526. case utils.DATA_SOURCE_CALCULATE_HBZ, utils.DATA_SOURCE_PREDICT_CALCULATE_HBZ:
  527. ruleTitle = fmt.Sprintf("N数值环比值计算(N=%s)", edbInfo.CalculateFormula)
  528. case utils.DATA_SOURCE_CALCULATE_HCZ, utils.DATA_SOURCE_PREDICT_CALCULATE_HCZ:
  529. ruleTitle = fmt.Sprintf("N数值环差值计算(N=%s)", edbInfo.CalculateFormula)
  530. case utils.DATA_SOURCE_CALCULATE_TIME_SHIFT, utils.DATA_SOURCE_PREDICT_CALCULATE_TIME_SHIFT:
  531. moveType := `领先`
  532. if edbInfo.MoveType == 2 {
  533. moveType = "滞后"
  534. }
  535. ruleTitle = fmt.Sprintf("时间移位计算(%s%s%s)", moveType, edbInfo.CalculateFormula, edbInfo.MoveFrequency)
  536. case utils.DATA_SOURCE_CALCULATE_BP, utils.DATA_SOURCE_PREDICT_CALCULATE_BP: // 变频
  537. childFrequency := ``
  538. if len(childList) > 0 {
  539. if childEdbInfo, ok := edbInfoMap[childList[0].EdbInfoId]; ok {
  540. childFrequency = childEdbInfo.Frequency
  541. }
  542. }
  543. ruleTitle = fmt.Sprintf("升频计算(%s转%s)", childFrequency, edbInfo.Frequency)
  544. case utils.DATA_SOURCE_CALCULATE_ZJPJ, utils.DATA_SOURCE_PREDICT_CALCULATE_ZJPJ: // 直接拼接
  545. ruleTitle = fmt.Sprintf("直接拼接计算(%s)", edbInfo.CalculateFormula)
  546. case utils.DATA_SOURCE_CALCULATE_LJZTBPJ, utils.DATA_SOURCE_PREDICT_CALCULATE_LJZTBPJ: // 累计值同比拼
  547. ruleTitle = fmt.Sprintf("累计值同比值拼接计算(%s)", edbInfo.CalculateFormula)
  548. case utils.DATA_SOURCE_PYTHON:
  549. ruleTitle = `代码运算`
  550. case utils.DATA_SOURCE_CALCULATE_CJJX, utils.DATA_SOURCE_PREDICT_CALCULATE_CJJX:
  551. ruleTitle = fmt.Sprintf("超季节性计算(N=%s,%s)", edbInfo.CalculateFormula, edbInfo.Calendar)
  552. case utils.DATA_SOURCE_CALCULATE_NHCC, utils.DATA_SOURCE_PREDICT_CALCULATE_NHCC: //计算指标(拟合残差)
  553. var startDate, endDate string
  554. dateList := strings.Split(edbInfo.CalculateFormula, ",")
  555. if len(dateList) == 2 {
  556. startDate = dateList[0]
  557. endDate = dateList[1]
  558. }
  559. ruleTitle = fmt.Sprintf("拟合残差计算(%s至%s)", startDate, endDate)
  560. case utils.DATA_SOURCE_CALCULATE_ADJUST:
  561. ruleTitle = `数据调整`
  562. case utils.DATA_SOURCE_CALCULATE_NH, utils.DATA_SOURCE_PREDICT_CALCULATE_NH:
  563. ruleTitle = `年化计算`
  564. case utils.DATA_SOURCE_CALCULATE_KSZS, utils.DATA_SOURCE_PREDICT_CALCULATE_KSZS: // 扩散指数->53
  565. type KszsConfig struct {
  566. DateType int `description:"扩散指标日期;1:全部指标日期并集;2:部分指标日期并集"`
  567. CheckList []string `description:"选中的数据,A,B,C"`
  568. }
  569. var config KszsConfig
  570. err := json.Unmarshal([]byte(edbInfo.CalculateFormula), &config)
  571. if err != nil {
  572. return
  573. }
  574. var startDate, endDate time.Time
  575. childEdbInfoIdList := make([]int, 0)
  576. if config.DateType == 1 {
  577. for _, v := range childList {
  578. childEdbInfoIdList = append(childEdbInfoIdList, v.EdbInfoId)
  579. }
  580. } else {
  581. if parentEdbInfo != nil {
  582. edbMappingList, ok := edbMappingMap[parentEdbInfo.EdbInfoId]
  583. if !ok {
  584. edbMappingList = []*models.EdbInfoCalculateMappingInfo{}
  585. }
  586. tagMap := make(map[string]int)
  587. for _, v := range edbMappingList {
  588. tagMap[v.FromTag] = v.FromEdbInfoId
  589. }
  590. for _, v := range config.CheckList {
  591. if tmpEdbInfoId, ok := tagMap[v]; ok {
  592. childEdbInfoIdList = append(childEdbInfoIdList, tmpEdbInfoId)
  593. }
  594. }
  595. }
  596. }
  597. for _, v := range childEdbInfoIdList {
  598. if childEdbInfo, ok := edbInfoMap[v]; ok {
  599. tmpStartDate, tmpErr := time.ParseInLocation(utils.FormatDate, childEdbInfo.StartDate, time.Local)
  600. if tmpErr != nil {
  601. return
  602. }
  603. if startDate.IsZero() || startDate.After(tmpStartDate) {
  604. startDate = tmpStartDate
  605. }
  606. tmpEndDate, tmpErr := time.ParseInLocation(utils.FormatDate, childEdbInfo.EndDate, time.Local)
  607. if tmpErr != nil {
  608. return
  609. }
  610. if endDate.IsZero() || endDate.Before(tmpEndDate) {
  611. endDate = tmpEndDate
  612. }
  613. }
  614. }
  615. ruleTitle = fmt.Sprintf("扩散指数计算(%s至%s)", startDate.Format(utils.FormatDate), endDate.Format(utils.FormatDate))
  616. case utils.DATA_SOURCE_STOCK_PLANT:
  617. ruleTitle = `来源于装置分析`
  618. case utils.DATA_SOURCE_CALCULATE_CORRELATION:
  619. type EdbCalculateFormula struct {
  620. BaseCalculateValue int `description:"基础计算窗口"`
  621. BaseCalculateUnit string `description:"基础计算频度"`
  622. LeadValue int `description:"领先期数"`
  623. LeadUnit string `description:"频度"`
  624. CalculateValue int `description:"计算窗口"`
  625. CalculateUnit string `description:"计算频度"`
  626. }
  627. var correlationConf EdbCalculateFormula
  628. err := json.Unmarshal([]byte(edbInfo.CalculateFormula), &correlationConf)
  629. if err != nil {
  630. return
  631. }
  632. ruleTitle = fmt.Sprintf("滚动相关性(计算窗口%d%s,B领先A%d%s)", correlationConf.CalculateValue, correlationConf.CalculateUnit, correlationConf.LeadValue, correlationConf.LeadUnit)
  633. case utils.DATA_SOURCE_CALCULATE_JP, utils.DATA_SOURCE_PREDICT_CALCULATE_JP:
  634. childFrequency := ``
  635. if len(childList) > 0 {
  636. if childEdbInfo, ok := edbInfoMap[childList[0].EdbInfoId]; ok {
  637. childFrequency = childEdbInfo.Frequency
  638. }
  639. }
  640. ruleTitle = fmt.Sprintf("降频计算(%s转%s,%s)", childFrequency, edbInfo.Frequency, edbInfo.CalculateFormula)
  641. case utils.DATA_SOURCE_CALCULATE_STANDARD_DEVIATION:
  642. ruleTitle = fmt.Sprintf("标准差(滚动%s期)", edbInfo.CalculateFormula)
  643. case utils.DATA_SOURCE_CALCULATE_PERCENTILE, utils.DATA_SOURCE_PREDICT_CALCULATE_PERCENTILE:
  644. type TempCalculate struct {
  645. CalculateValue int `description:"计算窗口"`
  646. CalculateUnit string `description:"计算频度"`
  647. }
  648. cf := TempCalculate{}
  649. if e := json.Unmarshal([]byte(edbInfo.CalculateFormula), &cf); e != nil {
  650. return
  651. }
  652. ruleTitle = fmt.Sprintf("百分位(时间长度%d%s)", cf.CalculateValue, cf.CalculateUnit)
  653. case utils.DATA_SOURCE_CALCULATE_ZSXY, utils.DATA_SOURCE_PREDICT_CALCULATE_ZSXY:
  654. ruleTitle = `指数修匀计算`
  655. }
  656. return
  657. }