predict_edb_data_calculate_tbz.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713
  1. package models
  2. import (
  3. "errors"
  4. "eta_gn/eta_index_lib/global"
  5. "eta_gn/eta_index_lib/utils"
  6. "fmt"
  7. "github.com/shopspring/decimal"
  8. "gorm.io/gorm"
  9. "reflect"
  10. "strconv"
  11. "strings"
  12. "time"
  13. )
  14. // PredictTb 预测同比值
  15. type PredictTb struct {
  16. }
  17. // Add 添加
  18. func (obj PredictTb) Add(params BatchSaveCalculateBatchParams) (edbInfo *EdbInfo, latestDateStr string, latestValue float64, err error, errMsg string) {
  19. req := params.Req
  20. fromEdbInfo := params.FromEdbInfo
  21. edbCode := params.EdbCode
  22. to := global.DEFAULT_DmSQL.Begin()
  23. defer func() {
  24. if err != nil {
  25. to.Rollback()
  26. } else {
  27. to.Commit()
  28. }
  29. }()
  30. edbInfo = &EdbInfo{
  31. //EdbInfoId: 0,
  32. SourceName: obj.GetSourceName(),
  33. Source: obj.GetSource(),
  34. EdbCode: edbCode,
  35. EdbName: req.EdbName,
  36. EdbNameSource: req.EdbName,
  37. Frequency: req.Frequency,
  38. Unit: req.Unit,
  39. StartDate: "",
  40. EndDate: "",
  41. ClassifyId: req.ClassifyId,
  42. SysUserId: params.SysUserId,
  43. SysUserRealName: params.SysUserRealName,
  44. UniqueCode: params.UniqueCode,
  45. CreateTime: time.Now(),
  46. ModifyTime: time.Now(),
  47. MinValue: 0,
  48. MaxValue: 0,
  49. CalculateFormula: req.Formula,
  50. EdbType: 2,
  51. Sort: GetAddEdbMaxSortByClassifyId(req.ClassifyId, utils.PREDICT_EDB_INFO_TYPE),
  52. MoveType: 0,
  53. MoveFrequency: "",
  54. NoUpdate: 0,
  55. ServerUrl: "",
  56. EdbInfoType: 1,
  57. EdbNameEn: req.EdbName,
  58. UnitEn: req.Unit,
  59. LatestDate: "",
  60. LatestValue: 0,
  61. ChartImage: "",
  62. Calendar: "",
  63. }
  64. tmpErr := to.Create(edbInfo).Error
  65. if tmpErr != nil {
  66. err = tmpErr
  67. return
  68. }
  69. //关联关系
  70. {
  71. calculateMappingItem := new(EdbInfoCalculateMapping)
  72. calculateMappingItem.CreateTime = time.Now()
  73. calculateMappingItem.ModifyTime = time.Now()
  74. calculateMappingItem.Sort = 1
  75. calculateMappingItem.EdbCode = edbCode
  76. calculateMappingItem.EdbInfoId = edbInfo.EdbInfoId
  77. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  78. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  79. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  80. calculateMappingItem.FromSource = fromEdbInfo.Source
  81. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  82. calculateMappingItem.FromTag = ""
  83. calculateMappingItem.Source = edbInfo.Source
  84. calculateMappingItem.SourceName = edbInfo.SourceName
  85. err = to.Create(calculateMappingItem).Error
  86. if err != nil {
  87. return
  88. }
  89. }
  90. //计算数据
  91. latestDateStr, latestValue, err = obj.refresh(to, edbInfo, fromEdbInfo, edbInfo.EdbCode, "")
  92. return
  93. }
  94. // Edit 编辑
  95. func (obj PredictTb) Edit(params BatchSaveCalculateBatchParams) (latestDateStr string, latestValue float64, err error, errMsg string) {
  96. req := params.Req
  97. edbInfo := params.EdbInfo
  98. fromEdbInfo := params.FromEdbInfo
  99. latestDateStr = edbInfo.LatestDate
  100. latestValue = edbInfo.LatestValue
  101. if fromEdbInfo.Frequency != req.Frequency {
  102. errMsg = "生成指标频度与原指标频度不同"
  103. err = errors.New(errMsg)
  104. return
  105. }
  106. if edbInfo.EdbInfoType != 1 {
  107. errMsg = `该指标不是预测指标`
  108. err = errors.New(errMsg)
  109. return
  110. }
  111. latestDateStr = edbInfo.LatestDate
  112. latestValue = edbInfo.LatestValue
  113. to := global.DEFAULT_DmSQL.Begin()
  114. defer func() {
  115. if err != nil {
  116. to.Rollback()
  117. } else {
  118. to.Commit()
  119. }
  120. }()
  121. tableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
  122. //修改指标信息
  123. switch params.Lang {
  124. case utils.EnLangVersion:
  125. edbInfo.EdbNameEn = req.EdbName
  126. edbInfo.UnitEn = req.Unit
  127. default:
  128. edbInfo.EdbName = req.EdbName
  129. edbInfo.Unit = req.Unit
  130. edbInfo.EdbNameSource = req.EdbName
  131. }
  132. edbInfo.Frequency = req.Frequency
  133. edbInfo.ClassifyId = req.ClassifyId
  134. edbInfo.ModifyTime = time.Now()
  135. err = to.Model(edbInfo).Select([]string{"EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime", "EdbNameEn", "UnitEn"}).Updates(edbInfo).Error
  136. if err != nil {
  137. return
  138. }
  139. //判断计算指标是否被更换
  140. var existCondition string
  141. var existPars []interface{}
  142. existCondition += " AND edb_info_id=? AND from_edb_info_id=? "
  143. existPars = append(existPars, edbInfo.EdbInfoId, fromEdbInfo.EdbInfoId)
  144. var count int
  145. count, err = GetEdbInfoCalculateCountByCondition(existCondition, existPars)
  146. if err != nil {
  147. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  148. return
  149. }
  150. if count > 0 { // 指标未被替换,无需重新计算
  151. return
  152. }
  153. //删除,计算指标关联的,基础指标的关联关系
  154. sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
  155. err = to.Exec(sql, edbInfo.EdbInfoId).Error
  156. if err != nil {
  157. return
  158. }
  159. // 删除该指标所有的同比数据
  160. sql = ` DELETE FROM ` + tableName + ` WHERE edb_info_id = ? `
  161. err = to.Exec(sql, edbInfo.EdbInfoId).Error
  162. if err != nil {
  163. return
  164. }
  165. //关联关系
  166. {
  167. calculateMappingItem := &EdbInfoCalculateMapping{
  168. EdbInfoCalculateMappingId: 0,
  169. EdbInfoId: edbInfo.EdbInfoId,
  170. Source: obj.GetSource(),
  171. SourceName: obj.GetSourceName(),
  172. EdbCode: edbInfo.EdbCode,
  173. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  174. FromEdbCode: fromEdbInfo.EdbCode,
  175. FromEdbName: fromEdbInfo.EdbName,
  176. FromSource: fromEdbInfo.Source,
  177. FromSourceName: fromEdbInfo.SourceName,
  178. FromTag: "",
  179. Sort: 1,
  180. CreateTime: time.Now(),
  181. ModifyTime: time.Now(),
  182. }
  183. err = to.Create(calculateMappingItem).Error
  184. if err != nil {
  185. return
  186. }
  187. }
  188. //计算数据
  189. latestDateStr, latestValue, err = obj.refresh(to, edbInfo, fromEdbInfo, edbInfo.EdbCode, "")
  190. return
  191. }
  192. // Refresh 刷新
  193. func (obj PredictTb) Refresh(params RefreshParams) (latestDateStr string, latestValue float64, err error, errMsg string) {
  194. calculateMapping, err := GetEdbInfoCalculateMappingDetail(params.EdbInfo.EdbInfoId)
  195. if err != nil {
  196. return
  197. }
  198. fromEdbInfo, err := GetEdbInfoById(calculateMapping.FromEdbInfoId)
  199. if err != nil {
  200. errMsg = "GetEdbInfoById Err:" + err.Error()
  201. return
  202. }
  203. to := global.DEFAULT_DmSQL.Begin()
  204. defer func() {
  205. if err != nil {
  206. to.Rollback()
  207. } else {
  208. to.Commit()
  209. }
  210. }()
  211. // 计算数据
  212. latestDateStr, latestValue, err = obj.refresh(to, params.EdbInfo, fromEdbInfo, params.EdbInfo.EdbCode, params.StartDate)
  213. return
  214. }
  215. // GetSource 获取来源编码id
  216. func (obj PredictTb) GetSource() int {
  217. return utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ
  218. }
  219. // GetSourceName 获取来源名称
  220. func (obj PredictTb) GetSourceName() string {
  221. return utils.DATA_SOURCE_NAME_PREDICT_CALCULATE_TBZ
  222. }
  223. func (obj PredictTb) refresh(to *gorm.DB, edbInfo, fromEdbInfo *EdbInfo, edbCode, startDate string) (latestDateStr string, latestValue float64, err error) {
  224. edbInfoId := edbInfo.EdbInfoId
  225. dataTableName := GetEdbDataTableName(edbInfo.Source, edbInfo.SubSource)
  226. edbInfoIdStr := strconv.Itoa(edbInfoId)
  227. // 获取关联指标数据
  228. dataList, err := GetPredictEdbDataListAllByStartDate(fromEdbInfo, 0, "")
  229. if err != nil {
  230. return
  231. }
  232. // 来源指标数据参与计算
  233. calculateDataList, err := EdbInfoSearchDataToEdbInfoData(dataList)
  234. if err != nil {
  235. return
  236. }
  237. baseCalculate := BaseCalculate{
  238. DataList: calculateDataList,
  239. Frequency: fromEdbInfo.Frequency,
  240. Source: 3,
  241. }
  242. dateDataMap, err, _ := baseCalculate.Tbz()
  243. if err != nil {
  244. return
  245. }
  246. // 真实数据的最后日期
  247. latestDateStr = fromEdbInfo.LatestDate
  248. //获取指标所有数据
  249. existDataList := make([]*EdbData, 0)
  250. //fmt.Println("dataTableName:", dataTableName)
  251. sql := `SELECT * FROM %s WHERE edb_info_id=? `
  252. sql = fmt.Sprintf(sql, dataTableName)
  253. err = to.Raw(sql, edbInfoId).Scan(&existDataList).Error
  254. if err != nil {
  255. return
  256. }
  257. existDataMap := make(map[string]string)
  258. removeDataTimeMap := make(map[string]int) //需要移除的日期数据
  259. for _, v := range existDataList {
  260. existDataMap[v.DataTime] = v.Value
  261. removeDataTimeMap[v.DataTime] = 1
  262. }
  263. //fmt.Println("existDataMap:", existDataMap)
  264. addSql := ` INSERT INTO edb_data_predict_calculate_tbz (edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  265. var isAdd bool
  266. existAddDataMap := make(map[string]string)
  267. for currentDate, valFloat := range dateDataMap {
  268. currentDateStr := currentDate.Format(utils.FormatDate)
  269. timestamp := currentDate.UnixNano() / 1e6
  270. timestampStr := fmt.Sprintf("%d", timestamp)
  271. val := decimal.NewFromFloat(valFloat).String()
  272. //校验待删除日期数据里面是否存在该元素,如果存在的话,那么移除该日期
  273. delete(removeDataTimeMap, currentDateStr)
  274. if existVal, ok := existDataMap[currentDateStr]; !ok {
  275. if _, existOk := existAddDataMap[currentDateStr]; !existOk {
  276. addSql += GetAddSql(edbInfoIdStr, edbCode, currentDateStr, timestampStr, val)
  277. isAdd = true
  278. }
  279. existAddDataMap[currentDateStr] = currentDateStr
  280. } else {
  281. existValDecimal, tmpErr := decimal.NewFromString(existVal)
  282. if tmpErr != nil {
  283. err = tmpErr
  284. return
  285. }
  286. existStr := existValDecimal.String()
  287. if existStr != val {
  288. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  289. sql = fmt.Sprintf(sql, dataTableName)
  290. err = to.Exec(sql, val, edbInfoId, currentDateStr).Error
  291. if err != nil {
  292. return
  293. }
  294. }
  295. }
  296. // 实际数据的值
  297. if fromEdbInfo.LatestDate == currentDateStr {
  298. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  299. if tmpErr != nil {
  300. return
  301. }
  302. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  303. }
  304. }
  305. //删除已经不存在的指标数据(由于该指标当日的数据删除了)
  306. {
  307. removeDateList := make([]string, 0)
  308. for dateTime := range removeDataTimeMap {
  309. removeDateList = append(removeDateList, dateTime)
  310. }
  311. removeNum := len(removeDateList)
  312. if removeNum > 0 {
  313. sql := fmt.Sprintf(` DELETE FROM %s WHERE edb_info_id = ? and data_time in (`+utils.GetOrmInReplace(removeNum)+`) `, dataTableName)
  314. err = to.Exec(sql, edbInfo.EdbInfoId, removeDateList).Error
  315. if err != nil {
  316. fmt.Println(reflect.TypeOf(obj).Name(), " add data ;delete Err", err.Error())
  317. err = fmt.Errorf("删除不存在的指标数据失败,Err:" + err.Error())
  318. return
  319. }
  320. }
  321. }
  322. // 批量插入
  323. if isAdd {
  324. addSql = strings.TrimRight(addSql, ",")
  325. err = to.Exec(addSql).Error
  326. if err != nil {
  327. return
  328. }
  329. }
  330. //确定实际数据的最终值
  331. {
  332. finalLast, tmpErr := GetFinalLastByTo(to, edbInfoId, edbInfo.Source, edbInfo.SubSource, fromEdbInfo.LatestDate)
  333. if tmpErr != nil && tmpErr.Error() != utils.ErrNoRow() {
  334. return
  335. }
  336. if tmpErr == nil {
  337. latestDateStr = finalLast.DataTime
  338. latestValue = finalLast.Value
  339. }
  340. }
  341. return
  342. }
  343. // SavePredictCalculateTbz 新增/编辑 预测同比指标
  344. func SavePredictCalculateTbz(reqEdbInfoId, classifyId int, edbName, frequency, unit, formula string, fromEdbInfo *EdbInfo, edbCode, uniqueCode string, sysUserId int, sysUserRealName string) (edbInfo *EdbInfo, latestDateStr string, latestValue float64, err error, errMsg string) {
  345. to := global.DEFAULT_DmSQL.Begin()
  346. defer func() {
  347. if err != nil {
  348. to.Rollback()
  349. } else {
  350. to.Commit()
  351. }
  352. }()
  353. if reqEdbInfoId <= 0 {
  354. edbInfo = &EdbInfo{
  355. //EdbInfoId: 0,
  356. EdbInfoType: 1,
  357. SourceName: "预测同比",
  358. Source: utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ,
  359. EdbCode: edbCode,
  360. EdbName: edbName,
  361. EdbNameSource: edbName,
  362. Frequency: frequency,
  363. Unit: unit,
  364. //StartDate: "",
  365. //EndDate: "",
  366. ClassifyId: classifyId,
  367. SysUserId: sysUserId,
  368. SysUserRealName: sysUserRealName,
  369. UniqueCode: uniqueCode,
  370. CreateTime: time.Now(),
  371. ModifyTime: time.Now(),
  372. MinValue: 0,
  373. MaxValue: 0,
  374. CalculateFormula: formula,
  375. EdbType: 2,
  376. Sort: 0,
  377. MoveType: 0,
  378. MoveFrequency: "",
  379. NoUpdate: 0,
  380. ServerUrl: "",
  381. EdbNameEn: "",
  382. UnitEn: "",
  383. LatestDate: "",
  384. LatestValue: 0,
  385. ChartImage: "",
  386. }
  387. tmpErr := to.Create(edbInfo).Error
  388. if tmpErr != nil {
  389. err = tmpErr
  390. return
  391. }
  392. // 添加关联关系
  393. {
  394. calculateMappingItem := &EdbInfoCalculateMapping{
  395. EdbInfoCalculateMappingId: 0,
  396. EdbInfoId: edbInfo.EdbInfoId,
  397. Source: edbInfo.Source,
  398. SourceName: edbInfo.SourceName,
  399. EdbCode: edbCode,
  400. FromEdbInfoId: fromEdbInfo.EdbInfoId,
  401. FromEdbCode: fromEdbInfo.EdbCode,
  402. FromEdbName: fromEdbInfo.EdbName,
  403. FromSource: fromEdbInfo.Source,
  404. FromSourceName: fromEdbInfo.SourceName,
  405. FromTag: "",
  406. Sort: 1,
  407. CreateTime: time.Now(),
  408. ModifyTime: time.Now(),
  409. }
  410. err = to.Create(calculateMappingItem).Error
  411. if err != nil {
  412. return
  413. }
  414. }
  415. } else {
  416. edbInfo, err = GetEdbInfoById(reqEdbInfoId)
  417. if err != nil {
  418. if err.Error() == utils.ErrNoRow() {
  419. errMsg = `获取指标信息失败`
  420. }
  421. return
  422. }
  423. if edbInfo.EdbInfoType != 1 {
  424. errMsg = `该指标不是预测指标`
  425. err = errors.New(errMsg)
  426. return
  427. }
  428. latestDateStr = edbInfo.LatestDate
  429. latestValue = edbInfo.LatestValue
  430. //修改指标信息
  431. edbInfo.EdbName = edbName
  432. edbInfo.EdbNameSource = edbName
  433. edbInfo.Frequency = frequency
  434. edbInfo.Unit = unit
  435. edbInfo.ClassifyId = classifyId
  436. edbInfo.ModifyTime = time.Now()
  437. err = to.Model(edbInfo).Select([]string{"EdbName", "EdbNameSource", "Frequency", "Unit", "ClassifyId", "ModifyTime"}).Updates(edbInfo).Error
  438. if err != nil {
  439. return
  440. }
  441. //判断计算指标是否被更换
  442. var existCondition string
  443. var existPars []interface{}
  444. existCondition += " AND edb_info_id=? AND from_edb_info_id=? "
  445. existPars = append(existPars, edbInfo.EdbInfoId, fromEdbInfo.EdbInfoId)
  446. var count int
  447. count, err = GetEdbInfoCalculateCountByCondition(existCondition, existPars)
  448. if err != nil {
  449. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  450. return
  451. }
  452. if count > 0 { // 指标未被替换,无需重新计算
  453. return
  454. }
  455. //删除,计算指标关联的,基础指标的关联关系
  456. sql := ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
  457. err = to.Exec(sql, edbInfo.EdbInfoId).Error
  458. if err != nil {
  459. return
  460. }
  461. // 删除该指标所有的同比数据
  462. dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_PREDICT_CALCULATE_TBZ, utils.DATA_SUB_SOURCE_EDB)
  463. deleteSql := ` DELETE FROM %s WHERE edb_info_id=? `
  464. deleteSql = fmt.Sprintf(deleteSql, dataTableName)
  465. err = to.Exec(deleteSql, edbInfo.EdbInfoId).Error
  466. if err != nil {
  467. return
  468. }
  469. }
  470. edbInfoIdStr := strconv.Itoa(edbInfo.EdbInfoId)
  471. // 获取源指标数据
  472. dataList, err := GetPredictEdbDataListAll(fromEdbInfo, 0)
  473. if err != nil {
  474. return
  475. }
  476. latestDateStr = fromEdbInfo.LatestDate
  477. // 计算数据
  478. var dateArr []string
  479. dataMap := make(map[string]*EdbInfoSearchData)
  480. for _, v := range dataList {
  481. dateArr = append(dateArr, v.DataTime)
  482. dataMap[v.DataTime] = v
  483. }
  484. addSql := ` INSERT INTO edb_data_predict_calculate_tbz(edb_info_id,edb_code,data_time,value,create_time,modify_time,data_timestamp) values `
  485. var isAdd bool
  486. existMap := make(map[string]string)
  487. for _, av := range dateArr {
  488. //fmt.Println(ak, av)
  489. currentItem := dataMap[av]
  490. if currentItem != nil {
  491. //当前日期
  492. currentDate, tmpErr := time.ParseInLocation(utils.FormatDate, av, time.Local)
  493. if tmpErr != nil {
  494. err = tmpErr
  495. return
  496. }
  497. //上一年的日期
  498. preDate := currentDate.AddDate(-1, 0, 0)
  499. preDateStr := preDate.Format(utils.FormatDate)
  500. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期找到
  501. //dataTime, _ := time.ParseInLocation(utils.FormatDate, date, time.Local)
  502. if _, ok := existMap[edbCode+av]; !ok {
  503. timestamp := currentDate.UnixNano() / 1e6
  504. timestampStr := fmt.Sprintf("%d", timestamp)
  505. val := TbzDiv(currentItem.Value, findItem.Value)
  506. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  507. isAdd = true
  508. utils.FileLog.Info("同期找到:" + av + ";" + preDateStr)
  509. // 实际数据的值
  510. if fromEdbInfo.LatestDate == av {
  511. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  512. if tmpErr != nil {
  513. return
  514. }
  515. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  516. }
  517. }
  518. existMap[edbCode+av] = av
  519. } else {
  520. if fromEdbInfo.Frequency == "月度" { //向上和向下,各找一个月
  521. nextDateDay := preDate
  522. preDateDay := preDate
  523. for i := 0; i <= 35; i++ {
  524. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  525. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  526. if _, ok := existMap[edbCode+av]; !ok {
  527. timestamp := currentDate.UnixNano() / 1e6
  528. timestampStr := fmt.Sprintf("%d", timestamp)
  529. val := TbzDiv(currentItem.Value, findItem.Value)
  530. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  531. isAdd = true
  532. // 实际数据的值
  533. if fromEdbInfo.LatestDate == av {
  534. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  535. if tmpErr != nil {
  536. return
  537. }
  538. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  539. }
  540. }
  541. existMap[edbCode+av] = av
  542. break
  543. } else {
  544. preDateDayStr := preDateDay.Format(utils.FormatDate)
  545. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  546. if _, ok := existMap[edbCode+av]; !ok {
  547. timestamp := currentDate.UnixNano() / 1e6
  548. timestampStr := fmt.Sprintf("%d", timestamp)
  549. val := TbzDiv(currentItem.Value, findItem.Value)
  550. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  551. isAdd = true
  552. // 实际数据的值
  553. if fromEdbInfo.LatestDate == av {
  554. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  555. if tmpErr != nil {
  556. return
  557. }
  558. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  559. }
  560. }
  561. existMap[edbCode+av] = av
  562. break
  563. }
  564. }
  565. nextDateDay = nextDateDay.AddDate(0, 0, 1)
  566. preDateDay = preDateDay.AddDate(0, 0, -1)
  567. }
  568. } else if fromEdbInfo.Frequency == "季度" || fromEdbInfo.Frequency == "年度" {
  569. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  570. if _, ok := existMap[edbCode+av]; !ok {
  571. timestamp := currentDate.UnixNano() / 1e6
  572. timestampStr := fmt.Sprintf("%d", timestamp)
  573. val := TbzDiv(currentItem.Value, findItem.Value)
  574. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  575. isAdd = true
  576. // 实际数据的值
  577. if fromEdbInfo.LatestDate == av {
  578. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  579. if tmpErr != nil {
  580. return
  581. }
  582. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  583. }
  584. }
  585. existMap[edbCode+av] = av
  586. break
  587. }
  588. } else {
  589. nextDateDay := preDate
  590. preDateDay := preDate
  591. for i := 0; i < 35; i++ {
  592. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  593. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  594. if _, ok := existMap[edbCode+av]; !ok {
  595. timestamp := currentDate.UnixNano() / 1e6
  596. timestampStr := fmt.Sprintf("%d", timestamp)
  597. val := TbzDiv(currentItem.Value, findItem.Value)
  598. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  599. isAdd = true
  600. // 实际数据的值
  601. if fromEdbInfo.LatestDate == av {
  602. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  603. if tmpErr != nil {
  604. return
  605. }
  606. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  607. }
  608. }
  609. existMap[edbCode+av] = av
  610. break
  611. } else {
  612. preDateDayStr := preDateDay.Format(utils.FormatDate)
  613. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  614. if _, ok := existMap[edbCode+av]; !ok {
  615. timestamp := currentDate.UnixNano() / 1e6
  616. timestampStr := fmt.Sprintf("%d", timestamp)
  617. val := TbzDiv(currentItem.Value, findItem.Value)
  618. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  619. isAdd = true
  620. // 实际数据的值
  621. if fromEdbInfo.LatestDate == av {
  622. latestValueDecimal, tmpErr := decimal.NewFromString(val)
  623. if tmpErr != nil {
  624. return
  625. }
  626. latestValue, _ = latestValueDecimal.Truncate(4).Float64() //保留4位小数
  627. }
  628. }
  629. existMap[edbCode+av] = av
  630. break
  631. } else {
  632. //fmt.Println("pre not find:", preDateStr, "i:", i)
  633. }
  634. }
  635. nextDateDay = nextDateDay.AddDate(0, 0, 1)
  636. preDateDay = preDateDay.AddDate(0, 0, -1)
  637. }
  638. }
  639. }
  640. }
  641. }
  642. if isAdd {
  643. addSql = strings.TrimRight(addSql, ",")
  644. err = to.Exec(addSql).Error
  645. if err != nil {
  646. return
  647. }
  648. }
  649. return
  650. }