oilchem_data.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813
  1. package controllers
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/models"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/system"
  7. "eta/eta_api/services/data"
  8. etaTrialService "eta/eta_api/services/eta_trial"
  9. "eta/eta_api/utils"
  10. "fmt"
  11. "github.com/rdlucklib/rdluck_tools/paging"
  12. "github.com/tealeg/xlsx"
  13. "os"
  14. "path/filepath"
  15. "strconv"
  16. "strings"
  17. "time"
  18. )
  19. // OilchemClassify
  20. // @title 获取隆众资讯分类列表
  21. // @Description 获取隆众资讯分类列表
  22. // @Success 200 {object} models.
  23. // @router /data/oilchem/classify [get]
  24. func (this *TradeCommonController) OilchemClassify() {
  25. br := new(models.BaseResponse).Init()
  26. defer func() {
  27. this.Data["json"] = br
  28. this.ServeJSON()
  29. }()
  30. classifies, e := data_manage.GetOilchemClassifyList()
  31. if e != nil {
  32. br.Msg = "获取失败"
  33. br.ErrMsg = "获取隆众资讯分类数据分类失败, Err: " + e.Error()
  34. return
  35. }
  36. br.Data = classifies
  37. br.Ret = 200
  38. br.Success = true
  39. br.Msg = "获取成功"
  40. }
  41. // OilchemIndexList
  42. // @title 获取隆众资讯指标列表
  43. // @Description 获取隆众资讯指标列表
  44. // @Success 200 {object} models.
  45. // @router /data/oilchem/indexList [get]
  46. func (this *TradeCommonController) OilchemIndexList() {
  47. br := new(models.BaseResponse).Init()
  48. defer func() {
  49. this.Data["json"] = br
  50. this.ServeJSON()
  51. }()
  52. pageSize, _ := this.GetInt("PageSize")
  53. currentIndex, _ := this.GetInt("CurrentIndex")
  54. var startSize int
  55. if pageSize <= 0 {
  56. pageSize = utils.PageSize20
  57. }
  58. if currentIndex <= 0 {
  59. currentIndex = 1
  60. }
  61. startSize = utils.StartIndex(currentIndex, pageSize)
  62. classifyId, _ := this.GetInt("ClassifyId")
  63. var condition string
  64. var pars []interface{}
  65. if classifyId > 0 {
  66. condition += ` AND b.classify_id=? `
  67. pars = append(pars, classifyId)
  68. }
  69. //keyword := this.GetString("KeyWord")
  70. //if keyword != "" {
  71. // condition += ` AND (index_code =? OR index_name LIKE ?) `
  72. // pars = append(pars, keyword)
  73. // pars = append(pars, "%"+keyword+"%")
  74. //}
  75. indexList, e := data_manage.GetOilchemIndexViewList(condition, pars, startSize, pageSize)
  76. if e != nil {
  77. br.Msg = "获取失败"
  78. br.ErrMsg = "获取隆众资讯分类数据分类失败, Err: " + e.Error()
  79. return
  80. }
  81. total, err := data_manage.GetOilchemIndexViewListCount(condition, pars)
  82. if err != nil {
  83. br.Msg = "获取失败"
  84. br.ErrMsg = "获取隆众资讯分类数据分类失败, Err: " + err.Error()
  85. return
  86. }
  87. page := paging.GetPaging(currentIndex, pageSize, total)
  88. resp := data_manage.BaseFromOilchemIndexListResp{}
  89. resp.List = indexList
  90. resp.Paging = page
  91. br.Data = resp
  92. br.Ret = 200
  93. br.Success = true
  94. br.Msg = "获取成功"
  95. }
  96. // OilchemMineData
  97. // @title 获取隆众资讯详细数据列表
  98. // @Description 获取隆众资讯详细数据接口
  99. // @Param ClassifyId query int true "数据id"
  100. // @Param GroupName query string true "分组名"
  101. // @Param Frequency query string true "频度"
  102. // @Param PageSize query int true "每页数据条数"
  103. // @Param CurrentIndex query int true "当前页页码,从1开始"
  104. // @Success 200 {object} []data_manage.CoalmineDataResp
  105. // @router /data/oilchem/data [get]
  106. func (this *TradeCommonController) OilchemData() {
  107. br := new(models.BaseResponse).Init()
  108. defer func() {
  109. this.Data["json"] = br
  110. this.ServeJSON()
  111. }()
  112. pageSize, _ := this.GetInt("PageSize")
  113. currentIndex, _ := this.GetInt("CurrentIndex")
  114. var startSize int
  115. if pageSize <= 0 {
  116. pageSize = utils.PageSize20
  117. }
  118. if currentIndex <= 0 {
  119. currentIndex = 1
  120. }
  121. startSize = utils.StartIndex(currentIndex, pageSize)
  122. indexCode := this.GetString("IndexCode")
  123. if indexCode == "" {
  124. br.Msg = "请选择指标"
  125. br.ErrMsg = "请选择指标"
  126. return
  127. }
  128. //获取指标
  129. var condition string
  130. var pars []interface{}
  131. condition += ` AND index_code =? `
  132. pars = append(pars, indexCode)
  133. index, err := data_manage.GetOilchemIndexByCode(indexCode)
  134. if err != nil {
  135. br.Msg = "获取数据失败"
  136. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  137. return
  138. }
  139. dataList, err := data_manage.GetOilchemIndexData(condition, pars, startSize, pageSize)
  140. if err != nil {
  141. br.Msg = "获取数据失败"
  142. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  143. return
  144. }
  145. resp := data_manage.BaseFromOilchemIndexList{}
  146. resp.BaseFromOilchemIndexId = index.BaseFromOilchemIndexId
  147. resp.ClassifyId = index.ClassifyId
  148. resp.IndexCode = index.IndexCode
  149. resp.IndexName = index.IndexName
  150. resp.Frequency = index.Frequency
  151. resp.Unit = index.Unit
  152. total, err := data_manage.GetOilchemDataListCount(condition, pars)
  153. page := paging.GetPaging(currentIndex, pageSize, total)
  154. if len(dataList) > 0 {
  155. resp.ModifyTime = dataList[0].ModifyTime
  156. }
  157. resp.Paging = page
  158. resp.DataList = dataList
  159. br.Ret = 200
  160. br.Success = true
  161. br.Msg = "获取成功"
  162. br.Data = resp
  163. }
  164. // CoalSearchList
  165. // @Title 隆众资讯模糊搜索
  166. // @Description 隆众资讯模糊搜索
  167. // @Param Keyword query string ture "关键字搜索"
  168. // @Success 200 {object} models.BaseResponse
  169. // @router /data/oilchem/search [get]
  170. func (this *TradeCommonController) OilchemSearchList() {
  171. br := new(models.BaseResponse).Init()
  172. defer func() {
  173. this.Data["json"] = br
  174. this.ServeJSON()
  175. }()
  176. //关键字
  177. keyword := this.GetString("Keyword")
  178. list, err := data_manage.GetOilchemItemList(keyword)
  179. if err != nil {
  180. br.ErrMsg = "获取失败,Err:" + err.Error()
  181. br.Msg = "获取失败"
  182. return
  183. }
  184. br.Ret = 200
  185. br.Success = true
  186. br.Msg = "获取成功"
  187. br.Data = list
  188. }
  189. // BatchIndexList
  190. // @Title 隆众资讯批量操作指标列表
  191. // @Description 隆众资讯批量操作指标列表
  192. // @Param request body request.SciHqDataBatchListReq true "type json string"
  193. // @Success 200 {object} data_manage.BaseFromMysteelChemicalIndexResp
  194. // @router /data/oilchem/batch_list [post]
  195. func (this *TradeCommonController) BatchIndexList() {
  196. br := new(models.BaseResponse).Init()
  197. defer func() {
  198. this.Data["json"] = br
  199. this.ServeJSON()
  200. }()
  201. var req data_manage.OilchemDataBatchListReq
  202. if err := json.Unmarshal(this.Ctx.Input.RequestBody, &req); err != nil {
  203. br.Msg = "参数解析异常!"
  204. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  205. return
  206. }
  207. var condition string
  208. var pars []interface{}
  209. if req.KeyWord != "" {
  210. condition += `AND (index_code=? OR index_name=?)`
  211. pars = utils.GetLikeKeywordPars(pars, req.KeyWord, 2)
  212. }
  213. if req.IsSelectAll {
  214. if len(req.SelectedId) > 0 {
  215. condition += ` AND base_from_oilchem_index_id NOT IN (` + utils.GetOrmInReplace(len(req.SelectedId)) + `)`
  216. pars = append(pars, req.SelectedId)
  217. }
  218. } else {
  219. if len(req.SelectedId) > 0 {
  220. condition += ` AND base_from_oilchem_index_id IN (` + utils.GetOrmInReplace(len(req.SelectedId)) + `)`
  221. pars = append(pars, req.SelectedId)
  222. }
  223. }
  224. if req.ClassifyId >= 0 {
  225. condition += ` AND classify_id=?`
  226. pars = append(pars, req.ClassifyId)
  227. }
  228. count, err := data_manage.GetOilchemIndexListCount(condition, pars)
  229. if err != nil {
  230. br.Msg = "获取失败"
  231. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  232. return
  233. }
  234. MaxLen := 30
  235. if count > MaxLen {
  236. br.Msg = "批量操作数量不能超过" + strconv.Itoa(MaxLen)
  237. return
  238. }
  239. indexList, err := data_manage.GetOilchemIndexByCondition(condition, pars)
  240. if err != nil {
  241. br.Msg = "获取失败"
  242. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  243. return
  244. }
  245. br.Ret = 200
  246. br.Success = true
  247. br.Msg = "获取成功"
  248. br.Data = indexList
  249. }
  250. // AddCheck
  251. // @Title 新增校验
  252. // @Description 新增校验
  253. // @Param request body request.BusinessDataBatchAddCheckReq true "type json string"
  254. // @Success 200 string "操作成功"
  255. // @router /data/oilchem/add_check [post]
  256. func (this *TradeCommonController) AddCheck() {
  257. br := new(models.BaseResponse).Init()
  258. defer func() {
  259. if br.ErrMsg == "" {
  260. br.IsSendEmail = false
  261. }
  262. this.Data["json"] = br
  263. this.ServeJSON()
  264. }()
  265. sysUser := this.SysUser
  266. if sysUser == nil {
  267. br.Msg = "请登录"
  268. br.ErrMsg = "请登录,SysUser Is Empty"
  269. br.Ret = 408
  270. return
  271. }
  272. var req *data_manage.OilchemDataBatchAddCheckReq
  273. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  274. br.Msg = "参数解析异常!"
  275. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  276. return
  277. }
  278. codeMax := 30
  279. codeLen := len(req.IndexCodes)
  280. if len(req.IndexCodes) == 0 {
  281. br.Msg = "请选择指标"
  282. return
  283. }
  284. if codeLen > codeMax {
  285. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  286. return
  287. }
  288. // 获取指标库已有指标
  289. existsEdb, e := data_manage.GetEdbCodesBySource(utils.DATA_SOURCE_OILCHEM)
  290. if e != nil {
  291. br.Msg = "获取失败"
  292. br.ErrMsg = "获取自有数据已添加的指标失败, Err: " + e.Error()
  293. return
  294. }
  295. existMap := make(map[string]*data_manage.EdbInfo)
  296. for _, v := range existsEdb {
  297. existMap[v.EdbCode] = v
  298. }
  299. // 查询选中的指标
  300. cond := fmt.Sprintf(` AND index_code IN (%s)`, utils.GetOrmInReplace(codeLen))
  301. pars := make([]interface{}, 0)
  302. pars = append(pars, req.IndexCodes)
  303. list, err := data_manage.GetOilchemIndexAndEdbInfoByCondition(cond, pars)
  304. if err != nil {
  305. br.Msg = "获取失败"
  306. br.ErrMsg = "获取卓创红期原始指标列表失败, Err: " + err.Error()
  307. return
  308. }
  309. resp := make([]*data_manage.BaseFromOilchemIndexView, 0)
  310. for _, v := range list {
  311. if v.EdbInfoId > 0 {
  312. v.EdbExist = 1
  313. }
  314. resp = append(resp, v)
  315. }
  316. br.Data = resp
  317. br.Msg = "校验成功"
  318. br.Ret = 200
  319. br.Success = true
  320. }
  321. // NameCheck
  322. // @Title 重名校验
  323. // @Description 批量新增
  324. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  325. // @Success 200 string "操作成功"
  326. // @router /data/oilchem/name_check [post]
  327. func (c *TradeCommonController) NameCheck() {
  328. br := new(models.BaseResponse).Init()
  329. defer func() {
  330. if br.ErrMsg == "" {
  331. br.IsSendEmail = false
  332. }
  333. c.Data["json"] = br
  334. c.ServeJSON()
  335. }()
  336. sysUser := c.SysUser
  337. if sysUser == nil {
  338. br.Msg = "请登录"
  339. br.ErrMsg = "请登录,SysUser Is Empty"
  340. br.Ret = 408
  341. return
  342. }
  343. var req []*data_manage.NameCheckEdbInfoReq
  344. if err := json.Unmarshal(c.Ctx.Input.RequestBody, &req); err != nil {
  345. br.Msg = "参数解析异常!"
  346. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  347. return
  348. }
  349. if len(req) == 0 {
  350. br.Msg = "请选择指标"
  351. return
  352. }
  353. codeMax := 30
  354. codeLen := len(req)
  355. if codeLen > codeMax {
  356. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  357. return
  358. }
  359. type NameCheckResult struct {
  360. EdbCode string
  361. EdbName string
  362. Exist bool
  363. }
  364. indexNames := make([]string, 0)
  365. resp := make([]*NameCheckResult, 0)
  366. for _, v := range req {
  367. v.EdbCode = strings.TrimSpace(v.EdbCode)
  368. if v.EdbCode == "" {
  369. br.Msg = "指标ID不可为空"
  370. return
  371. }
  372. v.EdbName = strings.TrimSpace(v.EdbName)
  373. if v.EdbName == "" {
  374. br.Msg = "请输入指标名称"
  375. return
  376. }
  377. indexNames = append(indexNames, v.EdbName)
  378. resp = append(resp, &NameCheckResult{
  379. EdbCode: v.EdbCode,
  380. EdbName: v.EdbName,
  381. })
  382. dataItems, err := data_manage.GetEdbDataAllByEdbCode(v.EdbCode, utils.DATA_SOURCE_OILCHEM, 0, utils.EDB_DATA_LIMIT)
  383. if err != nil && err.Error() != utils.ErrNoRow() {
  384. br.Msg = "获取失败"
  385. br.ErrMsg = "获取隆众资讯已存在信息失败,Err:" + err.Error()
  386. return
  387. }
  388. if len(dataItems) <= 0 {
  389. respItem, err := data.AddEdbData(utils.DATA_SOURCE_OILCHEM, v.EdbCode, v.Frequency)
  390. if err != nil {
  391. br.Msg = "获取失败"
  392. br.ErrMsg = "获取失败,Err:" + err.Error()
  393. return
  394. }
  395. if respItem.Ret != 200 {
  396. br.Msg = "未搜索到该指标"
  397. br.ErrMsg = respItem.ErrMsg + ";EdbCode:" + v.EdbCode
  398. return
  399. }
  400. }
  401. }
  402. // 重名校验
  403. edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
  404. if e != nil {
  405. br.Msg = "操作失败"
  406. br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
  407. return
  408. }
  409. nameExists := make(map[string]bool)
  410. for _, v := range edbList {
  411. nameExists[v.EdbName] = true
  412. }
  413. if len(nameExists) > 0 {
  414. for _, v := range resp {
  415. v.Exist = nameExists[v.EdbName]
  416. }
  417. }
  418. br.Data = resp
  419. br.Msg = "校验成功"
  420. br.Ret = 200
  421. br.Success = true
  422. }
  423. // BatchAdd
  424. // @Title 批量新增
  425. // @Description 批量新增
  426. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  427. // @Success 200 string "操作成功"
  428. // @router /data/oilchem/batch_add [post]
  429. func (this *TradeCommonController) BatchAdd() {
  430. br := new(models.BaseResponse).Init()
  431. defer func() {
  432. if br.ErrMsg == "" {
  433. br.IsSendEmail = false
  434. }
  435. this.Data["json"] = br
  436. this.ServeJSON()
  437. }()
  438. sysUser := this.SysUser
  439. if sysUser == nil {
  440. br.Msg = "请登录"
  441. br.ErrMsg = "请登录,SysUser Is Empty"
  442. br.Ret = 408
  443. return
  444. }
  445. deleteCache := true
  446. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_OILCHEM_" + strconv.Itoa(sysUser.AdminId)
  447. defer func() {
  448. if deleteCache {
  449. _ = utils.Rc.Delete(cacheKey)
  450. }
  451. }()
  452. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  453. deleteCache = false
  454. br.Msg = "系统处理中,请稍后重试!"
  455. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  456. return
  457. }
  458. var req []*data_manage.AddEdbInfoReq
  459. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  460. br.Msg = "参数解析异常!"
  461. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  462. return
  463. }
  464. if len(req) == 0 {
  465. br.Msg = "请选择指标"
  466. return
  467. }
  468. if len(req) > 30 {
  469. br.Msg = "批量添加指标数量不得超过30个"
  470. return
  471. }
  472. for _, v := range req {
  473. v.EdbCode = strings.TrimSpace(v.EdbCode)
  474. if v.EdbCode == "" {
  475. br.Msg = "指标ID不可为空"
  476. return
  477. }
  478. v.EdbName = strings.TrimSpace(v.EdbName)
  479. if v.EdbName == "" {
  480. br.Msg = "请输入指标名称"
  481. return
  482. }
  483. v.Frequency = strings.TrimSpace(v.Frequency)
  484. if v.Frequency == "" {
  485. br.Msg = "请选择频度"
  486. return
  487. }
  488. v.Unit = strings.TrimSpace(v.Unit)
  489. if v.Unit == "" {
  490. br.Msg = "请输入单位"
  491. return
  492. }
  493. if v.ClassifyId <= 0 {
  494. br.Msg = "请选择分类"
  495. return
  496. }
  497. }
  498. // 限定同一时间最多批量新增30个指标
  499. for _, v := range req {
  500. var r data.OilchemIndexSource2EdbReq
  501. r.EdbCode = v.EdbCode
  502. r.EdbName = v.EdbName
  503. r.Frequency = v.Frequency
  504. r.Unit = v.Unit
  505. r.ClassifyId = v.ClassifyId
  506. r.AdminId = sysUser.AdminId
  507. r.AdminRealName = sysUser.RealName
  508. edbInfo, errMsg, skip, e := data.OilchemIndexSource2Edb(r, this.Lang)
  509. if e != nil {
  510. br.Msg = "操作失败"
  511. if errMsg != "" {
  512. br.Msg = errMsg
  513. }
  514. br.ErrMsg = e.Error()
  515. return
  516. }
  517. if skip {
  518. continue
  519. }
  520. // 试用平台更新用户累计新增指标数
  521. if utils.BusinessCode == utils.BusinessCodeSandbox {
  522. go func() {
  523. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  524. if e != nil {
  525. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  526. utils.FileLog.Info(tips)
  527. return
  528. }
  529. if adminItem.DepartmentName != "ETA试用客户" {
  530. return
  531. }
  532. var ur etaTrialService.EtaTrialUserReq
  533. ur.Mobile = adminItem.Mobile
  534. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  535. }()
  536. }
  537. // 新增操作日志
  538. {
  539. edbLog := new(data_manage.EdbInfoLog)
  540. edbLog.EdbInfoId = edbInfo.EdbInfoId
  541. edbLog.SourceName = edbInfo.SourceName
  542. edbLog.Source = edbInfo.Source
  543. edbLog.EdbCode = edbInfo.EdbCode
  544. edbLog.EdbName = edbInfo.EdbName
  545. edbLog.ClassifyId = edbInfo.ClassifyId
  546. edbLog.SysUserId = sysUser.AdminId
  547. edbLog.SysUserRealName = sysUser.RealName
  548. edbLog.CreateTime = time.Now()
  549. edbLog.Content = string(this.Ctx.Input.RequestBody)
  550. edbLog.Status = "新增指标"
  551. edbLog.Method = this.Ctx.Input.URI()
  552. go data_manage.AddEdbInfoLog(edbLog)
  553. }
  554. }
  555. br.Msg = "操作成功"
  556. br.Ret = 200
  557. br.Success = true
  558. br.IsAddLog = true
  559. }
  560. // ExportoilchemList
  561. // @Title 导出隆众资讯数据
  562. // @Description 导出隆众资讯数据
  563. // @Param request body request.ExportOilchemExcelReq true "type json string"
  564. // @Success 200 导出成功
  565. // @router /data/export/oilchemList [post]
  566. func (this *TradeCommonController) ExportoilchemList() {
  567. br := new(models.BaseResponse).Init()
  568. defer func() {
  569. this.Data["json"] = br
  570. this.ServeJSON()
  571. }()
  572. sysUser := this.SysUser
  573. if sysUser == nil {
  574. br.Msg = "请重新登录"
  575. return
  576. }
  577. var req data_manage.ExportOilchemExcelReq
  578. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  579. br.Msg = "参数解析异常!"
  580. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  581. return
  582. }
  583. secNameList := make([]*models.EdbdataExportList, 0)
  584. dir, _ := os.Executable()
  585. exPath := filepath.Dir(dir)
  586. downLoadnFilePath := exPath + "/" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
  587. xlsxFile := xlsx.NewFile()
  588. var condition string
  589. var pars []interface{}
  590. if req.KeyWord != "" {
  591. condition += `AND (index_code=? OR index_name=?)`
  592. pars = utils.GetLikeKeywordPars(pars, req.KeyWord, 2)
  593. }
  594. if req.IsSelectedAll {
  595. if len(req.IndexCode) > 0 {
  596. condition += ` AND index_code NOT IN (` + utils.GetOrmInReplace(len(req.IndexCode)) + `)`
  597. pars = append(pars, req.IndexCode)
  598. }
  599. } else {
  600. if len(req.IndexCode) > 0 {
  601. condition += ` AND index_code IN (` + utils.GetOrmInReplace(len(req.IndexCode)) + `)`
  602. pars = append(pars, req.IndexCode)
  603. }
  604. }
  605. if req.ClassifyId > 0 {
  606. condition += ` AND classify_id=?`
  607. pars = append(pars, req.ClassifyId)
  608. }
  609. frequencies, err := data_manage.GetOilchemFrequencyByCondition(condition, pars)
  610. if err != nil {
  611. fmt.Println("GetSciFrequency err:", err.Error())
  612. utils.FileLog.Info("GetSciFrequency err:" + err.Error())
  613. return
  614. }
  615. for _, frequency := range frequencies {
  616. //获取指标
  617. secNameList, err := data_manage.GetOilchemIndexByConditionAndFrequency(condition, *frequency, pars)
  618. if err != nil {
  619. fmt.Println("获取数据失败,Err:" + err.Error())
  620. return
  621. }
  622. if len(secNameList) <= 0 {
  623. fmt.Println("secNameList长度为0")
  624. return
  625. }
  626. sheetNew, err := xlsxFile.AddSheet(*frequency)
  627. if err != nil {
  628. fmt.Println("新增Sheet失败", err.Error())
  629. return
  630. }
  631. //sheetNew.SetColWidth()
  632. //获取指标数据
  633. windRow := sheetNew.AddRow()
  634. secNameRow := sheetNew.AddRow()
  635. indexCodeRow := sheetNew.AddRow()
  636. frequencyRow := sheetNew.AddRow()
  637. unitRow := sheetNew.AddRow()
  638. lastModifyDateRow := sheetNew.AddRow()
  639. //获取分类下指标最大数据量
  640. dataMax, err := data_manage.GetOilchemDataMaxCount(req.ClassifyId)
  641. if err != nil {
  642. fmt.Println("获取指标最大数据量失败", err.Error())
  643. return
  644. }
  645. fmt.Println("dataMax:", dataMax)
  646. setRowIndex := 6
  647. for k, sv := range secNameList {
  648. //获取数据
  649. dataList, err := data_manage.GetOilchemIndexDataByCode(sv.IndexCode)
  650. if err != nil {
  651. br.Msg = "获取数据失败"
  652. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  653. return
  654. }
  655. if len(dataList) > 0 {
  656. windRow.AddCell().SetValue("隆众资讯")
  657. secNameRow.AddCell().SetValue("指标名称")
  658. indexCodeRow.AddCell().SetValue("指标ID")
  659. frequencyRow.AddCell().SetValue("频率")
  660. unitRow.AddCell().SetValue("单位")
  661. lastModifyDateRow.AddCell().SetValue("更新时间")
  662. secNameRow.AddCell().SetValue(sv.IndexName)
  663. indexCodeRow.AddCell().SetValue(sv.IndexCode)
  664. frequencyRow.AddCell().SetValue(sv.Frequency)
  665. unitRow.AddCell().SetValue(sv.Unit)
  666. lastModifyDateRow.AddCell().SetValue(sv.ModifyTime)
  667. windRow.AddCell()
  668. windRow.AddCell()
  669. secNameRow.AddCell()
  670. indexCodeRow.AddCell()
  671. frequencyRow.AddCell()
  672. unitRow.AddCell()
  673. lastModifyDateRow.AddCell()
  674. min := k * 3
  675. sheetNew.SetColWidth(min, min, 15)
  676. if len(dataList) <= 0 {
  677. for n := 0; n < dataMax; n++ {
  678. rowIndex := setRowIndex + n
  679. row := sheetNew.Row(rowIndex)
  680. row.AddCell()
  681. row.AddCell()
  682. row.AddCell()
  683. }
  684. } else {
  685. endRowIndex := 0
  686. for rk, dv := range dataList {
  687. rowIndex := setRowIndex + rk
  688. row := sheetNew.Row(rowIndex)
  689. displayDate, _ := time.Parse(utils.FormatDate, dv.DataTime)
  690. displayDateCell := row.AddCell()
  691. style := new(xlsx.Style)
  692. style.ApplyAlignment = true
  693. style.Alignment.WrapText = true
  694. displayDateCell.SetStyle(style)
  695. displayDateCell.SetDate(displayDate)
  696. row.AddCell().SetValue(dv.Value)
  697. row.AddCell()
  698. endRowIndex = rowIndex
  699. }
  700. if len(dataList) < dataMax {
  701. dataLen := dataMax - len(dataList)
  702. for n := 0; n < dataLen; n++ {
  703. rowIndex := (endRowIndex + 1) + n
  704. row := sheetNew.Row(rowIndex)
  705. row.AddCell()
  706. row.AddCell()
  707. row.AddCell()
  708. }
  709. }
  710. }
  711. }
  712. }
  713. }
  714. err = xlsxFile.Save(downLoadnFilePath)
  715. if err != nil {
  716. //有指标无数据时先导出一遍空表
  717. sheet, err := xlsxFile.AddSheet("无数据")
  718. if err != nil {
  719. br.Msg = "新增Sheet失败"
  720. br.ErrMsg = "新增Sheet失败,Err:" + err.Error()
  721. return
  722. }
  723. rowSecName := sheet.AddRow()
  724. celSecName := rowSecName.AddCell()
  725. celSecName.SetValue("")
  726. err = xlsxFile.Save(downLoadnFilePath)
  727. if err != nil {
  728. br.Msg = "保存文件失败"
  729. br.ErrMsg = "保存文件失败"
  730. return
  731. }
  732. }
  733. fileName := `隆众资讯`
  734. if len(secNameList) > 0 {
  735. fileName = secNameList[0].ClassifyName
  736. }
  737. fileName += time.Now().Format("2006.01.02") + `.xlsx` //文件名称
  738. this.Ctx.Output.Download(downLoadnFilePath, fileName)
  739. defer func() {
  740. os.Remove(downLoadnFilePath)
  741. }()
  742. br.Ret = 200
  743. br.Success = true
  744. br.Msg = "success"
  745. }