edb_data_calculate_ljztbpj.go 28 KB

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