edb_data_calculate_ljztbpj.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932
  1. package models
  2. import (
  3. "errors"
  4. "fmt"
  5. "github.com/beego/beego/v2/client/orm"
  6. "github.com/shopspring/decimal"
  7. "hongze/hongze_edb_lib/utils"
  8. "sort"
  9. "strings"
  10. "time"
  11. )
  12. // EdbDataCalculateLjztbpj 累计值同比拼接数据结构体
  13. type EdbDataCalculateLjztbpj struct {
  14. EdbDataId int `orm:"column(edb_data_id);pk"`
  15. EdbInfoId int
  16. EdbCode string
  17. DataTime string
  18. Value float64
  19. Status int
  20. CreateTime time.Time
  21. ModifyTime time.Time
  22. DataTimestamp int64
  23. }
  24. // AddCalculateLjztbpj 新增累计值同比拼接数据
  25. func AddCalculateLjztbpj(req *EdbInfoCalculateBatchSaveReq, firstEdbInfo, secondEdbInfo *EdbInfo, edbCode, uniqueCode string, sysUserId int, sysUserRealName string) (edbInfoId int, err error) {
  26. o := orm.NewOrm()
  27. to, err := o.Begin()
  28. if err != nil {
  29. return
  30. }
  31. defer func() {
  32. if err != nil {
  33. fmt.Println("AddCalculateLjztbpj,Err:" + err.Error())
  34. _ = to.Rollback()
  35. } else {
  36. _ = to.Commit()
  37. }
  38. }()
  39. //待拼接指标map
  40. pjDataMap := make(map[string]float64)
  41. //拼接指标的日期切片数据
  42. pjEdbDataTimeList := make([]string, 0)
  43. //最近开始的时间
  44. var lastDateTime time.Time
  45. //获取待拼接指标
  46. {
  47. var condition string
  48. var pars []interface{}
  49. //获取待拼接指标最近的个12月31日有值的年份
  50. condition += " AND data_time like ? AND edb_info_id=? "
  51. pars = append(pars, "%12-31", firstEdbInfo.EdbInfoId)
  52. lastEdbData, tmpErr := GetLastEdbData(condition, pars, firstEdbInfo.Source)
  53. if tmpErr != nil {
  54. return edbInfoId, tmpErr
  55. }
  56. lastDateTime, _ = time.ParseInLocation(utils.FormatDate, lastEdbData.DataTime, time.Local)
  57. //获取待拼接指标的数据列表
  58. condition = ``
  59. pars = make([]interface{}, 0)
  60. condition += " AND data_time <= ? AND edb_info_id=? "
  61. pars = append(pars, lastEdbData.DataTime, firstEdbInfo.EdbInfoId)
  62. firstDataList, tmpErr := GetEdbDataListAll(condition, pars, firstEdbInfo.Source, 0)
  63. if tmpErr != nil {
  64. return edbInfoId, tmpErr
  65. }
  66. for _, v := range firstDataList {
  67. pjDataMap[v.DataTime] = v.Value
  68. pjEdbDataTimeList = append(pjEdbDataTimeList, v.DataTime)
  69. }
  70. }
  71. var edbInfo *EdbInfo
  72. if req.EdbInfoId <= 0 {
  73. edbInfo = &EdbInfo{
  74. SourceName: "累计值同比拼接",
  75. Source: utils.DATA_SOURCE_CALCULATE_LJZTBPJ,
  76. EdbCode: edbCode,
  77. EdbName: req.EdbName,
  78. EdbNameSource: req.EdbName,
  79. Frequency: req.Frequency,
  80. Unit: req.Unit,
  81. StartDate: firstEdbInfo.StartDate,
  82. EndDate: firstEdbInfo.EndDate,
  83. ClassifyId: req.ClassifyId,
  84. SysUserId: sysUserId,
  85. SysUserRealName: sysUserRealName,
  86. UniqueCode: uniqueCode,
  87. CreateTime: time.Now(),
  88. ModifyTime: time.Now(),
  89. CalculateFormula: lastDateTime.Format(utils.FormatDate),
  90. EdbType: 2,
  91. }
  92. newEdbInfoId, err := to.Insert(edbInfo)
  93. if err != nil {
  94. return edbInfoId, err
  95. }
  96. edbInfoId = int(newEdbInfoId)
  97. } else {
  98. edbInfoId = req.EdbInfoId
  99. //查询
  100. tmpEdbInfo, tmpErr := GetEdbInfoById(edbInfoId)
  101. if tmpErr != nil {
  102. err = tmpErr
  103. return
  104. }
  105. tmpEdbInfo.EdbName = req.EdbName
  106. tmpEdbInfo.ClassifyId = req.ClassifyId
  107. tmpEdbInfo.Frequency = req.Frequency
  108. tmpEdbInfo.Unit = req.Unit
  109. tmpEdbInfo.CalculateFormula = req.Formula
  110. edbInfo = tmpEdbInfo
  111. //删除指标数据
  112. dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_CALCULATE_LJZTBPJ)
  113. fmt.Println("dataTableName:" + dataTableName)
  114. deleteSql := ` DELETE FROM %s WHERE edb_info_id=? `
  115. deleteSql = fmt.Sprintf(deleteSql, dataTableName)
  116. _, err = to.Raw(deleteSql, req.EdbInfoId).Exec()
  117. if err != nil {
  118. return 0, err
  119. }
  120. //删除指标关系
  121. sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id=? `
  122. _, err = to.Raw(sql, edbInfoId).Exec()
  123. }
  124. //关联关系
  125. //拼接指标
  126. {
  127. calculateMappingItem := new(EdbInfoCalculateMapping)
  128. calculateMappingItem.CreateTime = time.Now()
  129. calculateMappingItem.ModifyTime = time.Now()
  130. calculateMappingItem.Sort = 1
  131. calculateMappingItem.EdbCode = edbCode
  132. calculateMappingItem.EdbInfoId = edbInfoId
  133. calculateMappingItem.FromEdbInfoId = firstEdbInfo.EdbInfoId
  134. calculateMappingItem.FromEdbCode = firstEdbInfo.EdbCode
  135. calculateMappingItem.FromEdbName = firstEdbInfo.EdbName
  136. calculateMappingItem.FromSource = firstEdbInfo.Source
  137. calculateMappingItem.FromSourceName = firstEdbInfo.SourceName
  138. calculateMappingItem.FromTag = "A"
  139. calculateMappingItem.Source = edbInfo.Source
  140. calculateMappingItem.SourceName = edbInfo.SourceName
  141. _, err = to.Insert(calculateMappingItem)
  142. if err != nil {
  143. return
  144. }
  145. }
  146. //同比值指标
  147. {
  148. calculateMappingItem := new(EdbInfoCalculateMapping)
  149. calculateMappingItem.CreateTime = time.Now()
  150. calculateMappingItem.ModifyTime = time.Now()
  151. calculateMappingItem.Sort = 1
  152. calculateMappingItem.EdbCode = edbCode
  153. calculateMappingItem.EdbInfoId = edbInfoId
  154. calculateMappingItem.FromEdbInfoId = secondEdbInfo.EdbInfoId
  155. calculateMappingItem.FromEdbCode = secondEdbInfo.EdbCode
  156. calculateMappingItem.FromEdbName = secondEdbInfo.EdbName
  157. calculateMappingItem.FromSource = secondEdbInfo.Source
  158. calculateMappingItem.FromSourceName = secondEdbInfo.SourceName
  159. calculateMappingItem.FromTag = "B"
  160. calculateMappingItem.Source = edbInfo.Source
  161. calculateMappingItem.SourceName = edbInfo.SourceName
  162. _, err = to.Insert(calculateMappingItem)
  163. if err != nil {
  164. return
  165. }
  166. }
  167. //同比值指标map
  168. tbzEdbDataMap := make(map[string]float64)
  169. //同比值日期切片列表
  170. tbzEdbDataTimeList := make([]string, 0)
  171. //同比值指标
  172. {
  173. var condition string
  174. var pars []interface{}
  175. condition += " AND data_time > ? AND edb_info_id = ? "
  176. pars = append(pars, lastDateTime, secondEdbInfo.EdbInfoId)
  177. //同比值指标的数据列表
  178. secondDataList, tmpErr := GetEdbDataListAll(condition, pars, secondEdbInfo.Source, 0)
  179. if tmpErr != nil {
  180. return edbInfoId, tmpErr
  181. }
  182. for _, v := range secondDataList {
  183. tbzEdbDataMap[v.DataTime] = v.Value
  184. tbzEdbDataTimeList = append(tbzEdbDataTimeList, v.DataTime)
  185. }
  186. }
  187. sort.Strings(tbzEdbDataTimeList)
  188. for _, v := range tbzEdbDataTimeList {
  189. tbzDataTime, _ := time.ParseInLocation(utils.FormatDate, v, time.Local)
  190. //获取拼接指标上一年同一天的数据
  191. var pjDataTime time.Time
  192. if tbzDataTime.Month() == 2 {
  193. pjDataTime = tbzDataTime.AddDate(0, -11, 0)
  194. pjDataTime = time.Date(pjDataTime.Year(), pjDataTime.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 0, -1)
  195. } else {
  196. pjDataTime = tbzDataTime.AddDate(-1, 0, 0)
  197. }
  198. //如果存在数据,那么就去计算当前的数据
  199. if pjDataValue, ok := pjDataMap[pjDataTime.Format(utils.FormatDate)]; ok {
  200. tbzDataValue := tbzEdbDataMap[v] //同比值
  201. currValue := pjDataValue * (1 + tbzDataValue/100)
  202. currValue, _ = decimal.NewFromFloat(currValue).Truncate(4).Float64()
  203. //将计算后的数据存入待拼接指标map里面,以便后续计算
  204. pjDataMap[v] = currValue
  205. pjEdbDataTimeList = append(pjEdbDataTimeList, v)
  206. }
  207. }
  208. addDataList := make([]*EdbDataCalculateLjztbpj, 0)
  209. //日期排序下
  210. sort.Strings(pjEdbDataTimeList)
  211. //这么做的目的是为了让数据插入的时候,可以正序插入(业务上没啥卵用,就是为了让我看数据的时候舒服点,手动狗头-_-|)
  212. for _, dataTime := range pjEdbDataTimeList {
  213. if dataValue, ok := pjDataMap[dataTime]; ok {
  214. //时间戳
  215. currentDate, _ := time.Parse(utils.FormatDate, dataTime)
  216. timestamp := currentDate.UnixNano() / 1e6
  217. edbDataLjztbpj := &EdbDataCalculateLjztbpj{
  218. EdbInfoId: edbInfoId,
  219. EdbCode: edbInfo.EdbCode,
  220. DataTime: dataTime,
  221. Value: dataValue,
  222. Status: 1,
  223. CreateTime: time.Now(),
  224. ModifyTime: time.Now(),
  225. DataTimestamp: timestamp,
  226. }
  227. addDataList = append(addDataList, edbDataLjztbpj)
  228. }
  229. }
  230. //数据入库
  231. tmpAddDataList := make([]*EdbDataCalculateLjztbpj, 0)
  232. for _, v := range addDataList {
  233. tmpAddDataList = append(tmpAddDataList, v)
  234. if len(tmpAddDataList) >= 200 {
  235. _, tmpErr := to.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  236. if tmpErr != nil {
  237. err = tmpErr
  238. return
  239. }
  240. //重新初始化需要加入的数据切片
  241. tmpAddDataList = make([]*EdbDataCalculateLjztbpj, 0)
  242. }
  243. }
  244. //最后如果还有需要新增的数据,那么就统一入库
  245. if len(tmpAddDataList) > 0 {
  246. _, tmpErr := to.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  247. if tmpErr != nil {
  248. err = tmpErr
  249. return
  250. }
  251. }
  252. return
  253. }
  254. // EditCalculateLjztbpj 编辑累计值同比拼接数据
  255. func EditCalculateLjztbpj(req *EdbInfoCalculateBatchEditReq, nowEdbInfo, firstEdbInfo, secondEdbInfo *EdbInfo) (edbInfoId int, err error) {
  256. edbInfoId = req.EdbInfoId
  257. o := orm.NewOrm()
  258. to, err := o.Begin()
  259. if err != nil {
  260. return
  261. }
  262. defer func() {
  263. if err != nil {
  264. fmt.Println("EditCalculateLjztbpj,Err:" + err.Error())
  265. _ = to.Rollback()
  266. } else {
  267. _ = to.Commit()
  268. }
  269. }()
  270. sql := ``
  271. var existCondition string
  272. var existPars []interface{}
  273. existCondition += " AND edb_info_id=? "
  274. existPars = append(existPars, edbInfoId)
  275. //查询出所有的关联指标
  276. existList, err := GetEdbInfoCalculateListByCondition(existCondition, existPars)
  277. if err != nil {
  278. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  279. return
  280. }
  281. var existItemA, existItemB *EdbInfoCalculateMapping
  282. for _, existItem := range existList {
  283. if existItem.FromTag == "A" {
  284. existItemA = existItem
  285. } else if existItem.FromTag == "B" {
  286. existItemB = existItem
  287. }
  288. }
  289. // 原数据开始计算日期
  290. startCalculationDate, _ := time.ParseInLocation(utils.FormatDate, nowEdbInfo.CalculateFormula, time.Local)
  291. //待拼接指标map
  292. pjDataMap := make(map[string]float64) //需要入库的数据
  293. nowEdbDataMap := make(map[string]float64) //当前指标的数据(已经在库里了,不需要重新)
  294. //拼接指标的日期切片数据
  295. pjEdbDataTimeList := make([]string, 0)
  296. //最近开始的时间
  297. lastDateTime := startCalculationDate
  298. //待拼接指标数据
  299. {
  300. //如果拼接指标变更了,那么需要删除所有的指标进行重新拼接
  301. if existItemA.FromEdbInfoId != firstEdbInfo.EdbInfoId {
  302. //删除旧的指标数据
  303. {
  304. //删除之前的A指标关联关系
  305. sql = ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? and from_edb_info_id = ?`
  306. _, err = o.Raw(sql, edbInfoId, existItemA.FromEdbInfoId).Exec()
  307. if err != nil {
  308. err = errors.New("删除拼接日期之前的指标关联关系失败,Err:" + err.Error())
  309. return
  310. }
  311. //如果拼接指标变更了,那么需要删除所有的指标数据
  312. tableName := GetEdbDataTableName(nowEdbInfo.Source)
  313. sql = fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? `, tableName)
  314. _, err = o.Raw(sql, edbInfoId).Exec()
  315. if err != nil {
  316. err = errors.New("删除所有的累计值同比拼接指标数据失败,Err:" + err.Error())
  317. return
  318. }
  319. }
  320. //添加新的指标关系
  321. {
  322. calculateMappingItem := new(EdbInfoCalculateMapping)
  323. calculateMappingItem.CreateTime = time.Now()
  324. calculateMappingItem.ModifyTime = time.Now()
  325. calculateMappingItem.Sort = 1
  326. calculateMappingItem.EdbCode = nowEdbInfo.EdbCode
  327. calculateMappingItem.EdbInfoId = edbInfoId
  328. calculateMappingItem.FromEdbInfoId = firstEdbInfo.EdbInfoId
  329. calculateMappingItem.FromEdbCode = firstEdbInfo.EdbCode
  330. calculateMappingItem.FromEdbName = firstEdbInfo.EdbName
  331. calculateMappingItem.FromSource = firstEdbInfo.Source
  332. calculateMappingItem.FromSourceName = firstEdbInfo.SourceName
  333. calculateMappingItem.FromTag = "A"
  334. calculateMappingItem.Source = nowEdbInfo.Source
  335. calculateMappingItem.SourceName = nowEdbInfo.SourceName
  336. _, err = o.Insert(calculateMappingItem)
  337. if err != nil {
  338. return
  339. }
  340. }
  341. var condition string
  342. var pars []interface{}
  343. //获取待拼接指标最近的个12月31日有值的年份
  344. condition += " AND data_time like ? AND edb_info_id=? "
  345. pars = append(pars, "%12-31", firstEdbInfo.EdbInfoId)
  346. //获取最新的待拼接指标
  347. lastEdbData, tmpErr := GetLastEdbData(condition, pars, firstEdbInfo.Source)
  348. if tmpErr != nil {
  349. return edbInfoId, tmpErr
  350. }
  351. lastDateTime, _ = time.ParseInLocation(utils.FormatDate, lastEdbData.DataTime, time.Local)
  352. //获取待拼接指标的数据列表
  353. condition = ``
  354. pars = make([]interface{}, 0)
  355. condition += " AND data_time <= ? AND edb_info_id=? "
  356. pars = append(pars, lastEdbData.DataTime, firstEdbInfo.EdbInfoId)
  357. firstDataList, tmpErr := GetEdbDataListAll(condition, pars, firstEdbInfo.Source, 0)
  358. if tmpErr != nil {
  359. return edbInfoId, tmpErr
  360. }
  361. for _, v := range firstDataList {
  362. pjDataMap[v.DataTime] = v.Value
  363. pjEdbDataTimeList = append(pjEdbDataTimeList, v.DataTime)
  364. //将新的数据存入已入库指标map里面,以便后续计算
  365. nowEdbDataMap[v.DataTime] = v.Value
  366. }
  367. } else {
  368. // 获取当前指标的所有数据
  369. var condition string
  370. var pars []interface{}
  371. condition += " AND edb_info_id=? and data_time <= ?"
  372. pars = append(pars, nowEdbInfo.EdbInfoId, nowEdbInfo.CalculateFormula)
  373. nowEdbDataList, tmpErr := GetEdbDataListAll(condition, pars, nowEdbInfo.Source, 0)
  374. if tmpErr != nil {
  375. return edbInfoId, tmpErr
  376. }
  377. for _, v := range nowEdbDataList {
  378. nowEdbDataMap[v.DataTime] = v.Value
  379. pjEdbDataTimeList = append(pjEdbDataTimeList, v.DataTime)
  380. }
  381. }
  382. }
  383. //同比值指标map
  384. tbzEdbDataMap := make(map[string]float64)
  385. //同比值日期切片列表
  386. tbzEdbDataTimeList := make([]string, 0)
  387. //同比值指标
  388. {
  389. if existItemB.FromEdbInfoId != secondEdbInfo.EdbInfoId {
  390. //删除通过B指标生成的数据
  391. {
  392. //删除之前的B指标关联关系
  393. sql = ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? and from_edb_info_id = ?`
  394. _, err = o.Raw(sql, edbInfoId, existItemB.FromEdbInfoId).Exec()
  395. if err != nil {
  396. err = errors.New("删除拼接日期之后的指标关联关系失败,Err:" + err.Error())
  397. return
  398. }
  399. //如果同比值指标变更了,那么需要删除所有计算出来的指标数据
  400. tableName := GetEdbDataTableName(nowEdbInfo.Source)
  401. sql = fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time > ? `, tableName)
  402. _, err = o.Raw(sql, edbInfoId, nowEdbInfo.CalculateFormula).Exec()
  403. if err != nil {
  404. err = errors.New("删除计算出来的累计值同比拼接指标数据失败,Err:" + err.Error())
  405. return
  406. }
  407. }
  408. //添加新的指标关系
  409. {
  410. calculateMappingItem := new(EdbInfoCalculateMapping)
  411. calculateMappingItem.CreateTime = time.Now()
  412. calculateMappingItem.ModifyTime = time.Now()
  413. calculateMappingItem.Sort = 1
  414. calculateMappingItem.EdbCode = nowEdbInfo.EdbCode
  415. calculateMappingItem.EdbInfoId = edbInfoId
  416. calculateMappingItem.FromEdbInfoId = secondEdbInfo.EdbInfoId
  417. calculateMappingItem.FromEdbCode = secondEdbInfo.EdbCode
  418. calculateMappingItem.FromEdbName = secondEdbInfo.EdbName
  419. calculateMappingItem.FromSource = secondEdbInfo.Source
  420. calculateMappingItem.FromSourceName = secondEdbInfo.SourceName
  421. calculateMappingItem.FromTag = "B"
  422. calculateMappingItem.Source = nowEdbInfo.Source
  423. calculateMappingItem.SourceName = nowEdbInfo.SourceName
  424. _, err = o.Insert(calculateMappingItem)
  425. if err != nil {
  426. return
  427. }
  428. }
  429. }
  430. var condition string
  431. var pars []interface{}
  432. condition += " AND data_time > ? AND edb_info_id = ? "
  433. pars = append(pars, lastDateTime, secondEdbInfo.EdbInfoId)
  434. //同比值指标的数据列表
  435. secondDataList, tmpErr := GetEdbDataListAll(condition, pars, secondEdbInfo.Source, 0)
  436. if tmpErr != nil {
  437. return edbInfoId, tmpErr
  438. }
  439. for _, v := range secondDataList {
  440. tbzEdbDataMap[v.DataTime] = v.Value
  441. tbzEdbDataTimeList = append(tbzEdbDataTimeList, v.DataTime)
  442. }
  443. }
  444. sort.Strings(tbzEdbDataTimeList)
  445. // 遍历现有的数据,判断拼接指标中是否存在该日期数据,如果拼接指标无此数据,那么需要删除该日期数据(日期的判断:需要在开始计算日期之后)
  446. removeDateList := make([]string, 0)
  447. for nowEdbDate := range nowEdbDataMap {
  448. nowEdbDateTime, _ := time.ParseInLocation(utils.FormatDate, nowEdbDate, time.Local)
  449. //校验日期 需要 大于 拼接前日期
  450. if startCalculationDate.Before(nowEdbDateTime) {
  451. if _, ok := tbzEdbDataMap[nowEdbDate]; !ok {
  452. // 同比指标中,不存在该日期数据,那么需要移除 现有数据 中该日期的数据
  453. removeDateList = append(removeDateList, nowEdbDate)
  454. }
  455. }
  456. }
  457. //待修改的指标数据map(index:日期,value:值)
  458. updateEdbDataMap := make(map[string]float64)
  459. for _, v := range tbzEdbDataTimeList {
  460. tbzDataTime, _ := time.ParseInLocation(utils.FormatDate, v, time.Local)
  461. //获取拼接指标上一年同一天的数据
  462. var pjDataTime time.Time
  463. if tbzDataTime.Month() == 2 {
  464. pjDataTime = tbzDataTime.AddDate(0, -11, 0)
  465. pjDataTime = time.Date(pjDataTime.Year(), pjDataTime.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 0, -1)
  466. } else {
  467. pjDataTime = tbzDataTime.AddDate(-1, 0, 0)
  468. }
  469. //校验现有数据中,是否存在该日期的数据,如果存在的话,那么就要去校验 最新计算数据 与 现有数据 是否一致
  470. if nowEdbDataValue, isHas := nowEdbDataMap[v]; isHas {
  471. //获取去年今日的数据,获取到后,然后是去修改该日期的数据
  472. if lastYearEdbDataValue, ok := nowEdbDataMap[pjDataTime.Format(utils.FormatDate)]; ok {
  473. tbzDataValue := tbzEdbDataMap[v] //同比值
  474. currValue := lastYearEdbDataValue * (1 + tbzDataValue/100)
  475. currValue, _ = decimal.NewFromFloat(currValue).Truncate(4).Float64() //保留4位小数
  476. //如果计算出来的值与库里面的值不匹配,那么就去修改该值
  477. if nowEdbDataValue != currValue {
  478. //将计算后的数据存入待拼接指标map里面,以便后续计算
  479. updateEdbDataMap[v] = currValue
  480. }
  481. }
  482. } else {
  483. //因为 现有数据中 不存在该日期数据,那么需要做新增数据处理
  484. //如果去年今日存在该数据,那么就去计算当前的数据
  485. if pjDataValue, ok := nowEdbDataMap[pjDataTime.Format(utils.FormatDate)]; ok {
  486. tbzDataValue := tbzEdbDataMap[v] //同比值
  487. currValue := pjDataValue * (1 + tbzDataValue/100)
  488. currValue, _ = decimal.NewFromFloat(currValue).Truncate(4).Float64()
  489. //将计算后的数据存入已入库指标map里面,以便后续计算
  490. nowEdbDataMap[v] = currValue
  491. //将计算后的数据存入待拼接指标map里面,以便后续入库
  492. pjDataMap[v] = currValue
  493. pjEdbDataTimeList = append(pjEdbDataTimeList, v)
  494. }
  495. }
  496. }
  497. //新增的数据入库
  498. {
  499. addDataList := make([]*EdbDataCalculateLjztbpj, 0)
  500. for dataTime, dataValue := range pjDataMap {
  501. //时间戳
  502. currentDate, _ := time.Parse(utils.FormatDate, dataTime)
  503. timestamp := currentDate.UnixNano() / 1e6
  504. edbDataLjztbpj := &EdbDataCalculateLjztbpj{
  505. EdbInfoId: edbInfoId,
  506. EdbCode: nowEdbInfo.EdbCode,
  507. DataTime: dataTime,
  508. Value: dataValue,
  509. Status: 1,
  510. CreateTime: time.Now(),
  511. ModifyTime: time.Now(),
  512. DataTimestamp: timestamp,
  513. }
  514. addDataList = append(addDataList, edbDataLjztbpj)
  515. }
  516. tmpAddDataList := make([]*EdbDataCalculateLjztbpj, 0)
  517. for _, v := range addDataList {
  518. tmpAddDataList = append(tmpAddDataList, v)
  519. if len(tmpAddDataList) >= 200 {
  520. _, tmpErr := o.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  521. if tmpErr != nil {
  522. err = tmpErr
  523. return
  524. }
  525. //重新初始化需要加入的数据切片
  526. tmpAddDataList = make([]*EdbDataCalculateLjztbpj, 0)
  527. }
  528. }
  529. //最后如果还有需要新增的数据,那么就统一入库
  530. if len(tmpAddDataList) > 0 {
  531. _, tmpErr := o.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  532. if tmpErr != nil {
  533. err = tmpErr
  534. return
  535. }
  536. }
  537. }
  538. //删除已经不存在的累计同比拼接指标数据(由于同比值当日的数据删除了)
  539. {
  540. if len(removeDateList) > 0 {
  541. removeDateStr := strings.Join(removeDateList, `","`)
  542. removeDateStr = `"` + removeDateStr + `"`
  543. //如果拼接指标变更了,那么需要删除所有的指标数据
  544. tableName := GetEdbDataTableName(nowEdbInfo.Source)
  545. sql = fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr)
  546. _, err = o.Raw(sql, edbInfoId).Exec()
  547. if err != nil {
  548. err = errors.New("删除不存在的累计值同比拼接指标数据失败,Err:" + err.Error())
  549. return
  550. }
  551. }
  552. }
  553. //修改现有的数据中对应的值
  554. {
  555. tableName := GetEdbDataTableName(nowEdbInfo.Source)
  556. for edbDate, edbDataValue := range updateEdbDataMap {
  557. sql = fmt.Sprintf(` UPDATE %s set value = %f,modify_time=now() WHERE edb_info_id = ? and data_time = %s `, tableName, edbDataValue, edbDate)
  558. _, err = o.Raw(sql, edbInfoId).Exec()
  559. if err != nil {
  560. err = errors.New("更新现有的累计值同比拼接指标数据失败,Err:" + err.Error())
  561. return
  562. }
  563. }
  564. }
  565. //修改指标信息
  566. sql = ` UPDATE edb_info SET
  567. edb_name =?,
  568. edb_name_source=?,
  569. frequency = ?,
  570. unit = ?,
  571. classify_id = ?,
  572. calculate_formula=?,
  573. modify_time = NOW()
  574. WHERE edb_info_id = ? `
  575. _, err = o.Raw(sql, req.EdbName, req.EdbName, req.Frequency, req.Unit, req.ClassifyId, lastDateTime.Format(utils.FormatDate), edbInfoId).Exec()
  576. if err != nil {
  577. return
  578. }
  579. return
  580. }
  581. // GetAllEdbDataCalculateLjztbpjByEdbInfoId 根据指标id获取全部的数据
  582. func GetAllEdbDataCalculateLjztbpjByEdbInfoId(edbInfoId int) (items []*EdbDataCalculateLjztbpj, err error) {
  583. o := orm.NewOrm()
  584. sql := ` SELECT * FROM edb_data_calculate_ljztbpj WHERE edb_info_id=? ORDER BY data_time DESC `
  585. _, err = o.Raw(sql, edbInfoId).QueryRows(&items)
  586. return
  587. }
  588. // RefreshAllCalculateLjztbpj 刷新所有 累计值同比拼接 数据
  589. func RefreshAllCalculateLjztbpj(edbInfo *EdbInfo) (err error) {
  590. o := orm.NewOrm()
  591. to, err := o.Begin()
  592. if err != nil {
  593. return
  594. }
  595. defer func() {
  596. if err != nil {
  597. fmt.Println("RefreshAllCalculateLjztbpj,Err:" + err.Error())
  598. _ = to.Rollback()
  599. } else {
  600. _ = to.Commit()
  601. }
  602. }()
  603. //查询当前指标现有的数据
  604. var condition string
  605. var pars []interface{}
  606. condition += " AND edb_info_id=? "
  607. pars = append(pars, edbInfo.EdbInfoId)
  608. //所有的数据
  609. dataList, err := GetAllEdbDataCalculateLjztbpjByEdbInfoId(edbInfo.EdbInfoId)
  610. if err != nil {
  611. return err
  612. }
  613. //待拼接指标map
  614. pjDataMap := make(map[string]float64) //需要入库的数据
  615. nowEdbDataMap := make(map[string]float64) //当前指标的数据(已经在库里了,不需要重新)
  616. //拼接指标的日期切片数据
  617. pjEdbDataTimeList := make([]string, 0)
  618. dataMap := make(map[string]*EdbDataCalculateLjztbpj)
  619. for _, v := range dataList {
  620. pjEdbDataTimeList = append(pjEdbDataTimeList, v.DataTime)
  621. dataMap[v.DataTime] = v
  622. nowEdbDataMap[v.DataTime] = v.Value
  623. }
  624. //查询关联指标信息
  625. var existCondition string
  626. var existPars []interface{}
  627. existCondition += " AND edb_info_id=? "
  628. existPars = append(existPars, edbInfo.EdbInfoId)
  629. existList, err := GetEdbInfoCalculateListByCondition(existCondition, existPars)
  630. if err != nil {
  631. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  632. return
  633. }
  634. var existItemA, existItemB *EdbInfoCalculateMapping
  635. for _, existItem := range existList {
  636. if existItem.FromTag == "A" {
  637. existItemA = existItem
  638. } else if existItem.FromTag == "B" {
  639. existItemB = existItem
  640. }
  641. }
  642. // 原数据开始计算日期
  643. startCalculationDate, _ := time.ParseInLocation(utils.FormatDate, edbInfo.CalculateFormula, time.Local)
  644. //待拼接指标
  645. {
  646. var condition string
  647. var pars []interface{}
  648. condition += " AND data_time <= ? AND edb_info_id=? "
  649. pars = append(pars, startCalculationDate, existItemA.FromEdbInfoId)
  650. //第一个指标的数据列表
  651. firstDataList, tmpErr := GetEdbDataListAll(condition, pars, existItemA.FromSource, 0)
  652. if tmpErr != nil {
  653. return tmpErr
  654. }
  655. for _, v := range firstDataList {
  656. //时间戳
  657. if edbData, ok := dataMap[v.DataTime]; ok {
  658. if edbData.Value != v.Value {
  659. //更新指标数据
  660. edbData.Value = v.Value
  661. _, _ = o.Update(edbData, "Value")
  662. //将新的数据存入已入库指标map里面,以便后续计算
  663. nowEdbDataMap[edbData.DataTime] = v.Value
  664. }
  665. }
  666. }
  667. }
  668. //同比值指标map
  669. tbzEdbDataMap := make(map[string]float64)
  670. //同比值日期切片列表
  671. tbzEdbDataTimeList := make([]string, 0)
  672. //同比值指标
  673. {
  674. var condition string
  675. var pars []interface{}
  676. condition += " AND data_time > ? AND edb_info_id = ? "
  677. pars = append(pars, startCalculationDate, existItemB.FromEdbInfoId)
  678. //第二个指标的数据列表
  679. secondDataList, tmpErr := GetEdbDataListAll(condition, pars, existItemB.FromSource, 0)
  680. if tmpErr != nil {
  681. return tmpErr
  682. }
  683. for _, v := range secondDataList {
  684. tbzEdbDataMap[v.DataTime] = v.Value
  685. tbzEdbDataTimeList = append(tbzEdbDataTimeList, v.DataTime)
  686. }
  687. }
  688. sort.Strings(tbzEdbDataTimeList)
  689. // 遍历现有的数据,判断拼接指标中是否存在该日期数据,如果拼接指标无此数据,那么需要删除该日期数据(日期的判断:需要在开始计算日期之后)
  690. removeDateList := make([]string, 0)
  691. for nowEdbDate := range nowEdbDataMap {
  692. nowEdbDateTime, _ := time.ParseInLocation(utils.FormatDate, nowEdbDate, time.Local)
  693. //校验日期 需要 大于 拼接前日期
  694. if startCalculationDate.Before(nowEdbDateTime) {
  695. if _, ok := tbzEdbDataMap[nowEdbDate]; !ok {
  696. // 同比指标中,不存在该日期数据,那么需要移除 现有数据 中该日期的数据
  697. removeDateList = append(removeDateList, nowEdbDate)
  698. }
  699. }
  700. }
  701. //待修改的指标数据map(index:日期,value:值)
  702. updateEdbDataMap := make(map[string]float64)
  703. for _, v := range tbzEdbDataTimeList {
  704. tbzDataTime, _ := time.ParseInLocation(utils.FormatDate, v, time.Local)
  705. //获取拼接指标上一年同一天的数据
  706. var pjDataTime time.Time
  707. if tbzDataTime.Month() == 2 {
  708. pjDataTime = tbzDataTime.AddDate(0, -11, 0)
  709. pjDataTime = time.Date(pjDataTime.Year(), pjDataTime.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 0, -1)
  710. } else {
  711. pjDataTime = tbzDataTime.AddDate(-1, 0, 0)
  712. }
  713. //校验现有数据中,是否存在该日期的数据,如果存在的话,那么就要去校验 最新计算数据 与 现有数据 是否一致
  714. if nowEdbDataValue, isHas := nowEdbDataMap[v]; isHas {
  715. //获取去年今日的数据,获取到后,然后是去修改该日期的数据
  716. if lastYearEdbDataValue, ok := nowEdbDataMap[pjDataTime.Format(utils.FormatDate)]; ok {
  717. if v == "2021-08-31" {
  718. fmt.Println("进来了")
  719. }
  720. tbzDataValue := tbzEdbDataMap[v] //同比值
  721. currValue := lastYearEdbDataValue * (1 + tbzDataValue/100)
  722. currValue, _ = decimal.NewFromFloat(currValue).Truncate(4).Float64() //保留4位小数
  723. //如果计算出来的值与库里面的值不匹配,那么就去修改该值
  724. if nowEdbDataValue != currValue {
  725. //将计算后的数据存入待拼接指标map里面,以便后续计算
  726. updateEdbDataMap[v] = currValue
  727. }
  728. }
  729. } else {
  730. //因为 现有数据中 不存在该日期数据,那么需要做新增数据处理
  731. //如果去年今日存在该数据,那么就去计算当前的数据
  732. if pjDataValue, ok := nowEdbDataMap[pjDataTime.Format(utils.FormatDate)]; ok {
  733. tbzDataValue := tbzEdbDataMap[v] //同比值
  734. currValue := pjDataValue * (1 + tbzDataValue/100)
  735. currValue, _ = decimal.NewFromFloat(currValue).Truncate(4).Float64()
  736. //将计算后的数据存入已入库指标map里面,以便后续计算
  737. nowEdbDataMap[v] = currValue
  738. //将计算后的数据存入待拼接指标map里面,以便后续入库
  739. pjDataMap[v] = currValue
  740. pjEdbDataTimeList = append(pjEdbDataTimeList, v)
  741. }
  742. }
  743. }
  744. //新增的数据入库
  745. {
  746. addDataList := make([]*EdbDataCalculateLjztbpj, 0)
  747. for dataTime, dataValue := range pjDataMap {
  748. //时间戳
  749. currentDate, _ := time.Parse(utils.FormatDate, dataTime)
  750. timestamp := currentDate.UnixNano() / 1e6
  751. edbDataLjztbpj := &EdbDataCalculateLjztbpj{
  752. EdbInfoId: edbInfo.EdbInfoId,
  753. EdbCode: edbInfo.EdbCode,
  754. DataTime: dataTime,
  755. Value: dataValue,
  756. Status: 1,
  757. CreateTime: time.Now(),
  758. ModifyTime: time.Now(),
  759. DataTimestamp: timestamp,
  760. }
  761. addDataList = append(addDataList, edbDataLjztbpj)
  762. }
  763. tmpAddDataList := make([]*EdbDataCalculateLjztbpj, 0)
  764. for _, v := range addDataList {
  765. tmpAddDataList = append(tmpAddDataList, v)
  766. if len(tmpAddDataList) >= 200 {
  767. _, tmpErr := o.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  768. if tmpErr != nil {
  769. err = tmpErr
  770. return
  771. }
  772. //重新初始化需要加入的数据切片
  773. tmpAddDataList = make([]*EdbDataCalculateLjztbpj, 0)
  774. }
  775. }
  776. //最后如果还有需要新增的数据,那么就统一入库
  777. if len(tmpAddDataList) > 0 {
  778. _, tmpErr := o.InsertMulti(len(tmpAddDataList), tmpAddDataList)
  779. if tmpErr != nil {
  780. err = tmpErr
  781. return
  782. }
  783. }
  784. }
  785. //删除已经不存在的累计同比拼接指标数据(由于同比值当日的数据删除了)
  786. {
  787. if len(removeDateList) > 0 {
  788. removeDateStr := strings.Join(removeDateList, `","`)
  789. removeDateStr = `"` + removeDateStr + `"`
  790. //如果拼接指标变更了,那么需要删除所有的指标数据
  791. tableName := GetEdbDataTableName(edbInfo.Source)
  792. sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (%s) `, tableName, removeDateStr)
  793. _, err = o.Raw(sql, edbInfo.EdbInfoId).Exec()
  794. if err != nil {
  795. err = errors.New("删除不存在的累计值同比拼接指标数据失败,Err:" + err.Error())
  796. return
  797. }
  798. }
  799. }
  800. //修改现有的数据中对应的值
  801. {
  802. tableName := GetEdbDataTableName(edbInfo.Source)
  803. for edbDate, edbDataValue := range updateEdbDataMap {
  804. sql := fmt.Sprintf(` UPDATE %s set value = ?,modify_time=now() WHERE edb_info_id = ? and data_time = ? `, tableName)
  805. _, err = o.Raw(sql, edbDataValue, edbInfo.EdbInfoId, edbDate).Exec()
  806. if err != nil {
  807. err = errors.New("更新现有的累计值同比拼接指标数据失败,Err:" + err.Error())
  808. return
  809. }
  810. }
  811. }
  812. return
  813. }