edb_data_calculate_ljz.go 22 KB

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