mysteel_chemical_data.go 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153
  1. package data_manage
  2. import (
  3. "encoding/json"
  4. "eta/eta_api/models"
  5. "eta/eta_api/models/data_manage"
  6. "eta/eta_api/models/data_manage/request"
  7. "eta/eta_api/models/data_manage/response"
  8. "eta/eta_api/services/alarm_msg"
  9. "eta/eta_api/services/data"
  10. "eta/eta_api/utils"
  11. "fmt"
  12. "github.com/rdlucklib/rdluck_tools/paging"
  13. "github.com/tealeg/xlsx"
  14. "os"
  15. "path/filepath"
  16. "sort"
  17. "strings"
  18. "time"
  19. )
  20. // MysteelChemicalClassify
  21. // @Title 钢联化工数据分类
  22. // @Description 钢联化工分类接口
  23. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  24. // @router /mysteel_chemical/classify [get]
  25. func (this *EdbInfoController) MysteelChemicalClassify() {
  26. br := new(models.BaseResponse).Init()
  27. defer func() {
  28. this.Data["json"] = br
  29. this.ServeJSON()
  30. }()
  31. sysUser := this.SysUser
  32. if sysUser == nil {
  33. br.Msg = "请登录"
  34. br.ErrMsg = "请登录,SysUser Is Empty"
  35. br.Ret = 408
  36. return
  37. }
  38. rootList, err := data_manage.GetBaseFromMysteelChemicalClassifyByParentId(0)
  39. if err != nil && err.Error() != utils.ErrNoRow() {
  40. br.Msg = "获取失败"
  41. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  42. return
  43. }
  44. classifyAll, err := data_manage.GetAllBaseFromMysteelChemicalClassify()
  45. if err != nil && err.Error() != utils.ErrNoRow() {
  46. br.Msg = "获取失败"
  47. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  48. return
  49. }
  50. rootChildMap := make(map[int][]*data_manage.BaseFromMysteelChemicalClassifyItems)
  51. for _, v := range classifyAll {
  52. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId)
  53. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  54. button.AddButton = false //不管有没有权限,二级分类都是没有添加按钮的
  55. v.Button = button
  56. rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v)
  57. }
  58. nodeAll := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  59. for k, v := range rootList {
  60. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  61. rootList[k].Button = button
  62. rootList[k].UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId)
  63. if existItems, ok := rootChildMap[v.BaseFromMysteelChemicalClassifyId]; ok {
  64. v.Children = existItems
  65. } else {
  66. items := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  67. v.Children = items
  68. }
  69. nodeAll = append(nodeAll, v)
  70. }
  71. br.Ret = 200
  72. br.Success = true
  73. br.Msg = "获取成功"
  74. br.Data = nodeAll
  75. }
  76. // MysteelChemicalIndexList
  77. // @Title 钢联化工数据指标列表
  78. // @Description 钢联化工数据指标列表接口
  79. // @Success 200 {object} data_manage.BaseFromMysteelChemicalIndexResp
  80. // @router /mysteel_chemical/index/list [get]
  81. func (this *EdbInfoController) MysteelChemicalIndexList() {
  82. br := new(models.BaseResponse).Init()
  83. defer func() {
  84. this.Data["json"] = br
  85. this.ServeJSON()
  86. }()
  87. sysUser := this.SysUser
  88. if sysUser == nil {
  89. br.Msg = "请登录"
  90. br.ErrMsg = "请登录,SysUser Is Empty"
  91. br.Ret = 408
  92. return
  93. }
  94. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId")
  95. if classifyId <= 0 {
  96. br.Msg = "请选择正确的分类"
  97. return
  98. }
  99. allBaseFromMysteelChemicalIndex, err := data_manage.GetMysteelChemicalIndexByClassifyId(classifyId)
  100. if err != nil && err.Error() != utils.ErrNoRow() {
  101. br.Msg = "获取失败"
  102. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  103. return
  104. }
  105. nodeAll := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  106. for _, v := range allBaseFromMysteelChemicalIndex {
  107. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.BaseFromMysteelChemicalIndexId)
  108. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  109. button.AddButton = false //不管有没有权限,指标都是没有添加按钮的
  110. v.Button = button
  111. nodeAll = append(nodeAll, v)
  112. }
  113. ret := &data_manage.BaseFromMysteelChemicalIndexResp{
  114. List: nodeAll,
  115. }
  116. br.Ret = 200
  117. br.Success = true
  118. br.Msg = "获取成功"
  119. br.Data = ret
  120. }
  121. // AddMysteelChemicalClassify
  122. // @Title 新增分类
  123. // @Description 新增分类接口
  124. // @Param request body data_manage.AddBaseFromMysteelChemicalClassifyReq true "type json string"
  125. // @Success 200 Ret=200 保存成功
  126. // @router /mysteel_chemical/classify/add [post]
  127. func (this *EdbClassifyController) AddMysteelChemicalClassify() {
  128. br := new(models.BaseResponse).Init()
  129. defer func() {
  130. this.Data["json"] = br
  131. this.ServeJSON()
  132. }()
  133. var req request.AddBaseFromMysteelChemicalClassifyReq
  134. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  135. if err != nil {
  136. br.Msg = "参数解析异常!"
  137. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  138. return
  139. }
  140. if req.ClassifyName == "" {
  141. br.Msg = "请输入分类名称"
  142. br.IsSendEmail = false
  143. return
  144. }
  145. req.ClassifyName = utils.TrimStr(req.ClassifyName)
  146. req.ClassifyName = strings.Replace(req.ClassifyName, "/t", "", -1)
  147. //添加指标
  148. mysteelChemicalClassifyInfo, err, errMsg := data.AddMysteelChemicalClassify(req.ClassifyName, req.ParentId, req.Level, this.SysUser.AdminId, this.SysUser.RealName)
  149. if errMsg != `` {
  150. br.Msg = errMsg
  151. br.ErrMsg = errMsg
  152. if err != nil {
  153. br.ErrMsg = errMsg + ";Err:" + err.Error()
  154. } else {
  155. br.IsSendEmail = false
  156. }
  157. return
  158. }
  159. // 如果是添加一级分类,那么默认加上 价格、供应、需求、库存 四个二级分类
  160. if mysteelChemicalClassifyInfo.ParentId == 0 {
  161. secondClassifyList := []string{"价格", "供应", "需求", "库存"}
  162. for _, v := range secondClassifyList {
  163. _, _, tmpErrMsg := data.AddMysteelChemicalClassify(v, mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId, mysteelChemicalClassifyInfo.Level, this.SysUser.AdminId, this.SysUser.RealName)
  164. if tmpErrMsg != `` {
  165. go alarm_msg.SendAlarmMsg("钢联化工-添加一级分类时,默认添加二级分类失败,一级分类名称:"+req.ClassifyName+",二级分类名称:"+v+", Err:"+tmpErrMsg, 3)
  166. return
  167. }
  168. }
  169. }
  170. resp := response.AddMysteelChemicalClassifyResp{
  171. BaseFromMysteelChemicalClassifyId: mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId,
  172. UniqueCode: fmt.Sprint(mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId),
  173. }
  174. br.Data = resp
  175. br.Ret = 200
  176. br.Msg = "保存成功"
  177. br.Success = true
  178. br.IsAddLog = true
  179. }
  180. // EditMysteelChemicalClassify
  181. // @Title 修改分类
  182. // @Description 修改分类接口
  183. // @Param request body data_manage.EditBaseFromMysteelChemicalClassifyReq true "type json string"
  184. // @Success 200 Ret=200 修改成功
  185. // @router /mysteel_chemical/classify/edit [post]
  186. func (this *EdbClassifyController) EditMysteelChemicalClassify() {
  187. br := new(models.BaseResponse).Init()
  188. defer func() {
  189. this.Data["json"] = br
  190. this.ServeJSON()
  191. }()
  192. var req request.EditBaseFromMysteelChemicalClassifyReq
  193. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  194. if err != nil {
  195. br.Msg = "参数解析异常!"
  196. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  197. return
  198. }
  199. if req.ClassifyName == "" {
  200. br.Msg = "请输入分类名称"
  201. br.IsSendEmail = false
  202. return
  203. }
  204. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  205. br.Msg = "参数错误"
  206. br.IsSendEmail = false
  207. return
  208. }
  209. err, errMsg := data.EditMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, req.ClassifyName, this.SysUser)
  210. if errMsg != `` {
  211. br.Msg = errMsg
  212. br.ErrMsg = errMsg
  213. if err != nil {
  214. br.ErrMsg = errMsg + ";Err:" + err.Error()
  215. } else {
  216. br.IsSendEmail = false
  217. }
  218. return
  219. }
  220. br.Ret = 200
  221. br.Msg = "保存成功"
  222. br.Success = true
  223. br.IsAddLog = true
  224. }
  225. // MoveMysteelChemicalClassify
  226. // @Title 分类移动接口
  227. // @Description 分类移动接口
  228. // @Success 200 {object} data_manage.MoveBaseFromMysteelChemicalClassifyReq
  229. // @router /mysteel_chemical/classify/move [post]
  230. func (this *EdbClassifyController) MoveMysteelChemicalClassify() {
  231. br := new(models.BaseResponse).Init()
  232. defer func() {
  233. this.Data["json"] = br
  234. this.ServeJSON()
  235. }()
  236. sysUser := this.SysUser
  237. if sysUser == nil {
  238. br.Msg = "请登录"
  239. br.ErrMsg = "请登录,SysUser Is Empty"
  240. br.Ret = 408
  241. return
  242. }
  243. var req request.MoveBaseFromMysteelChemicalClassifyReq
  244. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  245. if err != nil {
  246. br.Msg = "参数解析异常!"
  247. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  248. return
  249. }
  250. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  251. br.Msg = "参数错误"
  252. br.ErrMsg = "分类id小于等于0"
  253. br.IsSendEmail = false
  254. return
  255. }
  256. err, errMsg := data.MoveMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, req.ParentBaseFromMysteelChemicalClassifyId, req.PrevBaseFromMysteelChemicalClassifyId, req.NextBaseFromMysteelChemicalClassifyId, this.SysUser)
  257. if errMsg != `` {
  258. br.Msg = errMsg
  259. br.ErrMsg = errMsg
  260. if err != nil {
  261. br.ErrMsg = err.Error()
  262. } else {
  263. br.IsSendEmail = false
  264. }
  265. return
  266. }
  267. br.Ret = 200
  268. br.Success = true
  269. br.IsAddLog = true
  270. br.Msg = "移动成功"
  271. }
  272. // DeleteMysteelChemicalClassify
  273. // @Title 删除分类
  274. // @Description 删除分类接口
  275. // @Param request body data_manage.DelBaseFromMysteelChemicalClassifyReq true "type json string"
  276. // @Success 200 Ret=200 删除成功
  277. // @router /mysteel_chemical/classify/del [post]
  278. func (this *EdbClassifyController) DeleteMysteelChemicalClassify() {
  279. br := new(models.BaseResponse).Init()
  280. defer func() {
  281. this.Data["json"] = br
  282. this.ServeJSON()
  283. }()
  284. var req request.DelBaseFromMysteelChemicalClassifyReq
  285. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  286. if err != nil {
  287. br.Msg = "参数解析异常!"
  288. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  289. return
  290. }
  291. if req.BaseFromMysteelChemicalClassifyId < 0 {
  292. br.Msg = "参数错误"
  293. br.IsSendEmail = false
  294. return
  295. }
  296. err, errMsg := data.DelMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, this.SysUser)
  297. if errMsg != `` {
  298. br.Msg = errMsg
  299. br.ErrMsg = errMsg
  300. if err != nil {
  301. br.ErrMsg = errMsg + ";Err:" + err.Error()
  302. } else {
  303. br.IsSendEmail = false
  304. }
  305. return
  306. }
  307. br.Ret = 200
  308. br.Msg = "删除成功"
  309. br.Success = true
  310. br.IsAddLog = true
  311. }
  312. // MysteelChemicalSearch
  313. // @Title 钢联化工指标查询
  314. // @Description 钢联化工指标查询
  315. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  316. // @Param Keyword query string true "名称关键词"
  317. // @Success 200 {object} data_manage.LzFrequency
  318. // @router /mysteel_chemical/search [get]
  319. func (this *EdbInfoController) MysteelChemicalSearch() {
  320. br := new(models.BaseResponse).Init()
  321. defer func() {
  322. this.Data["json"] = br
  323. this.ServeJSON()
  324. }()
  325. sysUser := this.SysUser
  326. if sysUser == nil {
  327. br.Msg = "请登录"
  328. br.ErrMsg = "请登录,SysUser Is Empty"
  329. br.Ret = 408
  330. return
  331. }
  332. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId")
  333. var condition string
  334. var pars []interface{}
  335. if classifyId > 0 {
  336. condition += " AND base_from_mysteel_chemical_classify_id=? "
  337. pars = append(pars, classifyId)
  338. }
  339. keyword := this.GetString("Keyword")
  340. if keyword != `` {
  341. condition += " AND (index_name like ? OR index_code like ?) "
  342. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  343. }
  344. list, err := data_manage.GetMysteelChemicalIndex(condition, pars)
  345. if err != nil {
  346. br.Msg = "获取失败"
  347. br.ErrMsg = "获取失败,Err:" + err.Error()
  348. return
  349. }
  350. for _, v := range list {
  351. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.Id)
  352. }
  353. br.Ret = 200
  354. br.Success = true
  355. br.Msg = "获取成功"
  356. br.Data = list
  357. }
  358. // MysteelChemicalFrequency
  359. // @Title 钢联化工数据频度
  360. // @Description 钢联化工数据频度接口
  361. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  362. // @Param BaseFromMysteelChemicalIndexId query int true "钢联化工指标id"
  363. // @Param Keyword query string true "名称关键词"
  364. // @Success 200 {object} data_manage.LzFrequency
  365. // @router /mysteel_chemical/frequency [get]
  366. func (this *EdbInfoController) MysteelChemicalFrequency() {
  367. br := new(models.BaseResponse).Init()
  368. defer func() {
  369. this.Data["json"] = br
  370. this.ServeJSON()
  371. }()
  372. sysUser := this.SysUser
  373. if sysUser == nil {
  374. br.Msg = "请登录"
  375. br.ErrMsg = "请登录,SysUser Is Empty"
  376. br.Ret = 408
  377. return
  378. }
  379. var condition string
  380. var pars []interface{}
  381. var classifyId int
  382. var frequency string
  383. // 指定指标
  384. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  385. if indexId > 0 {
  386. condition += " AND base_from_mysteel_chemical_index_id = ? "
  387. pars = append(pars, indexId)
  388. } else {
  389. classifyId, _ = this.GetInt("BaseFromMysteelChemicalClassifyId")
  390. if classifyId <= 0 {
  391. br.Msg = "请选择分类"
  392. br.ErrMsg = "请选择分类"
  393. return
  394. }
  395. }
  396. if classifyId > 0 {
  397. condition += " AND base_from_mysteel_chemical_classify_id=? "
  398. pars = append(pars, classifyId)
  399. }
  400. if frequency != `` {
  401. condition += " AND frequency=? "
  402. pars = append(pars, frequency)
  403. }
  404. // 关键词搜索
  405. keyword := this.GetString("Keyword")
  406. if keyword != `` {
  407. condition += " AND (index_name like ? OR index_code like ?) "
  408. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  409. }
  410. edbFrequencyList := make([]string, 0)
  411. frequencyList, err := data_manage.GetMysteelChemicalFrequency(condition, pars)
  412. for _, unit := range frequencyList {
  413. edbFrequencyList = append(edbFrequencyList, unit.Frequency)
  414. }
  415. edbFrequencyList = sortEdbFrequency(edbFrequencyList)
  416. for i, v := range edbFrequencyList {
  417. frequencyList[i].Frequency = v
  418. }
  419. if err != nil {
  420. br.Msg = "获取频度失败"
  421. br.ErrMsg = "获取频度失败,Err:" + err.Error()
  422. return
  423. }
  424. br.Ret = 200
  425. br.Success = true
  426. br.Msg = "获取成功"
  427. br.Data = frequencyList
  428. }
  429. // MysteelChemicalData
  430. // @Title 获取钢联化工数据
  431. // @Description 获取钢联化工数据
  432. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  433. // @Param BaseFromMysteelChemicalIndexId query int true "钢联化工指标id"
  434. // @Param Frequency query string true "频度名称"
  435. // @Param Keyword query string true "名称关键词"
  436. // @Param PageSize query int true "每页数据条数"
  437. // @Param CurrentIndex query int true "当前页页码,从1开始"
  438. // @Success 200 {object} data_manage.LzFrequency
  439. // @router /mysteel_chemical/data [get]
  440. func (this *EdbInfoController) MysteelChemicalData() {
  441. br := new(models.BaseResponse).Init()
  442. defer func() {
  443. this.Data["json"] = br
  444. this.ServeJSON()
  445. }()
  446. sysUser := this.SysUser
  447. if sysUser == nil {
  448. br.Msg = "请登录"
  449. br.ErrMsg = "请登录,SysUser Is Empty"
  450. br.Ret = 408
  451. return
  452. }
  453. pageSize, _ := this.GetInt("PageSize")
  454. currentIndex, _ := this.GetInt("CurrentIndex")
  455. var startSize int
  456. if pageSize <= 0 {
  457. pageSize = utils.PageSize20
  458. }
  459. if currentIndex <= 0 {
  460. currentIndex = 1
  461. }
  462. startSize = utils.StartIndex(currentIndex, pageSize)
  463. var condition string
  464. var pars []interface{}
  465. var classifyId int
  466. var frequency string
  467. // 指定指标
  468. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  469. if indexId > 0 {
  470. condition += " AND base_from_mysteel_chemical_index_id = ? "
  471. pars = append(pars, indexId)
  472. } else {
  473. classifyId, _ = this.GetInt("BaseFromMysteelChemicalClassifyId")
  474. if classifyId <= 0 {
  475. br.Msg = "请选择分类"
  476. br.ErrMsg = "请选择分类"
  477. br.IsSendEmail = false
  478. return
  479. }
  480. frequency = this.GetString("Frequency")
  481. if frequency == "" {
  482. br.Msg = "请选择频度"
  483. br.ErrMsg = "请选择频度"
  484. br.IsSendEmail = false
  485. return
  486. }
  487. }
  488. if classifyId > 0 {
  489. condition += " AND base_from_mysteel_chemical_classify_id=? "
  490. pars = append(pars, classifyId)
  491. }
  492. if frequency != `` {
  493. condition += " AND frequency=? "
  494. pars = append(pars, frequency)
  495. }
  496. // 关键词搜索
  497. keyword := this.GetString("Keyword")
  498. if keyword != `` {
  499. condition += " AND (index_name like ? OR index_code like ?) "
  500. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  501. }
  502. //获取指标
  503. list, err := data_manage.GetMysteelChemicalIndex(condition, pars)
  504. if err != nil {
  505. br.Msg = "获取数据失败"
  506. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  507. return
  508. }
  509. for k, v := range list {
  510. total, err := data_manage.GetMysteelChemicalIndexDataCount(v.IndexCode)
  511. page := paging.GetPaging(currentIndex, pageSize, total)
  512. dataList, err := data_manage.GetMysteelChemicalIndexData(v.IndexCode, startSize, pageSize)
  513. if err != nil {
  514. br.Msg = "获取数据失败"
  515. br.ErrMsg = "获取指标数据失败,Err:" + err.Error()
  516. return
  517. }
  518. if dataList == nil {
  519. dataList = make([]*data_manage.MysteelChemicalData, 0)
  520. }
  521. list[k].DataList = dataList
  522. list[k].Paging = page
  523. }
  524. br.Ret = 200
  525. br.Success = true
  526. br.Msg = "获取成功"
  527. br.Data = list
  528. }
  529. // MysteelChemicalExport
  530. // @Title 导出钢联化工数据
  531. // @Description 导出钢联化工数据
  532. // @Param BaseFromMysteelChemicalClassifyId query int true "分类"
  533. // @Param BaseFromMysteelChemicalIndexId query string true "钢联化工指标id"
  534. // @Param Keyword query string true "名称关键词"
  535. // @Success 200 导出成功
  536. // @router /mysteel_chemical/export/dataList [get]
  537. func (this *EdbClassifyController) MysteelChemicalExport() {
  538. br := new(models.BaseResponse).Init()
  539. defer func() {
  540. this.Data["json"] = br
  541. this.ServeJSON()
  542. }()
  543. sysUser := this.SysUser
  544. if sysUser == nil {
  545. br.Msg = "请重新登录"
  546. return
  547. }
  548. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId") //分类
  549. frequency := this.GetString("Frequency")
  550. dir, _ := os.Executable()
  551. exPath := filepath.Dir(dir)
  552. downLoadnFilePath := exPath + "/" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
  553. xlsxFile := xlsx.NewFile()
  554. var condition string
  555. var pars []interface{}
  556. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  557. if indexId > 0 {
  558. condition += " AND base_from_mysteel_chemical_index_id = ? "
  559. pars = append(pars, indexId)
  560. }
  561. if classifyId > 0 {
  562. condition += " AND base_from_mysteel_chemical_classify_id=? "
  563. pars = append(pars, classifyId)
  564. }
  565. if frequency != `` {
  566. condition += " AND frequency=? "
  567. pars = append(pars, frequency)
  568. }
  569. // 关键词搜索
  570. keyword := this.GetString("Keyword")
  571. if keyword != `` {
  572. condition += " AND (index_name like ? OR index_code like ?) "
  573. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  574. }
  575. var frequencies []string
  576. frequencyList, err := data_manage.GetMysteelChemicalFrequency(condition, pars)
  577. for _, v := range frequencyList {
  578. frequencies = append(frequencies, v.Frequency)
  579. }
  580. fileName := `钢联化工数据`
  581. if frequencies == nil {
  582. sheet, err := xlsxFile.AddSheet("无数据")
  583. if err != nil {
  584. br.Msg = "新增Sheet失败"
  585. br.ErrMsg = "新增Sheet失败,Err:" + err.Error()
  586. return
  587. }
  588. rowSecName := sheet.AddRow()
  589. celSecName := rowSecName.AddCell()
  590. celSecName.SetValue("")
  591. }
  592. for _, frequency := range frequencies {
  593. //获取指标
  594. var tmpCondition string
  595. var tmpPars []interface{}
  596. if indexId > 0 {
  597. tmpCondition += " AND base_from_mysteel_chemical_index_id = ? "
  598. tmpPars = append(pars, indexId)
  599. }
  600. if classifyId > 0 {
  601. tmpCondition += " AND base_from_mysteel_chemical_classify_id=? "
  602. tmpPars = append(pars, classifyId)
  603. }
  604. if frequency != `` {
  605. tmpCondition += " AND frequency=? "
  606. tmpPars = append(pars, frequency)
  607. }
  608. // 关键词搜索
  609. keyword := this.GetString("Keyword")
  610. if keyword != `` {
  611. tmpCondition += " AND (index_name like ? OR index_code like ?) "
  612. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  613. }
  614. //获取指标
  615. secNameList, err := data_manage.GetMysteelChemicalIndex(tmpCondition, tmpPars)
  616. if err != nil {
  617. fmt.Println("获取数据失败,Err:" + err.Error())
  618. return
  619. }
  620. if len(secNameList) <= 0 {
  621. fmt.Println("secNameList长度为0")
  622. return
  623. }
  624. sheetNew, err := xlsxFile.AddSheet(frequency)
  625. if err != nil {
  626. fmt.Println("新增Sheet失败", err.Error())
  627. return
  628. }
  629. //sheetNew.SetColWidth()
  630. //获取指标数据
  631. windRow := sheetNew.AddRow()
  632. secNameRow := sheetNew.AddRow()
  633. indexCodeRow := sheetNew.AddRow()
  634. frequencyRow := sheetNew.AddRow()
  635. unitRow := sheetNew.AddRow()
  636. lastModifyDateRow := sheetNew.AddRow()
  637. //获取分类下指标最大数据量
  638. dataMax, err := data_manage.GetBaseFromMysteelChemicalDataMaxCount(classifyId)
  639. if err != nil {
  640. fmt.Println("获取指标最大数据量失败", err.Error())
  641. return
  642. }
  643. fmt.Println("dataMax:", dataMax)
  644. setRowIndex := 6
  645. for k, sv := range secNameList {
  646. //获取数据
  647. dataList, err := data_manage.GetMysteelChemicalIndexDataByCode(sv.IndexCode)
  648. if err != nil {
  649. br.Msg = "获取数据失败"
  650. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  651. return
  652. }
  653. if len(dataList) > 0 {
  654. windRow.AddCell().SetValue("钢联")
  655. secNameRow.AddCell().SetValue("指标名称")
  656. indexCodeRow.AddCell().SetValue("指标ID")
  657. frequencyRow.AddCell().SetValue("频率")
  658. unitRow.AddCell().SetValue("单位")
  659. lastModifyDateRow.AddCell().SetValue("更新时间")
  660. secNameRow.AddCell().SetValue(sv.IndexName)
  661. indexCodeRow.AddCell().SetValue(sv.IndexCode)
  662. frequencyRow.AddCell().SetValue(sv.FrequencyName)
  663. unitRow.AddCell().SetValue(sv.UnitName)
  664. lastModifyDateRow.AddCell().SetValue(sv.UpdateTime)
  665. windRow.AddCell()
  666. windRow.AddCell()
  667. secNameRow.AddCell()
  668. indexCodeRow.AddCell()
  669. frequencyRow.AddCell()
  670. unitRow.AddCell()
  671. lastModifyDateRow.AddCell()
  672. min := k * 3
  673. sheetNew.SetColWidth(min, min, 15)
  674. if len(dataList) <= 0 {
  675. for n := 0; n < dataMax; n++ {
  676. rowIndex := setRowIndex + n
  677. row := sheetNew.Row(rowIndex)
  678. row.AddCell()
  679. row.AddCell()
  680. row.AddCell()
  681. }
  682. } else {
  683. endRowIndex := 0
  684. for rk, dv := range dataList {
  685. rowIndex := setRowIndex + rk
  686. row := sheetNew.Row(rowIndex)
  687. displayDate, _ := time.Parse(utils.FormatDate, dv.DataTime)
  688. displayDateCell := row.AddCell()
  689. style := new(xlsx.Style)
  690. style.ApplyAlignment = true
  691. style.Alignment.WrapText = true
  692. displayDateCell.SetStyle(style)
  693. displayDateCell.SetDate(displayDate)
  694. row.AddCell().SetValue(dv.InputValue)
  695. row.AddCell()
  696. endRowIndex = rowIndex
  697. }
  698. if len(dataList) < dataMax {
  699. dataLen := dataMax - len(dataList)
  700. for n := 0; n < dataLen; n++ {
  701. rowIndex := (endRowIndex + 1) + n
  702. row := sheetNew.Row(rowIndex)
  703. row.AddCell()
  704. row.AddCell()
  705. row.AddCell()
  706. }
  707. }
  708. }
  709. }
  710. }
  711. }
  712. err = xlsxFile.Save(downLoadnFilePath)
  713. if err != nil {
  714. //有指标无数据时先导出一遍空表
  715. sheet, err := xlsxFile.AddSheet("无数据")
  716. if err != nil {
  717. br.Msg = "新增Sheet失败"
  718. br.ErrMsg = "新增Sheet失败,Err:" + err.Error()
  719. return
  720. }
  721. rowSecName := sheet.AddRow()
  722. celSecName := rowSecName.AddCell()
  723. celSecName.SetValue("")
  724. err = xlsxFile.Save(downLoadnFilePath)
  725. if err != nil {
  726. br.Msg = "保存文件失败"
  727. br.ErrMsg = "保存文件失败"
  728. return
  729. }
  730. }
  731. fileName += time.Now().Format("06.01.02") + `.xlsx` //文件名称
  732. this.Ctx.Output.Download(downLoadnFilePath, fileName)
  733. defer func() {
  734. os.Remove(downLoadnFilePath)
  735. }()
  736. br.Ret = 200
  737. br.Success = true
  738. br.Msg = "success"
  739. }
  740. // AddMysteelChemical
  741. // @Title 新增钢联化工指标
  742. // @Description 新增钢联化工指标接口
  743. // @Param request body data_manage.AddEdbClassifyReq true "type json string"
  744. // @Success 200 Ret=200 保存成功
  745. // @router /mysteel_chemical/add [post]
  746. func (this *EdbClassifyController) AddMysteelChemical() {
  747. br := new(models.BaseResponse).Init()
  748. defer func() {
  749. this.Data["json"] = br
  750. this.ServeJSON()
  751. }()
  752. var req request.AddBaseFromMysteelChemicalReq
  753. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  754. if err != nil {
  755. br.Msg = "参数解析异常!"
  756. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  757. return
  758. }
  759. resp := new(response.AddMysteelChemicalIndexInfoResp)
  760. for i, v := range req.List {
  761. if v.IndexCode == "" {
  762. br.Msg = "请输入指标ID"
  763. br.IsSendEmail = false
  764. return
  765. }
  766. if v.BaseFromMysteelChemicalClassifyId <= 0 {
  767. br.Msg = "请选择分类"
  768. br.IsSendEmail = false
  769. return
  770. }
  771. v.IndexCode = utils.TrimStr(v.IndexCode)
  772. v.IndexCode = strings.Replace(v.IndexCode, "\t", "", -1)
  773. //添加指标
  774. mysteelChemicalIndexInfo, err, errMsg := data.AddMysteelChemicalIndex(v.BaseFromMysteelChemicalClassifyId, strings.ToUpper(v.IndexCode), v.UpdateWeek, v.UpdateTime, this.SysUser.AdminId, this.SysUser.RealName)
  775. if errMsg != `` {
  776. br.Msg = errMsg
  777. br.ErrMsg = errMsg
  778. if err != nil {
  779. br.ErrMsg = errMsg + ";Err:" + err.Error()
  780. } else {
  781. br.IsSendEmail = false
  782. }
  783. return
  784. }
  785. if i == 0 {
  786. resp.BaseFromMysteelChemicalIndexId = mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId
  787. resp.IndexCode = mysteelChemicalIndexInfo.IndexCode
  788. resp.UniqueCode = fmt.Sprint(mysteelChemicalIndexInfo.BaseFromMysteelChemicalClassifyId, "_", mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId)
  789. }
  790. }
  791. br.Data = resp
  792. br.Ret = 200
  793. br.Msg = "保存成功"
  794. br.Success = true
  795. br.IsAddLog = true
  796. }
  797. func sortEdbFrequency(frequencyList []string) (newFrequencyList []string) {
  798. var frequencyMap1 = map[string]int{
  799. "日度": 1,
  800. "周度": 2,
  801. "旬度": 3,
  802. "月度": 4,
  803. "季度": 5,
  804. "半年度": 6,
  805. "年度": 7,
  806. }
  807. var frequencyMap2 = map[int]string{
  808. 1: "日度",
  809. 2: "周度",
  810. 3: "旬度",
  811. 4: "月度",
  812. 5: "季度",
  813. 6: "半年度",
  814. 7: "年度",
  815. }
  816. idList := make([]int, 0)
  817. for _, frequency := range frequencyList {
  818. idList = append(idList, frequencyMap1[frequency])
  819. }
  820. sort.Ints(idList)
  821. for _, id := range idList {
  822. newFrequencyList = append(newFrequencyList, frequencyMap2[id])
  823. }
  824. return
  825. }
  826. // EditMysteelChemical
  827. // @Title 编辑钢联化工指标
  828. // @Description 新增钢联化工指标接口
  829. // @Param request body data_manage.AddEdbClassifyReq true "type json string"
  830. // @Success 200 Ret=200 保存成功
  831. // @router /mysteel_chemical/edit [post]
  832. func (this *EdbClassifyController) EditMysteelChemical() {
  833. br := new(models.BaseResponse).Init()
  834. defer func() {
  835. this.Data["json"] = br
  836. this.ServeJSON()
  837. }()
  838. var req request.AddBaseFromMysteelChemicalReqItem
  839. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  840. if err != nil {
  841. br.Msg = "参数解析异常!"
  842. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  843. return
  844. }
  845. if req.BaseFromMysteelChemicalIndexId <= 0 {
  846. br.Msg = "请选择指标"
  847. br.IsSendEmail = false
  848. return
  849. }
  850. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  851. br.Msg = "请选择分类"
  852. br.IsSendEmail = false
  853. return
  854. }
  855. //编辑指标
  856. mysteelChemicalIndexInfo, err, errMsg := data.EditMysteelChemicalIndex(req.BaseFromMysteelChemicalIndexId, req.BaseFromMysteelChemicalClassifyId, req.UpdateWeek, req.UpdateTime, this.SysUser)
  857. if errMsg != `` {
  858. br.Msg = errMsg
  859. br.ErrMsg = errMsg
  860. if err != nil {
  861. br.ErrMsg = errMsg + ";Err:" + err.Error()
  862. } else {
  863. br.IsSendEmail = false
  864. }
  865. return
  866. }
  867. resp := response.AddMysteelChemicalIndexInfoResp{
  868. BaseFromMysteelChemicalIndexId: mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId,
  869. IndexCode: mysteelChemicalIndexInfo.IndexCode,
  870. UniqueCode: fmt.Sprint(mysteelChemicalIndexInfo.BaseFromMysteelChemicalClassifyId, "_", mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId),
  871. }
  872. br.Data = resp
  873. br.Ret = 200
  874. br.Msg = "保存成功"
  875. br.Success = true
  876. br.IsAddLog = true
  877. }
  878. // DeleteMysteelChemical
  879. // @Title 删除指标
  880. // @Description 删除指标接口
  881. // @Param request body data_manage.DelBaseFromMysteelChemicalReq true "type json string"
  882. // @Success 200 Ret=200 删除成功
  883. // @router /mysteel_chemical/del [post]
  884. func (this *EdbClassifyController) DeleteMysteelChemical() {
  885. br := new(models.BaseResponse).Init()
  886. defer func() {
  887. this.Data["json"] = br
  888. this.ServeJSON()
  889. }()
  890. var req request.DelBaseFromMysteelChemicalReq
  891. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  892. if err != nil {
  893. br.Msg = "参数解析异常!"
  894. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  895. return
  896. }
  897. if req.BaseFromMysteelChemicalIndexId < 0 {
  898. br.Msg = "参数错误"
  899. br.IsSendEmail = false
  900. return
  901. }
  902. err, errMsg := data.DelMysteelChemical(req.BaseFromMysteelChemicalIndexId, this.SysUser)
  903. if errMsg != `` {
  904. br.Msg = errMsg
  905. br.ErrMsg = errMsg
  906. if err != nil {
  907. br.ErrMsg = errMsg + ";Err:" + err.Error()
  908. } else {
  909. br.IsSendEmail = false
  910. }
  911. return
  912. }
  913. br.Ret = 200
  914. br.Msg = "删除成功"
  915. br.Success = true
  916. br.IsAddLog = true
  917. }
  918. // MoveMysteelChemical
  919. // @Title 指标移动接口
  920. // @Description 分类移动接口
  921. // @Success 200 {object} data_manage.MoveBaseFromMysteelChemicalReq
  922. // @router /mysteel_chemical/move [post]
  923. func (this *EdbClassifyController) MoveMysteelChemical() {
  924. br := new(models.BaseResponse).Init()
  925. defer func() {
  926. this.Data["json"] = br
  927. this.ServeJSON()
  928. }()
  929. sysUser := this.SysUser
  930. if sysUser == nil {
  931. br.Msg = "请登录"
  932. br.ErrMsg = "请登录,SysUser Is Empty"
  933. br.Ret = 408
  934. return
  935. }
  936. var req request.MoveBaseFromMysteelChemicalReq
  937. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  938. if err != nil {
  939. br.Msg = "参数解析异常!"
  940. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  941. return
  942. }
  943. if req.BaseFromMysteelChemicalIndexId <= 0 {
  944. br.Msg = "参数错误"
  945. br.ErrMsg = "指标id小于等于0"
  946. br.IsSendEmail = false
  947. return
  948. }
  949. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  950. br.Msg = "请选择分类"
  951. br.ErrMsg = "请选择分类"
  952. br.IsSendEmail = false
  953. return
  954. }
  955. err, errMsg := data.MoveMysteelChemical(req.BaseFromMysteelChemicalIndexId, req.BaseFromMysteelChemicalClassifyId, req.PrevBaseFromMysteelChemicalIndexId, req.NextBaseFromMysteelChemicalIndexId, this.SysUser)
  956. if errMsg != `` {
  957. br.Msg = errMsg
  958. br.ErrMsg = errMsg
  959. if err != nil {
  960. br.ErrMsg = err.Error()
  961. } else {
  962. br.IsSendEmail = false
  963. }
  964. return
  965. }
  966. br.Ret = 200
  967. br.Success = true
  968. br.IsAddLog = true
  969. br.Msg = "移动成功"
  970. }
  971. // MysteelChemicalDetail
  972. // @Title 钢联化工指标详情
  973. // @Description 钢联化工指标详情接口
  974. // @Param BaseFromMysteelChemicalIndexId query string true "钢联化工指标id"
  975. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  976. // @router /mysteel_chemical/detail [get]
  977. func (this *EdbInfoController) MysteelChemicalDetail() {
  978. br := new(models.BaseResponse).Init()
  979. defer func() {
  980. this.Data["json"] = br
  981. this.ServeJSON()
  982. }()
  983. sysUser := this.SysUser
  984. if sysUser == nil {
  985. br.Msg = "请登录"
  986. br.ErrMsg = "请登录,SysUser Is Empty"
  987. br.Ret = 408
  988. return
  989. }
  990. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  991. if indexId <= 0 {
  992. br.Msg = "请选择指标"
  993. br.IsSendEmail = false
  994. return
  995. }
  996. item, err := data_manage.GetBaseFromMysteelChemicalIndexByIndexId(indexId)
  997. if err != nil {
  998. br.Msg = "获取失败"
  999. br.ErrMsg = "获取失败,Err:" + err.Error()
  1000. }
  1001. if item.UpdateTime2 != `` {
  1002. item.UpdateTime = item.UpdateTime + "," + item.UpdateTime2
  1003. }
  1004. br.Ret = 200
  1005. br.Success = true
  1006. br.Msg = "获取成功"
  1007. br.Data = item
  1008. }
  1009. // MysteelChemicalDetail
  1010. // @Title 钢联化工指标详情
  1011. // @Description 钢联化工指标详情接口
  1012. // @Param BaseFromMysteelChemicalIndexId query string true "钢联化工指标id"
  1013. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  1014. // @router /mysteel_chemical/refresh [get]
  1015. func (this *EdbInfoController) MysteelChemicalRefresh() {
  1016. br := new(models.BaseResponse).Init()
  1017. defer func() {
  1018. this.Data["json"] = br
  1019. this.ServeJSON()
  1020. }()
  1021. sysUser := this.SysUser
  1022. if sysUser == nil {
  1023. br.Msg = "请登录"
  1024. br.ErrMsg = "请登录,SysUser Is Empty"
  1025. br.Ret = 408
  1026. return
  1027. }
  1028. // 每次刷新都需要两分钟的等待时间
  1029. setNxKey := utils.CACHE_KEY_MYSTEEL_REFRESH
  1030. if !utils.Rc.SetNX(setNxKey, 1, 2*60*time.Second) {
  1031. br.Msg = "系统正在处理数据生成,请2分钟之后刷新"
  1032. br.IsSendEmail = false
  1033. return
  1034. }
  1035. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  1036. if indexId <= 0 {
  1037. br.Msg = "请选择指标"
  1038. br.IsSendEmail = false
  1039. return
  1040. }
  1041. item, err := data_manage.GetBaseFromMysteelChemicalIndexByIndexId(indexId)
  1042. if err != nil {
  1043. br.Msg = "获取失败"
  1044. br.ErrMsg = "获取失败,Err:" + err.Error()
  1045. }
  1046. if item.MergeFilePath == "" {
  1047. br.Msg = "系统正在处理数据生成,请5分钟之后刷新"
  1048. br.IsSendEmail = false
  1049. return
  1050. }
  1051. go data.CurlRefreshMysteelChemicalData(item.MergeFilePath)
  1052. br.Ret = 200
  1053. br.Success = true
  1054. br.Msg = "刷新成功"
  1055. br.Data = item
  1056. }