national_data.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675
  1. package national_data
  2. import (
  3. "context"
  4. "fmt"
  5. "hongze/hongze_data_crawler/models"
  6. "hongze/hongze_data_crawler/services/alarm_msg"
  7. "hongze/hongze_data_crawler/utils"
  8. "strings"
  9. "time"
  10. )
  11. // RefreshNationalData 刷新统计局数据
  12. func RefreshNationalData(cont context.Context) (err error) {
  13. utils.FileLog.Info("开始刷新统计局数据")
  14. _ = SyncXDateYQuotaDb()
  15. _ = SyncXDateYQuotaZRegDb()
  16. _ = SyncXRegYDateZQuotaDb()
  17. utils.FileLog.Info("统计局数据刷新成功")
  18. return
  19. }
  20. // SyncXDateYQuotaDb 同步两维度X轴-日期, Y轴-指标数据库(月度/季度/年度数据指标)
  21. func SyncXDateYQuotaDb() (err error) {
  22. defer func() {
  23. if err != nil {
  24. utils.FileLog.Error("统计局-同步月度/季度/年度数据失败, ErrMsg: %s", err.Error())
  25. go alarm_msg.SendAlarmMsg(fmt.Sprintf("统计局-同步月度/季度/年度数据失败, ErrMsg: %s", err.Error()), 3)
  26. return
  27. }
  28. utils.FileLog.Info("统计局-同步月度/季度/年度数据成功")
  29. }()
  30. // 查询无父级的指标分类
  31. classifyOB := new(models.BaseFromNationalStatisticsClassify)
  32. classifyCond := ` AND is_parent = 0 AND dbcode IN ('hgyd', 'hgjd', 'hgnd')`
  33. classifyPars := make([]interface{}, 0)
  34. classifyOrder := ` base_from_national_statistics_classify_id ASC`
  35. classifyList, e := classifyOB.GetItemsByCondition(classifyCond, classifyPars, []string{}, classifyOrder)
  36. if e != nil {
  37. err = fmt.Errorf("获取指标分类列表失败, Err: %s", e.Error())
  38. return
  39. }
  40. utils.FileLog.Info("分类长度: %d\n", len(classifyList))
  41. // 同步指标和数据
  42. for _, c := range classifyList {
  43. if e = SyncXDateYQuotaData(c.BaseFromNationalStatisticsClassifyId, c.Dbcode, c.Id); e != nil {
  44. err = fmt.Errorf("同步指标数据失败, DbCode: %s, ClassifyId: %d, Err: %s", c.Dbcode, c.BaseFromNationalStatisticsClassifyId, e.Error())
  45. return
  46. }
  47. }
  48. return
  49. }
  50. // SyncXDateYQuotaData 同步两维度X轴-日期, Y轴-指标的数据
  51. func SyncXDateYQuotaData(classifyId int, dbCode, classifyCode string) (err error) {
  52. defer func() {
  53. if err != nil {
  54. utils.FileLog.Error("同步数据库DbCode: %s, 分类ClassifyId: %d失败, ErrMsg: %s", dbCode, classifyId, err.Error())
  55. }
  56. }()
  57. // yd-月度 jd-季度 nd-年度
  58. frequency := ""
  59. timeParam := ""
  60. if strings.Contains(dbCode, "yd") {
  61. timeParam = "LAST36" // 最近36个月
  62. frequency = "月度"
  63. }
  64. if strings.Contains(dbCode, "jd") {
  65. timeParam = "LAST18" // 最近18个季度
  66. frequency = "季度"
  67. }
  68. if strings.Contains(dbCode, "nd") {
  69. timeParam = "LAST20" // 最近20年
  70. frequency = "年度"
  71. }
  72. var dataReq DataApiReq
  73. dataReq.DbCode = dbCode
  74. dataReq.DfwdsList = append(dataReq.DfwdsList, Wds{
  75. WdCode: "zb",
  76. ValueCode: classifyCode,
  77. }, Wds{
  78. WdCode: "sj",
  79. ValueCode: timeParam,
  80. })
  81. resp, e := CommonDataApiRequest(dataReq)
  82. if e != nil {
  83. err = fmt.Errorf("请求数据接口失败, Err: %s", e.Error())
  84. return
  85. }
  86. // 数据集
  87. dataNodes := resp.ReturnData.DataNodes
  88. dataMap := make(map[string]QuotaDataNode)
  89. for _, d := range dataNodes {
  90. dataMap[d.Code] = d
  91. }
  92. // 取出指标(Y轴), 日期(X轴)
  93. wdNodes := resp.ReturnData.WdNodes
  94. var quotaNodes, dateNodes []QuotaWdNodeData
  95. for _, w := range wdNodes {
  96. if w.WdCode == "zb" {
  97. quotaNodes = w.Nodes
  98. continue
  99. }
  100. if w.WdCode == "sj" {
  101. dateNodes = w.Nodes
  102. }
  103. }
  104. // 指标编码去重, 指标编码+日期数据去重
  105. indexOB := new(models.BaseFromNationalStatisticsIndex)
  106. indexCond := ``
  107. indexPars := make([]interface{}, 0)
  108. indexList, e := indexOB.GetItemsByCondition(indexCond, indexPars, []string{"index_code"}, "")
  109. if e != nil {
  110. err = fmt.Errorf("获取指标列表失败, Err: %s", e.Error())
  111. return
  112. }
  113. indexExistMap := make(map[string]bool)
  114. for _, v := range indexList {
  115. indexExistMap[v.IndexCode] = true
  116. }
  117. dataOB := new(models.BaseFromNationalStatisticsData)
  118. dataCond := ``
  119. dataPars := make([]interface{}, 0)
  120. dataList, e := dataOB.GetItemsByCondition(dataCond, dataPars, []string{"index_code", "data_time"}, "")
  121. if e != nil {
  122. err = fmt.Errorf("获取指标数据列表失败, Err: %s", e.Error())
  123. return
  124. }
  125. dataExistMap := make(map[string]bool)
  126. for _, v := range dataList {
  127. dataExistMap[fmt.Sprintf("%s%s", v.IndexCode, v.DataTime.Format(utils.FormatDate))] = true
  128. }
  129. // 遍历XY轴
  130. indexDataList := make([]*models.SaveNationalStatisticsIndexAndDataReq, 0)
  131. indexDataMap := make(map[string][]*models.BaseFromNationalStatisticsData)
  132. for _, q := range quotaNodes {
  133. indexCode := fmt.Sprintf("%s%s", dbCode, q.Code)
  134. // 指标
  135. r := new(models.SaveNationalStatisticsIndexAndDataReq)
  136. r.Index = &models.BaseFromNationalStatisticsIndex{
  137. BaseFromNationalStatisticsClassifyId: classifyId,
  138. Dbcode: dbCode,
  139. IndexCode: indexCode,
  140. IndexName: q.Name,
  141. Frequency: frequency,
  142. CreateTime: time.Now().Local(),
  143. ModifyTime: time.Now().Local(),
  144. }
  145. if indexExistMap[indexCode] {
  146. r.IndexExist = true
  147. }
  148. // 数据
  149. for _, d := range dateNodes {
  150. k := fmt.Sprintf("%s.%s_%s.%s", "zb", q.Code, "sj", d.Code)
  151. v := dataMap[k]
  152. if !v.Data.HasData {
  153. continue
  154. }
  155. // 日期去重
  156. t, e := formatMonth2YearDateCode(d.Code)
  157. if e != nil {
  158. err = fmt.Errorf("格式化日期code失败, Err: %s", e.Error())
  159. return
  160. }
  161. existKey := fmt.Sprintf("%s%s", indexCode, t.Format(utils.FormatDate))
  162. if dataExistMap[existKey] {
  163. continue
  164. }
  165. // 数据map
  166. if indexDataMap[indexCode] == nil {
  167. indexDataMap[indexCode] = make([]*models.BaseFromNationalStatisticsData, 0)
  168. }
  169. indexDataMap[indexCode] = append(indexDataMap[indexCode], &models.BaseFromNationalStatisticsData{
  170. IndexCode: indexCode,
  171. DataTime: t,
  172. Value: v.Data.Data,
  173. CreateTime: time.Now().Local(),
  174. ModifyTime: time.Now().Local(),
  175. })
  176. }
  177. indexDataList = append(indexDataList, r)
  178. }
  179. // 保存指标
  180. for _, v := range indexDataList {
  181. ds := indexDataMap[v.Index.IndexCode]
  182. if ds == nil || (ds != nil && len(ds) == 0) {
  183. continue
  184. }
  185. v.DataList = ds
  186. if e := models.SaveNationalStatisticsIndexAndData(v); e != nil {
  187. err = fmt.Errorf("保存指标和数据失败, Err: %s", e.Error())
  188. return
  189. }
  190. }
  191. return
  192. }
  193. // SyncXDateYQuotaZRegDb 同步三维度X轴-日期, Y轴-指标, Z轴-地区的数据库(分省月季年度、主要城市月年度、国际市场月度商品价格)
  194. func SyncXDateYQuotaZRegDb() (err error) {
  195. defer func() {
  196. if err != nil {
  197. utils.FileLog.Error("统计局-同步分省月季年度、主要城市月年度、国际市场月度商品价格失败, ErrMsg: %s", err.Error())
  198. go alarm_msg.SendAlarmMsg(fmt.Sprintf("统计局-同步分省月季年度、主要城市月年度、国际市场月度商品价格失败, ErrMsg: %s", err.Error()), 3)
  199. return
  200. }
  201. utils.FileLog.Info("统计局-同步分省月季年度、主要城市月年度、国际市场月度商品价格成功")
  202. }()
  203. // 需要同步的数据库
  204. dbCodes := []string{"fsyd", "fsjd", "fsnd", "csyd", "csnd", "gjydsc"}
  205. for _, d := range dbCodes {
  206. classifyOB := new(models.BaseFromNationalStatisticsClassify)
  207. // 注意此处只需要同步分类中为指标的即可, 分类为地区的数据在指标中均有包含
  208. classifyCond := ` AND is_parent = 0 AND wdcode = 'zb' AND dbcode = ?`
  209. classifyPars := make([]interface{}, 0)
  210. classifyPars = append(classifyPars, d)
  211. classifyOrder := ` base_from_national_statistics_classify_id ASC`
  212. classifyList, e := classifyOB.GetItemsByCondition(classifyCond, classifyPars, []string{}, classifyOrder)
  213. if e != nil {
  214. err = fmt.Errorf("获取%s分类列表失败, Err: %s", d, e.Error())
  215. return
  216. }
  217. utils.FileLog.Info("%s分类长度: %d\n", d, len(classifyList))
  218. // 查询其他维度-地区
  219. wdList, e := GetOtherWd(d, "", "")
  220. var regList []OtherWdNodes
  221. for _, wd := range wdList {
  222. if wd.WdCode == "reg" {
  223. regList = wd.Nodes
  224. break
  225. }
  226. }
  227. if len(regList) == 0 {
  228. err = fmt.Errorf("其他维度为空, DbCode: %s", d)
  229. return
  230. }
  231. // 同步指标和数据
  232. for _, c := range classifyList {
  233. if e = SyncXDateYQuotaZRegData(c.BaseFromNationalStatisticsClassifyId, c.Dbcode, c.Id, regList); e != nil {
  234. err = fmt.Errorf("同步指标数据失败, DbCode: %s, ClassifyId: %d, Err: %s", c.Dbcode, c.BaseFromNationalStatisticsClassifyId, e.Error())
  235. return
  236. }
  237. }
  238. }
  239. return
  240. }
  241. // SyncXDateYQuotaZRegData 同步三维度X轴-日期, Y轴-指标, Z轴-地区的数据
  242. func SyncXDateYQuotaZRegData(classifyId int, dbCode, classifyCode string, regList []OtherWdNodes) (err error) {
  243. defer func() {
  244. if err != nil {
  245. utils.FileLog.Error("同步数据库DbCode: %s, 分类ClassifyId: %d失败, ErrMsg: %s", dbCode, classifyId, err.Error())
  246. }
  247. }()
  248. // 根据DbCode判断频度和查询的时间区间
  249. frequency := ""
  250. timeParam := ""
  251. if strings.Contains(dbCode, "yd") {
  252. timeParam = "LAST36" // 最近36个月
  253. frequency = "月度"
  254. }
  255. if strings.Contains(dbCode, "jd") {
  256. timeParam = "LAST18" // 最近18个季度
  257. frequency = "季度"
  258. }
  259. if strings.Contains(dbCode, "nd") {
  260. timeParam = "LAST20" // 最近20年
  261. frequency = "年度"
  262. }
  263. // 遍历地区维度, 查询指标和数据
  264. for _, reg := range regList {
  265. var dataReq DataApiReq
  266. dataReq.DbCode = dbCode
  267. dataReq.WdsList = append(dataReq.WdsList, Wds{
  268. WdCode: "reg",
  269. ValueCode: reg.Code,
  270. })
  271. dataReq.DfwdsList = append(dataReq.DfwdsList, Wds{
  272. WdCode: "zb",
  273. ValueCode: classifyCode,
  274. }, Wds{
  275. WdCode: "sj",
  276. ValueCode: timeParam,
  277. })
  278. resp, e := CommonDataApiRequest(dataReq)
  279. if e != nil {
  280. err = fmt.Errorf("获取分类下的指标数据失败, Err: %s", e.Error())
  281. return
  282. }
  283. // 数据集
  284. dataNodes := resp.ReturnData.DataNodes
  285. dataMap := make(map[string]QuotaDataNode)
  286. for _, d := range dataNodes {
  287. dataMap[d.Code] = d
  288. }
  289. // 取出指标(Y轴), 日期(X轴)
  290. wdNodes := resp.ReturnData.WdNodes
  291. var quotaNodes, dateNodes []QuotaWdNodeData
  292. for _, w := range wdNodes {
  293. if w.WdCode == "zb" {
  294. quotaNodes = w.Nodes
  295. continue
  296. }
  297. if w.WdCode == "sj" {
  298. dateNodes = w.Nodes
  299. }
  300. }
  301. // 指标编码去重, 指标编码+日期数据去重
  302. indexOB := new(models.BaseFromNationalStatisticsIndex)
  303. indexCond := ``
  304. indexPars := make([]interface{}, 0)
  305. indexList, e := indexOB.GetItemsByCondition(indexCond, indexPars, []string{"index_code"}, "")
  306. if e != nil {
  307. err = fmt.Errorf("获取指标列表失败, Err: %s", e.Error())
  308. return
  309. }
  310. indexExistMap := make(map[string]bool)
  311. for _, v := range indexList {
  312. indexExistMap[v.IndexCode] = true
  313. }
  314. dataOB := new(models.BaseFromNationalStatisticsData)
  315. dataCond := ``
  316. dataPars := make([]interface{}, 0)
  317. dataList, e := dataOB.GetItemsByCondition(dataCond, dataPars, []string{"index_code", "data_time"}, "")
  318. if e != nil {
  319. err = fmt.Errorf("获取指标数据列表失败, Err: %s", e.Error())
  320. return
  321. }
  322. dataExistMap := make(map[string]bool)
  323. for _, v := range dataList {
  324. dataExistMap[fmt.Sprintf("%s%s", v.IndexCode, v.DataTime.Format(utils.FormatDate))] = true
  325. }
  326. // 遍历XY轴
  327. indexDataList := make([]*models.SaveNationalStatisticsIndexAndDataReq, 0)
  328. indexDataMap := make(map[string][]*models.BaseFromNationalStatisticsData)
  329. for _, q := range quotaNodes {
  330. // dbcode+指标code+地区code
  331. indexCode := fmt.Sprintf("%s%s%s", dbCode, q.Code, reg.Code)
  332. // 指标
  333. r := new(models.SaveNationalStatisticsIndexAndDataReq)
  334. r.Index = &models.BaseFromNationalStatisticsIndex{
  335. BaseFromNationalStatisticsClassifyId: classifyId,
  336. Dbcode: dbCode,
  337. IndexCode: indexCode,
  338. IndexName: q.Name,
  339. Frequency: frequency,
  340. Reg: reg.Name,
  341. CreateTime: time.Now().Local(),
  342. ModifyTime: time.Now().Local(),
  343. }
  344. if indexExistMap[indexCode] {
  345. r.IndexExist = true
  346. }
  347. // 数据
  348. // zb.A01010201_reg.110000_sj.201608
  349. for _, d := range dateNodes {
  350. k := fmt.Sprintf("%s.%s_%s.%s_%s.%s", "zb", q.Code, "reg", reg.Code, "sj", d.Code)
  351. v := dataMap[k]
  352. if !v.Data.HasData {
  353. continue
  354. }
  355. // 日期去重
  356. t, e := formatMonth2YearDateCode(d.Code)
  357. if e != nil {
  358. err = fmt.Errorf("格式化日期code失败, Err: %s", e.Error())
  359. return
  360. }
  361. existKey := fmt.Sprintf("%s%s", indexCode, t.Format(utils.FormatDate))
  362. if dataExistMap[existKey] {
  363. continue
  364. }
  365. // 数据map
  366. if indexDataMap[indexCode] == nil {
  367. indexDataMap[indexCode] = make([]*models.BaseFromNationalStatisticsData, 0)
  368. }
  369. indexDataMap[indexCode] = append(indexDataMap[indexCode], &models.BaseFromNationalStatisticsData{
  370. IndexCode: indexCode,
  371. DataTime: t,
  372. Value: v.Data.Data,
  373. CreateTime: time.Now().Local(),
  374. ModifyTime: time.Now().Local(),
  375. })
  376. }
  377. indexDataList = append(indexDataList, r)
  378. }
  379. // 保存指标
  380. for _, v := range indexDataList {
  381. ds := indexDataMap[v.Index.IndexCode]
  382. if ds == nil || (ds != nil && len(ds) == 0) {
  383. continue
  384. }
  385. v.DataList = ds
  386. if e := models.SaveNationalStatisticsIndexAndData(v); e != nil {
  387. err = fmt.Errorf("保存指标和数据失败, Err: %s", e.Error())
  388. return
  389. }
  390. }
  391. }
  392. return
  393. }
  394. // SyncXRegYDateZQuotaDb 同步三维度X轴-地区, Y轴-日期的数据库(港澳台、国际数据指标)
  395. func SyncXRegYDateZQuotaDb() (err error) {
  396. defer func() {
  397. if err != nil {
  398. utils.FileLog.Error("统计局-同步港澳台、国际数据指标失败, ErrMsg: %s", err.Error())
  399. go alarm_msg.SendAlarmMsg(fmt.Sprintf("统计局-同步港澳台、国际数据指标失败, ErrMsg: %s", err.Error()), 3)
  400. return
  401. }
  402. utils.FileLog.Info("统计局-同步港澳台、国际数据指标成功")
  403. }()
  404. // 需要同步的数据库
  405. dbCodes := []string{"gatyd", "gatnd", "gjyd", "gjydsdj", "gjnd"}
  406. for _, d := range dbCodes {
  407. classifyOB := new(models.BaseFromNationalStatisticsClassify)
  408. // 注意此处只需要同步分类中为指标的即可, 分类为地区的数据在指标中均有包含
  409. classifyCond := ` AND is_parent = 0 AND wdcode = 'zb' AND dbcode = ?`
  410. classifyPars := make([]interface{}, 0)
  411. classifyPars = append(classifyPars, d)
  412. classifyOrder := ` base_from_national_statistics_classify_id ASC`
  413. classifyList, e := classifyOB.GetItemsByCondition(classifyCond, classifyPars, []string{}, classifyOrder)
  414. if e != nil {
  415. err = fmt.Errorf("获取%s分类列表失败, Err: %s", d, e.Error())
  416. return
  417. }
  418. utils.FileLog.Info("%s分类长度: %d\n", d, len(classifyList))
  419. // 同步指标和数据
  420. for _, c := range classifyList {
  421. if e = SyncXRegYDateZQuotaDbData(c.BaseFromNationalStatisticsClassifyId, c.Dbcode, c.Id); e != nil {
  422. err = fmt.Errorf("同步指标数据失败, DbCode: %s, ClassifyId: %d, Err: %s", c.Dbcode, c.BaseFromNationalStatisticsClassifyId, e.Error())
  423. return
  424. }
  425. }
  426. }
  427. return
  428. }
  429. // SyncXRegYDateZQuotaDbData 同步三维度X轴-地区, Y轴-日期, Z轴-指标的数据
  430. func SyncXRegYDateZQuotaDbData(classifyId int, dbCode, classifyCode string) (err error) {
  431. defer func() {
  432. if err != nil {
  433. utils.FileLog.Error("同步数据库DbCode: %s, 分类ClassifyId: %d失败, ErrMsg: %s", dbCode, classifyId, err.Error())
  434. }
  435. }()
  436. // 根据DbCode判断频度和查询的时间区间
  437. frequency := ""
  438. timeParam := ""
  439. if strings.Contains(dbCode, "yd") {
  440. timeParam = "LAST36" // 最近36个月
  441. frequency = "月度"
  442. }
  443. if strings.Contains(dbCode, "jd") {
  444. timeParam = "LAST18" // 最近18个季度
  445. frequency = "季度"
  446. }
  447. if strings.Contains(dbCode, "nd") {
  448. timeParam = "LAST20" // 最近20年
  449. frequency = "年度"
  450. }
  451. // 先以指标作为行进行默认查询, 取出其中的指标作为当前分类的指标
  452. //f := url.Values{}
  453. //f.Add("m", "QueryData")
  454. //f.Add("dbcode", "gatyd")
  455. //f.Add("rowcode", "zb")
  456. //f.Add("colcode", "reg")
  457. //f.Add("wds", `[{"wdcode":"sj","valuecode":"LAST36"}]`)
  458. //f.Add("dfwds", `[{"wdcode":"zb","valuecode":"A01"}]`)
  459. var defaultReq DataApiReq
  460. defaultReq.DbCode = dbCode
  461. defaultReq.RowCode = "zb"
  462. defaultReq.ColCode = "reg"
  463. defaultReq.WdsList = append(defaultReq.WdsList, Wds{
  464. WdCode: "sj",
  465. ValueCode: timeParam,
  466. })
  467. defaultReq.DfwdsList = append(defaultReq.DfwdsList, Wds{
  468. WdCode: "zb",
  469. ValueCode: classifyCode,
  470. })
  471. defaultResult, e := CommonDataApiRequest(defaultReq)
  472. if e != nil {
  473. err = fmt.Errorf("默认查询数据失败, Err: %s", e.Error())
  474. return
  475. }
  476. var quotaWds []QuotaWdNodeData
  477. for _, n := range defaultResult.ReturnData.WdNodes {
  478. if n.WdCode == "zb" {
  479. quotaWds = n.Nodes
  480. break
  481. }
  482. }
  483. // 遍历指标维度
  484. for _, quota := range quotaWds {
  485. //f := url.Values{}
  486. //f.Add("m", "QueryData")
  487. //f.Add("dbcode", "gatyd")
  488. //f.Add("rowcode", "sj")
  489. //f.Add("colcode", "reg")
  490. //f.Add("wds", `[{"wdcode":"zb","valuecode":"A010A"}]`)
  491. //f.Add("dfwds", `[{"wdcode":"sj","valuecode":"LAST36"}]`)
  492. var dataReq DataApiReq
  493. dataReq.DbCode = dbCode
  494. dataReq.RowCode = "sj"
  495. dataReq.ColCode = "reg"
  496. dataReq.WdsList = append(defaultReq.WdsList, Wds{
  497. WdCode: "zb",
  498. ValueCode: quota.Code,
  499. })
  500. dataReq.DfwdsList = append(defaultReq.DfwdsList, Wds{
  501. WdCode: "sj",
  502. ValueCode: timeParam,
  503. })
  504. resp, e := CommonDataApiRequest(dataReq)
  505. if e != nil {
  506. err = fmt.Errorf("查询数据失败, Err: %s", e.Error())
  507. return
  508. }
  509. // 数据集
  510. dataNodes := resp.ReturnData.DataNodes
  511. dataMap := make(map[string]QuotaDataNode)
  512. for _, d := range dataNodes {
  513. dataMap[d.Code] = d
  514. }
  515. // 取出指标(Y轴), 日期(X轴)
  516. wdNodes := resp.ReturnData.WdNodes
  517. //var quotaNodes, dateNodes, regNodes []QuotaWdNodeData
  518. var dateNodes, regNodes []QuotaWdNodeData
  519. for _, w := range wdNodes {
  520. switch w.WdCode {
  521. case "zb":
  522. //quotaNodes = w.Nodes
  523. break
  524. case "sj":
  525. dateNodes = w.Nodes
  526. break
  527. case "reg":
  528. regNodes = w.Nodes
  529. break
  530. }
  531. }
  532. // 指标编码去重, 指标编码+日期数据去重
  533. indexOB := new(models.BaseFromNationalStatisticsIndex)
  534. indexCond := ``
  535. indexPars := make([]interface{}, 0)
  536. indexList, e := indexOB.GetItemsByCondition(indexCond, indexPars, []string{"index_code"}, "")
  537. if e != nil {
  538. err = fmt.Errorf("获取指标列表失败, Err: %s", e.Error())
  539. return
  540. }
  541. indexExistMap := make(map[string]bool)
  542. for _, v := range indexList {
  543. indexExistMap[v.IndexCode] = true
  544. }
  545. dataOB := new(models.BaseFromNationalStatisticsData)
  546. dataCond := ``
  547. dataPars := make([]interface{}, 0)
  548. dataList, e := dataOB.GetItemsByCondition(dataCond, dataPars, []string{"index_code", "data_time"}, "")
  549. if e != nil {
  550. err = fmt.Errorf("获取指标数据列表失败, Err: %s", e.Error())
  551. return
  552. }
  553. dataExistMap := make(map[string]bool)
  554. for _, v := range dataList {
  555. dataExistMap[fmt.Sprintf("%s%s", v.IndexCode, v.DataTime.Format(utils.FormatDate))] = true
  556. }
  557. // 遍历XY轴
  558. indexDataList := make([]*models.SaveNationalStatisticsIndexAndDataReq, 0)
  559. indexDataMap := make(map[string][]*models.BaseFromNationalStatisticsData)
  560. // 遍历X轴-地区
  561. for _, reg := range regNodes {
  562. // 指标: dbcode+指标code+地区code
  563. indexCode := fmt.Sprintf("%s%s%s", dbCode, quota.Code, reg.Code)
  564. r := new(models.SaveNationalStatisticsIndexAndDataReq)
  565. r.Index = &models.BaseFromNationalStatisticsIndex{
  566. BaseFromNationalStatisticsClassifyId: classifyId,
  567. Dbcode: dbCode,
  568. IndexCode: indexCode,
  569. IndexName: quota.Name,
  570. Frequency: frequency,
  571. Reg: reg.Name,
  572. CreateTime: time.Now().Local(),
  573. ModifyTime: time.Now().Local(),
  574. }
  575. if indexExistMap[indexCode] {
  576. r.IndexExist = true
  577. }
  578. // 遍历Y轴-日期
  579. for _, d := range dateNodes {
  580. k := fmt.Sprintf("%s.%s_%s.%s_%s.%s", "zb", quota.Code, "reg", reg.Code, "sj", d.Code)
  581. v := dataMap[k]
  582. if !v.Data.HasData {
  583. continue
  584. }
  585. // 日期去重
  586. t, e := formatMonth2YearDateCode(d.Code)
  587. if e != nil {
  588. err = fmt.Errorf("格式化日期code失败, Err: %s", e.Error())
  589. return
  590. }
  591. existKey := fmt.Sprintf("%s%s", indexCode, t.Format(utils.FormatDate))
  592. if dataExistMap[existKey] {
  593. continue
  594. }
  595. // 数据map
  596. if indexDataMap[indexCode] == nil {
  597. indexDataMap[indexCode] = make([]*models.BaseFromNationalStatisticsData, 0)
  598. }
  599. indexDataMap[indexCode] = append(indexDataMap[indexCode], &models.BaseFromNationalStatisticsData{
  600. IndexCode: indexCode,
  601. DataTime: t,
  602. Value: v.Data.Data,
  603. CreateTime: time.Now().Local(),
  604. ModifyTime: time.Now().Local(),
  605. })
  606. }
  607. indexDataList = append(indexDataList, r)
  608. }
  609. // 保存指标
  610. for _, v := range indexDataList {
  611. ds := indexDataMap[v.Index.IndexCode]
  612. if ds == nil || (ds != nil && len(ds) == 0) {
  613. continue
  614. }
  615. v.DataList = ds
  616. if e := models.SaveNationalStatisticsIndexAndData(v); e != nil {
  617. err = fmt.Errorf("保存指标和数据失败, Err: %s", e.Error())
  618. return
  619. }
  620. }
  621. }
  622. return
  623. }