edb_data_calculate_ljz.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747
  1. package models
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "eta/eta_index_lib/utils"
  6. "fmt"
  7. "github.com/beego/beego/v2/client/orm"
  8. "github.com/shopspring/decimal"
  9. "reflect"
  10. "strconv"
  11. "strings"
  12. "time"
  13. )
  14. // Ljz 累计值
  15. type Ljz struct {
  16. }
  17. // Add 添加
  18. func (obj Ljz) Add(params AddCalculateBatchParams) (edbInfo *EdbInfo, err error, errMsg string) {
  19. req := params.Req
  20. fromEdbInfo := params.FromEdbInfo
  21. edbCode := params.EdbCode
  22. frequencyInt := utils.CheckFrequency(fromEdbInfo.Frequency, req.Frequency)
  23. if frequencyInt < 0 {
  24. errMsg = "频度异常,不允许低频转为高频"
  25. err = errors.New(errMsg)
  26. return
  27. }
  28. if frequencyInt == 0 {
  29. errMsg = "频度异常,不允许同频计算累计值"
  30. err = errors.New(errMsg)
  31. return
  32. }
  33. o := orm.NewOrm()
  34. to, err := o.Begin()
  35. if err != nil {
  36. return
  37. }
  38. defer func() {
  39. if err != nil {
  40. _ = to.Rollback()
  41. fmt.Println(reflect.TypeOf(obj).Name(), ";Add,Err:"+err.Error())
  42. } else {
  43. _ = to.Commit()
  44. }
  45. }()
  46. edbInfo = &EdbInfo{
  47. //EdbInfoId: 0,
  48. SourceName: obj.GetSourceName(),
  49. Source: obj.GetSource(),
  50. EdbCode: edbCode,
  51. EdbName: req.EdbName,
  52. EdbNameSource: req.EdbName,
  53. Frequency: req.Frequency,
  54. Unit: req.Unit,
  55. StartDate: "",
  56. EndDate: "",
  57. ClassifyId: req.ClassifyId,
  58. SysUserId: params.SysUserId,
  59. SysUserRealName: params.SysUserRealName,
  60. UniqueCode: params.UniqueCode,
  61. CreateTime: time.Now(),
  62. ModifyTime: time.Now(),
  63. MinValue: 0,
  64. MaxValue: 0,
  65. CalculateFormula: req.Formula,
  66. EdbType: obj.GetEdbType(),
  67. Sort: 0,
  68. MoveType: 0,
  69. MoveFrequency: "",
  70. NoUpdate: 0,
  71. ServerUrl: "",
  72. EdbInfoType: 0,
  73. EdbNameEn: req.EdbName,
  74. UnitEn: req.Unit,
  75. LatestDate: "",
  76. LatestValue: 0,
  77. ChartImage: "",
  78. Calendar: "",
  79. Extra: req.Extra,
  80. }
  81. newEdbInfoId, tmpErr := to.Insert(edbInfo)
  82. if tmpErr != nil {
  83. err = tmpErr
  84. return
  85. }
  86. edbInfo.EdbInfoId = int(newEdbInfoId)
  87. //关联关系
  88. {
  89. calculateMappingItem := new(EdbInfoCalculateMapping)
  90. calculateMappingItem.CreateTime = time.Now()
  91. calculateMappingItem.ModifyTime = time.Now()
  92. calculateMappingItem.Sort = 1
  93. calculateMappingItem.EdbCode = edbCode
  94. calculateMappingItem.EdbInfoId = edbInfo.EdbInfoId
  95. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  96. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  97. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  98. calculateMappingItem.FromSource = fromEdbInfo.Source
  99. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  100. calculateMappingItem.FromTag = ""
  101. calculateMappingItem.Source = edbInfo.Source
  102. calculateMappingItem.SourceName = edbInfo.SourceName
  103. calculateMappingItem.FromSubSource = fromEdbInfo.SubSource
  104. _, err = to.Insert(calculateMappingItem)
  105. if err != nil {
  106. return
  107. }
  108. }
  109. //计算数据
  110. err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
  111. return
  112. }
  113. // Edit 编辑
  114. func (obj Ljz) Edit(params EditCalculateBatchParams) (err error, errMsg string) {
  115. req := params.Req
  116. edbInfo := params.EdbInfo
  117. fromEdbInfo := params.FromEdbInfo
  118. frequencyInt := utils.CheckFrequency(fromEdbInfo.Frequency, req.Frequency)
  119. if frequencyInt < 0 {
  120. errMsg = "频度异常,不允许低频转为高频"
  121. err = errors.New(errMsg)
  122. return
  123. }
  124. if frequencyInt == 0 {
  125. errMsg = "频度异常,不允许同频计算累计值"
  126. err = errors.New(errMsg)
  127. return
  128. }
  129. o := orm.NewOrm()
  130. to, err := o.Begin()
  131. if err != nil {
  132. return
  133. }
  134. defer func() {
  135. if err != nil {
  136. _ = to.Rollback()
  137. fmt.Println(reflect.TypeOf(obj).Name(), ";Edit,Err:"+err.Error())
  138. } else {
  139. _ = to.Commit()
  140. }
  141. }()
  142. tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
  143. var isRecalculate bool
  144. if edbInfo.Frequency != req.Frequency || edbInfo.Extra != req.Extra {
  145. isRecalculate = true
  146. }
  147. //修改指标信息
  148. edbInfo.EdbName = req.EdbName
  149. edbInfo.EdbNameSource = req.EdbName
  150. edbInfo.Frequency = req.Frequency
  151. edbInfo.Unit = req.Unit
  152. edbInfo.ClassifyId = req.ClassifyId
  153. edbInfo.Extra = req.Extra
  154. edbInfo.EdbNameEn = req.EdbNameEn
  155. edbInfo.UnitEn = req.UnitEn
  156. edbInfo.ModifyTime = time.Now()
  157. _, err = to.Update(edbInfo, "EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime", "Extra", "EdbNameEn", "UnitEn")
  158. if err != nil {
  159. return
  160. }
  161. var existCondition string
  162. var existPars []interface{}
  163. existCondition += " AND edb_info_id=? AND from_edb_info_id=? "
  164. existPars = append(existPars, edbInfo.EdbInfoId, req.FromEdbInfoId)
  165. //判断计算指标是否被更换
  166. count, err := GetEdbInfoCalculateCountByCondition(existCondition, existPars)
  167. if err != nil {
  168. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  169. return
  170. }
  171. if count > 0 { // 指标未被替换,无需删除关联数据
  172. // todo 频度被换了,需要重新计算
  173. if isRecalculate {
  174. err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", req.Extra)
  175. }
  176. return
  177. }
  178. //删除,计算指标关联的,基础指标的关联关系
  179. sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
  180. _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
  181. if err != nil {
  182. return
  183. }
  184. //清空原有数据
  185. sql = ` DELETE FROM ` + tableName + ` WHERE edb_info_id = ? `
  186. _, err = to.Raw(sql, edbInfo.EdbInfoId).Exec()
  187. if err != nil {
  188. return
  189. }
  190. //关联关系
  191. {
  192. calculateMappingItem := &EdbInfoCalculateMapping{
  193. EdbInfoCalculateMappingId: 0,
  194. EdbInfoId: edbInfo.EdbInfoId,
  195. Source: obj.GetSource(),
  196. SourceName: obj.GetSourceName(),
  197. EdbCode: edbInfo.EdbCode,
  198. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  199. FromEdbCode: fromEdbInfo.EdbCode,
  200. FromEdbName: fromEdbInfo.EdbName,
  201. FromSource: fromEdbInfo.Source,
  202. FromSourceName: fromEdbInfo.SourceName,
  203. FromTag: "",
  204. Sort: 1,
  205. CreateTime: time.Now(),
  206. ModifyTime: time.Now(),
  207. FromSubSource: fromEdbInfo.SubSource,
  208. }
  209. _, err = to.Insert(calculateMappingItem)
  210. if err != nil {
  211. return
  212. }
  213. }
  214. //计算数据
  215. err = obj.refresh(to, edbInfo.EdbInfoId, edbInfo.Source, edbInfo.SubSource, edbInfo, fromEdbInfo, edbInfo.EdbCode, "", edbInfo.Extra)
  216. return
  217. }
  218. // Refresh 刷新
  219. func (obj Ljz) Refresh(params RefreshParams) (err error, errMsg string) {
  220. calculateMapping, err := GetEdbInfoCalculateMappingDetail(params.EdbInfo.EdbInfoId)
  221. if err != nil {
  222. errMsg = "GetEdbInfoCalculateLjzDetail Err:" + err.Error()
  223. return
  224. }
  225. fromEdbInfo, err := GetEdbInfoById(calculateMapping.FromEdbInfoId)
  226. if err != nil {
  227. errMsg = "GetEdbInfoById Err:" + err.Error()
  228. return
  229. }
  230. o := orm.NewOrm()
  231. to, err := o.Begin()
  232. if err != nil {
  233. return
  234. }
  235. defer func() {
  236. if err != nil {
  237. _ = to.Rollback()
  238. fmt.Println(reflect.TypeOf(obj).Name(), ";Refresh,Err:"+err.Error())
  239. } else {
  240. _ = to.Commit()
  241. }
  242. }()
  243. // 计算数据
  244. err = obj.refresh(to, params.EdbInfo.EdbInfoId, params.EdbInfo.Source, params.EdbInfo.SubSource, params.EdbInfo, fromEdbInfo, params.EdbInfo.EdbCode, params.StartDate, params.EdbInfo.Extra)
  245. return
  246. }
  247. // GetSource 获取来源编码id
  248. func (obj Ljz) GetSource() int {
  249. return utils.DATA_SOURCE_CALCULATE_LJZ
  250. }
  251. // GetSourceName 获取来源名称
  252. func (obj Ljz) GetSourceName() string {
  253. return utils.DATA_SOURCE_NAME_CALCULATE_LJZ
  254. }
  255. // GetEdbType 获取指标类型
  256. func (obj Ljz) GetEdbType() int {
  257. return utils.CALCULATE_EDB_TYPE
  258. }
  259. func (obj Ljz) refresh(to orm.TxOrmer, edbInfoId, source, subSource int, edbInfo, fromEdbInfo *EdbInfo, edbCode, startDate string, extra string) (err error) {
  260. dataTableName := GetEdbDataTableName(source, subSource)
  261. edbInfoIdStr := strconv.Itoa(edbInfoId)
  262. //计算数据
  263. var condition string
  264. var pars []interface{}
  265. condition += " AND edb_info_id=? "
  266. pars = append(pars, fromEdbInfo.EdbInfoId)
  267. //if startDate != "" {
  268. // condition += " AND data_time>=? "
  269. // pars = append(pars, startDate)
  270. //}
  271. var isWeekData bool // 是否周度数据,如果是周度数据的话,是需要变频的,最后结果还需要除以7
  272. // 周度数据需要先变成日度的
  273. if fromEdbInfo.Frequency == `周度` {
  274. isWeekData = true
  275. }
  276. dataList, err := GetEdbDataListAllByTo(to, condition, pars, fromEdbInfo.Source, fromEdbInfo.SubSource, 1)
  277. if err != nil {
  278. return err
  279. }
  280. fromEdbDataMap := make(map[string]float64)
  281. if isWeekData {
  282. dataList, err = HandleDataByLinearRegression(dataList, fromEdbDataMap)
  283. if err != nil {
  284. return
  285. }
  286. }
  287. var lastValType int
  288. if extra != "" {
  289. var lastValConfig CalculateLjzEdbExtra
  290. err = json.Unmarshal([]byte(extra), &lastValConfig)
  291. if err != nil {
  292. err = fmt.Errorf("refreshAllCalculate,extra解析失败,Err:%s", err.Error())
  293. return
  294. }
  295. lastValType = lastValConfig.LastValType
  296. }
  297. //日度转周度:日期选周五,计算上周六到本周五的日度值的加总,最新日期为最新值对应的周五。
  298. //日度转旬度:日期选每个旬的最后一天,计算当旬所有日度值的加总,最新日期为最新值对应当旬的最后一天。
  299. //日度转月度:日期选每个月最后一天,计算当月所有日度值的加总,最新日期为最新值对应当月最后一天。
  300. //日度转季度、年度:方法类似转月度。
  301. //周度转月度/季度/年度:将周度值转成日度,空值用插值法插值,计算当月/当季/当年所有值的加总,然后除以7。
  302. //月度转季度/年度: 当季/当年月度值相加。
  303. dateList := make([]time.Time, 0)
  304. valueMap := make(map[time.Time]float64)
  305. switch edbInfo.Frequency {
  306. case "年度":
  307. yearMap := make(map[int]float64)
  308. yearList := make([]int, 0)
  309. var lastNewDate time.Time
  310. for _, item := range dataList {
  311. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  312. if tmpErr != nil {
  313. err = tmpErr
  314. return
  315. }
  316. year := itemDate.Year()
  317. yearVal, ok := yearMap[year]
  318. if ok {
  319. yearMap[year] = item.Value + yearVal
  320. } else {
  321. yearList = append(yearList, year)
  322. yearMap[year] = item.Value
  323. }
  324. }
  325. for _, v := range yearList {
  326. currTime := time.Date(v, 12, 31, 0, 0, 0, 0, time.Local)
  327. dateList = append(dateList, currTime)
  328. valueMap[currTime] = yearMap[v]
  329. lastNewDate = currTime
  330. }
  331. // 根据配置处理最新值, 1 表示均值填充
  332. if lastValType == 1 {
  333. lastItem := dataList[len(dataList)-1]
  334. // 最后一天的累计值
  335. initVal := valueMap[lastNewDate]
  336. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
  337. if tmpErr != nil {
  338. err = tmpErr
  339. return
  340. }
  341. //获取当季的总天数, 当季最后一天减去当季第一天
  342. // 获取当年的总天数
  343. startDateT := time.Date(itemDate.Year(), 1, 1, 0, 0, 0, 0, time.Local)
  344. allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
  345. //获取距离当年第一天的天数
  346. days := int(itemDate.Sub(startDateT).Hours() / 24)
  347. daysT := decimal.NewFromInt(int64(days))
  348. initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
  349. val, _ := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
  350. fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
  351. valueMap[lastNewDate] = val
  352. }
  353. case "半年度":
  354. yearMonthMap := make(map[string]float64)
  355. yearMonthList := make([]string, 0)
  356. var lastNewDate time.Time
  357. for _, item := range dataList {
  358. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  359. if tmpErr != nil {
  360. err = tmpErr
  361. return
  362. }
  363. year := itemDate.Year()
  364. var tmpK string
  365. if itemDate.Month() <= 6 {
  366. tmpK = fmt.Sprint(year, "06")
  367. } else {
  368. tmpK = fmt.Sprint(year, "12")
  369. }
  370. yearVal, ok := yearMonthMap[tmpK]
  371. if ok {
  372. yearMonthMap[tmpK] = item.Value + yearVal
  373. } else {
  374. yearMonthList = append(yearMonthList, tmpK)
  375. yearMonthMap[tmpK] = item.Value
  376. }
  377. }
  378. for _, v := range yearMonthList {
  379. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  380. if tmpErr != nil {
  381. err = tmpErr
  382. return
  383. }
  384. currTime = currTime.AddDate(0, 1, -1)
  385. dateList = append(dateList, currTime)
  386. valueMap[currTime] = yearMonthMap[v]
  387. lastNewDate = currTime
  388. }
  389. // 根据配置处理最新值, 1 表示均值填充
  390. if lastValType == 1 {
  391. lastItem := dataList[len(dataList)-1]
  392. // 最后一天的累计值
  393. initVal := valueMap[lastNewDate]
  394. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
  395. if tmpErr != nil {
  396. err = tmpErr
  397. return
  398. }
  399. //获取上半年或者下半年的总天数, 半年度最后一天减去半年度第一天
  400. startDateT := lastNewDate.AddDate(0, -6, 0)
  401. allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
  402. //获取距离半年度第一天的天数
  403. days := int(itemDate.Sub(startDateT).Hours() / 24)
  404. daysT := decimal.NewFromInt(int64(days))
  405. initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
  406. val, _ := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
  407. fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
  408. valueMap[lastNewDate] = val
  409. }
  410. case "季度":
  411. yearMonthMap := make(map[string]float64)
  412. yearMonthList := make([]string, 0)
  413. var lastNewDate time.Time
  414. for _, item := range dataList {
  415. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  416. if tmpErr != nil {
  417. err = tmpErr
  418. return
  419. }
  420. year := itemDate.Year()
  421. var tmpK string
  422. if itemDate.Month() <= 3 {
  423. tmpK = fmt.Sprint(year, "03")
  424. } else if itemDate.Month() <= 6 {
  425. tmpK = fmt.Sprint(year, "06")
  426. } else if itemDate.Month() <= 9 {
  427. tmpK = fmt.Sprint(year, "09")
  428. } else {
  429. tmpK = fmt.Sprint(year, "12")
  430. }
  431. yearVal, ok := yearMonthMap[tmpK]
  432. if ok {
  433. yearMonthMap[tmpK] = item.Value + yearVal
  434. } else {
  435. yearMonthList = append(yearMonthList, tmpK)
  436. yearMonthMap[tmpK] = item.Value
  437. }
  438. }
  439. for _, v := range yearMonthList {
  440. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  441. if tmpErr != nil {
  442. err = tmpErr
  443. return
  444. }
  445. currTime = currTime.AddDate(0, 1, -1)
  446. dateList = append(dateList, currTime)
  447. valueMap[currTime] = yearMonthMap[v]
  448. lastNewDate = currTime
  449. }
  450. // 根据配置处理最新值, 1 表示均值填充
  451. if lastValType == 1 {
  452. lastItem := dataList[len(dataList)-1]
  453. // 最后一天的累计值
  454. initVal := valueMap[lastNewDate]
  455. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
  456. if tmpErr != nil {
  457. err = tmpErr
  458. return
  459. }
  460. //获取当季的总天数, 当季最后一天减去当季第一天
  461. startDateT := lastNewDate.AddDate(0, -3, 0)
  462. allDays := int(lastNewDate.Sub(startDateT).Hours() / 24)
  463. //获取距离当季第一天的天数
  464. days := int(itemDate.Sub(startDateT).Hours() / 24)
  465. daysT := decimal.NewFromInt(int64(days))
  466. initValAndMonthT := decimal.NewFromFloat(initVal * float64(allDays))
  467. val, _ := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
  468. fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", allDays, initVal, days, val)
  469. valueMap[lastNewDate] = val
  470. }
  471. case "月度":
  472. yearMonthMap := make(map[string]float64)
  473. yearMonthList := make([]string, 0)
  474. var lastNewDate time.Time
  475. for _, item := range dataList {
  476. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  477. if tmpErr != nil {
  478. err = tmpErr
  479. return
  480. }
  481. year := itemDate.Year()
  482. var tmpK string
  483. tmpK = fmt.Sprint(year*100 + int(itemDate.Month()))
  484. yearVal, ok := yearMonthMap[tmpK]
  485. if ok {
  486. yearMonthMap[tmpK] = item.Value + yearVal
  487. fmt.Printf("%s 累加值:%f = %.6f + %.6f\n", tmpK, yearMonthMap[tmpK], item.Value, yearVal)
  488. } else {
  489. yearMonthList = append(yearMonthList, tmpK)
  490. yearMonthMap[tmpK] = item.Value
  491. }
  492. }
  493. for _, v := range yearMonthList {
  494. currTime, tmpErr := time.ParseInLocation(utils.FormatYearMonthUnSpace, v, time.Local)
  495. if tmpErr != nil {
  496. err = tmpErr
  497. return
  498. }
  499. currTime = currTime.AddDate(0, 1, -1)
  500. dateList = append(dateList, currTime)
  501. valueMap[currTime] = yearMonthMap[v]
  502. lastNewDate = currTime
  503. }
  504. // 根据配置处理最新值, 1 表示均值填充
  505. if lastValType == 1 {
  506. lastItem := dataList[len(dataList)-1]
  507. // 最后一天的累计值
  508. initVal := valueMap[lastNewDate]
  509. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, lastItem.DataTime, time.Local)
  510. if tmpErr != nil {
  511. err = tmpErr
  512. return
  513. }
  514. //获取当月的天数
  515. monthDays := lastNewDate.Day()
  516. //获取自然日的天数
  517. days := itemDate.Day()
  518. daysT := decimal.NewFromInt(int64(days))
  519. initValAndMonthT := decimal.NewFromFloat(initVal * float64(monthDays))
  520. val, _ := initValAndMonthT.Div(daysT).RoundCeil(4).Float64()
  521. fmt.Printf("最新值 计算公式:%d*%f/(%d) = %f\n", monthDays, initVal, days, val)
  522. valueMap[lastNewDate] = val
  523. }
  524. case "旬度":
  525. tmpDateDataMap := make(map[time.Time]float64)
  526. tmpDateList := make([]time.Time, 0)
  527. for _, item := range dataList {
  528. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  529. if tmpErr != nil {
  530. err = tmpErr
  531. return
  532. }
  533. dayInt := itemDate.Year()*100 + int(itemDate.Month())
  534. var currTime time.Time
  535. if itemDate.Day() <= 10 {
  536. //本月上旬
  537. tmpK := fmt.Sprint(dayInt, "10")
  538. currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
  539. if err != nil {
  540. return
  541. }
  542. } else if itemDate.Day() <= 20 {
  543. // 本月中旬
  544. tmpK := fmt.Sprint(dayInt, "20")
  545. currTime, err = time.ParseInLocation(utils.FormatDateUnSpace, tmpK, time.Local)
  546. if err != nil {
  547. return
  548. }
  549. } else {
  550. // 本月下旬
  551. currTime, err = time.ParseInLocation(utils.FormatYearMonthUnSpace, fmt.Sprint(dayInt), time.Local)
  552. if err != nil {
  553. return
  554. }
  555. currTime = currTime.AddDate(0, 1, -1)
  556. }
  557. yearVal, ok := tmpDateDataMap[currTime]
  558. if ok {
  559. tmpDateDataMap[currTime] = item.Value + yearVal
  560. } else {
  561. tmpDateList = append(tmpDateList, currTime)
  562. tmpDateDataMap[currTime] = item.Value
  563. }
  564. }
  565. for _, currTime := range tmpDateList {
  566. dateList = append(dateList, currTime)
  567. valueMap[currTime] = tmpDateDataMap[currTime]
  568. }
  569. case "周度":
  570. tmpDateDataMap := make(map[time.Time]float64)
  571. tmpDateList := make([]time.Time, 0)
  572. for _, item := range dataList {
  573. itemDate, tmpErr := time.ParseInLocation(utils.FormatDate, item.DataTime, time.Local)
  574. if tmpErr != nil {
  575. err = tmpErr
  576. return
  577. }
  578. var currTime time.Time
  579. // 周六周日,这是下一个周五的数据
  580. if itemDate.Weekday() == 0 {
  581. currTime = itemDate.AddDate(0, 0, 5)
  582. } else if itemDate.Weekday() == 6 {
  583. currTime = itemDate.AddDate(0, 0, 6)
  584. } else {
  585. currTime = itemDate.AddDate(0, 0, 5-int(itemDate.Weekday()))
  586. }
  587. yearVal, ok := tmpDateDataMap[currTime]
  588. if ok {
  589. tmpDateDataMap[currTime] = item.Value + yearVal
  590. } else {
  591. tmpDateList = append(tmpDateList, currTime)
  592. tmpDateDataMap[currTime] = item.Value
  593. }
  594. }
  595. for _, currTime := range tmpDateList {
  596. dateList = append(dateList, currTime)
  597. valueMap[currTime] = tmpDateDataMap[currTime]
  598. }
  599. }
  600. //获取指标所有数据
  601. existDataList := make([]*EdbData, 0)
  602. sql := `SELECT * FROM %s WHERE edb_info_id=? `
  603. sql = fmt.Sprintf(sql, dataTableName)
  604. _, err = to.Raw(sql, edbInfoId).QueryRows(&existDataList)
  605. if err != nil {
  606. return err
  607. }
  608. existDataMap := make(map[string]string)
  609. removeDataTimeMap := make(map[string]int) //需要移除的日期数据
  610. for _, v := range existDataList {
  611. existDataMap[v.DataTime] = v.Value
  612. removeDataTimeMap[v.DataTime] = 1
  613. }
  614. needAddDateMap := make(map[time.Time]int)
  615. addSql := ` INSERT INTO ` + dataTableName + `(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  616. var isAdd bool
  617. for _, currTime := range dateList {
  618. currDateStr := currTime.Format(utils.FormatDate)
  619. existVal, ok := existDataMap[currDateStr]
  620. tmpVal, ok2 := valueMap[currTime]
  621. if !ok2 {
  622. err = errors.New("数据异常,date:" + currDateStr)
  623. return
  624. }
  625. var saveValue string
  626. if isWeekData { //周度指标转的话,最后结果要除以7
  627. saveValue = decimal.NewFromFloat(tmpVal).Div(decimal.NewFromInt(7)).Round(4).String()
  628. } else {
  629. saveValue = decimal.NewFromFloat(tmpVal).Round(4).String()
  630. }
  631. // 如果库中已经存在该数据的话,那么就进行值的变更操作
  632. if ok {
  633. //校验待删除日期数据里面是否存在该元素,如果存在的话,那么移除该日期
  634. delete(removeDataTimeMap, currDateStr)
  635. if existVal != saveValue {
  636. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  637. sql = fmt.Sprintf(sql, dataTableName)
  638. _, err = to.Raw(sql, saveValue, edbInfoId, currDateStr).Exec()
  639. if err != nil {
  640. return
  641. }
  642. }
  643. continue
  644. }
  645. // 库中不存在该日期的数据
  646. timestamp := currTime.UnixNano() / 1e6
  647. timeStr := fmt.Sprintf("%d", timestamp)
  648. if _, existOk := needAddDateMap[currTime]; !existOk {
  649. addSql += GetAddSql(edbInfoIdStr, edbCode, currDateStr, timeStr, saveValue)
  650. isAdd = true
  651. }
  652. needAddDateMap[currTime] = 1
  653. }
  654. //删除已经不存在的指标数据(由于该指标当日的数据删除了)
  655. {
  656. removeDateList := make([]string, 0)
  657. for dateTime := range removeDataTimeMap {
  658. removeDateList = append(removeDateList, dateTime)
  659. }
  660. removeNum := len(removeDateList)
  661. if removeNum > 0 {
  662. sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (`+utils.GetOrmInReplace(removeNum)+`) `, dataTableName)
  663. _, err = to.Raw(sql, edbInfo.EdbInfoId, removeDateList).Exec()
  664. if err != nil {
  665. fmt.Println(reflect.TypeOf(obj).Name(), " add data ;delete Err", err.Error())
  666. err = fmt.Errorf("删除不存在的指标数据失败,Err:" + err.Error())
  667. return
  668. }
  669. }
  670. }
  671. if isAdd {
  672. addSql = strings.TrimRight(addSql, ",")
  673. _, err = to.Raw(addSql).Exec()
  674. if err != nil {
  675. fmt.Println(reflect.TypeOf(obj).Name(), " add data Err", err.Error())
  676. return
  677. }
  678. }
  679. return
  680. }