edb_info.go 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206
  1. package data
  2. import (
  3. "errors"
  4. "eta/eta_chart_lib/models"
  5. "eta/eta_chart_lib/models/data_manage"
  6. "eta/eta_chart_lib/services/alarm_msg"
  7. "eta/eta_chart_lib/utils"
  8. "fmt"
  9. "sort"
  10. "time"
  11. )
  12. // EdbInfoRefreshAllFromBaseBak
  13. // @author Roc
  14. // @datetime 2022-09-16 11:04:44
  15. // @description 全部刷新指标(切换到edb_lib服务),将原有的单个指标刷新,调整为批量多个指标刷新
  16. // DeprecatedTime 2023-10-23 09:38:19废弃
  17. // Deprecated
  18. func EdbInfoRefreshAllFromBaseBak(edbInfoIdList []int, refreshAll bool) (err error, errmsg string) {
  19. defer func() {
  20. if err != nil {
  21. fmt.Println("EdbInfoRefreshAllFromBaseV2 Err:" + err.Error() + ";errmsg:" + errmsg)
  22. go alarm_msg.SendAlarmMsg("EdbInfoRefreshFromBaseV2,Err"+err.Error()+";errMsg:"+errmsg, 3)
  23. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "EdbInfoRefreshFromBase:"+errmsg, utils.EmailSendToUsers)
  24. }
  25. }()
  26. // 获取关联的基础指标
  27. newBaseEdbInfoArr, newBasePredictEdbInfoArr, newCalculateMap, newPredictCalculateMap, calculateArr, predictCalculateArr, err, errmsg := getRefreshEdbInfoListByIds(edbInfoIdList)
  28. if err != nil {
  29. return
  30. }
  31. var startDate string
  32. for _, bv := range newBaseEdbInfoArr {
  33. //source := bv.Source
  34. //edbInfoId := bv.EdbInfoId
  35. //edbCode := bv.EdbCode
  36. if bv.StartDate == "0000-00-00" {
  37. continue
  38. }
  39. //开始时间
  40. startDate = ``
  41. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  42. sTime, tmpErr := time.Parse(utils.FormatDate, bv.StartDate)
  43. if tmpErr != nil {
  44. err = tmpErr
  45. errmsg = err.Error()
  46. return
  47. }
  48. startDate = sTime.Format(utils.FormatDate)
  49. } else {
  50. sTime, tmpErr := time.Parse(utils.FormatDate, bv.EndDate)
  51. if tmpErr != nil {
  52. err = tmpErr
  53. errmsg = err.Error()
  54. return
  55. }
  56. frequency := bv.Frequency
  57. var limitDay int
  58. switch frequency {
  59. case "日度":
  60. limitDay = utils.DATA_REFRESH
  61. case "周度":
  62. limitDay = utils.DATA_REFRESH * 7
  63. case "月度":
  64. limitDay = utils.DATA_REFRESH * 30
  65. case "季度":
  66. limitDay = utils.DATA_REFRESH * 90
  67. case "年度":
  68. limitDay = utils.DATA_REFRESH * 365
  69. default:
  70. limitDay = utils.DATA_REFRESH
  71. }
  72. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  73. }
  74. result, tmpErr := RefreshEdbData(bv.EdbInfoId, bv.Source, bv.EdbCode, startDate)
  75. if tmpErr != nil {
  76. err = tmpErr
  77. fmt.Println(bv.EdbInfoId, "RefreshBaseEdbData err", time.Now())
  78. errmsg = "RefreshBaseEdbData Err:" + tmpErr.Error()
  79. return
  80. }
  81. if result.Ret != 200 {
  82. fmt.Println(bv.EdbInfoId, "RefreshBaseEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  83. errmsg = fmt.Sprint(bv.EdbInfoId, "RefreshBaseEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  84. err = fmt.Errorf("刷新失败, err:", errmsg)
  85. return
  86. }
  87. //maxAndMinItem, err := data_manage.GetEdbInfoMaxAndMinInfo(source, edbCode)
  88. //if err != nil {
  89. // if err.Error() == utils.ErrNoRow() { //找不到数据,那么就进入到下一条数据做处理
  90. // continue
  91. // }
  92. // return err
  93. //}
  94. //if maxAndMinItem != nil {
  95. // err = data_manage.ModifyEdbInfoMaxAndMinInfo(edbInfoId, maxAndMinItem)
  96. // if err != nil {
  97. // return err
  98. // }
  99. //}
  100. fmt.Println("end newBaseEdbInfoArr:", bv, time.Now())
  101. }
  102. //刷新相关普通计算指标
  103. for _, v := range calculateArr {
  104. edbInfo := newCalculateMap[v]
  105. if edbInfo == nil {
  106. return
  107. }
  108. startDate = edbInfo.StartDate
  109. source := edbInfo.Source
  110. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  111. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  112. } else {
  113. if source == utils.DATA_SOURCE_CALCULATE {
  114. startDate = ``
  115. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  116. startDate = edbInfo.StartDate
  117. } else {
  118. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  119. if tmpErr != nil {
  120. err = tmpErr
  121. errmsg = tmpErr.Error()
  122. return
  123. }
  124. frequency := edbInfo.Frequency
  125. var limitDay int
  126. switch frequency {
  127. case "日度":
  128. limitDay = utils.DATA_REFRESH
  129. case "周度":
  130. limitDay = utils.DATA_REFRESH * 7
  131. case "月度":
  132. limitDay = utils.DATA_REFRESH * 30
  133. case "季度":
  134. limitDay = utils.DATA_REFRESH * 90
  135. case "年度":
  136. limitDay = utils.DATA_REFRESH * 365
  137. default:
  138. limitDay = utils.DATA_REFRESH
  139. }
  140. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  141. }
  142. }
  143. }
  144. result, tmpErr := RefreshEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  145. if tmpErr != nil {
  146. err = tmpErr
  147. fmt.Println(v, "RefreshEdbCalculateData err", time.Now())
  148. errmsg = "RefreshEdbCalculateData Err:" + tmpErr.Error()
  149. return
  150. }
  151. if result.Ret != 200 {
  152. fmt.Println(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  153. errmsg = fmt.Sprint(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  154. err = fmt.Errorf("刷新失败")
  155. return
  156. }
  157. }
  158. //刷新相关预测计算指标
  159. for _, v := range predictCalculateArr {
  160. edbInfo := newPredictCalculateMap[v]
  161. if edbInfo == nil {
  162. return
  163. }
  164. startDate = edbInfo.StartDate
  165. source := edbInfo.Source
  166. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  167. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  168. } else {
  169. if source == utils.DATA_SOURCE_PREDICT_CALCULATE {
  170. startDate = ``
  171. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  172. startDate = edbInfo.StartDate
  173. } else {
  174. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  175. if tmpErr != nil {
  176. err = tmpErr
  177. errmsg = tmpErr.Error()
  178. return
  179. }
  180. frequency := edbInfo.Frequency
  181. var limitDay int
  182. switch frequency {
  183. case "日度":
  184. limitDay = utils.DATA_REFRESH
  185. case "周度":
  186. limitDay = utils.DATA_REFRESH * 7
  187. case "月度":
  188. limitDay = utils.DATA_REFRESH * 30
  189. case "季度":
  190. limitDay = utils.DATA_REFRESH * 90
  191. case "年度":
  192. limitDay = utils.DATA_REFRESH * 365
  193. default:
  194. limitDay = utils.DATA_REFRESH
  195. }
  196. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  197. }
  198. }
  199. }
  200. result, tmpErr := RefreshPredictEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  201. if tmpErr != nil {
  202. err = tmpErr
  203. fmt.Println(v, "RefreshPredictEdbCalculateData err", time.Now())
  204. errmsg = "RefreshPredictEdbCalculateData Err:" + tmpErr.Error()
  205. return
  206. }
  207. if result.Ret != 200 {
  208. fmt.Println(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  209. errmsg = fmt.Sprint(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  210. err = fmt.Errorf("刷新失败")
  211. return
  212. }
  213. }
  214. // 4、更新动态环差值
  215. {
  216. predictEdbInfoIdList := make([]int, 0)
  217. for _, v := range newBasePredictEdbInfoArr {
  218. predictEdbInfoIdList = append(predictEdbInfoIdList, v.EdbInfoId)
  219. }
  220. fmt.Println("predictEdbConfCalculateMappingDetailList:", predictEdbInfoIdList)
  221. predictEdbConfCalculateMappingDetailList, tmpErr := data_manage.GetPredictEdbConfCalculateMappingDetailListByEdbInfoId(predictEdbInfoIdList)
  222. if tmpErr != nil {
  223. err = tmpErr
  224. errmsg = fmt.Sprint("更新动态环差值时,获取数据失败:" + err.Error())
  225. return
  226. }
  227. for _, bv := range predictEdbConfCalculateMappingDetailList {
  228. result, tmpErr := RefreshEdbData(bv.EdbInfoId, utils.DATA_SOURCE_PREDICT, bv.EdbCode, "")
  229. if tmpErr != nil {
  230. err = tmpErr
  231. fmt.Println(bv.EdbInfoId, "RefreshBasePredictEdbData err", time.Now())
  232. errmsg = "RefreshBasePredictEdbData Err:" + err.Error()
  233. return
  234. }
  235. if result.Ret != 200 {
  236. fmt.Println(bv.EdbInfoId, "RefreshBasePredictEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  237. errmsg = fmt.Sprint(bv.EdbInfoId, "RefreshBasePredictEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  238. err = errors.New("刷新失败:" + errmsg)
  239. return
  240. }
  241. fmt.Println("end predictEdbConfCalculateMappingDetailList:", bv, time.Now())
  242. }
  243. }
  244. //5、刷新相关预测计算指标
  245. for _, v := range predictCalculateArr {
  246. edbInfo := newPredictCalculateMap[v]
  247. if edbInfo == nil {
  248. return
  249. }
  250. startDate = edbInfo.StartDate
  251. source := edbInfo.Source
  252. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  253. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  254. } else {
  255. if source == utils.DATA_SOURCE_PREDICT_CALCULATE {
  256. startDate = ``
  257. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  258. startDate = edbInfo.StartDate
  259. } else {
  260. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  261. if tmpErr != nil {
  262. err = tmpErr
  263. errmsg = tmpErr.Error()
  264. return
  265. }
  266. frequency := edbInfo.Frequency
  267. var limitDay int
  268. switch frequency {
  269. case "日度":
  270. limitDay = utils.DATA_REFRESH
  271. case "周度":
  272. limitDay = utils.DATA_REFRESH * 7
  273. case "月度":
  274. limitDay = utils.DATA_REFRESH * 30
  275. case "季度":
  276. limitDay = utils.DATA_REFRESH * 90
  277. case "年度":
  278. limitDay = utils.DATA_REFRESH * 365
  279. default:
  280. limitDay = utils.DATA_REFRESH
  281. }
  282. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  283. }
  284. }
  285. }
  286. result, tmpErr := RefreshPredictEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  287. if tmpErr != nil {
  288. err = tmpErr
  289. fmt.Println(v, "RefreshPredictEdbCalculateData err", time.Now())
  290. errmsg = "RefreshPredictEdbCalculateData Err:" + tmpErr.Error()
  291. return
  292. }
  293. if result.Ret != 200 {
  294. fmt.Println(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  295. errmsg = fmt.Sprint(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  296. err = fmt.Errorf("刷新失败")
  297. return
  298. }
  299. }
  300. return
  301. }
  302. // getRefreshEdbInfoList 获取待更新的指标(普通基础指标、普通运算指标,预测运算指标)
  303. func getRefreshEdbInfoListByIds(edbInfoIdList []int) (newBaseEdbInfoArr, newBasePredictEdbInfoArr []*data_manage.EdbInfo, newCalculateMap, newPredictCalculateMap map[int]*data_manage.EdbInfo, calculateArr, predictCalculateArr []int, err error, errMsg string) {
  304. calculateList, err := data_manage.GetEdbInfoAllCalculateByEdbInfoIdList(edbInfoIdList)
  305. if err != nil && err.Error() != utils.ErrNoRow() {
  306. err = errors.New("GetEdbInfoAllCalculate Err:" + err.Error())
  307. return
  308. }
  309. // 获取指标信息
  310. edbInfoList, err := data_manage.GetEdbInfoByIdList(edbInfoIdList)
  311. if err != nil {
  312. err = errors.New("GetEdbInfoAllCalGetEdbInfoByIdr:" + err.Error())
  313. return
  314. }
  315. baseEdbInfoArr := make([]*data_manage.EdbInfo, 0) // 基础普通指标
  316. calculateInfoArr := make([]*data_manage.EdbInfo, 0) //基础运算指标
  317. basePredictEdbInfoArr := make([]*data_manage.EdbInfo, 0) // 预测普通指标
  318. predictCalculateInfoArr := make([]*data_manage.EdbInfo, 0) //预测运算指标
  319. for _, baseEdbInfo := range edbInfoList {
  320. if baseEdbInfo.EdbInfoType == 0 { //普通指标
  321. switch baseEdbInfo.EdbType {
  322. case 1: //基础指标
  323. baseEdbInfoArr = append(baseEdbInfoArr, baseEdbInfo)
  324. case 2: //计算指标
  325. calculateList = append(calculateList, baseEdbInfo)
  326. }
  327. } else { //预测指标
  328. switch baseEdbInfo.EdbType {
  329. case 1: //基础指标
  330. basePredictEdbInfoArr = append(basePredictEdbInfoArr, baseEdbInfo)
  331. case 2: //计算指标
  332. predictCalculateInfoArr = append(predictCalculateInfoArr, baseEdbInfo)
  333. }
  334. }
  335. }
  336. // 计算指标
  337. for _, v := range calculateList {
  338. getBaseEdbInfoArr, getCalculateInfoArr, tmpErr := data_manage.GetRefreshEdbInfoFromBase(v.EdbInfoId, v.Source)
  339. if tmpErr != nil {
  340. err = tmpErr
  341. fmt.Println("GetRefreshEdbInfoFromBase err:" + err.Error())
  342. errMsg = "获取需要刷新的指标失败:Err:" + err.Error()
  343. return
  344. }
  345. for _, tmpEdbInfo := range getBaseEdbInfoArr {
  346. switch tmpEdbInfo.EdbInfoType {
  347. case 0: //普通基础指标
  348. baseEdbInfoArr = append(baseEdbInfoArr, tmpEdbInfo)
  349. case 1: //预测基础指标
  350. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  351. }
  352. }
  353. for _, tmpEdbInfo := range getCalculateInfoArr {
  354. switch tmpEdbInfo.EdbInfoType {
  355. case 0: //普通运算指标
  356. calculateInfoArr = append(calculateInfoArr, tmpEdbInfo)
  357. case 1: //预测运算指标
  358. //predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  359. if tmpEdbInfo.EdbType == 1 { //预测普通指标
  360. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  361. } else {
  362. predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  363. }
  364. }
  365. }
  366. switch v.EdbInfoType {
  367. case 0: //普通运算指标
  368. calculateInfoArr = append(calculateInfoArr, v)
  369. case 1: //预测运算指标
  370. if v.EdbType == 1 { //预测普通指标
  371. basePredictEdbInfoArr = append(basePredictEdbInfoArr, v)
  372. } else {
  373. predictCalculateInfoArr = append(predictCalculateInfoArr, v)
  374. }
  375. }
  376. }
  377. // 预测计算指标
  378. for _, v := range predictCalculateInfoArr {
  379. getBaseEdbInfoArr, getCalculateInfoArr, tmpErr := data_manage.GetRefreshEdbInfoFromBase(v.EdbInfoId, v.Source)
  380. if tmpErr != nil {
  381. err = tmpErr
  382. fmt.Println("GetRefreshEdbInfoFromBase err:" + err.Error())
  383. errMsg = "获取需要刷新的指标失败:Err:" + err.Error()
  384. return
  385. }
  386. for _, tmpEdbInfo := range getBaseEdbInfoArr {
  387. switch tmpEdbInfo.EdbInfoType {
  388. case 0: //普通基础指标
  389. baseEdbInfoArr = append(baseEdbInfoArr, tmpEdbInfo)
  390. case 1: //预测基础指标
  391. if tmpEdbInfo.EdbType == 1 { //预测普通指标
  392. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  393. } else {
  394. predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  395. }
  396. //basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  397. }
  398. }
  399. for _, tmpEdbInfo := range getCalculateInfoArr {
  400. switch tmpEdbInfo.EdbInfoType {
  401. case 0: //普通运算指标
  402. calculateInfoArr = append(calculateInfoArr, tmpEdbInfo)
  403. case 1: //预测运算指标
  404. if tmpEdbInfo.EdbType == 1 { //预测普通指标
  405. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  406. } else {
  407. predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  408. }
  409. //predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  410. }
  411. }
  412. switch v.EdbInfoType {
  413. case 0: //普通运算指标
  414. calculateInfoArr = append(calculateInfoArr, v)
  415. case 1: //预测运算指标
  416. if v.EdbType == 1 { //预测普通指标
  417. basePredictEdbInfoArr = append(basePredictEdbInfoArr, v)
  418. } else {
  419. predictCalculateInfoArr = append(predictCalculateInfoArr, v)
  420. }
  421. }
  422. }
  423. // 第一次的计算指标map
  424. newCalculateMap = make(map[int]*data_manage.EdbInfo)
  425. for _, v := range calculateInfoArr {
  426. if _, ok := newCalculateMap[v.EdbInfoId]; !ok {
  427. calculateArr = append(calculateArr, v.EdbInfoId)
  428. }
  429. newCalculateMap[v.EdbInfoId] = v
  430. }
  431. // 基础预测指标
  432. if len(basePredictEdbInfoArr) > 0 {
  433. basePredictEdbInfoIdList := make([]int, 0)
  434. for _, v := range basePredictEdbInfoArr {
  435. basePredictEdbInfoIdList = append(basePredictEdbInfoIdList, v.EdbInfoId)
  436. }
  437. baseEdbInfoList, tmpErr := data_manage.GetPredictEdbInfoAllCalculate(basePredictEdbInfoIdList)
  438. if tmpErr != nil {
  439. err = tmpErr
  440. return
  441. }
  442. tmpCalculateList := make([]*data_manage.EdbInfo, 0)
  443. for _, v := range baseEdbInfoList {
  444. switch v.EdbType {
  445. case 1:
  446. baseEdbInfoArr = append(baseEdbInfoArr, v)
  447. case 2:
  448. // 如果不在第一次里面,那么在第二次 刷新预测指标的时候 需要加进去
  449. if _, ok := newCalculateMap[v.EdbInfoId]; !ok {
  450. tmpCalculateList = append(tmpCalculateList, v)
  451. }
  452. }
  453. }
  454. // 这里是查询预测指标用到的 普通计算指标
  455. for _, v := range tmpCalculateList {
  456. getBaseEdbInfoArr, getCalculateInfoArr, tmpErr := data_manage.GetRefreshEdbInfoFromBase(v.EdbInfoId, v.Source)
  457. if tmpErr != nil {
  458. err = tmpErr
  459. fmt.Println("GetRefreshEdbInfoFromBase err:" + err.Error())
  460. errMsg = "获取需要刷新的指标失败:Err:" + err.Error()
  461. return
  462. }
  463. // 基础指标
  464. for _, tmpEdbInfo := range getBaseEdbInfoArr {
  465. switch tmpEdbInfo.EdbInfoType {
  466. case 0: //普通基础指标
  467. baseEdbInfoArr = append(baseEdbInfoArr, tmpEdbInfo)
  468. case 1: //预测基础指标
  469. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  470. }
  471. }
  472. // 计算指标
  473. for _, tmpEdbInfo := range getCalculateInfoArr {
  474. switch tmpEdbInfo.EdbInfoType {
  475. case 0: //普通运算指标
  476. calculateInfoArr = append(calculateInfoArr, tmpEdbInfo)
  477. case 1: //预测运算指标
  478. if tmpEdbInfo.EdbType == 1 { //预测普通指标
  479. basePredictEdbInfoArr = append(basePredictEdbInfoArr, tmpEdbInfo)
  480. } else {
  481. predictCalculateInfoArr = append(predictCalculateInfoArr, tmpEdbInfo)
  482. }
  483. }
  484. }
  485. switch v.EdbInfoType {
  486. case 0: //普通运算指标
  487. calculateInfoArr = append(calculateInfoArr, v)
  488. case 1: //预测运算指标
  489. if v.EdbType == 1 { //预测普通指标
  490. basePredictEdbInfoArr = append(basePredictEdbInfoArr, v)
  491. } else {
  492. predictCalculateInfoArr = append(predictCalculateInfoArr, v)
  493. }
  494. }
  495. }
  496. // 第二次计算指标的map
  497. for _, v := range calculateInfoArr {
  498. if _, ok := newCalculateMap[v.EdbInfoId]; !ok {
  499. calculateArr = append(calculateArr, v.EdbInfoId)
  500. }
  501. newCalculateMap[v.EdbInfoId] = v
  502. }
  503. }
  504. // 普通基础指标去重
  505. newBaseEdbInfoArr = make([]*data_manage.EdbInfo, 0)
  506. baseMap := make(map[int]int)
  507. for _, v := range baseEdbInfoArr {
  508. if _, ok := baseMap[v.EdbInfoId]; !ok {
  509. newBaseEdbInfoArr = append(newBaseEdbInfoArr, v)
  510. }
  511. baseMap[v.EdbInfoId] = v.EdbInfoId
  512. }
  513. // 普通计算指标的id
  514. sort.Ints(calculateArr)
  515. // 普通预测指标去重
  516. newBasePredictEdbInfoArr = make([]*data_manage.EdbInfo, 0)
  517. basePredictMap := make(map[int]int)
  518. for _, v := range basePredictEdbInfoArr {
  519. if _, ok := basePredictMap[v.EdbInfoId]; !ok {
  520. newBasePredictEdbInfoArr = append(newBasePredictEdbInfoArr, v)
  521. }
  522. basePredictMap[v.EdbInfoId] = v.EdbInfoId
  523. }
  524. // 预测计算指标去重
  525. newPredictCalculateMap = make(map[int]*data_manage.EdbInfo)
  526. for _, v := range predictCalculateInfoArr {
  527. if _, ok := newPredictCalculateMap[v.EdbInfoId]; !ok {
  528. predictCalculateArr = append(predictCalculateArr, v.EdbInfoId)
  529. }
  530. newPredictCalculateMap[v.EdbInfoId] = v
  531. }
  532. // 预测计算指标的id
  533. sort.Ints(predictCalculateArr)
  534. return
  535. }
  536. // GetEdbSourceByEdbInfoIdList 获取关联指标的来源
  537. func GetEdbSourceByEdbInfoIdList(chartEdbInfoMappingList []*models.ChartEdbInfoMapping) (sourceNameList, sourceNameEnList []string) {
  538. sourceNameList = make([]string, 0)
  539. sourceNameEnList = make([]string, 0)
  540. sourceMap := make(map[int]string)
  541. for _, v := range chartEdbInfoMappingList {
  542. // 指标类型:1:基础指标,2:计算指标
  543. if v.EdbType == 2 || v.EdbInfoCategoryType == 1 {
  544. //sourceMap[0] = "弘则研究"
  545. baseEdbInfoArr, _, _ := data_manage.GetRefreshEdbInfoFromBase(v.EdbInfoId, v.Source)
  546. for _, baseEdbInfo := range baseEdbInfoArr {
  547. if baseEdbInfo.EdbInfoType == 0 { //普通指标才参与,预测指标不参与
  548. sourceMap[baseEdbInfo.Source] = baseEdbInfo.SourceName
  549. }
  550. }
  551. } else {
  552. sourceMap[v.Source] = v.SourceName
  553. }
  554. }
  555. for source, sourceName := range sourceMap {
  556. if utils.InArrayByInt([]int{utils.DATA_SOURCE_MANUAL, utils.DATA_SOURCE_MYSTEEL_CHEMICAL}, source) {
  557. continue
  558. }
  559. sourceNameList = append(sourceNameList, sourceName)
  560. sourceNameEn, ok := utils.DataSourceEnMap[source]
  561. if !ok {
  562. sourceNameEn = sourceName
  563. }
  564. sourceNameEnList = append(sourceNameEnList, sourceNameEn)
  565. }
  566. //sourceNameList = append(sourceNameList, utils.ChartDefaultNameCn)
  567. //sourceNameEnList = append(sourceNameEnList, utils.ChartDefaultNameEn)
  568. // 图表来源
  569. conf, e := models.GetBusinessConf()
  570. if e != nil {
  571. return
  572. }
  573. if conf[models.BusinessConfCompanyName] != "" {
  574. sourceNameList = append(sourceNameList, conf[models.BusinessConfCompanyName])
  575. sourceNameEnList = append(sourceNameEnList, conf[models.BusinessConfCompanyName])
  576. }
  577. return
  578. }
  579. // edbInfoRefreshAll
  580. //
  581. // @Description: 指标刷新服务
  582. // @author: Roc
  583. // @datetime2023-10-23 11:04:30
  584. // @param refreshAll bool
  585. // @param newBaseEdbInfoArr []*data_manage.EdbInfo
  586. // @param newBasePredictEdbInfoArr []*data_manage.EdbInfo
  587. // @param newCalculateMap map[int]*data_manage.EdbInfo
  588. // @param newPredictCalculateMap map[int]*data_manage.EdbInfo
  589. // @param calculateArr []int
  590. // @param predictCalculateArr []int
  591. // @return err error
  592. // @return errmsg string
  593. func edbInfoRefreshAll(refreshAll bool, newBaseEdbInfoArr, newBasePredictEdbInfoArr []*data_manage.EdbInfo, newCalculateMap, newPredictCalculateMap map[int]*data_manage.EdbInfo, calculateArr, predictCalculateArr []int) (err error, errmsg string) {
  594. defer func() {
  595. if err != nil {
  596. fmt.Println("EdbInfoRefreshAllFromBaseV2 Err:" + err.Error() + ";errmsg:" + errmsg)
  597. go alarm_msg.SendAlarmMsg("EdbInfoRefreshFromBaseV2,Err"+err.Error()+";errMsg:"+errmsg, 3)
  598. //go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "EdbInfoRefreshFromBase:"+errmsg, utils.EmailSendToUsers)
  599. }
  600. }()
  601. var startDate string
  602. for _, bv := range newBaseEdbInfoArr {
  603. //source := bv.Source
  604. //edbInfoId := bv.EdbInfoId
  605. //edbCode := bv.EdbCode
  606. if bv.StartDate == "0000-00-00" {
  607. continue
  608. }
  609. //开始时间
  610. startDate = ``
  611. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  612. sTime, tmpErr := time.Parse(utils.FormatDate, bv.StartDate)
  613. if tmpErr != nil {
  614. err = tmpErr
  615. errmsg = err.Error()
  616. return
  617. }
  618. startDate = sTime.Format(utils.FormatDate)
  619. } else {
  620. sTime, tmpErr := time.Parse(utils.FormatDate, bv.EndDate)
  621. if tmpErr != nil {
  622. err = tmpErr
  623. errmsg = err.Error()
  624. return
  625. }
  626. frequency := bv.Frequency
  627. var limitDay int
  628. switch frequency {
  629. case "日度":
  630. limitDay = utils.DATA_REFRESH
  631. case "周度":
  632. limitDay = utils.DATA_REFRESH * 7
  633. case "月度":
  634. limitDay = utils.DATA_REFRESH * 30
  635. case "季度":
  636. limitDay = utils.DATA_REFRESH * 90
  637. case "年度":
  638. limitDay = utils.DATA_REFRESH * 365
  639. default:
  640. limitDay = utils.DATA_REFRESH
  641. }
  642. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  643. }
  644. result, tmpErr := RefreshEdbData(bv.EdbInfoId, bv.Source, bv.EdbCode, startDate)
  645. if tmpErr != nil {
  646. err = tmpErr
  647. fmt.Println(bv.EdbInfoId, "RefreshBaseEdbData err", time.Now())
  648. errmsg = "RefreshBaseEdbData Err:" + tmpErr.Error()
  649. return
  650. }
  651. if result.Ret != 200 {
  652. fmt.Println(bv.EdbInfoId, "RefreshBaseEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  653. errmsg = fmt.Sprint(bv.EdbInfoId, "RefreshBaseEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  654. err = fmt.Errorf("刷新失败, err:", errmsg)
  655. return
  656. }
  657. //maxAndMinItem, err := data_manage.GetEdbInfoMaxAndMinInfo(source, edbCode)
  658. //if err != nil {
  659. // if err.Error() == utils.ErrNoRow() { //找不到数据,那么就进入到下一条数据做处理
  660. // continue
  661. // }
  662. // return err
  663. //}
  664. //if maxAndMinItem != nil {
  665. // err = data_manage.ModifyEdbInfoMaxAndMinInfo(edbInfoId, maxAndMinItem)
  666. // if err != nil {
  667. // return err
  668. // }
  669. //}
  670. fmt.Println("end newBaseEdbInfoArr:", bv, time.Now())
  671. }
  672. //刷新相关普通计算指标
  673. for _, v := range calculateArr {
  674. edbInfo := newCalculateMap[v]
  675. if edbInfo == nil {
  676. return
  677. }
  678. startDate = edbInfo.StartDate
  679. source := edbInfo.Source
  680. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  681. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  682. } else {
  683. if source == utils.DATA_SOURCE_CALCULATE {
  684. startDate = ``
  685. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  686. startDate = edbInfo.StartDate
  687. } else {
  688. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  689. if tmpErr != nil {
  690. err = tmpErr
  691. errmsg = tmpErr.Error()
  692. return
  693. }
  694. frequency := edbInfo.Frequency
  695. var limitDay int
  696. switch frequency {
  697. case "日度":
  698. limitDay = utils.DATA_REFRESH
  699. case "周度":
  700. limitDay = utils.DATA_REFRESH * 7
  701. case "月度":
  702. limitDay = utils.DATA_REFRESH * 30
  703. case "季度":
  704. limitDay = utils.DATA_REFRESH * 90
  705. case "年度":
  706. limitDay = utils.DATA_REFRESH * 365
  707. default:
  708. limitDay = utils.DATA_REFRESH
  709. }
  710. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  711. }
  712. }
  713. }
  714. result, tmpErr := RefreshEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  715. if tmpErr != nil {
  716. err = tmpErr
  717. fmt.Println(v, "RefreshEdbCalculateData err", time.Now())
  718. errmsg = "RefreshEdbCalculateData Err:" + tmpErr.Error()
  719. return
  720. }
  721. if result.Ret != 200 {
  722. fmt.Println(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  723. errmsg = fmt.Sprint(v, "RefreshEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  724. err = fmt.Errorf("刷新失败")
  725. return
  726. }
  727. }
  728. //刷新相关预测计算指标
  729. for _, v := range predictCalculateArr {
  730. edbInfo := newPredictCalculateMap[v]
  731. if edbInfo == nil {
  732. return
  733. }
  734. startDate = edbInfo.StartDate
  735. source := edbInfo.Source
  736. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  737. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  738. } else {
  739. if source == utils.DATA_SOURCE_PREDICT_CALCULATE {
  740. startDate = ``
  741. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  742. startDate = edbInfo.StartDate
  743. } else {
  744. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  745. if tmpErr != nil {
  746. err = tmpErr
  747. errmsg = tmpErr.Error()
  748. return
  749. }
  750. frequency := edbInfo.Frequency
  751. var limitDay int
  752. switch frequency {
  753. case "日度":
  754. limitDay = utils.DATA_REFRESH
  755. case "周度":
  756. limitDay = utils.DATA_REFRESH * 7
  757. case "月度":
  758. limitDay = utils.DATA_REFRESH * 30
  759. case "季度":
  760. limitDay = utils.DATA_REFRESH * 90
  761. case "年度":
  762. limitDay = utils.DATA_REFRESH * 365
  763. default:
  764. limitDay = utils.DATA_REFRESH
  765. }
  766. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  767. }
  768. }
  769. }
  770. result, tmpErr := RefreshPredictEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  771. if tmpErr != nil {
  772. err = tmpErr
  773. fmt.Println(v, "RefreshPredictEdbCalculateData err", time.Now())
  774. errmsg = "RefreshPredictEdbCalculateData Err:" + tmpErr.Error()
  775. return
  776. }
  777. if result.Ret != 200 {
  778. fmt.Println(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  779. errmsg = fmt.Sprint(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  780. err = fmt.Errorf("刷新失败")
  781. return
  782. }
  783. }
  784. // 4、更新动态环差值
  785. {
  786. predictEdbInfoIdList := make([]int, 0)
  787. for _, v := range newBasePredictEdbInfoArr {
  788. predictEdbInfoIdList = append(predictEdbInfoIdList, v.EdbInfoId)
  789. }
  790. fmt.Println("predictEdbConfCalculateMappingDetailList:", predictEdbInfoIdList)
  791. predictEdbConfCalculateMappingDetailList, tmpErr := data_manage.GetPredictEdbConfCalculateMappingDetailListByEdbInfoId(predictEdbInfoIdList)
  792. if tmpErr != nil {
  793. err = tmpErr
  794. errmsg = fmt.Sprint("更新动态环差值时,获取数据失败:" + err.Error())
  795. return
  796. }
  797. for _, bv := range predictEdbConfCalculateMappingDetailList {
  798. result, tmpErr := RefreshEdbData(bv.EdbInfoId, utils.DATA_SOURCE_PREDICT, bv.EdbCode, "")
  799. if tmpErr != nil {
  800. err = tmpErr
  801. fmt.Println(bv.EdbInfoId, "RefreshBasePredictEdbData err", time.Now())
  802. errmsg = "RefreshBasePredictEdbData Err:" + err.Error()
  803. return
  804. }
  805. if result.Ret != 200 {
  806. fmt.Println(bv.EdbInfoId, "RefreshBasePredictEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  807. errmsg = fmt.Sprint(bv.EdbInfoId, "RefreshBasePredictEdbData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  808. err = errors.New("刷新失败:" + errmsg)
  809. return
  810. }
  811. fmt.Println("end predictEdbConfCalculateMappingDetailList:", bv, time.Now())
  812. }
  813. }
  814. //5、刷新相关预测计算指标
  815. for _, v := range predictCalculateArr {
  816. edbInfo := newPredictCalculateMap[v]
  817. if edbInfo == nil {
  818. return
  819. }
  820. startDate = edbInfo.StartDate
  821. source := edbInfo.Source
  822. if startDate == "" || startDate == "0000-00-00" { //如果没有开始日期,说明还没有计算出来数据,那么就往前面推40年吧(也意味着重新计算了)
  823. startDate = time.Now().AddDate(-40, 0, 0).Format(utils.FormatDate)
  824. } else {
  825. if source == utils.DATA_SOURCE_PREDICT_CALCULATE {
  826. startDate = ``
  827. if refreshAll { //刷新所有数据,用开始时间作为起始日期去刷新
  828. startDate = edbInfo.StartDate
  829. } else {
  830. sTime, tmpErr := time.Parse(utils.FormatDate, edbInfo.EndDate)
  831. if tmpErr != nil {
  832. err = tmpErr
  833. errmsg = tmpErr.Error()
  834. return
  835. }
  836. frequency := edbInfo.Frequency
  837. var limitDay int
  838. switch frequency {
  839. case "日度":
  840. limitDay = utils.DATA_REFRESH
  841. case "周度":
  842. limitDay = utils.DATA_REFRESH * 7
  843. case "月度":
  844. limitDay = utils.DATA_REFRESH * 30
  845. case "季度":
  846. limitDay = utils.DATA_REFRESH * 90
  847. case "年度":
  848. limitDay = utils.DATA_REFRESH * 365
  849. default:
  850. limitDay = utils.DATA_REFRESH
  851. }
  852. startDate = sTime.AddDate(0, 0, -limitDay).Format(utils.FormatDate)
  853. }
  854. }
  855. }
  856. result, tmpErr := RefreshPredictEdbCalculateData(edbInfo.EdbInfoId, edbInfo.EdbCode, startDate)
  857. if tmpErr != nil {
  858. err = tmpErr
  859. fmt.Println(v, "RefreshPredictEdbCalculateData err", time.Now())
  860. errmsg = "RefreshPredictEdbCalculateData Err:" + tmpErr.Error()
  861. return
  862. }
  863. if result.Ret != 200 {
  864. fmt.Println(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  865. errmsg = fmt.Sprint(v, "RefreshPredictEdbCalculateData err;msg:", result.Msg, ";errMsg:", result.ErrMsg)
  866. err = fmt.Errorf("刷新失败")
  867. return
  868. }
  869. }
  870. return
  871. }
  872. // EdbInfoRefreshAllFromBaseV3
  873. //
  874. // @Description: 全部刷新指标(切换到edb_lib服务)
  875. // @author: Roc
  876. // @datetime2023-10-23 09:57:55
  877. // @param edbInfoIdList []int
  878. // @param refreshAll bool
  879. // @param isSync bool
  880. // @return err error
  881. // @return isAsync bool
  882. func EdbInfoRefreshAllFromBase(edbInfoIdList []int, refreshAll bool) (err error, errMsg string) {
  883. defer func() {
  884. if err != nil {
  885. fmt.Println("EdbInfoRefreshAllFromBaseV3 Err:" + err.Error() + ";errMsg:" + errMsg)
  886. go alarm_msg.SendAlarmMsg("EdbInfoRefreshAllFromBaseV3,Err"+err.Error()+";errMsg:"+errMsg, 3)
  887. }
  888. }()
  889. traceEdbInfoList, err := TraceEdbInfoByEdbInfoIdList(edbInfoIdList)
  890. if err != nil {
  891. return
  892. }
  893. // existEdbInfoIdMap 已经处理了的指标id map
  894. existEdbInfoIdMap := make(map[int]int)
  895. // 基础指标
  896. newBaseEdbInfoArr := make([]*data_manage.EdbInfo, 0)
  897. newBasePredictEdbInfoArr := make([]*data_manage.EdbInfo, 0)
  898. newBaseMap := make(map[int]*data_manage.EdbInfo)
  899. newPredictBaseMap := make(map[int]*data_manage.EdbInfo)
  900. // 计算指标
  901. newCalculateMap := make(map[int]*data_manage.EdbInfo)
  902. newPredictCalculateMap := make(map[int]*data_manage.EdbInfo)
  903. calculateArr := make([]int, 0)
  904. predictCalculateArr := make([]int, 0)
  905. // 获取关联指标
  906. for _, traceEdbInfo := range traceEdbInfoList {
  907. tmpBaseEdbInfoArr, tmpBasePredictEdbInfoArr, tmpCalculateMap, tmpPredictCalculateMap, _, _ := getRefreshEdbInfoListByTraceEdbInfo(traceEdbInfo, existEdbInfoIdMap)
  908. // 普通基础指标
  909. for _, edbInfo := range tmpBaseEdbInfoArr {
  910. if _, ok := newBaseMap[edbInfo.EdbInfoId]; !ok {
  911. newBaseMap[edbInfo.EdbInfoId] = edbInfo
  912. newBaseEdbInfoArr = append(newBaseEdbInfoArr, edbInfo)
  913. }
  914. }
  915. // 预测基础指标
  916. for _, edbInfo := range tmpBasePredictEdbInfoArr {
  917. if _, ok := newPredictBaseMap[edbInfo.EdbInfoId]; !ok {
  918. newPredictBaseMap[edbInfo.EdbInfoId] = edbInfo
  919. newBasePredictEdbInfoArr = append(newBasePredictEdbInfoArr, edbInfo)
  920. }
  921. }
  922. // 普通计算指标
  923. for _, edbInfo := range tmpCalculateMap {
  924. if _, ok := newCalculateMap[edbInfo.EdbInfoId]; !ok {
  925. newCalculateMap[edbInfo.EdbInfoId] = edbInfo
  926. calculateArr = append(calculateArr, edbInfo.EdbInfoId)
  927. }
  928. }
  929. // 预测计算指标
  930. for _, edbInfo := range tmpPredictCalculateMap {
  931. if _, ok := newPredictCalculateMap[edbInfo.EdbInfoId]; !ok {
  932. newPredictCalculateMap[edbInfo.EdbInfoId] = edbInfo
  933. predictCalculateArr = append(predictCalculateArr, edbInfo.EdbInfoId)
  934. }
  935. }
  936. }
  937. // 普通计算指标的id
  938. sort.Ints(calculateArr)
  939. // 预测计算指标的id
  940. sort.Ints(predictCalculateArr)
  941. // 需要刷新的指标数量
  942. //totalEdbInfo := len(newBaseEdbInfoArr) + len(calculateArr) + len(predictCalculateArr)
  943. //if totalEdbInfo <= 20{
  944. // err,errMsg = edbInfoRefreshAll(refreshAll, newBaseEdbInfoArr, newBasePredictEdbInfoArr, newCalculateMap, newPredictCalculateMap, calculateArr, predictCalculateArr)
  945. //} else {
  946. // go edbInfoRefreshAll(refreshAll, newBaseEdbInfoArr, newBasePredictEdbInfoArr, newCalculateMap, newPredictCalculateMap, calculateArr, predictCalculateArr)
  947. //}
  948. err, errMsg = edbInfoRefreshAll(refreshAll, newBaseEdbInfoArr, newBasePredictEdbInfoArr, newCalculateMap, newPredictCalculateMap, calculateArr, predictCalculateArr)
  949. return
  950. }
  951. // getRefreshEdbInfoListByTraceEdbInfo
  952. //
  953. // @Description: 根据溯源获取关联指标(去掉重复指标id)
  954. // @author: Roc
  955. // @datetime2023-10-23 11:22:50
  956. // @param traceEdbInfo data_manage.TraceEdbInfoResp
  957. // @param existEdbInfoIdMap map[int]int
  958. // @return newBaseEdbInfoArr []*data_manage.EdbInfo
  959. // @return newBasePredictEdbInfoArr []*data_manage.EdbInfo
  960. // @return newCalculateMap map[int]*data_manage.EdbInfo
  961. // @return newPredictCalculateMap map[int]*data_manage.EdbInfo
  962. // @return calculateArr []int
  963. // @return predictCalculateArr []int
  964. func getRefreshEdbInfoListByTraceEdbInfo(traceEdbInfo data_manage.TraceEdbInfoResp, existEdbInfoIdMap map[int]int) (newBaseEdbInfoArr, newBasePredictEdbInfoArr []*data_manage.EdbInfo, newCalculateMap, newPredictCalculateMap map[int]*data_manage.EdbInfo, calculateArr, predictCalculateArr []int) {
  965. newBaseEdbInfoArr = make([]*data_manage.EdbInfo, 0)
  966. newBasePredictEdbInfoArr = make([]*data_manage.EdbInfo, 0)
  967. newCalculateMap = make(map[int]*data_manage.EdbInfo)
  968. newPredictCalculateMap = make(map[int]*data_manage.EdbInfo)
  969. calculateArr = make([]int, 0)
  970. predictCalculateArr = make([]int, 0)
  971. _, ok := existEdbInfoIdMap[traceEdbInfo.EdbInfoId]
  972. if ok {
  973. return
  974. }
  975. existEdbInfoIdMap[traceEdbInfo.EdbInfoId] = traceEdbInfo.EdbInfoId
  976. switch traceEdbInfo.EdbInfoType {
  977. //0-普通指标; 1-预测指标
  978. case 0: // 0-普通指标
  979. if traceEdbInfo.EdbType == 1 { //1-基础指标
  980. newBaseEdbInfoArr = append(newBaseEdbInfoArr, traceEdbInfo.EdbInfo)
  981. } else if traceEdbInfo.EdbType == 2 { //2-计算指标
  982. newCalculateMap[traceEdbInfo.EdbInfoId] = traceEdbInfo.EdbInfo
  983. calculateArr = append(calculateArr, traceEdbInfo.EdbInfoId)
  984. }
  985. case 1: // 1-预测指标
  986. if traceEdbInfo.EdbType == 1 { //1-基础指标
  987. newBasePredictEdbInfoArr = append(newBasePredictEdbInfoArr, traceEdbInfo.EdbInfo)
  988. } else if traceEdbInfo.EdbType == 2 { //2-计算指标
  989. newPredictCalculateMap[traceEdbInfo.EdbInfoId] = traceEdbInfo.EdbInfo
  990. predictCalculateArr = append(predictCalculateArr, traceEdbInfo.EdbInfoId)
  991. }
  992. }
  993. if traceEdbInfo.Child != nil && len(traceEdbInfo.Child) > 0 {
  994. for _, v := range traceEdbInfo.Child {
  995. tmpBaseEdbInfoArr, tmpPredictEdbInfoArr, tmpCalculateMap, tmpPredictCalculateMap, tmpCalculateArr, tmpPredictCalculateArr := getRefreshEdbInfoListByTraceEdbInfo(v, existEdbInfoIdMap)
  996. newBaseEdbInfoArr = append(newBaseEdbInfoArr, tmpBaseEdbInfoArr...)
  997. newBasePredictEdbInfoArr = append(newBasePredictEdbInfoArr, tmpPredictEdbInfoArr...)
  998. for k, tmpEdbInfo := range tmpCalculateMap {
  999. newCalculateMap[k] = tmpEdbInfo
  1000. }
  1001. for k, tmpEdbInfo := range tmpPredictCalculateMap {
  1002. newPredictCalculateMap[k] = tmpEdbInfo
  1003. }
  1004. calculateArr = append(calculateArr, tmpCalculateArr...)
  1005. predictCalculateArr = append(predictCalculateArr, tmpPredictCalculateArr...)
  1006. }
  1007. }
  1008. return
  1009. }
  1010. // TraceEdbInfoByEdbInfoIdList
  1011. //
  1012. // @Description: 根据指标id列表返回指标追溯的列表
  1013. // @author: Roc
  1014. // @datetime2023-10-23 11:22:21
  1015. // @param edbInfoIdList []int
  1016. // @return traceEdbInfoList []data_manage.TraceEdbInfoResp
  1017. // @return err error
  1018. func TraceEdbInfoByEdbInfoIdList(edbInfoIdList []int) (traceEdbInfoList []data_manage.TraceEdbInfoResp, err error) {
  1019. traceEdbInfoList = make([]data_manage.TraceEdbInfoResp, 0)
  1020. edbInfoList, err := data_manage.GetEdbInfoByIdList(edbInfoIdList)
  1021. if err != nil {
  1022. return
  1023. }
  1024. edbInfoRuleMap := make(map[int]string, 0)
  1025. edbMappingMap := make(map[int][]*data_manage.EdbInfoCalculateMappingInfo)
  1026. findIdMap := make(map[int]int)
  1027. existMap := make(map[int]data_manage.TraceEdbInfoResp)
  1028. for _, edbInfo := range edbInfoList {
  1029. findIdMap[edbInfo.EdbInfoId] = edbInfo.EdbInfoId
  1030. //edbInfoRuleMap[edbInfoId] = getEdbRuleTitle(edbInfo)
  1031. traceEdbInfo := data_manage.TraceEdbInfoResp{
  1032. //EdbInfoId: edbInfo.EdbInfoId,
  1033. EdbInfoId: edbInfo.EdbInfoId,
  1034. EdbInfoType: edbInfo.EdbInfoType,
  1035. EdbName: edbInfo.EdbName,
  1036. EdbType: edbInfo.EdbType,
  1037. //Source: edbInfo.Source,
  1038. UniqueCode: edbInfo.UniqueCode,
  1039. ClassifyId: edbInfo.ClassifyId,
  1040. EdbInfo: edbInfo,
  1041. }
  1042. traceEdbInfo.Child, err = traceEdbInfoByEdbInfoId(edbInfo.EdbInfoId, traceEdbInfo, edbInfoRuleMap, findIdMap, existMap, edbMappingMap)
  1043. traceEdbInfoList = append(traceEdbInfoList, traceEdbInfo)
  1044. }
  1045. //findEdbInfoIdList := make([]int, 0)
  1046. //for _, v := range findIdMap {
  1047. // findEdbInfoIdList = append(findEdbInfoIdList, v)
  1048. //}
  1049. //findEdbInfoList, err := data_manage.GetEdbInfoByIdList(findEdbInfoIdList)
  1050. //if err != nil {
  1051. // return
  1052. //}
  1053. //edbInfoMap := make(map[int]*data_manage.EdbInfo)
  1054. //for _, tmpEdbInfo := range findEdbInfoList {
  1055. // edbInfoMap[tmpEdbInfo.EdbInfoId] = tmpEdbInfo
  1056. //}
  1057. //for k, traceEdbInfo := range traceEdbInfoList {
  1058. // traceEdbInfoList[k], err = handleTraceEdbInfo(traceEdbInfo, 0, edbInfoMap, edbMappingMap)
  1059. //}
  1060. return
  1061. }
  1062. // traceEdbInfoByEdbInfoId
  1063. //
  1064. // @Description: 指标追溯
  1065. // @author: Roc
  1066. // @datetime2023-10-23 11:21:54
  1067. // @param edbInfoId int
  1068. // @param traceEdbInfo data_manage.TraceEdbInfoResp
  1069. // @param edbInfoRuleMap map[int]string
  1070. // @param findIdMap map[int]int
  1071. // @param existMap map[int]data_manage.TraceEdbInfoResp
  1072. // @param edbMappingMap map[int][]*data_manage.EdbInfoCalculateMappingInfo
  1073. // @return child []data_manage.TraceEdbInfoResp
  1074. // @return err error
  1075. func traceEdbInfoByEdbInfoId(edbInfoId int, traceEdbInfo data_manage.TraceEdbInfoResp, edbInfoRuleMap map[int]string, findIdMap map[int]int, existMap map[int]data_manage.TraceEdbInfoResp, edbMappingMap map[int][]*data_manage.EdbInfoCalculateMappingInfo) (child []data_manage.TraceEdbInfoResp, err error) {
  1076. traceEdbInfo, ok := existMap[edbInfoId]
  1077. if ok {
  1078. return
  1079. }
  1080. child = make([]data_manage.TraceEdbInfoResp, 0)
  1081. edbInfoMappingList, e := data_manage.GetEdbInfoCalculateMappingListByEdbInfoId(edbInfoId)
  1082. if e != nil {
  1083. err = fmt.Errorf("GetEdbInfoCalculateMappingListByEdbInfoId err: %s", e.Error())
  1084. return
  1085. }
  1086. // 指标信息map
  1087. edbInfoMap := make(map[int]*data_manage.EdbInfo)
  1088. if len(edbInfoMappingList) > 0 {
  1089. fromEdbInfoIdList := make([]int, 0)
  1090. for _, v := range edbInfoMappingList {
  1091. fromEdbInfoIdList = append(fromEdbInfoIdList, v.FromEdbInfoId)
  1092. }
  1093. edbInfoList, tmpErr := data_manage.GetEdbInfoByIdList(fromEdbInfoIdList)
  1094. if tmpErr != nil {
  1095. err = fmt.Errorf("traceEdbInfoByEdbInfoId GetEdbInfoByIdList err: %s", tmpErr.Error())
  1096. return
  1097. }
  1098. for _, v := range edbInfoList {
  1099. edbInfoMap[v.EdbInfoId] = v
  1100. }
  1101. }
  1102. edbMappingMap[edbInfoId] = edbInfoMappingList
  1103. for _, v := range edbInfoMappingList {
  1104. tmpEdbInfoId := v.FromEdbInfoId
  1105. tmpTraceEdbInfo := data_manage.TraceEdbInfoResp{
  1106. EdbInfoId: tmpEdbInfoId,
  1107. EdbInfoType: v.FromEdbInfoType,
  1108. EdbType: v.FromEdbType,
  1109. UniqueCode: v.FromUniqueCode,
  1110. ClassifyId: v.FromClassifyId,
  1111. EdbInfo: edbInfoMap[v.FromEdbInfoId],
  1112. }
  1113. // 计算指标/预测指标继续溯源
  1114. if v.FromEdbType == 2 || v.FromEdbInfoType == 1 {
  1115. // 查过了就不查了
  1116. if _, ok2 := findIdMap[tmpEdbInfoId]; !ok2 {
  1117. tmpTraceEdbInfo.Child, e = traceEdbInfoByEdbInfoId(tmpEdbInfoId, tmpTraceEdbInfo, edbInfoRuleMap, findIdMap, existMap, edbMappingMap)
  1118. if e != nil {
  1119. err = fmt.Errorf("traceEdbInfoByEdbInfoId err: %s", e.Error())
  1120. return
  1121. }
  1122. }
  1123. }
  1124. child = append(child, tmpTraceEdbInfo)
  1125. findIdMap[tmpEdbInfoId] = tmpEdbInfoId
  1126. }
  1127. existMap[edbInfoId] = traceEdbInfo
  1128. return
  1129. }