edb_data_calculate_bp.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702
  1. package models
  2. import (
  3. "errors"
  4. "eta/eta_index_lib/utils"
  5. "fmt"
  6. "github.com/beego/beego/v2/client/orm"
  7. "github.com/shopspring/decimal"
  8. "strconv"
  9. "strings"
  10. "time"
  11. )
  12. // AddCalculateBp 升频
  13. func AddCalculateBp(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbInfo, edbCode, uniqueCode string, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, err error) {
  14. o := orm.NewOrm()
  15. to, err := o.Begin()
  16. if err != nil {
  17. return
  18. }
  19. defer func() {
  20. if err != nil {
  21. fmt.Println("AddCalculateBp,Err:" + err.Error())
  22. _ = to.Rollback()
  23. } else {
  24. _ = to.Commit()
  25. }
  26. }()
  27. if req.EdbInfoId <= 0 {
  28. edbInfo = new(EdbInfo)
  29. edbInfo.Source = utils.DATA_SOURCE_CALCULATE_BP
  30. edbInfo.SourceName = utils.DATA_SOURCE_NAME_CALCULATE_BP
  31. edbInfo.EdbCode = edbCode
  32. edbInfo.EdbName = req.EdbName
  33. edbInfo.EdbNameSource = req.EdbName
  34. edbInfo.Frequency = req.Frequency
  35. edbInfo.Unit = req.Unit
  36. edbInfo.ClassifyId = req.ClassifyId
  37. edbInfo.SysUserId = sysUserId
  38. edbInfo.SysUserRealName = sysUserRealName
  39. edbInfo.CreateTime = time.Now()
  40. edbInfo.ModifyTime = time.Now()
  41. edbInfo.UniqueCode = uniqueCode
  42. edbInfo.CalculateFormula = req.Formula
  43. edbInfo.EdbType = 2
  44. newEdbInfoId, tmpErr := to.Insert(edbInfo)
  45. if tmpErr != nil {
  46. err = tmpErr
  47. return
  48. }
  49. edbInfo.EdbInfoId = int(newEdbInfoId)
  50. //关联关系
  51. {
  52. calculateMappingItem := new(EdbInfoCalculateMapping)
  53. calculateMappingItem.CreateTime = time.Now()
  54. calculateMappingItem.ModifyTime = time.Now()
  55. calculateMappingItem.Sort = 1
  56. calculateMappingItem.EdbCode = edbCode
  57. calculateMappingItem.EdbInfoId = edbInfo.EdbInfoId
  58. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  59. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  60. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  61. calculateMappingItem.FromSource = fromEdbInfo.Source
  62. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  63. calculateMappingItem.FromTag = ""
  64. calculateMappingItem.Source = edbInfo.Source
  65. calculateMappingItem.SourceName = edbInfo.SourceName
  66. calculateMappingItem.FromSubSource = edbInfo.SubSource
  67. _, err = to.Insert(calculateMappingItem)
  68. if err != nil {
  69. return
  70. }
  71. }
  72. } else {
  73. edbInfo, err = GetEdbInfoById(req.EdbInfoId)
  74. if err != nil {
  75. return
  76. }
  77. dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_CALCULATE_BP, utils.DATA_SUB_SOURCE_EDB)
  78. fmt.Println("dataTableName:" + dataTableName)
  79. deleteSql := ` DELETE FROM %s WHERE edb_info_id=? `
  80. deleteSql = fmt.Sprintf(deleteSql, dataTableName)
  81. _, err = to.Raw(deleteSql, req.EdbInfoId).Exec()
  82. if err != nil {
  83. return
  84. }
  85. }
  86. //计算数据
  87. err = refreshAllCalculateBp(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, fromEdbInfo, edbInfo.EdbCode, "", "", 0)
  88. return
  89. }
  90. // EditCalculateBp 修改升频数据
  91. func EditCalculateBp(edbInfo *EdbInfo, req *EdbInfoCalculateBatchEditReq, fromEdbInfo *EdbInfo) (err error) {
  92. o := orm.NewOrm()
  93. to, err := o.Begin()
  94. if err != nil {
  95. return
  96. }
  97. defer func() {
  98. if err != nil {
  99. fmt.Println("EditCalculateBp,Err:" + err.Error())
  100. _ = to.Rollback()
  101. } else {
  102. _ = to.Commit()
  103. }
  104. }()
  105. //修改指标信息
  106. edbInfo.EdbName = req.EdbName
  107. edbInfo.EdbNameSource = req.EdbName
  108. edbInfo.Frequency = req.Frequency
  109. edbInfo.Unit = req.Unit
  110. edbInfo.ClassifyId = req.ClassifyId
  111. edbInfo.ModifyTime = time.Now()
  112. _, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime")
  113. if err != nil {
  114. return
  115. }
  116. //判断计算指标是否被更换
  117. var existCondition string
  118. var existPars []interface{}
  119. existCondition += " AND edb_info_id=? AND from_edb_info_id=? "
  120. existPars = append(existPars, edbInfo.EdbInfoId, req.FromEdbInfoId)
  121. count, err := GetEdbInfoCalculateCountByCondition(existCondition, existPars)
  122. if err != nil {
  123. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  124. return
  125. }
  126. if count > 0 { // 指标未被替换,无需重新计算
  127. return
  128. }
  129. //删除,计算指标关联的,基础指标的关联关系
  130. sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
  131. _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
  132. if err != nil {
  133. err = errors.New("删除计算指标关联关系失败,Err:" + err.Error())
  134. return
  135. }
  136. //清空原有数据
  137. sql = ` DELETE FROM edb_data_calculate_bp WHERE edb_info_id = ? `
  138. _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
  139. if err != nil {
  140. return
  141. }
  142. //关联关系
  143. {
  144. calculateMappingItem := &EdbInfoCalculateMapping{
  145. EdbInfoCalculateMappingId: 0,
  146. EdbInfoId: edbInfo.EdbInfoId,
  147. Source: utils.DATA_SOURCE_CALCULATE_BP,
  148. SourceName: utils.DATA_SOURCE_NAME_CALCULATE_BP,
  149. EdbCode: edbInfo.EdbCode,
  150. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  151. FromEdbCode: fromEdbInfo.EdbCode,
  152. FromEdbName: fromEdbInfo.EdbName,
  153. FromSource: fromEdbInfo.Source,
  154. FromSourceName: fromEdbInfo.SourceName,
  155. FromTag: "",
  156. Sort: 1,
  157. CreateTime: time.Now(),
  158. ModifyTime: time.Now(),
  159. FromSubSource: fromEdbInfo.SubSource,
  160. }
  161. _, err = to.Insert(calculateMappingItem)
  162. if err != nil {
  163. return
  164. }
  165. }
  166. //计算数据
  167. err = refreshAllCalculateBp(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, fromEdbInfo, edbInfo.EdbCode, "", "", 0)
  168. return
  169. }
  170. // RefreshAllCalculateBpBak 刷新所有升频数据
  171. func RefreshAllCalculateBpBak(edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (err error) {
  172. o := orm.NewOrm()
  173. to, err := o.Begin()
  174. if err != nil {
  175. return
  176. }
  177. defer func() {
  178. if err != nil {
  179. fmt.Println("RefreshAllCalculateBp,Err:" + err.Error())
  180. _ = to.Rollback()
  181. } else {
  182. _ = to.Commit()
  183. }
  184. }()
  185. if err != nil {
  186. return
  187. }
  188. edbInfoIdStr := strconv.Itoa(edbInfoId)
  189. //计算数据
  190. //计算数据
  191. var condition string
  192. var pars []interface{}
  193. condition += " AND edb_info_id=? "
  194. pars = append(pars, fromEdbInfo.EdbInfoId)
  195. if startDate != "" {
  196. condition += " AND data_time>=? "
  197. pars = append(pars, startDate)
  198. }
  199. if endDate != "" {
  200. condition += " AND data_time<=? "
  201. pars = append(pars, endDate)
  202. }
  203. dataList, err := GetEdbDataListAllByTo(to, condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource, 0)
  204. if err != nil {
  205. return err
  206. }
  207. var dateArr []string
  208. dataMap := make(map[string]*EdbInfoSearchData)
  209. for _, v := range dataList {
  210. dateArr = append(dateArr, v.DataTime)
  211. dataMap[v.DataTime] = v
  212. }
  213. fmt.Println("source:", source)
  214. //获取指标所有数据
  215. existDataList := make([]*EdbData, 0)
  216. dataTableName := GetEdbDataTableName(source, subSource)
  217. fmt.Println("dataTableName:", dataTableName)
  218. sql := `SELECT * FROM %s WHERE edb_info_id=? `
  219. sql = fmt.Sprintf(sql, dataTableName)
  220. _, err = to.Raw(sql, edbInfoId).QueryRows(&existDataList)
  221. if err != nil {
  222. return err
  223. }
  224. existDataMap := make(map[string]string)
  225. for _, v := range existDataList {
  226. existDataMap[v.DataTime] = v.Value
  227. }
  228. fmt.Println("existDataMap:", existDataMap)
  229. addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  230. var isAdd bool
  231. existMap := make(map[string]string)
  232. dataLen := len(dataList)
  233. for i := 0; i < dataLen; i++ {
  234. //当期
  235. currentItem := dataList[i]
  236. currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
  237. var day int
  238. var preItem *EdbInfoSearchData
  239. var preDate time.Time
  240. if i == 0 {
  241. day = int(time.Now().Sub(currentDate).Hours() / float64(24))
  242. preDate = time.Now()
  243. } else {
  244. j := i + 1
  245. if j < dataLen {
  246. preItem = dataList[j]
  247. preDate, _ = time.ParseInLocation(utils.FormatDate, preItem.DataTime, time.Local)
  248. day = int(currentDate.Sub(preDate).Hours() / float64(24))
  249. }
  250. }
  251. for k := 0; k <= day; k++ {
  252. needDay := preDate.AddDate(0, 0, -k)
  253. needDayStr := needDay.Format(utils.FormatDate)
  254. existKey := edbCode + needDayStr
  255. if _, ok := existMap[existKey]; !ok {
  256. timestamp := needDay.UnixNano() / 1e6
  257. timestampStr := fmt.Sprintf("%d", timestamp)
  258. valStr := decimal.NewFromFloat(currentItem.Value).String()
  259. if existVal, ok := existDataMap[needDayStr]; !ok {
  260. addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, valStr)
  261. } else {
  262. if existVal != valStr {
  263. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  264. sql = fmt.Sprintf(sql, dataTableName)
  265. _, err = to.Raw(sql, valStr, edbInfoId, needDay).Exec()
  266. if err != nil {
  267. return err
  268. }
  269. }
  270. }
  271. }
  272. existMap[existKey] = needDayStr
  273. }
  274. existKey := edbCode + currentItem.DataTime
  275. if _, ok := existMap[existKey]; !ok {
  276. currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
  277. timestamp := currentDate.UnixNano() / 1e6
  278. timestampStr := fmt.Sprintf("%d", timestamp)
  279. valStr := decimal.NewFromFloat(currentItem.Value).String()
  280. if existVal, ok := existDataMap[currentItem.DataTime]; !ok {
  281. addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr)
  282. } else {
  283. if existVal != valStr {
  284. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  285. sql = fmt.Sprintf(sql, dataTableName)
  286. _, err = to.Raw(sql, valStr, edbInfoId, currentItem.DataTime).Exec()
  287. if err != nil {
  288. return err
  289. }
  290. }
  291. }
  292. }
  293. existMap[existKey] = currentItem.DataTime
  294. }
  295. if isAdd {
  296. addSql = strings.TrimRight(addSql, ",")
  297. _, err = to.Raw(addSql).Exec()
  298. if err != nil {
  299. return err
  300. }
  301. }
  302. return
  303. }
  304. func RefreshAllCalculateBp(edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (err error) {
  305. o := orm.NewOrm()
  306. to, err := o.Begin()
  307. if err != nil {
  308. return
  309. }
  310. defer func() {
  311. if err != nil {
  312. fmt.Println("RefreshAllCalculateBp,Err:" + err.Error())
  313. _ = to.Rollback()
  314. } else {
  315. _ = to.Commit()
  316. }
  317. }()
  318. // 计算数据
  319. err = refreshAllCalculateBp(to, edbInfoId, source, subSource, fromEdbInfo, edbCode, startDate, endDate, 1)
  320. return
  321. }
  322. // refreshAllCalculateBp 刷新升频数据
  323. //func refreshAllCalculateBp(to orm.TxOrmer, edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string, order int) (err error) {
  324. // edbInfoIdStr := strconv.Itoa(edbInfoId)
  325. //
  326. // //计算数据
  327. // var condition string
  328. // var pars []interface{}
  329. // condition += " AND edb_info_id=? "
  330. // pars = append(pars, fromEdbInfo.EdbInfoId)
  331. //
  332. // //if startDate != "" {
  333. // // condition += " AND data_time>=? "
  334. // // pars = append(pars, startDate)
  335. // //}
  336. // //if endDate != "" {
  337. // // condition += " AND data_time<=? "
  338. // // pars = append(pars, endDate)
  339. // //}
  340. // //获取来源指标的数据
  341. // dataList, err := GetEdbDataListAllByTo(to, condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource, order)
  342. // if err != nil {
  343. // return err
  344. // }
  345. // var dateArr []string
  346. // dataMap := make(map[string]*EdbInfoSearchData)
  347. // fromDataMap := make(map[string]float64)
  348. // //来源指指标数据
  349. // for _, v := range dataList {
  350. // dateArr = append(dateArr, v.DataTime)
  351. // dataMap[v.DataTime] = v
  352. // fromDataMap[v.DataTime] = v.Value
  353. // }
  354. // fmt.Println("source:", source)
  355. //
  356. // //获取升频指标所有数据
  357. // existDataList, err := GetAllEdbDataListByTo(to, edbInfoId, source, subSource)
  358. // if err != nil {
  359. // return
  360. // }
  361. // //计算指标的map
  362. // existDataMap := make(map[string]*EdbData, 0)
  363. //
  364. // addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  365. // var isAdd bool
  366. //
  367. // var lastValue float64 //最后数据的值(float64)
  368. // var lastValueStr string //最后数据的值(string)
  369. // //待删除的日期
  370. // removeDateList := make([]string, 0)
  371. // if len(existDataList) > 0 {
  372. // //第一个已经入库的日期
  373. // firstExistDataTimeStr := existDataList[0].DataTime //计算指标数据第一条的日期字符串
  374. // if len(dateArr) > 0 {
  375. // firstFromDataTimeStr := dateArr[0] //来源数据第一条的日期字符串
  376. // firstExistDataTime, _ := time.ParseInLocation(utils.FormatDate, firstExistDataTimeStr, time.Local) //计算指标数据第一条的日期(time类型)
  377. // firstFromDataTime, _ := time.ParseInLocation(utils.FormatDate, firstFromDataTimeStr, time.Local) //来源数据第一条的日期(time类型)
  378. // nowDateStr := time.Now().Format(utils.FormatDate) //当天日期字符串
  379. // nowDate, _ := time.ParseInLocation(utils.FormatDate, nowDateStr, firstFromDataTime.Location()) //当天日期(time类型)
  380. //
  381. // lastValue = fromDataMap[firstFromDataTimeStr]
  382. // lastValueStr = decimal.NewFromFloat(lastValue).String()
  383. // //第一步: 判断来源指标的开始时间与计算指标的开始时间是否相等,相等的话,那么就不需要对两个时间之间的数据做处理
  384. // if firstExistDataTimeStr != firstFromDataTimeStr {
  385. // if firstExistDataTime.Before(firstFromDataTime) { //如果计算指标第一条数据的开始时间 早于 来源指标的第一条开始时间,那么需要对两个时间之间的计算指标数据做 删除处理
  386. // for _, v := range existDataList {
  387. // if v.DataTime == firstFromDataTimeStr {
  388. // if tmpLastValue, ok := fromDataMap[firstFromDataTimeStr]; ok { //来源指标当天的数据
  389. // lastValue = tmpLastValue
  390. // lastValueStr = decimal.NewFromFloat(lastValue).String()
  391. // }
  392. // break
  393. // }
  394. // removeDateList = append(removeDateList, v.DataTime)
  395. // }
  396. // } else {
  397. // for _, v := range dateArr { //如果计算指标第一条数据的开始时间 晚于 来源指标的第一条开始时间,那么需要对两个时间之间的计算指标数据做 新增处理
  398. // vDataTime, _ := time.ParseInLocation(utils.FormatDate, v, time.Local) //当前日期(time类型)
  399. // if firstExistDataTime.Equal(vDataTime) || firstExistDataTime.Before(vDataTime) {
  400. // if tmpLastValue, ok := fromDataMap[v]; ok { //来源指标当天的数据
  401. // lastValue = tmpLastValue
  402. // lastValueStr = decimal.NewFromFloat(lastValue).String()
  403. // }
  404. // break
  405. // }
  406. //
  407. // currentDate, _ := time.ParseInLocation(utils.FormatDate, v, time.Local)
  408. // timestamp := currentDate.UnixNano() / 1e6
  409. // timestampStr := fmt.Sprintf("%d", timestamp)
  410. // addSql += GetAddSql(edbInfoIdStr, edbCode, v, timestampStr, lastValueStr)
  411. //
  412. // isAdd = true
  413. // }
  414. // }
  415. // }
  416. //
  417. // //第二步 剩余数据每天修改
  418. //
  419. // day := int(nowDate.Sub(firstExistDataTime).Hours() / float64(24))
  420. //
  421. // //第三步: 已经入库的数据处理
  422. // for _, v := range existDataList {
  423. // existDataMap[v.DataTime] = v
  424. // }
  425. //
  426. // for k := day; k >= 0; k-- {
  427. // needDay := nowDate.AddDate(0, 0, -k)
  428. // needDayStr := needDay.Format(utils.FormatDate)
  429. // tmpExistData, ok := existDataMap[needDayStr]
  430. // if ok {
  431. // if tmpLastValue, ok := fromDataMap[tmpExistData.DataTime]; ok { //来源指标当天的数据
  432. // lastValue = tmpLastValue
  433. // //lastValueStr = decimal.NewFromFloat(lastValue).String()
  434. // lastValueStr = fmt.Sprintf("%.4f", lastValue)
  435. // }
  436. // //如果对应的值不匹配
  437. // if tmpExistData.Value != lastValueStr {
  438. // err = ModifyEdbDataById(source, subSource, tmpExistData.EdbDataId, lastValueStr)
  439. // if err != nil {
  440. // return err
  441. // }
  442. // }
  443. // } else {
  444. // timestamp := needDay.UnixNano() / 1e6
  445. // timestampStr := fmt.Sprintf("%d", timestamp)
  446. // addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, lastValueStr)
  447. //
  448. // isAdd = true
  449. // }
  450. // }
  451. // } else {
  452. // //如果没有来源指标数据,那么已经入库的计算指标数据需要全部删除
  453. // tableName := GetEdbDataTableName(source, subSource)
  454. // sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ?`, tableName)
  455. // _, err = to.Raw(sql, edbInfoId).Exec()
  456. // if err != nil {
  457. // err = fmt.Errorf("删除所有的升频指标数据失败,Err:" + err.Error())
  458. // return
  459. // }
  460. //
  461. // //for _, v := range existDataList {
  462. // // removeDateList = append(removeDateList, v.DataTime)
  463. // //}
  464. // }
  465. // } else {
  466. // existMap := make(map[string]string)
  467. // dataLen := len(dataList)
  468. //
  469. // for i := 0; i < dataLen; i++ {
  470. // //当期
  471. // currentItem := dataList[i]
  472. // currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
  473. // var day int
  474. // var preItem *EdbInfoSearchData
  475. // var preDate time.Time
  476. // if i == 0 {
  477. // day = int(time.Now().Sub(currentDate).Hours() / float64(24))
  478. // preDate = time.Now()
  479. // } else {
  480. // j := i - 1
  481. // if j < dataLen {
  482. // preItem = dataList[j]
  483. // preDate, _ = time.ParseInLocation(utils.FormatDate, preItem.DataTime, time.Local)
  484. // day = int(preDate.Sub(currentDate).Hours() / float64(24))
  485. // utils.FileLog.Info("preItem.DataTime:" + preItem.DataTime + ";currentItem.DataTime" + currentItem.DataTime)
  486. // }
  487. // }
  488. // for k := 0; k <= day; k++ {
  489. // needDay := preDate.AddDate(0, 0, -k)
  490. // needDayStr := needDay.Format(utils.FormatDate)
  491. // existKey := edbCode + needDayStr
  492. // if _, ok := existMap[existKey]; !ok {
  493. // timestamp := needDay.UnixNano() / 1e6
  494. // timestampStr := fmt.Sprintf("%d", timestamp)
  495. // valStr := decimal.NewFromFloat(currentItem.Value).String()
  496. // addSql += GetAddSql(edbInfoIdStr, edbCode, needDayStr, timestampStr, valStr)
  497. // isAdd = true
  498. // }
  499. // existMap[existKey] = needDayStr
  500. // }
  501. // existKey := edbCode + currentItem.DataTime
  502. // if _, ok := existMap[existKey]; !ok {
  503. // currentDate, _ := time.ParseInLocation(utils.FormatDate, currentItem.DataTime, time.Local)
  504. // timestamp := currentDate.UnixNano() / 1e6
  505. // timestampStr := fmt.Sprintf("%d", timestamp)
  506. // valStr := decimal.NewFromFloat(currentItem.Value).String()
  507. // addSql += GetAddSql(edbInfoIdStr, edbCode, currentItem.DataTime, timestampStr, valStr)
  508. // isAdd = true
  509. // }
  510. // existMap[existKey] = currentItem.DataTime
  511. // }
  512. // }
  513. //
  514. // // 删除不需要的指标数据
  515. // if len(removeDateList) > 0 {
  516. // removeDateStr := strings.Join(removeDateList, `","`)
  517. // removeDateStr = `"` + removeDateStr + `"`
  518. // //如果拼接指标变更了,那么需要删除所有的指标数据
  519. // tableName := GetEdbDataTableName(source, subSource)
  520. // sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr)
  521. //
  522. // _, err = to.Raw(sql, edbInfoId).Exec()
  523. // if err != nil {
  524. // err = fmt.Errorf("删除不存在的升频指标数据失败,Err:" + err.Error())
  525. // return
  526. // }
  527. // }
  528. //
  529. // if isAdd {
  530. // addSql = strings.TrimRight(addSql, ",")
  531. // _, err = to.Raw(addSql).Exec()
  532. // }
  533. // return
  534. //}
  535. // refreshAllCalculateBp 刷新升频数据
  536. func refreshAllCalculateBp(to orm.TxOrmer, edbInfoId, source, subSource int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string, order int) (err error) {
  537. edbInfoIdStr := strconv.Itoa(edbInfoId)
  538. //计算数据
  539. var condition string
  540. var pars []interface{}
  541. condition += " AND edb_info_id=? "
  542. pars = append(pars, fromEdbInfo.EdbInfoId)
  543. //if startDate != "" {
  544. // condition += " AND data_time>=? "
  545. // pars = append(pars, startDate)
  546. //}
  547. //if endDate != "" {
  548. // condition += " AND data_time<=? "
  549. // pars = append(pars, endDate)
  550. //}
  551. //获取来源指标的数据
  552. dataList, err := GetEdbDataListAllByTo(to, condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource, order)
  553. if err != nil {
  554. return err
  555. }
  556. // 来源指标没有数据,那么需要删除所有的计算指标数据
  557. if len(dataList) <= 0 {
  558. // todo 删除所有的计算指标数据
  559. return
  560. }
  561. // 来源指标的第一个日期
  562. fromFirstDate, err := time.ParseInLocation(utils.FormatDate, dataList[0].DataTime, time.Local)
  563. if err != nil {
  564. return
  565. }
  566. fromFirstDate = time.Date(fromFirstDate.Year(), fromFirstDate.Month(), fromFirstDate.Day(), 0, 0, 0, 0, time.Local)
  567. // 变频计算
  568. newDataList, err := EdbInfoSearchDataToData(dataList)
  569. if err != nil {
  570. return
  571. }
  572. baseCalculate := BaseCalculate{
  573. DataList: newDataList,
  574. Frequency: "",
  575. Formula: nil,
  576. Calendar: "",
  577. MoveType: 0,
  578. MoveFrequency: "",
  579. FromFrequency: "",
  580. Source: source,
  581. }
  582. dateDataMap, err, _ := baseCalculate.UpFrequency()
  583. if err != nil {
  584. return
  585. }
  586. // 获取升频指所有已经存在的计算指标数据
  587. existDataList, err := GetAllEdbDataListByTo(to, edbInfoId, source, subSource)
  588. if err != nil {
  589. return
  590. }
  591. //计算指标的map
  592. existDataMap := make(map[string]*EdbData, 0)
  593. for _, v := range existDataList {
  594. existDataMap[v.DataTime] = v
  595. }
  596. addSql := ` INSERT INTO edb_data_calculate_bp(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  597. var isAdd bool
  598. now := time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day(), 0, 0, 0, 0, time.Local)
  599. for currDate := fromFirstDate; !currDate.After(now); currDate = currDate.AddDate(0, 0, 1) {
  600. currDateStr := currDate.Format(utils.FormatDate)
  601. timestamp := currDate.UnixNano() / 1e6
  602. timestampStr := fmt.Sprintf("%d", timestamp)
  603. // 当前计算的值
  604. currValue, ok := dateDataMap[currDate]
  605. if !ok {
  606. // 没有计算成功就过滤
  607. continue
  608. }
  609. lastValueStr := decimal.NewFromFloat(currValue).Round(4).String()
  610. // 已经入库的值
  611. existData, ok := existDataMap[currDateStr]
  612. if !ok {
  613. // 没有入库那么就插入添加
  614. isAdd = true
  615. addSql += GetAddSql(edbInfoIdStr, edbCode, currDateStr, timestampStr, lastValueStr)
  616. continue
  617. }
  618. // 将已经入库的值转换为decimal类型,然后再保留4位小数,目的是为了做匹配,要不然取出来的数据与计算的数据不一致
  619. existDataValueDec, tmpErr := decimal.NewFromString(existData.Value)
  620. if tmpErr != nil {
  621. err = tmpErr
  622. return
  623. }
  624. existDataValueStr := existDataValueDec.Round(4).String()
  625. // 如果该日期已经入库了,且两个值不匹配,那么就更新
  626. if lastValueStr != existDataValueStr {
  627. err = ModifyEdbDataById(source, subSource, existData.EdbDataId, lastValueStr)
  628. if err != nil {
  629. return err
  630. }
  631. }
  632. // 该日期已经处理过了,所以需要移除,如果后面该map还有数据,那么需要删除该map里面的日期数据
  633. delete(existDataMap, currDateStr)
  634. }
  635. // 删除不需要的指标数据
  636. if len(existDataMap) > 0 {
  637. //待删除的日期
  638. removeDateList := make([]string, 0)
  639. for date := range existDataMap {
  640. removeDateList = append(removeDateList, date)
  641. }
  642. removeDateStr := strings.Join(removeDateList, `","`)
  643. removeDateStr = `"` + removeDateStr + `"`
  644. //如果拼接指标变更了,那么需要删除所有的指标数据
  645. tableName := GetEdbDataTableName(source, subSource)
  646. sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr)
  647. _, err = to.Raw(sql, edbInfoId).Exec()
  648. if err != nil {
  649. err = fmt.Errorf("删除不存在的升频指标数据失败,Err:" + err.Error())
  650. return
  651. }
  652. }
  653. // 新增的数据值
  654. if isAdd {
  655. addSql = strings.TrimRight(addSql, ",")
  656. _, err = to.Raw(addSql).Exec()
  657. }
  658. return
  659. }