edb_data_calculate_tcz.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761
  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. "strconv"
  9. "strings"
  10. "time"
  11. )
  12. type EdbInfoCalculateTczDetail struct {
  13. EdbInfoCalculateTczId int `orm:"column(edb_info_calculate_tcz_id);pk"`
  14. EdbInfoId int `description:"指标id"`
  15. EdbCode string `description:"指标编码"`
  16. FromEdbInfoId int `description:"计算指标id"`
  17. FromEdbCode string `description:"计算指标编码"`
  18. FromEdbName string `description:"计算指标名称"`
  19. FromSource int `description:"计算指标来源"`
  20. FromSourceName string `description:"计算指标来源名称"`
  21. FromTag string `description:"来源指标标签"`
  22. Sort int `description:"计算指标名称排序"`
  23. CreateTime time.Time `description:"创建时间"`
  24. ModifyTime time.Time `description:"修改时间"`
  25. StartDate string `description:"开始日期"`
  26. EndDate string `description:"结束日期"`
  27. }
  28. func GetEdbInfoCalculateTczDetail(edbInfoId int) (item *EdbInfoCalculateTczDetail, err error) {
  29. o := orm.NewOrm()
  30. sql := ` SELECT a.*,b.start_date,b.end_date FROM edb_info_calculate_mapping AS a
  31. INNER JOIN edb_info AS b ON a.from_edb_info_id=b.edb_info_id
  32. WHERE a.edb_info_id=? `
  33. err = o.Raw(sql, edbInfoId).QueryRow(&item)
  34. return
  35. }
  36. // AddCalculateTcz 同差值
  37. func AddCalculateTcz(req *EdbInfoCalculateBatchSaveReq, fromEdbInfo *EdbInfo, edbCode, uniqueCode string, sysUserId int, sysUserRealName string) (edbInfoId int, err error) {
  38. o := orm.NewOrm()
  39. to, err := o.Begin()
  40. if err != nil {
  41. return
  42. }
  43. defer func() {
  44. if err != nil {
  45. fmt.Println("AddCalculateTcz,Err:" + err.Error())
  46. _ = to.Rollback()
  47. } else {
  48. _ = to.Commit()
  49. }
  50. }()
  51. fmt.Println("req.EdbInfoId:", req.EdbInfoId)
  52. if req.EdbInfoId <= 0 {
  53. edbInfo := new(EdbInfo)
  54. edbInfo.Source = utils.DATA_SOURCE_CALCULATE_TCZ
  55. edbInfo.SourceName = "同差值"
  56. edbInfo.EdbCode = edbCode
  57. edbInfo.EdbName = req.EdbName
  58. edbInfo.EdbNameSource = req.EdbName
  59. edbInfo.Frequency = req.Frequency
  60. edbInfo.Unit = req.Unit
  61. edbInfo.ClassifyId = req.ClassifyId
  62. edbInfo.SysUserId = sysUserId
  63. edbInfo.SysUserRealName = sysUserRealName
  64. edbInfo.CreateTime = time.Now()
  65. edbInfo.ModifyTime = time.Now()
  66. edbInfo.UniqueCode = uniqueCode
  67. edbInfo.CalculateFormula = req.Formula
  68. edbInfo.EdbType = 2
  69. newEdbInfoId, tmpErr := to.Insert(edbInfo)
  70. if tmpErr != nil {
  71. return edbInfoId, tmpErr
  72. }
  73. edbInfoId = int(newEdbInfoId)
  74. //关联关系
  75. {
  76. calculateMappingItem := new(EdbInfoCalculateMapping)
  77. calculateMappingItem.CreateTime = time.Now()
  78. calculateMappingItem.ModifyTime = time.Now()
  79. calculateMappingItem.Sort = 1
  80. calculateMappingItem.EdbCode = edbCode
  81. calculateMappingItem.EdbInfoId = edbInfoId
  82. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  83. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  84. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  85. calculateMappingItem.FromSource = fromEdbInfo.Source
  86. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  87. calculateMappingItem.FromTag = ""
  88. calculateMappingItem.Source = edbInfo.Source
  89. calculateMappingItem.SourceName = edbInfo.SourceName
  90. _, err = to.Insert(calculateMappingItem)
  91. if err != nil {
  92. return
  93. }
  94. }
  95. } else {
  96. edbInfoId = req.EdbInfoId
  97. dataTableName := GetEdbDataTableName(utils.DATA_SOURCE_CALCULATE_TCZ)
  98. fmt.Println("dataTableName:", dataTableName)
  99. deleteSql := ` DELETE FROM %s WHERE edb_info_id=? `
  100. deleteSql = fmt.Sprintf(deleteSql, dataTableName)
  101. _, err = to.Raw(deleteSql, req.EdbInfoId).Exec()
  102. if err != nil {
  103. return edbInfoId, err
  104. }
  105. }
  106. edbInfoIdStr := strconv.Itoa(edbInfoId)
  107. //计算数据
  108. var condition string
  109. var pars []interface{}
  110. condition += " AND edb_info_id=? "
  111. if req.EdbInfoId <= 0 {
  112. pars = append(pars, req.FromEdbInfoId)
  113. } else {
  114. pars = append(pars, fromEdbInfo.EdbInfoId)
  115. }
  116. fmt.Println("EdbInfoId:", req.FromEdbInfoId)
  117. dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0)
  118. if err != nil {
  119. return edbInfoId, err
  120. }
  121. var dateArr []string
  122. dataMap := make(map[string]*EdbInfoSearchData)
  123. for _, v := range dataList {
  124. dateArr = append(dateArr, v.DataTime)
  125. dataMap[v.DataTime] = v
  126. }
  127. fmt.Println("Frequency:", fromEdbInfo.Frequency)
  128. addSql := ` INSERT INTO edb_data_calculate_tcz(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values `
  129. var isAdd bool
  130. existMap := make(map[string]string)
  131. for _, av := range dateArr {
  132. //fmt.Println("dateArr:", ak, av)
  133. currentItem := dataMap[av]
  134. if currentItem != nil {
  135. //当前日期
  136. currentDate, err := time.Parse(utils.FormatDate, av)
  137. if err != nil {
  138. return edbInfoId, err
  139. }
  140. //上一年的日期
  141. preDate := currentDate.AddDate(-1, 0, 0)
  142. preDateStr := preDate.Format(utils.FormatDate)
  143. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期找到
  144. //dataTime, _ := time.Parse(utils.FormatDate, date)
  145. if _, ok := existMap[edbCode+av]; !ok {
  146. timestamp := currentDate.UnixNano() / 1e6
  147. timestampStr := fmt.Sprintf("%d", timestamp)
  148. val := TczSub(currentItem.Value, findItem.Value)
  149. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  150. isAdd = true
  151. }
  152. existMap[edbCode+av] = av
  153. } else {
  154. if fromEdbInfo.Frequency == "月度" { //向上和向下,各找一个月
  155. nextDateDay := preDate
  156. preDateDay := preDate
  157. for i := 0; i <= 35; i++ {
  158. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  159. if findItem, ok := dataMap[nextDateDayStr]; ok { //下一年同期->下一个月找到
  160. if _, ok := existMap[edbCode+av]; !ok {
  161. timestamp := currentDate.UnixNano() / 1e6
  162. timestampStr := fmt.Sprintf("%d", timestamp)
  163. val := TczSub(currentItem.Value, findItem.Value)
  164. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  165. isAdd = true
  166. }
  167. existMap[edbCode+av] = av
  168. break
  169. } else {
  170. preDateDayStr := preDateDay.Format(utils.FormatDate)
  171. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  172. if _, ok := existMap[edbCode+av]; !ok {
  173. timestamp := currentDate.UnixNano() / 1e6
  174. timestampStr := fmt.Sprintf("%d", timestamp)
  175. val := TczSub(currentItem.Value, findItem.Value)
  176. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  177. isAdd = true
  178. }
  179. existMap[edbCode+av] = av
  180. break
  181. }
  182. }
  183. nextDateDay = nextDateDay.AddDate(0, 0, 1)
  184. preDateDay = preDateDay.AddDate(0, 0, -1)
  185. }
  186. } else if fromEdbInfo.Frequency == "季度" || fromEdbInfo.Frequency == "年度" {
  187. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  188. if _, ok := existMap[edbCode+av]; !ok {
  189. timestamp := currentDate.UnixNano() / 1e6
  190. timestampStr := fmt.Sprintf("%d", timestamp)
  191. val := TczSub(currentItem.Value, findItem.Value)
  192. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  193. isAdd = true
  194. }
  195. existMap[edbCode+av] = av
  196. break
  197. }
  198. } else {
  199. nextDateDay := preDate
  200. preDateDay := preDate
  201. for i := 0; i < 35; i++ {
  202. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  203. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  204. if _, ok := existMap[edbCode+av]; !ok {
  205. timestamp := currentDate.UnixNano() / 1e6
  206. timestampStr := fmt.Sprintf("%d", timestamp)
  207. val := TczSub(currentItem.Value, findItem.Value)
  208. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  209. isAdd = true
  210. }
  211. existMap[edbCode+av] = av
  212. break
  213. } else {
  214. preDateDayStr := preDateDay.Format(utils.FormatDate)
  215. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  216. if _, ok := existMap[edbCode+av]; !ok {
  217. timestamp := currentDate.UnixNano() / 1e6
  218. timestampStr := fmt.Sprintf("%d", timestamp)
  219. val := TczSub(currentItem.Value, findItem.Value)
  220. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  221. isAdd = true
  222. }
  223. existMap[edbCode+av] = av
  224. break
  225. } else {
  226. }
  227. }
  228. nextDateDay = nextDateDay.AddDate(0, 0, 1)
  229. preDateDay = preDateDay.AddDate(0, 0, -1)
  230. }
  231. }
  232. }
  233. }
  234. }
  235. if isAdd {
  236. addSql = strings.TrimRight(addSql, ",")
  237. _, err = to.Raw(addSql).Exec()
  238. if err != nil {
  239. return edbInfoId, err
  240. }
  241. }
  242. return
  243. }
  244. // EditCalculateTcz 同差值
  245. func EditCalculateTcz(req *EdbInfoCalculateBatchEditReq, fromEdbInfo *EdbInfo, edbCode string) (edbInfoId int, err error) {
  246. edbInfoId = req.EdbInfoId
  247. o := orm.NewOrm()
  248. to, err := o.Begin()
  249. if err != nil {
  250. return
  251. }
  252. defer func() {
  253. if err != nil {
  254. fmt.Println("EditCalculateTcz,Err:" + err.Error())
  255. _ = to.Rollback()
  256. } else {
  257. _ = to.Commit()
  258. }
  259. }()
  260. //修改指标信息
  261. sql := ` UPDATE edb_info
  262. SET
  263. edb_name =?,
  264. edb_name_source=?,
  265. frequency = ?,
  266. unit = ?,
  267. classify_id = ?,
  268. modify_time = NOW()
  269. WHERE edb_info_id = ? `
  270. _, err = o.Raw(sql, req.EdbName, req.EdbName, req.Frequency, req.Unit, req.ClassifyId, edbInfoId).Exec()
  271. if err != nil {
  272. return
  273. }
  274. var existCondition string
  275. var existPars []interface{}
  276. existCondition += " AND edb_info_id=? "
  277. existPars = append(existPars, edbInfoId)
  278. existCondition += " AND from_edb_info_id=? "
  279. existPars = append(existPars, req.FromEdbInfoId)
  280. //判断计算指标是否被更换
  281. count, err := GetEdbInfoCalculateCountByCondition(existCondition, existPars)
  282. if err != nil {
  283. err = errors.New("判断指标是否改变失败,Err:" + err.Error())
  284. return
  285. }
  286. if count <= 0 {
  287. //删除指标关联计算指标
  288. //sql := ` DELETE FROM edb_info_calculate_tcz WHERE edb_info_id = ? `
  289. //_, err = o.Raw(sql, edbInfoId).Exec()
  290. //if err != nil {
  291. // return
  292. //}
  293. //
  294. //calculateItem := new(EdbInfoCalculateTcz)
  295. //calculateItem.CreateTime = time.Now()
  296. //calculateItem.ModifyTime = time.Now()
  297. //calculateItem.Sort = 1
  298. //calculateItem.EdbCode = edbCode
  299. //calculateItem.EdbInfoId = edbInfoId
  300. //calculateItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  301. //calculateItem.FromEdbCode = fromEdbInfo.EdbCode
  302. //calculateItem.FromEdbName = fromEdbInfo.EdbName
  303. //calculateItem.FromSource = fromEdbInfo.Source
  304. //calculateItem.FromSourceName = fromEdbInfo.SourceName
  305. //
  306. //_, err = o.Insert(calculateItem)
  307. //if err != nil {
  308. // return
  309. //}
  310. //删除,计算指标关联的,基础指标的关联关系
  311. sql = ` DELETE FROM edb_info_calculate_mapping WHERE edb_info_id = ? `
  312. _, err = o.Raw(sql, edbInfoId).Exec()
  313. if err != nil {
  314. return
  315. }
  316. //清空原有数据
  317. sql = ` DELETE FROM edb_data_calculate_tcz WHERE edb_info_id = ? `
  318. _, err = o.Raw(sql, edbInfoId).Exec()
  319. if err != nil {
  320. return edbInfoId, err
  321. }
  322. //关联关系
  323. {
  324. calculateMappingItem := new(EdbInfoCalculateMapping)
  325. calculateMappingItem.CreateTime = time.Now()
  326. calculateMappingItem.ModifyTime = time.Now()
  327. calculateMappingItem.Sort = 1
  328. calculateMappingItem.EdbCode = edbCode
  329. calculateMappingItem.EdbInfoId = edbInfoId
  330. calculateMappingItem.FromEdbInfoId = fromEdbInfo.EdbInfoId
  331. calculateMappingItem.FromEdbCode = fromEdbInfo.EdbCode
  332. calculateMappingItem.FromEdbName = fromEdbInfo.EdbName
  333. calculateMappingItem.FromSource = fromEdbInfo.Source
  334. calculateMappingItem.FromSourceName = fromEdbInfo.SourceName
  335. calculateMappingItem.FromTag = ""
  336. calculateMappingItem.Source = utils.DATA_SOURCE_CALCULATE_TCZ
  337. calculateMappingItem.SourceName = "同差值"
  338. _, err = o.Insert(calculateMappingItem)
  339. if err != nil {
  340. return
  341. }
  342. }
  343. edbInfoIdStr := strconv.Itoa(edbInfoId)
  344. //计算数据
  345. var condition string
  346. var pars []interface{}
  347. condition += " AND edb_info_id=? "
  348. pars = append(pars, req.FromEdbInfoId)
  349. fmt.Println("EdbInfoId:", req.FromEdbInfoId)
  350. dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0)
  351. if err != nil {
  352. return edbInfoId, err
  353. }
  354. var dateArr []string
  355. dataMap := make(map[string]*EdbInfoSearchData)
  356. for _, v := range dataList {
  357. dateArr = append(dateArr, v.DataTime)
  358. dataMap[v.DataTime] = v
  359. }
  360. addSql := ` INSERT INTO edb_data_calculate_tcz(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values `
  361. var isAdd bool
  362. for ak, av := range dateArr {
  363. fmt.Println(ak, av)
  364. currentItem := dataMap[av]
  365. if currentItem != nil {
  366. //当前日期
  367. currentDate, err := time.Parse(utils.FormatDate, av)
  368. if err != nil {
  369. return edbInfoId, err
  370. }
  371. //上一年的日期
  372. preDate := currentDate.AddDate(-1, 0, 0)
  373. preDateStr := preDate.Format(utils.FormatDate)
  374. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期找到
  375. //dataTime, _ := time.Parse(utils.FormatDate, date)
  376. timestamp := currentDate.UnixNano() / 1e6
  377. timestampStr := fmt.Sprintf("%d", timestamp)
  378. val := TczSub(currentItem.Value, findItem.Value)
  379. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  380. isAdd = true
  381. utils.FileLog.Info("同期找到:" + av + ";" + preDateStr)
  382. continue
  383. } else {
  384. if fromEdbInfo.Frequency == "月度" { //向上和向下,各找一个月
  385. nextDateDay := preDate.AddDate(0, 1, 0)
  386. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  387. preDateDay := preDate.AddDate(0, -1, 0)
  388. preDateDayStr := preDateDay.Format(utils.FormatDate)
  389. for i := 0; i <= 6; i++ {
  390. if i >= 1 {
  391. nextDateDay = nextDateDay.AddDate(0, 0, i)
  392. nextDateDayStr = nextDateDay.Format(utils.FormatDate)
  393. }
  394. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  395. timestamp := currentDate.UnixNano() / 1e6
  396. timestampStr := fmt.Sprintf("%d", timestamp)
  397. val := TczSub(currentItem.Value, findItem.Value)
  398. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  399. isAdd = true
  400. break
  401. } else {
  402. if i >= 1 {
  403. preDateDay = preDate.AddDate(0, 0, -i)
  404. preDateDayStr = nextDateDay.Format(utils.FormatDate)
  405. }
  406. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  407. timestamp := currentDate.UnixNano() / 1e6
  408. timestampStr := fmt.Sprintf("%d", timestamp)
  409. val := TczSub(currentItem.Value, findItem.Value)
  410. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  411. isAdd = true
  412. break
  413. }
  414. }
  415. }
  416. } else if fromEdbInfo.Frequency == "季度" || fromEdbInfo.Frequency == "年度" {
  417. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  418. timestamp := currentDate.UnixNano() / 1e6
  419. timestampStr := fmt.Sprintf("%d", timestamp)
  420. val := TczSub(currentItem.Value, findItem.Value)
  421. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  422. isAdd = true
  423. break
  424. }
  425. } else {
  426. nextDateDay := preDate.AddDate(0, 0, 1)
  427. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  428. preDateDay := preDate.AddDate(0, 0, -1)
  429. preDateDayStr := preDateDay.Format(utils.FormatDate)
  430. for i := 0; i < 35; i++ {
  431. if i >= 1 {
  432. nextDateDay = nextDateDay.AddDate(0, 0, i)
  433. nextDateDayStr = nextDateDay.Format(utils.FormatDate)
  434. }
  435. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  436. timestamp := currentDate.UnixNano() / 1e6
  437. timestampStr := fmt.Sprintf("%d", timestamp)
  438. val := TczSub(currentItem.Value, findItem.Value)
  439. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  440. isAdd = true
  441. break
  442. } else {
  443. if i >= 1 {
  444. preDateDay = preDate.AddDate(0, 0, -i)
  445. preDateDayStr = nextDateDay.Format(utils.FormatDate)
  446. }
  447. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  448. timestamp := currentDate.UnixNano() / 1e6
  449. timestampStr := fmt.Sprintf("%d", timestamp)
  450. val := TczSub(currentItem.Value, findItem.Value)
  451. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  452. isAdd = true
  453. break
  454. }
  455. }
  456. }
  457. }
  458. }
  459. }
  460. }
  461. if isAdd {
  462. addSql = strings.TrimRight(addSql, ",")
  463. _, err = o.Raw(addSql).Exec()
  464. if err != nil {
  465. return edbInfoId, err
  466. }
  467. }
  468. }
  469. return
  470. }
  471. // RefreshAllCalculateTcz 刷新全部同差值数据
  472. func RefreshAllCalculateTcz(edbInfoId, source int, fromEdbInfo *EdbInfo, edbCode, startDate, endDate string) (err error) {
  473. o := orm.NewOrm()
  474. to, err := o.Begin()
  475. if err != nil {
  476. return
  477. }
  478. defer func() {
  479. if err != nil {
  480. fmt.Println("RefreshAllCalculateTcz,Err:" + err.Error())
  481. _ = to.Rollback()
  482. } else {
  483. _ = to.Commit()
  484. }
  485. }()
  486. if err != nil {
  487. return
  488. }
  489. edbInfoIdStr := strconv.Itoa(edbInfoId)
  490. //计算数据
  491. //计算数据
  492. var condition string
  493. var pars []interface{}
  494. condition += " AND edb_info_id=? "
  495. pars = append(pars, fromEdbInfo.EdbInfoId)
  496. if startDate != "" {
  497. condition += " AND data_time>=? "
  498. pars = append(pars, startDate)
  499. }
  500. if endDate != "" {
  501. condition += " AND data_time<=? "
  502. pars = append(pars, endDate)
  503. }
  504. dataList, err := GetEdbDataListAll(condition, pars, fromEdbInfo.Source, 0)
  505. if err != nil {
  506. return err
  507. }
  508. var dateArr []string
  509. dataMap := make(map[string]*EdbInfoSearchData)
  510. for _, v := range dataList {
  511. dateArr = append(dateArr, v.DataTime)
  512. dataMap[v.DataTime] = v
  513. }
  514. //获取指标所有数据
  515. existDataList := make([]*EdbData, 0)
  516. dataTableName := GetEdbDataTableName(source)
  517. sql := `SELECT * FROM %s WHERE edb_info_id=? `
  518. sql = fmt.Sprintf(sql, dataTableName)
  519. _, err = o.Raw(sql, edbInfoId).QueryRows(&existDataList)
  520. if err != nil {
  521. return err
  522. }
  523. existDataMap := make(map[string]string)
  524. for _, v := range existDataList {
  525. existDataMap[v.DataTime] = v.Value
  526. }
  527. addSql := ` INSERT INTO edb_data_calculate_tcz(edb_info_id,edb_code,data_time,value,create_time,modify_time,status,data_timestamp) values `
  528. var isAdd bool
  529. existAddDataMap := make(map[string]string)
  530. for _, av := range dateArr {
  531. currentItem := dataMap[av]
  532. if currentItem != nil {
  533. //当前日期
  534. currentDate, err := time.Parse(utils.FormatDate, av)
  535. if err != nil {
  536. return err
  537. }
  538. //上一年的日期
  539. preDate := currentDate.AddDate(-1, 0, 0)
  540. preDateStr := preDate.Format(utils.FormatDate)
  541. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期找到
  542. //dataTime, _ := time.Parse(utils.FormatDate, date)
  543. timestamp := currentDate.UnixNano() / 1e6
  544. timestampStr := fmt.Sprintf("%d", timestamp)
  545. val := TczSub(currentItem.Value, findItem.Value)
  546. if existVal, ok := existDataMap[av]; !ok {
  547. if _, existOk := existAddDataMap[av]; !existOk {
  548. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  549. isAdd = true
  550. }
  551. existAddDataMap[av] = av
  552. } else {
  553. if existVal != val {
  554. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  555. sql = fmt.Sprintf(sql, dataTableName)
  556. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  557. if err != nil {
  558. return err
  559. }
  560. }
  561. }
  562. utils.FileLog.Info("同期找到:" + av + ";" + preDateStr)
  563. continue
  564. } else {
  565. if fromEdbInfo.Frequency == "月度" { //向上和向下,各找一个月
  566. for i := 0; i <= 35; i++ {
  567. nextDateDay := preDate.AddDate(0, 0, 1)
  568. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  569. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  570. timestamp := currentDate.UnixNano() / 1e6
  571. timestampStr := fmt.Sprintf("%d", timestamp)
  572. val := TczSub(currentItem.Value, findItem.Value)
  573. if existVal, ok := existDataMap[av]; !ok {
  574. if _, existOk := existAddDataMap[av]; !existOk {
  575. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  576. isAdd = true
  577. }
  578. existAddDataMap[av] = av
  579. } else {
  580. if existVal != val {
  581. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  582. sql = fmt.Sprintf(sql, dataTableName)
  583. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  584. if err != nil {
  585. return err
  586. }
  587. }
  588. }
  589. break
  590. } else {
  591. preDateDay := preDate.AddDate(0, 0, -1)
  592. preDateDayStr := preDateDay.Format(utils.FormatDate)
  593. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  594. timestamp := currentDate.UnixNano() / 1e6
  595. timestampStr := fmt.Sprintf("%d", timestamp)
  596. val := TczSub(currentItem.Value, findItem.Value)
  597. if existVal, ok := existDataMap[av]; !ok {
  598. if _, existOk := existAddDataMap[av]; !existOk {
  599. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  600. isAdd = true
  601. }
  602. existAddDataMap[av] = av
  603. } else {
  604. if existVal != val {
  605. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  606. sql = fmt.Sprintf(sql, dataTableName)
  607. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  608. if err != nil {
  609. return err
  610. }
  611. }
  612. }
  613. break
  614. }
  615. }
  616. }
  617. } else if fromEdbInfo.Frequency == "季度" || fromEdbInfo.Frequency == "年度" {
  618. if findItem, ok := dataMap[preDateStr]; ok { //上一年同期->下一个月找到
  619. timestamp := currentDate.UnixNano() / 1e6
  620. timestampStr := fmt.Sprintf("%d", timestamp)
  621. val := TczSub(currentItem.Value, findItem.Value)
  622. if existVal, ok := existDataMap[av]; !ok {
  623. if _, existOk := existAddDataMap[av]; !existOk {
  624. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  625. isAdd = true
  626. }
  627. existAddDataMap[av] = av
  628. } else {
  629. if existVal != val {
  630. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  631. sql = fmt.Sprintf(sql, dataTableName)
  632. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  633. if err != nil {
  634. return err
  635. }
  636. }
  637. }
  638. break
  639. }
  640. } else {
  641. for i := 0; i < 35; i++ {
  642. nextDateDay := preDate.AddDate(0, 0, 1)
  643. nextDateDayStr := nextDateDay.Format(utils.FormatDate)
  644. if findItem, ok := dataMap[nextDateDayStr]; ok { //上一年同期->下一个月找到
  645. timestamp := currentDate.UnixNano() / 1e6
  646. timestampStr := fmt.Sprintf("%d", timestamp)
  647. val := TczSub(currentItem.Value, findItem.Value)
  648. if existVal, ok := existDataMap[av]; !ok {
  649. if _, existOk := existAddDataMap[av]; !existOk {
  650. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  651. isAdd = true
  652. }
  653. existAddDataMap[av] = av
  654. } else {
  655. if existVal != val {
  656. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  657. sql = fmt.Sprintf(sql, dataTableName)
  658. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  659. if err != nil {
  660. return err
  661. }
  662. }
  663. }
  664. break
  665. } else {
  666. preDateDay := preDate.AddDate(0, 0, -1)
  667. preDateDayStr := preDateDay.Format(utils.FormatDate)
  668. if findItem, ok := dataMap[preDateDayStr]; ok { //上一年同期->上一个月找到
  669. timestamp := currentDate.UnixNano() / 1e6
  670. timestampStr := fmt.Sprintf("%d", timestamp)
  671. val := TczSub(currentItem.Value, findItem.Value)
  672. if existVal, ok := existDataMap[av]; !ok {
  673. if _, existOk := existAddDataMap[av]; !existOk {
  674. addSql += GetAddSql(edbInfoIdStr, edbCode, av, timestampStr, val)
  675. isAdd = true
  676. }
  677. existAddDataMap[av] = av
  678. } else {
  679. if existVal != val {
  680. sql := ` UPDATE %s SET value=?,modify_time=NOW() WHERE edb_info_id=? AND data_time=? `
  681. sql = fmt.Sprintf(sql, dataTableName)
  682. _, err = o.Raw(sql, val, edbInfoId, av).Exec()
  683. if err != nil {
  684. return err
  685. }
  686. }
  687. }
  688. break
  689. }
  690. }
  691. }
  692. }
  693. }
  694. }
  695. }
  696. if isAdd {
  697. addSql = strings.TrimRight(addSql, ",")
  698. _, err = o.Raw(addSql).Exec()
  699. if err != nil {
  700. return err
  701. }
  702. }
  703. return
  704. }
  705. // TczSub 计算同差值
  706. func TczSub(a, b float64) string {
  707. af := decimal.NewFromFloat(a)
  708. fmt.Println(af)
  709. bf := decimal.NewFromFloat(b)
  710. val, _ := af.Sub(bf).Float64()
  711. valStr := utils.SubFloatToString(val, 4)
  712. return valStr
  713. }