mysteel_chemical_data.go 52 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877
  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/models/system"
  9. "eta/eta_api/services/alarm_msg"
  10. "eta/eta_api/services/data"
  11. "eta/eta_api/services/data_stat"
  12. etaTrialService "eta/eta_api/services/eta_trial"
  13. "eta/eta_api/utils"
  14. "fmt"
  15. "github.com/shopspring/decimal"
  16. "os"
  17. "path/filepath"
  18. "sort"
  19. "strconv"
  20. "strings"
  21. "time"
  22. "github.com/rdlucklib/rdluck_tools/paging"
  23. "github.com/tealeg/xlsx"
  24. )
  25. // MysteelChemicalClassify
  26. // @Title 上海钢联数据分类
  27. // @Description 上海钢联分类接口
  28. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  29. // @router /mysteel_chemical/classify [get]
  30. func (this *EdbInfoController) MysteelChemicalClassify() {
  31. br := new(models.BaseResponse).Init()
  32. defer func() {
  33. this.Data["json"] = br
  34. this.ServeJSON()
  35. }()
  36. sysUser := this.SysUser
  37. if sysUser == nil {
  38. br.Msg = "请登录"
  39. br.ErrMsg = "请登录,SysUser Is Empty"
  40. br.Ret = 408
  41. return
  42. }
  43. rootList, err := data_manage.GetBaseFromMysteelChemicalClassifyByParentId(0)
  44. if err != nil && !utils.IsErrNoRow(err) {
  45. br.Msg = "获取失败"
  46. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  47. return
  48. }
  49. classifyAll, err := data_manage.GetAllBaseFromMysteelChemicalClassify()
  50. if err != nil && !utils.IsErrNoRow(err) {
  51. br.Msg = "获取失败"
  52. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  53. return
  54. }
  55. rootChildMap := make(map[int][]*data_manage.BaseFromMysteelChemicalClassifyItems)
  56. for _, v := range classifyAll {
  57. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId)
  58. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  59. button.AddButton = false //不管有没有权限,二级分类都是没有添加按钮的
  60. v.Button = button
  61. rootChildMap[v.ParentId] = append(rootChildMap[v.ParentId], v)
  62. }
  63. nodeAll := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  64. for k, v := range rootList {
  65. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  66. rootList[k].Button = button
  67. rootList[k].UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId)
  68. if existItems, ok := rootChildMap[v.BaseFromMysteelChemicalClassifyId]; ok {
  69. v.Children = existItems
  70. } else {
  71. items := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  72. v.Children = items
  73. }
  74. nodeAll = append(nodeAll, v)
  75. }
  76. br.Ret = 200
  77. br.Success = true
  78. br.Msg = "获取成功"
  79. br.Data = nodeAll
  80. }
  81. // MysteelChemicalIndexList
  82. // @Title 上海钢联数据指标列表
  83. // @Description 上海钢联数据指标列表接口
  84. // @Success 200 {object} data_manage.BaseFromMysteelChemicalIndexResp
  85. // @router /mysteel_chemical/index/list [get]
  86. func (this *EdbInfoController) MysteelChemicalIndexList() {
  87. br := new(models.BaseResponse).Init()
  88. defer func() {
  89. this.Data["json"] = br
  90. this.ServeJSON()
  91. }()
  92. sysUser := this.SysUser
  93. if sysUser == nil {
  94. br.Msg = "请登录"
  95. br.ErrMsg = "请登录,SysUser Is Empty"
  96. br.Ret = 408
  97. return
  98. }
  99. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId")
  100. if classifyId <= 0 {
  101. br.Msg = "请选择正确的分类"
  102. return
  103. }
  104. allBaseFromMysteelChemicalIndex, err := data_manage.GetMysteelChemicalIndexByClassifyId(classifyId)
  105. if err != nil && !utils.IsErrNoRow(err) {
  106. br.Msg = "获取失败"
  107. br.ErrMsg = "获取指标信息失败,Err:" + err.Error()
  108. return
  109. }
  110. nodeAll := make([]*data_manage.BaseFromMysteelChemicalClassifyItems, 0)
  111. for _, v := range allBaseFromMysteelChemicalIndex {
  112. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.BaseFromMysteelChemicalIndexId)
  113. button := data.GetMysteelChemicalOpButton(sysUser, v.SysUserId)
  114. button.AddButton = false //不管有没有权限,指标都是没有添加按钮的
  115. v.Button = button
  116. nodeAll = append(nodeAll, v)
  117. }
  118. ret := &data_manage.BaseFromMysteelChemicalIndexResp{
  119. List: nodeAll,
  120. }
  121. br.Ret = 200
  122. br.Success = true
  123. br.Msg = "获取成功"
  124. br.Data = ret
  125. }
  126. // AddMysteelChemicalClassify
  127. // @Title 新增分类
  128. // @Description 新增分类接口
  129. // @Param request body data_manage.AddBaseFromMysteelChemicalClassifyReq true "type json string"
  130. // @Success 200 Ret=200 保存成功
  131. // @router /mysteel_chemical/classify/add [post]
  132. func (this *EdbClassifyController) AddMysteelChemicalClassify() {
  133. br := new(models.BaseResponse).Init()
  134. defer func() {
  135. this.Data["json"] = br
  136. this.ServeJSON()
  137. }()
  138. var req request.AddBaseFromMysteelChemicalClassifyReq
  139. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  140. if err != nil {
  141. br.Msg = "参数解析异常!"
  142. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  143. return
  144. }
  145. if req.ClassifyName == "" {
  146. br.Msg = "请输入分类名称"
  147. br.IsSendEmail = false
  148. return
  149. }
  150. req.ClassifyName = utils.TrimStr(req.ClassifyName)
  151. req.ClassifyName = strings.Replace(req.ClassifyName, "/t", "", -1)
  152. //添加指标
  153. mysteelChemicalClassifyInfo, err, errMsg := data.AddMysteelChemicalClassify(req.ClassifyName, req.ParentId, req.Level, this.SysUser.AdminId, this.SysUser.RealName, this.Lang)
  154. if errMsg != `` {
  155. br.Msg = errMsg
  156. br.ErrMsg = errMsg
  157. if err != nil {
  158. br.ErrMsg = errMsg + ";Err:" + err.Error()
  159. } else {
  160. br.IsSendEmail = false
  161. }
  162. return
  163. }
  164. // 如果是添加一级分类,那么默认加上 价格、供应、需求、库存 四个二级分类
  165. if mysteelChemicalClassifyInfo.ParentId == 0 {
  166. secondClassifyList := []string{"价格", "供应", "需求", "库存"}
  167. for _, v := range secondClassifyList {
  168. _, _, tmpErrMsg := data.AddMysteelChemicalClassify(v, mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId, mysteelChemicalClassifyInfo.Level, this.SysUser.AdminId, this.SysUser.RealName, this.Lang)
  169. if tmpErrMsg != `` {
  170. go alarm_msg.SendAlarmMsg("上海钢联-添加一级分类时,默认添加二级分类失败,一级分类名称:"+req.ClassifyName+",二级分类名称:"+v+", Err:"+tmpErrMsg, 3)
  171. return
  172. }
  173. }
  174. }
  175. resp := response.AddMysteelChemicalClassifyResp{
  176. BaseFromMysteelChemicalClassifyId: mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId,
  177. UniqueCode: fmt.Sprint(mysteelChemicalClassifyInfo.BaseFromMysteelChemicalClassifyId),
  178. }
  179. br.Data = resp
  180. br.Ret = 200
  181. br.Msg = "保存成功"
  182. br.Success = true
  183. br.IsAddLog = true
  184. }
  185. // EditMysteelChemicalClassify
  186. // @Title 修改分类
  187. // @Description 修改分类接口
  188. // @Param request body data_manage.EditBaseFromMysteelChemicalClassifyReq true "type json string"
  189. // @Success 200 Ret=200 修改成功
  190. // @router /mysteel_chemical/classify/edit [post]
  191. func (this *EdbClassifyController) EditMysteelChemicalClassify() {
  192. br := new(models.BaseResponse).Init()
  193. defer func() {
  194. this.Data["json"] = br
  195. this.ServeJSON()
  196. }()
  197. var req request.EditBaseFromMysteelChemicalClassifyReq
  198. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  199. if err != nil {
  200. br.Msg = "参数解析异常!"
  201. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  202. return
  203. }
  204. if req.ClassifyName == "" {
  205. br.Msg = "请输入分类名称"
  206. br.IsSendEmail = false
  207. return
  208. }
  209. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  210. br.Msg = "参数错误"
  211. br.IsSendEmail = false
  212. return
  213. }
  214. err, errMsg := data.EditMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, req.ClassifyName, this.Lang, this.SysUser)
  215. if errMsg != `` {
  216. br.Msg = errMsg
  217. br.ErrMsg = errMsg
  218. if err != nil {
  219. br.ErrMsg = errMsg + ";Err:" + err.Error()
  220. } else {
  221. br.IsSendEmail = false
  222. }
  223. return
  224. }
  225. br.Ret = 200
  226. br.Msg = "保存成功"
  227. br.Success = true
  228. br.IsAddLog = true
  229. }
  230. // MoveMysteelChemicalClassify
  231. // @Title 分类移动接口
  232. // @Description 分类移动接口
  233. // @Success 200 {object} data_manage.MoveBaseFromMysteelChemicalClassifyReq
  234. // @router /mysteel_chemical/classify/move [post]
  235. func (this *EdbClassifyController) MoveMysteelChemicalClassify() {
  236. br := new(models.BaseResponse).Init()
  237. defer func() {
  238. this.Data["json"] = br
  239. this.ServeJSON()
  240. }()
  241. sysUser := this.SysUser
  242. if sysUser == nil {
  243. br.Msg = "请登录"
  244. br.ErrMsg = "请登录,SysUser Is Empty"
  245. br.Ret = 408
  246. return
  247. }
  248. var req request.MoveBaseFromMysteelChemicalClassifyReq
  249. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  250. if err != nil {
  251. br.Msg = "参数解析异常!"
  252. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  253. return
  254. }
  255. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  256. br.Msg = "参数错误"
  257. br.ErrMsg = "分类id小于等于0"
  258. br.IsSendEmail = false
  259. return
  260. }
  261. err, errMsg := data.MoveMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, req.ParentBaseFromMysteelChemicalClassifyId, req.PrevBaseFromMysteelChemicalClassifyId, req.NextBaseFromMysteelChemicalClassifyId, this.SysUser)
  262. if errMsg != `` {
  263. br.Msg = errMsg
  264. br.ErrMsg = errMsg
  265. if err != nil {
  266. br.ErrMsg = err.Error()
  267. } else {
  268. br.IsSendEmail = false
  269. }
  270. return
  271. }
  272. br.Ret = 200
  273. br.Success = true
  274. br.IsAddLog = true
  275. br.Msg = "移动成功"
  276. }
  277. // DeleteMysteelChemicalClassify
  278. // @Title 删除分类
  279. // @Description 删除分类接口
  280. // @Param request body data_manage.DelBaseFromMysteelChemicalClassifyReq true "type json string"
  281. // @Success 200 Ret=200 删除成功
  282. // @router /mysteel_chemical/classify/del [post]
  283. func (this *EdbClassifyController) DeleteMysteelChemicalClassify() {
  284. br := new(models.BaseResponse).Init()
  285. defer func() {
  286. this.Data["json"] = br
  287. this.ServeJSON()
  288. }()
  289. var req request.DelBaseFromMysteelChemicalClassifyReq
  290. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  291. if err != nil {
  292. br.Msg = "参数解析异常!"
  293. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  294. return
  295. }
  296. if req.BaseFromMysteelChemicalClassifyId < 0 {
  297. br.Msg = "参数错误"
  298. br.IsSendEmail = false
  299. return
  300. }
  301. err, errMsg := data.DelMysteelChemicalClassify(req.BaseFromMysteelChemicalClassifyId, this.SysUser)
  302. if errMsg != `` {
  303. br.Msg = errMsg
  304. br.ErrMsg = errMsg
  305. if err != nil {
  306. br.ErrMsg = errMsg + ";Err:" + err.Error()
  307. } else {
  308. br.IsSendEmail = false
  309. }
  310. return
  311. }
  312. br.Ret = 200
  313. br.Msg = "删除成功"
  314. br.Success = true
  315. br.IsAddLog = true
  316. }
  317. // MysteelChemicalSearch
  318. // @Title 上海钢联指标查询
  319. // @Description 上海钢联指标查询
  320. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  321. // @Param Keyword query string true "名称关键词"
  322. // @Success 200 {object} data_manage.LzFrequency
  323. // @router /mysteel_chemical/search [get]
  324. func (this *EdbInfoController) MysteelChemicalSearch() {
  325. br := new(models.BaseResponse).Init()
  326. defer func() {
  327. this.Data["json"] = br
  328. this.ServeJSON()
  329. }()
  330. sysUser := this.SysUser
  331. if sysUser == nil {
  332. br.Msg = "请登录"
  333. br.ErrMsg = "请登录,SysUser Is Empty"
  334. br.Ret = 408
  335. return
  336. }
  337. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId")
  338. var condition string
  339. var pars []interface{}
  340. if classifyId > 0 {
  341. condition += " AND base_from_mysteel_chemical_classify_id=? "
  342. pars = append(pars, classifyId)
  343. }
  344. keyword := this.GetString("Keyword")
  345. if keyword != `` {
  346. condition += " AND (index_name like ? OR index_code like ?) "
  347. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  348. }
  349. list, err := data_manage.GetMysteelChemicalIndex(condition, pars)
  350. if err != nil {
  351. br.Msg = "获取失败"
  352. br.ErrMsg = "获取失败,Err:" + err.Error()
  353. return
  354. }
  355. classifyIds := make([]int, 0)
  356. for _, v := range list {
  357. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.Id)
  358. if v.BaseFromMysteelChemicalClassifyId > 0 {
  359. classifyIds = append(classifyIds, v.BaseFromMysteelChemicalClassifyId)
  360. }
  361. }
  362. if len(classifyIds) > 0 {
  363. // 查询父级分类信息
  364. classifyList, e := data_manage.GetBaseFromMysteelChemicalClassifyByIds(classifyIds)
  365. if e != nil {
  366. br.Msg = "获取目录信息失败"
  367. br.ErrMsg = "获取失获取目录信息失败,Err:" + err.Error()
  368. return
  369. }
  370. classifyListMap := make(map[int]int, 0)
  371. for _, v := range classifyList {
  372. classifyListMap[v.BaseFromMysteelChemicalClassifyId] = v.ParentId
  373. }
  374. for _, v := range list {
  375. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.Id)
  376. if p, ok := classifyListMap[v.BaseFromMysteelChemicalClassifyId]; ok {
  377. v.ParentClassifyId = p
  378. }
  379. }
  380. }
  381. br.Ret = 200
  382. br.Success = true
  383. br.Msg = "获取成功"
  384. br.Data = list
  385. }
  386. // MysteelChemicalFrequency
  387. // @Title 上海钢联数据频度
  388. // @Description 上海钢联数据频度接口
  389. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  390. // @Param BaseFromMysteelChemicalIndexId query int true "上海钢联指标id"
  391. // @Param Keyword query string true "名称关键词"
  392. // @Success 200 {object} data_manage.LzFrequency
  393. // @router /mysteel_chemical/frequency [get]
  394. func (this *EdbInfoController) MysteelChemicalFrequency() {
  395. br := new(models.BaseResponse).Init()
  396. defer func() {
  397. this.Data["json"] = br
  398. this.ServeJSON()
  399. }()
  400. sysUser := this.SysUser
  401. if sysUser == nil {
  402. br.Msg = "请登录"
  403. br.ErrMsg = "请登录,SysUser Is Empty"
  404. br.Ret = 408
  405. return
  406. }
  407. var condition string
  408. var pars []interface{}
  409. var classifyId int
  410. var frequency string
  411. // 指定指标
  412. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  413. if indexId > 0 {
  414. condition += " AND base_from_mysteel_chemical_index_id = ? "
  415. pars = append(pars, indexId)
  416. } else {
  417. classifyId, _ = this.GetInt("BaseFromMysteelChemicalClassifyId")
  418. if classifyId <= 0 {
  419. br.Msg = "请选择分类"
  420. br.ErrMsg = "请选择分类"
  421. return
  422. }
  423. }
  424. if classifyId > 0 {
  425. condition += " AND base_from_mysteel_chemical_classify_id=? "
  426. pars = append(pars, classifyId)
  427. }
  428. if frequency != `` {
  429. condition += " AND frequency=? "
  430. pars = append(pars, frequency)
  431. }
  432. // 关键词搜索
  433. keyword := this.GetString("Keyword")
  434. if keyword != `` {
  435. condition += " AND (index_name like ? OR index_code like ?) "
  436. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  437. }
  438. edbFrequencyList := make([]string, 0)
  439. frequencyList, err := data_manage.GetMysteelChemicalFrequency(condition, pars)
  440. for _, unit := range frequencyList {
  441. edbFrequencyList = append(edbFrequencyList, unit.Frequency)
  442. }
  443. edbFrequencyList = sortEdbFrequency(edbFrequencyList)
  444. for i, v := range edbFrequencyList {
  445. frequencyList[i].Frequency = v
  446. }
  447. if err != nil {
  448. br.Msg = "获取频度失败"
  449. br.ErrMsg = "获取频度失败,Err:" + err.Error()
  450. return
  451. }
  452. br.Ret = 200
  453. br.Success = true
  454. br.Msg = "获取成功"
  455. br.Data = frequencyList
  456. }
  457. // MysteelChemicalData
  458. // @Title 获取上海钢联数据
  459. // @Description 获取上海钢联数据
  460. // @Param BaseFromMysteelChemicalClassifyId query int true "分类id"
  461. // @Param BaseFromMysteelChemicalIndexId query int true "上海钢联指标id"
  462. // @Param Frequency query string true "频度名称"
  463. // @Param Keyword query string true "名称关键词"
  464. // @Param PageSize query int true "每页数据条数"
  465. // @Param CurrentIndex query int true "当前页页码,从1开始"
  466. // @Success 200 {object} data_manage.LzFrequency
  467. // @router /mysteel_chemical/data [get]
  468. func (this *EdbInfoController) MysteelChemicalData() {
  469. br := new(models.BaseResponse).Init()
  470. defer func() {
  471. this.Data["json"] = br
  472. this.ServeJSON()
  473. }()
  474. sysUser := this.SysUser
  475. if sysUser == nil {
  476. br.Msg = "请登录"
  477. br.ErrMsg = "请登录,SysUser Is Empty"
  478. br.Ret = 408
  479. return
  480. }
  481. pageSize, _ := this.GetInt("PageSize")
  482. currentIndex, _ := this.GetInt("CurrentIndex")
  483. var startSize int
  484. if pageSize <= 0 {
  485. pageSize = utils.PageSize20
  486. }
  487. if currentIndex <= 0 {
  488. currentIndex = 1
  489. }
  490. startSize = utils.StartIndex(currentIndex, pageSize)
  491. var condition string
  492. var pars []interface{}
  493. var classifyId int
  494. var frequency string
  495. // 指定指标
  496. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  497. if indexId > 0 {
  498. condition += " AND base_from_mysteel_chemical_index_id = ? "
  499. pars = append(pars, indexId)
  500. } else {
  501. classifyId, _ = this.GetInt("BaseFromMysteelChemicalClassifyId")
  502. if classifyId <= 0 {
  503. br.Msg = "请选择分类"
  504. br.ErrMsg = "请选择分类"
  505. br.IsSendEmail = false
  506. return
  507. }
  508. frequency = this.GetString("Frequency")
  509. if frequency == "" {
  510. br.Msg = "请选择频度"
  511. br.ErrMsg = "请选择频度"
  512. br.IsSendEmail = false
  513. return
  514. }
  515. }
  516. if classifyId > 0 {
  517. condition += " AND base_from_mysteel_chemical_classify_id=? "
  518. pars = append(pars, classifyId)
  519. }
  520. if frequency != `` {
  521. condition += " AND frequency=? "
  522. pars = append(pars, frequency)
  523. }
  524. // 关键词搜索
  525. keyword := this.GetString("Keyword")
  526. if keyword != `` {
  527. condition += " AND (index_name like ? OR index_code like ?) "
  528. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  529. }
  530. //获取指标
  531. list, err := data_manage.GetMysteelChemicalIndex(condition, pars)
  532. if err != nil {
  533. br.Msg = "获取数据失败"
  534. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  535. return
  536. }
  537. for k, v := range list {
  538. total, err := data_manage.GetMysteelChemicalIndexDataCount(v.IndexCode)
  539. page := paging.GetPaging(currentIndex, pageSize, total)
  540. dataList := make([]*data_manage.MysteelChemicalData, 0)
  541. tmpDataList, err := data_manage.GetMysteelChemicalIndexData(v.IndexCode, startSize, pageSize)
  542. if err != nil {
  543. br.Msg = "获取数据失败"
  544. br.ErrMsg = "获取指标数据失败,Err:" + err.Error()
  545. return
  546. }
  547. for _, tmpData := range tmpDataList {
  548. dataList = append(dataList, &data_manage.MysteelChemicalData{
  549. DataTime: tmpData.DataTime,
  550. InputValue: decimal.NewFromFloat(tmpData.InputValue).String(),
  551. })
  552. }
  553. edbInfo, err := data_manage.GetEdbInfoByEdbCode(utils.DATA_SOURCE_MYSTEEL_CHEMICAL, v.IndexCode)
  554. if err != nil && !utils.IsErrNoRow(err) {
  555. br.Msg = "获取数据失败"
  556. br.ErrMsg = "获取指标库统计数据失败,Err:" + err.Error()
  557. return
  558. }
  559. if edbInfo == nil {
  560. list[k].EdbInfoId = 0
  561. } else {
  562. list[k].EdbInfoId = edbInfo.EdbInfoId
  563. }
  564. list[k].DataList = dataList
  565. list[k].Paging = page
  566. }
  567. br.Ret = 200
  568. br.Success = true
  569. br.Msg = "获取成功"
  570. br.Data = list
  571. }
  572. // MysteelChemicalExport
  573. // @Title 导出上海钢联数据
  574. // @Description 导出上海钢联数据
  575. // @Param BaseFromMysteelChemicalClassifyId query int true "分类"
  576. // @Param BaseFromMysteelChemicalIndexId query string true "上海钢联指标id"
  577. // @Param Keyword query string true "名称关键词"
  578. // @Success 200 导出成功
  579. // @router /mysteel_chemical/export/dataList [get]
  580. func (this *EdbClassifyController) MysteelChemicalExport() {
  581. br := new(models.BaseResponse).Init()
  582. defer func() {
  583. this.Data["json"] = br
  584. this.ServeJSON()
  585. }()
  586. sysUser := this.SysUser
  587. if sysUser == nil {
  588. br.Msg = "请重新登录"
  589. return
  590. }
  591. classifyId, _ := this.GetInt("BaseFromMysteelChemicalClassifyId") //分类
  592. frequency := this.GetString("Frequency")
  593. dir, _ := os.Executable()
  594. exPath := filepath.Dir(dir)
  595. downLoadnFilePath := exPath + "/" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
  596. xlsxFile := xlsx.NewFile()
  597. var condition string
  598. var pars []interface{}
  599. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  600. if indexId > 0 {
  601. condition += " AND base_from_mysteel_chemical_index_id = ? "
  602. pars = append(pars, indexId)
  603. }
  604. if classifyId > 0 {
  605. condition += " AND base_from_mysteel_chemical_classify_id=? "
  606. pars = append(pars, classifyId)
  607. }
  608. if frequency != `` {
  609. condition += " AND frequency=? "
  610. pars = append(pars, frequency)
  611. }
  612. // 关键词搜索
  613. keyword := this.GetString("Keyword")
  614. if keyword != `` {
  615. condition += " AND (index_name like ? OR index_code like ?) "
  616. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  617. }
  618. var frequencies []string
  619. frequencieSet := make(map[string]struct{})
  620. frequencyList, err := data_manage.GetMysteelChemicalFrequency(condition, pars)
  621. if err != nil {
  622. br.Msg = "获取频度失败"
  623. br.ErrMsg = "获取频度失败,Err:" + err.Error()
  624. return
  625. }
  626. for _, v := range frequencyList {
  627. if _, ok := frequencieSet[v.Frequency]; !ok {
  628. frequencies = append(frequencies, v.Frequency)
  629. frequencieSet[v.Frequency] = struct{}{}
  630. }
  631. }
  632. secNameList := make([]*data_manage.MysteelChemicalList, 0)
  633. for _, frequency := range frequencies {
  634. //获取指标
  635. var tmpCondition string
  636. var tmpPars []interface{}
  637. if indexId > 0 {
  638. tmpCondition += " AND base_from_mysteel_chemical_index_id = ? "
  639. tmpPars = append(pars, indexId)
  640. }
  641. if classifyId > 0 {
  642. tmpCondition += " AND base_from_mysteel_chemical_classify_id=? "
  643. tmpPars = append(pars, classifyId)
  644. }
  645. if frequency != `` {
  646. tmpCondition += " AND frequency=? "
  647. tmpPars = append(pars, frequency)
  648. }
  649. // 关键词搜索
  650. keyword := this.GetString("Keyword")
  651. if keyword != `` {
  652. tmpCondition += " AND (index_name like ? OR index_code like ?) "
  653. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  654. }
  655. //获取指标
  656. secNameList, err = data_manage.GetMysteelChemicalIndex(tmpCondition, tmpPars)
  657. if err != nil {
  658. fmt.Println("获取数据失败,Err:" + err.Error())
  659. return
  660. }
  661. if len(secNameList) <= 0 {
  662. fmt.Println("secNameList长度为0")
  663. return
  664. }
  665. sheetNew, err := xlsxFile.AddSheet(frequency)
  666. if err != nil {
  667. fmt.Println("新增Sheet失败", err.Error())
  668. return
  669. }
  670. //sheetNew.SetColWidth()
  671. //获取指标数据
  672. windRow := sheetNew.AddRow()
  673. secNameRow := sheetNew.AddRow()
  674. indexCodeRow := sheetNew.AddRow()
  675. frequencyRow := sheetNew.AddRow()
  676. unitRow := sheetNew.AddRow()
  677. lastModifyDateRow := sheetNew.AddRow()
  678. var indexIdList []int
  679. for _, sv := range secNameList {
  680. indexIdList = append(indexIdList, sv.Id)
  681. }
  682. dataTimeList, err := data_manage.GetBaseFromMysteelChemicalDataTimeByIndexId(indexIdList)
  683. if err != nil {
  684. fmt.Println("获取数据时间失败", err.Error())
  685. return
  686. }
  687. //获取分类下指标最大数据量
  688. setRowIndex := 6
  689. for rk, dv := range dataTimeList {
  690. rowIndex := setRowIndex + rk
  691. row := sheetNew.Row(rowIndex)
  692. displayDate, _ := time.Parse(utils.FormatDate, dv)
  693. displayDateCell := row.AddCell()
  694. style := new(xlsx.Style)
  695. style.ApplyAlignment = true
  696. style.Alignment.WrapText = true
  697. displayDateCell.SetStyle(style)
  698. displayDateCell.SetDate(displayDate)
  699. }
  700. for k, sv := range secNameList {
  701. //获取数据
  702. dataList, err := data_manage.GetMysteelChemicalIndexDataByCode(sv.IndexCode)
  703. if err != nil {
  704. br.Msg = "获取数据失败"
  705. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  706. return
  707. }
  708. if k == 0 {
  709. windRow.AddCell().SetValue("上海钢联")
  710. secNameRow.AddCell().SetValue("指标名称")
  711. indexCodeRow.AddCell().SetValue("指标ID")
  712. frequencyRow.AddCell().SetValue("频率")
  713. unitRow.AddCell().SetValue("单位")
  714. lastModifyDateRow.AddCell().SetValue("更新时间")
  715. }
  716. if len(dataList) == 0 {
  717. continue
  718. }
  719. secNameRow.AddCell().SetValue(sv.IndexName)
  720. indexCodeRow.AddCell().SetValue(sv.IndexCode)
  721. frequencyRow.AddCell().SetValue(sv.FrequencyName)
  722. unitRow.AddCell().SetValue(sv.UnitName)
  723. lastModifyDateRow.AddCell().SetValue(sv.UpdateTime)
  724. dataInfoMap := make(map[string]*data_manage.MysteelChemicalData)
  725. for _, dv := range dataList {
  726. dataInfoMap[dv.DataTime] = dv
  727. }
  728. for rk, dtv := range dataTimeList {
  729. rowIndex := setRowIndex + rk
  730. row := sheetNew.Row(rowIndex)
  731. displayDateCell := row.AddCell()
  732. tmpData, ok := dataInfoMap[dtv]
  733. if ok {
  734. displayDateCell.SetValue(tmpData.InputValue)
  735. }
  736. }
  737. }
  738. }
  739. err = xlsxFile.Save(downLoadnFilePath)
  740. if err != nil {
  741. //有指标无数据时先导出一遍空表
  742. sheet, err := xlsxFile.AddSheet("无数据")
  743. if err != nil {
  744. br.Msg = "新增Sheet失败"
  745. br.ErrMsg = "新增Sheet失败,Err:" + err.Error()
  746. return
  747. }
  748. rowSecName := sheet.AddRow()
  749. celSecName := rowSecName.AddCell()
  750. celSecName.SetValue("")
  751. err = xlsxFile.Save(downLoadnFilePath)
  752. if err != nil {
  753. br.Msg = "保存文件失败"
  754. br.ErrMsg = "保存文件失败"
  755. return
  756. }
  757. }
  758. fileName := "上海钢联"
  759. fileName += time.Now().Format("06.01.02") + `.xlsx` //文件名称
  760. this.Ctx.Output.Download(downLoadnFilePath, fileName)
  761. defer func() {
  762. os.Remove(downLoadnFilePath)
  763. }()
  764. br.Ret = 200
  765. br.Success = true
  766. br.Msg = "success"
  767. }
  768. // AddMysteelChemical
  769. // @Title 新增上海钢联指标
  770. // @Description 新增上海钢联指标接口
  771. // @Param request body data_manage.AddEdbClassifyReq true "type json string"
  772. // @Success 200 Ret=200 保存成功
  773. // @router /mysteel_chemical/add [post]
  774. func (this *EdbClassifyController) AddMysteelChemical() {
  775. br := new(models.BaseResponse).Init()
  776. defer func() {
  777. this.Data["json"] = br
  778. this.ServeJSON()
  779. }()
  780. var req request.AddBaseFromMysteelChemicalReq
  781. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  782. if err != nil {
  783. br.Msg = "参数解析异常!"
  784. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  785. return
  786. }
  787. resp := new(response.AddMysteelChemicalIndexInfoResp)
  788. conf, err := models.GetBusinessConf()
  789. if err != nil {
  790. br.Msg = "添加指标失败"
  791. br.ErrMsg = "获取业务配置失败,Err:" + err.Error()
  792. return
  793. }
  794. // 判断上海钢联的数据刷新方式
  795. if conf["MySteelDataMethod"] == "api" {
  796. if len(req.List) > 150 {
  797. br.Msg = "添加指标失败,指标数量不能超过150条"
  798. return
  799. }
  800. ok, errMsg, err := data.HealthCheckMysteelChemicalApi()
  801. if err != nil {
  802. br.Msg = "添加指标失败"
  803. br.ErrMsg = "上海钢联数据接口异常,Err:" + err.Error()
  804. return
  805. }
  806. if !ok {
  807. br.Msg = errMsg
  808. return
  809. }
  810. addMySteelList := make([]*data_manage.BaseFromMysteelChemicalIndex, 0)
  811. for _, v := range req.List {
  812. if v.IndexCode == "" {
  813. br.Msg = "请输入指标ID"
  814. br.IsSendEmail = false
  815. return
  816. }
  817. if v.BaseFromMysteelChemicalClassifyId <= 0 {
  818. br.Msg = "请选择分类"
  819. br.IsSendEmail = false
  820. return
  821. }
  822. var updateTime1, updateTime2 string
  823. if v.UpdateTime != `` {
  824. updateTimeList := strings.Split(v.UpdateTime, ",")
  825. updateTime1 = updateTimeList[0]
  826. if len(updateTimeList) > 1 {
  827. updateTime2 = updateTimeList[1]
  828. }
  829. }
  830. v.IndexCode = utils.TrimStr(v.IndexCode)
  831. v.IndexCode = strings.Replace(v.IndexCode, "\t", "", -1)
  832. addMySteelList = append(addMySteelList, &data_manage.BaseFromMysteelChemicalIndex{
  833. BaseFromMysteelChemicalClassifyId: v.BaseFromMysteelChemicalClassifyId,
  834. IndexCode: strings.ToUpper(v.IndexCode),
  835. IndexName: "",
  836. Unit: "",
  837. Frequency: "",
  838. StartDate: "",
  839. EndDate: "",
  840. Describe: "",
  841. Source: "api",
  842. TerminalCode: "",
  843. UpdateWeek: v.UpdateWeek,
  844. UpdateTime: updateTime1,
  845. UpdateTime2: updateTime2,
  846. SysUserId: this.SysUser.AdminId,
  847. SysUserRealName: this.SysUser.RealName,
  848. ModifyTime: time.Now(),
  849. CreateTime: time.Now(),
  850. })
  851. }
  852. mySteelChemicalIndexInfos, err, errMsg := data.BatchAddMysteelChemicalIndex(addMySteelList, this.Lang)
  853. if err != nil {
  854. br.Msg = errMsg
  855. br.ErrMsg = errMsg + ";Err:" + err.Error()
  856. return
  857. }
  858. if len(mySteelChemicalIndexInfos) > 0 {
  859. resp.BaseFromMysteelChemicalIndexId = mySteelChemicalIndexInfos[0].BaseFromMysteelChemicalIndexId
  860. resp.IndexCode = mySteelChemicalIndexInfos[0].IndexCode
  861. resp.UniqueCode = fmt.Sprint(mySteelChemicalIndexInfos[0].BaseFromMysteelChemicalClassifyId, "_", mySteelChemicalIndexInfos[0].BaseFromMysteelChemicalIndexId)
  862. }
  863. for _, v := range mySteelChemicalIndexInfos {
  864. err = utils.Rc.LPush(utils.CACHE_MYSTEEL_CHEMICAL_ADD_DATA, v.IndexCode)
  865. if err != nil {
  866. br.Msg = "添加指标失败" + v.IndexCode
  867. br.ErrMsg = "添加指标缓存失败" + v.IndexCode + ";Err:" + err.Error()
  868. return
  869. }
  870. }
  871. } else {
  872. for i, v := range req.List {
  873. if v.IndexCode == "" {
  874. br.Msg = "请输入指标ID"
  875. br.IsSendEmail = false
  876. return
  877. }
  878. if v.BaseFromMysteelChemicalClassifyId <= 0 {
  879. br.Msg = "请选择分类"
  880. br.IsSendEmail = false
  881. return
  882. }
  883. v.IndexCode = utils.TrimStr(v.IndexCode)
  884. v.IndexCode = strings.Replace(v.IndexCode, "\t", "", -1)
  885. //添加指标
  886. mysteelChemicalIndexInfo, err, errMsg := data.AddMysteelChemicalIndex(v.BaseFromMysteelChemicalClassifyId, strings.ToUpper(v.IndexCode), v.UpdateWeek, v.UpdateTime, this.SysUser.AdminId, this.SysUser.RealName, this.Lang)
  887. if errMsg != `` {
  888. br.Msg = errMsg
  889. br.ErrMsg = errMsg
  890. if err != nil {
  891. br.ErrMsg = errMsg + ";Err:" + err.Error()
  892. } else {
  893. br.IsSendEmail = false
  894. }
  895. return
  896. }
  897. if i == 0 {
  898. resp.BaseFromMysteelChemicalIndexId = mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId
  899. resp.IndexCode = mysteelChemicalIndexInfo.IndexCode
  900. resp.UniqueCode = fmt.Sprint(mysteelChemicalIndexInfo.BaseFromMysteelChemicalClassifyId, "_", mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId)
  901. }
  902. }
  903. }
  904. br.Data = resp
  905. br.Ret = 200
  906. br.Msg = "保存成功"
  907. br.Success = true
  908. br.IsAddLog = true
  909. }
  910. func sortEdbFrequency(frequencyList []string) (newFrequencyList []string) {
  911. var frequencyMap1 = map[string]int{
  912. "日度": 1,
  913. "周度": 2,
  914. "旬度": 3,
  915. "月度": 4,
  916. "季度": 5,
  917. "半年度": 6,
  918. "年度": 7,
  919. }
  920. var frequencyMap2 = map[int]string{
  921. 1: "日度",
  922. 2: "周度",
  923. 3: "旬度",
  924. 4: "月度",
  925. 5: "季度",
  926. 6: "半年度",
  927. 7: "年度",
  928. }
  929. idList := make([]int, 0)
  930. for _, frequency := range frequencyList {
  931. idList = append(idList, frequencyMap1[frequency])
  932. }
  933. sort.Ints(idList)
  934. for _, id := range idList {
  935. newFrequencyList = append(newFrequencyList, frequencyMap2[id])
  936. }
  937. return
  938. }
  939. // EditMysteelChemical
  940. // @Title 编辑上海钢联指标
  941. // @Description 新增上海钢联指标接口
  942. // @Param request body data_manage.AddEdbClassifyReq true "type json string"
  943. // @Success 200 Ret=200 保存成功
  944. // @router /mysteel_chemical/edit [post]
  945. func (this *EdbClassifyController) EditMysteelChemical() {
  946. br := new(models.BaseResponse).Init()
  947. defer func() {
  948. this.Data["json"] = br
  949. this.ServeJSON()
  950. }()
  951. var req request.AddBaseFromMysteelChemicalReqItem
  952. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  953. if err != nil {
  954. br.Msg = "参数解析异常!"
  955. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  956. return
  957. }
  958. if req.BaseFromMysteelChemicalIndexId <= 0 {
  959. br.Msg = "请选择指标"
  960. br.IsSendEmail = false
  961. return
  962. }
  963. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  964. br.Msg = "请选择分类"
  965. br.IsSendEmail = false
  966. return
  967. }
  968. //编辑指标
  969. mysteelChemicalIndexInfo, err, errMsg := data.EditMysteelChemicalIndex(req.BaseFromMysteelChemicalIndexId, req.BaseFromMysteelChemicalClassifyId, req.UpdateWeek, req.UpdateTime, this.SysUser)
  970. if errMsg != `` {
  971. br.Msg = errMsg
  972. br.ErrMsg = errMsg
  973. if err != nil {
  974. br.ErrMsg = errMsg + ";Err:" + err.Error()
  975. } else {
  976. br.IsSendEmail = false
  977. }
  978. return
  979. }
  980. resp := response.AddMysteelChemicalIndexInfoResp{
  981. BaseFromMysteelChemicalIndexId: mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId,
  982. IndexCode: mysteelChemicalIndexInfo.IndexCode,
  983. UniqueCode: fmt.Sprint(mysteelChemicalIndexInfo.BaseFromMysteelChemicalClassifyId, "_", mysteelChemicalIndexInfo.BaseFromMysteelChemicalIndexId),
  984. }
  985. br.Data = resp
  986. br.Ret = 200
  987. br.Msg = "保存成功"
  988. br.Success = true
  989. br.IsAddLog = true
  990. }
  991. // DeleteMysteelChemical
  992. // @Title 删除指标
  993. // @Description 删除指标接口
  994. // @Param request body data_manage.DelBaseFromMysteelChemicalReq true "type json string"
  995. // @Success 200 Ret=200 删除成功
  996. // @router /mysteel_chemical/del [post]
  997. func (this *EdbClassifyController) DeleteMysteelChemical() {
  998. br := new(models.BaseResponse).Init()
  999. defer func() {
  1000. this.Data["json"] = br
  1001. this.ServeJSON()
  1002. }()
  1003. var req request.DelBaseFromMysteelChemicalReq
  1004. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1005. if err != nil {
  1006. br.Msg = "参数解析异常!"
  1007. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1008. return
  1009. }
  1010. if req.BaseFromMysteelChemicalIndexId < 0 {
  1011. br.Msg = "参数错误"
  1012. br.IsSendEmail = false
  1013. return
  1014. }
  1015. err, errMsg := data.DelMysteelChemical(req.BaseFromMysteelChemicalIndexId, this.SysUser)
  1016. if errMsg != `` {
  1017. br.Msg = errMsg
  1018. br.ErrMsg = errMsg
  1019. if err != nil {
  1020. br.ErrMsg = errMsg + ";Err:" + err.Error()
  1021. } else {
  1022. br.IsSendEmail = false
  1023. }
  1024. return
  1025. }
  1026. br.Ret = 200
  1027. br.Msg = "删除成功"
  1028. br.Success = true
  1029. br.IsAddLog = true
  1030. }
  1031. // MoveMysteelChemical
  1032. // @Title 指标移动接口
  1033. // @Description 分类移动接口
  1034. // @Success 200 {object} data_manage.MoveBaseFromMysteelChemicalReq
  1035. // @router /mysteel_chemical/move [post]
  1036. func (this *EdbClassifyController) MoveMysteelChemical() {
  1037. br := new(models.BaseResponse).Init()
  1038. defer func() {
  1039. this.Data["json"] = br
  1040. this.ServeJSON()
  1041. }()
  1042. sysUser := this.SysUser
  1043. if sysUser == nil {
  1044. br.Msg = "请登录"
  1045. br.ErrMsg = "请登录,SysUser Is Empty"
  1046. br.Ret = 408
  1047. return
  1048. }
  1049. var req request.MoveBaseFromMysteelChemicalReq
  1050. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1051. if err != nil {
  1052. br.Msg = "参数解析异常!"
  1053. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1054. return
  1055. }
  1056. if req.BaseFromMysteelChemicalIndexId <= 0 {
  1057. br.Msg = "参数错误"
  1058. br.ErrMsg = "指标id小于等于0"
  1059. br.IsSendEmail = false
  1060. return
  1061. }
  1062. if req.BaseFromMysteelChemicalClassifyId <= 0 {
  1063. br.Msg = "请选择分类"
  1064. br.ErrMsg = "请选择分类"
  1065. br.IsSendEmail = false
  1066. return
  1067. }
  1068. err, errMsg := data.MoveMysteelChemical(req.BaseFromMysteelChemicalIndexId, req.BaseFromMysteelChemicalClassifyId, req.PrevBaseFromMysteelChemicalIndexId, req.NextBaseFromMysteelChemicalIndexId, this.SysUser)
  1069. if errMsg != `` {
  1070. br.Msg = errMsg
  1071. br.ErrMsg = errMsg
  1072. if err != nil {
  1073. br.ErrMsg = err.Error()
  1074. } else {
  1075. br.IsSendEmail = false
  1076. }
  1077. return
  1078. }
  1079. br.Ret = 200
  1080. br.Success = true
  1081. br.IsAddLog = true
  1082. br.Msg = "移动成功"
  1083. }
  1084. // MysteelChemicalDetail
  1085. // @Title 上海钢联指标详情
  1086. // @Description 上海钢联指标详情接口
  1087. // @Param BaseFromMysteelChemicalIndexId query string true "上海钢联指标id"
  1088. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  1089. // @router /mysteel_chemical/detail [get]
  1090. func (this *EdbInfoController) MysteelChemicalDetail() {
  1091. br := new(models.BaseResponse).Init()
  1092. defer func() {
  1093. this.Data["json"] = br
  1094. this.ServeJSON()
  1095. }()
  1096. sysUser := this.SysUser
  1097. if sysUser == nil {
  1098. br.Msg = "请登录"
  1099. br.ErrMsg = "请登录,SysUser Is Empty"
  1100. br.Ret = 408
  1101. return
  1102. }
  1103. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  1104. if indexId <= 0 {
  1105. br.Msg = "请选择指标"
  1106. br.IsSendEmail = false
  1107. return
  1108. }
  1109. item, err := data_manage.GetBaseFromMysteelChemicalIndexByIndexId(indexId)
  1110. if err != nil {
  1111. br.Msg = "获取失败"
  1112. br.ErrMsg = "获取失败,Err:" + err.Error()
  1113. }
  1114. if item.UpdateTime2 != `` {
  1115. item.UpdateTime = item.UpdateTime + "," + item.UpdateTime2
  1116. }
  1117. br.Ret = 200
  1118. br.Success = true
  1119. br.Msg = "获取成功"
  1120. br.Data = item
  1121. }
  1122. // MysteelChemicalDetail
  1123. // @Title 上海钢联指标详情
  1124. // @Description 上海钢联指标详情接口
  1125. // @Param BaseFromMysteelChemicalIndexId query string true "上海钢联指标id"
  1126. // @Success 200 {object} data_manage.BaseFromMysteelChemicalClassifyItems
  1127. // @router /mysteel_chemical/refresh [get]
  1128. func (this *EdbInfoController) MysteelChemicalRefresh() {
  1129. br := new(models.BaseResponse).Init()
  1130. defer func() {
  1131. this.Data["json"] = br
  1132. this.ServeJSON()
  1133. }()
  1134. sysUser := this.SysUser
  1135. if sysUser == nil {
  1136. br.Msg = "请登录"
  1137. br.ErrMsg = "请登录,SysUser Is Empty"
  1138. br.Ret = 408
  1139. return
  1140. }
  1141. // 每次刷新都需要两分钟的等待时间
  1142. setNxKey := utils.CACHE_KEY_MYSTEEL_REFRESH
  1143. if !utils.Rc.SetNX(setNxKey, 1, 2*60*time.Second) {
  1144. br.Msg = "系统正在处理数据生成,请2分钟之后刷新"
  1145. br.IsSendEmail = false
  1146. return
  1147. }
  1148. indexId, _ := this.GetInt("BaseFromMysteelChemicalIndexId")
  1149. if indexId <= 0 {
  1150. br.Msg = "请选择指标"
  1151. br.IsSendEmail = false
  1152. return
  1153. }
  1154. item, err := data_manage.GetBaseFromMysteelChemicalIndexByIndexId(indexId)
  1155. if err != nil {
  1156. br.Msg = "获取失败"
  1157. br.ErrMsg = "获取失败,Err:" + err.Error()
  1158. }
  1159. if item.MergeFilePath == "" {
  1160. br.Msg = "系统正在处理数据生成,请5分钟之后刷新"
  1161. br.IsSendEmail = false
  1162. return
  1163. }
  1164. go data.CurlRefreshMysteelChemicalData(item.MergeFilePath)
  1165. br.Ret = 200
  1166. br.Success = true
  1167. br.Msg = "刷新成功"
  1168. br.Data = item
  1169. }
  1170. // NameCheck
  1171. // @Title 重名校验
  1172. // @Description 批量新增
  1173. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  1174. // @Success 200 string "操作成功"
  1175. // @router /mysteel_chemical/name_check [post]
  1176. func (c *EdbInfoController) NameCheck() {
  1177. br := new(models.BaseResponse).Init()
  1178. defer func() {
  1179. if br.ErrMsg == "" {
  1180. br.IsSendEmail = false
  1181. }
  1182. c.Data["json"] = br
  1183. c.ServeJSON()
  1184. }()
  1185. sysUser := c.SysUser
  1186. if sysUser == nil {
  1187. br.Msg = "请登录"
  1188. br.ErrMsg = "请登录,SysUser Is Empty"
  1189. br.Ret = 408
  1190. return
  1191. }
  1192. var req []*data_manage.NameCheckEdbInfoReq
  1193. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  1194. br.Msg = "参数解析异常!"
  1195. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  1196. return
  1197. }
  1198. if len(req) == 0 {
  1199. br.Msg = "请选择指标"
  1200. return
  1201. }
  1202. codeMax := 30
  1203. codeLen := len(req)
  1204. if codeLen > codeMax {
  1205. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  1206. return
  1207. }
  1208. type NameCheckResult struct {
  1209. EdbCode string
  1210. EdbName string
  1211. Exist bool
  1212. }
  1213. indexNames := make([]string, 0)
  1214. resp := make([]*NameCheckResult, 0)
  1215. for _, v := range req {
  1216. v.EdbCode = strings.TrimSpace(v.EdbCode)
  1217. if v.EdbCode == "" {
  1218. br.Msg = "指标ID不可为空"
  1219. return
  1220. }
  1221. v.EdbName = strings.TrimSpace(v.EdbName)
  1222. if v.EdbName == "" {
  1223. br.Msg = "请输入指标名称"
  1224. return
  1225. }
  1226. indexNames = append(indexNames, v.EdbName)
  1227. resp = append(resp, &NameCheckResult{
  1228. EdbCode: v.EdbCode,
  1229. EdbName: v.EdbName,
  1230. })
  1231. dataItems, err := data_manage.GetEdbDataAllByEdbCode(v.EdbCode, utils.DATA_SOURCE_MYSTEEL_CHEMICAL, 0, utils.EDB_DATA_LIMIT)
  1232. if err != nil && !utils.IsErrNoRow(err) {
  1233. br.Msg = "获取失败"
  1234. br.ErrMsg = "获取钢联已存在信息失败,Err:" + err.Error()
  1235. return
  1236. }
  1237. if len(dataItems) <= 0 {
  1238. respItem, err := data.AddEdbData(utils.DATA_SOURCE_MYSTEEL_CHEMICAL, v.EdbCode, v.Frequency)
  1239. if err != nil {
  1240. br.Msg = "获取失败"
  1241. br.ErrMsg = "获取失败,Err:" + err.Error()
  1242. return
  1243. }
  1244. if respItem.Ret != 200 {
  1245. br.Msg = "未搜索到该指标"
  1246. br.ErrMsg = respItem.ErrMsg + ";EdbCode:" + v.EdbCode
  1247. return
  1248. }
  1249. }
  1250. }
  1251. // 重名校验
  1252. edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
  1253. if e != nil {
  1254. br.Msg = "操作失败"
  1255. br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
  1256. return
  1257. }
  1258. nameExists := make(map[string]bool)
  1259. for _, v := range edbList {
  1260. nameExists[v.EdbName] = true
  1261. }
  1262. if len(nameExists) > 0 {
  1263. for _, v := range resp {
  1264. v.Exist = nameExists[v.EdbName]
  1265. }
  1266. }
  1267. br.Data = resp
  1268. br.Msg = "校验成功"
  1269. br.Ret = 200
  1270. br.Success = true
  1271. }
  1272. // BatchAdd
  1273. // @Title 批量新增
  1274. // @Description 批量新增
  1275. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  1276. // @Success 200 string "操作成功"
  1277. // @router /mysteel_chemical/batch_add [post]
  1278. func (this *EdbInfoController) BatchAdd() {
  1279. br := new(models.BaseResponse).Init()
  1280. defer func() {
  1281. if br.ErrMsg == "" {
  1282. br.IsSendEmail = false
  1283. }
  1284. this.Data["json"] = br
  1285. this.ServeJSON()
  1286. }()
  1287. sysUser := this.SysUser
  1288. if sysUser == nil {
  1289. br.Msg = "请登录"
  1290. br.ErrMsg = "请登录,SysUser Is Empty"
  1291. br.Ret = 408
  1292. return
  1293. }
  1294. deleteCache := true
  1295. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_MYSTEEL_CHEMICAL_" + strconv.Itoa(sysUser.AdminId)
  1296. defer func() {
  1297. if deleteCache {
  1298. _ = utils.Rc.Delete(cacheKey)
  1299. }
  1300. }()
  1301. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  1302. deleteCache = false
  1303. br.Msg = "系统处理中,请稍后重试!"
  1304. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  1305. return
  1306. }
  1307. var req []*data_manage.AddEdbInfoReq
  1308. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  1309. br.Msg = "参数解析异常!"
  1310. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  1311. return
  1312. }
  1313. if len(req) == 0 {
  1314. br.Msg = "请选择指标"
  1315. return
  1316. }
  1317. if len(req) > 30 {
  1318. br.Msg = "批量添加指标数量不得超过30个"
  1319. return
  1320. }
  1321. for _, v := range req {
  1322. v.EdbCode = strings.TrimSpace(v.EdbCode)
  1323. if v.EdbCode == "" {
  1324. br.Msg = "指标ID不可为空"
  1325. return
  1326. }
  1327. v.EdbName = strings.TrimSpace(v.EdbName)
  1328. if v.EdbName == "" {
  1329. br.Msg = "请输入指标名称"
  1330. return
  1331. }
  1332. v.Frequency = strings.TrimSpace(v.Frequency)
  1333. if v.Frequency == "" {
  1334. br.Msg = "请选择频度"
  1335. return
  1336. }
  1337. v.Unit = strings.TrimSpace(v.Unit)
  1338. if v.Unit == "" {
  1339. br.Msg = "请输入单位"
  1340. return
  1341. }
  1342. if v.ClassifyId <= 0 {
  1343. br.Msg = "请选择分类"
  1344. return
  1345. }
  1346. }
  1347. // 限定同一时间最多批量新增30个指标
  1348. for _, v := range req {
  1349. var r data.MysteelChemicalIndexSource2EdbReq
  1350. r.EdbCode = v.EdbCode
  1351. r.EdbName = v.EdbName
  1352. r.Frequency = v.Frequency
  1353. r.Unit = v.Unit
  1354. r.ClassifyId = v.ClassifyId
  1355. r.AdminId = sysUser.AdminId
  1356. r.AdminRealName = sysUser.RealName
  1357. edbInfo, e, errMsg, skip := data.MysteelChemicalIndexSource2Edb(r, this.Lang)
  1358. if e != nil {
  1359. br.Msg = "操作失败"
  1360. if errMsg != "" {
  1361. br.Msg = errMsg
  1362. }
  1363. br.ErrMsg = e.Error()
  1364. return
  1365. }
  1366. if skip {
  1367. continue
  1368. }
  1369. // 试用平台更新用户累计新增指标数
  1370. if utils.BusinessCode == utils.BusinessCodeSandbox {
  1371. go func() {
  1372. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  1373. if e != nil {
  1374. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  1375. utils.FileLog.Info(tips)
  1376. return
  1377. }
  1378. if adminItem.DepartmentName != "ETA试用客户" {
  1379. return
  1380. }
  1381. var ur etaTrialService.EtaTrialUserReq
  1382. ur.Mobile = adminItem.Mobile
  1383. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  1384. }()
  1385. }
  1386. // 新增操作日志
  1387. {
  1388. edbLog := new(data_manage.EdbInfoLog)
  1389. edbLog.EdbInfoId = edbInfo.EdbInfoId
  1390. edbLog.SourceName = edbInfo.SourceName
  1391. edbLog.Source = edbInfo.Source
  1392. edbLog.EdbCode = edbInfo.EdbCode
  1393. edbLog.EdbName = edbInfo.EdbName
  1394. edbLog.ClassifyId = edbInfo.ClassifyId
  1395. edbLog.SysUserId = sysUser.AdminId
  1396. edbLog.SysUserRealName = sysUser.RealName
  1397. edbLog.CreateTime = time.Now()
  1398. edbLog.Content = string(this.Ctx.Input.RequestBody)
  1399. edbLog.Status = "新增指标"
  1400. edbLog.Method = this.Ctx.Input.URI()
  1401. go data_manage.AddEdbInfoLog(edbLog)
  1402. }
  1403. }
  1404. br.Msg = "操作成功"
  1405. br.Ret = 200
  1406. br.Success = true
  1407. br.IsAddLog = true
  1408. }
  1409. // EdbInfoAdd
  1410. // @Title 新增指标接口
  1411. // @Description 新增指标接口
  1412. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  1413. // @Success Ret=200 保存成功
  1414. // @router /mysteel_chemical/edb_info/add [post]
  1415. func (this *EdbInfoController) Add() {
  1416. br := new(models.BaseResponse).Init()
  1417. defer func() {
  1418. this.Data["json"] = br
  1419. this.ServeJSON()
  1420. }()
  1421. sysUser := this.SysUser
  1422. if sysUser == nil {
  1423. br.Msg = "请登录"
  1424. br.ErrMsg = "请登录,SysUser Is Empty"
  1425. br.Ret = 408
  1426. return
  1427. }
  1428. deleteCache := true
  1429. cacheKey := "CACHE_EDB_INFO_ADD_" + strconv.Itoa(sysUser.AdminId)
  1430. defer func() {
  1431. if deleteCache {
  1432. utils.Rc.Delete(cacheKey)
  1433. }
  1434. }()
  1435. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  1436. deleteCache = false
  1437. br.Msg = "系统处理中,请稍后重试!"
  1438. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  1439. return
  1440. }
  1441. var req data_manage.AddEdbInfoReq
  1442. err := json.Unmarshal(this.Ctx.Input.RequestBody, &req)
  1443. if err != nil {
  1444. br.Msg = "参数解析异常!"
  1445. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  1446. return
  1447. }
  1448. req.EdbName = strings.Trim(req.EdbName, " ")
  1449. req.EdbCode = strings.Trim(req.EdbCode, " ")
  1450. if req.EdbCode == "" {
  1451. br.Msg = "指标ID不能为空"
  1452. return
  1453. }
  1454. if req.EdbName == "" {
  1455. br.Msg = "指标名称不能为空"
  1456. return
  1457. }
  1458. if req.Frequency == "" {
  1459. br.Msg = "频率不能为空"
  1460. return
  1461. }
  1462. if req.Unit == "" {
  1463. br.Msg = "单位不能为空"
  1464. return
  1465. }
  1466. if req.ClassifyId <= 0 {
  1467. br.Msg = "请选择分类"
  1468. return
  1469. }
  1470. tmpInfo, err := data_manage.GetBaseFromMysteelChemicalIndexByCode(req.EdbCode)
  1471. if err != nil {
  1472. if !utils.IsErrNoRow(err) {
  1473. br.Msg = "获取失败"
  1474. br.ErrMsg = "获取失败,Err:" + err.Error()
  1475. return
  1476. }
  1477. }
  1478. var source int
  1479. if tmpInfo != nil {
  1480. source = utils.DATA_SOURCE_MYSTEEL_CHEMICAL
  1481. } else {
  1482. br.Msg = "指标不存在"
  1483. return
  1484. }
  1485. // 指标入库
  1486. edbInfo, err, errMsg, isSendEmail := data.EdbInfoAdd(source, utils.DATA_SUB_SOURCE_EDB, req.ClassifyId, req.EdbCode, req.EdbName, req.Frequency, req.Unit, req.StartDate, req.EndDate, sysUser.AdminId, sysUser.RealName, this.Lang)
  1487. if err != nil {
  1488. br.Msg = "保存失败"
  1489. if errMsg != `` {
  1490. br.Msg = errMsg
  1491. }
  1492. br.ErrMsg = err.Error()
  1493. br.IsSendEmail = isSendEmail
  1494. return
  1495. }
  1496. // 试用平台更新用户累计新增指标数
  1497. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  1498. if e != nil {
  1499. br.Msg = "操作失败"
  1500. br.ErrMsg = "获取系统用户数据失败,Err:" + e.Error()
  1501. return
  1502. }
  1503. if utils.BusinessCode == utils.BusinessCodeSandbox && adminItem.DepartmentName == "ETA试用客户" {
  1504. go func() {
  1505. var r etaTrialService.EtaTrialUserReq
  1506. r.Mobile = adminItem.Mobile
  1507. _, _ = etaTrialService.UpdateUserIndexNum(r)
  1508. }()
  1509. }
  1510. //新增操作日志
  1511. {
  1512. // 添加钢联指标更新日志
  1513. if edbInfo.Source == utils.DATA_SOURCE_MYSTEEL_CHEMICAL {
  1514. go data_stat.AddEdbInfoUpdateLog(edbInfo.EdbInfoId, 1, "", sysUser, 2)
  1515. }
  1516. edbLog := new(data_manage.EdbInfoLog)
  1517. edbLog.EdbInfoId = edbInfo.EdbInfoId
  1518. edbLog.SourceName = edbInfo.SourceName
  1519. edbLog.Source = edbInfo.Source
  1520. edbLog.EdbCode = edbInfo.EdbCode
  1521. edbLog.EdbName = edbInfo.EdbName
  1522. edbLog.ClassifyId = edbInfo.ClassifyId
  1523. edbLog.SysUserId = sysUser.AdminId
  1524. edbLog.SysUserRealName = sysUser.RealName
  1525. edbLog.CreateTime = time.Now()
  1526. edbLog.Content = string(this.Ctx.Input.RequestBody)
  1527. edbLog.Status = "新增指标"
  1528. edbLog.Method = this.Ctx.Input.URI()
  1529. go data_manage.AddEdbInfoLog(edbLog)
  1530. }
  1531. // 更新es
  1532. go data.AddOrEditEdbInfoToEs(edbInfo.EdbInfoId)
  1533. resp := new(data_manage.AddEdbInfoResp)
  1534. resp.EdbInfoId = edbInfo.EdbInfoId
  1535. resp.UniqueCode = edbInfo.UniqueCode
  1536. br.Ret = 200
  1537. br.Success = true
  1538. br.Msg = "保存成功"
  1539. br.Data = resp
  1540. br.IsAddLog = true
  1541. }
  1542. // AddCheck
  1543. // @Title 新增校验
  1544. // @Description 新增校验
  1545. // @Param request body request.BusinessDataBatchAddCheckReq true "type json string"
  1546. // @Success 200 string "操作成功"
  1547. // @router /mysteel_chemical/edb_info/add_check [post]
  1548. func (c *EdbInfoController) AddCheck() {
  1549. br := new(models.BaseResponse).Init()
  1550. defer func() {
  1551. if br.ErrMsg == "" {
  1552. br.IsSendEmail = false
  1553. }
  1554. c.Data["json"] = br
  1555. c.ServeJSON()
  1556. }()
  1557. sysUser := c.SysUser
  1558. if sysUser == nil {
  1559. br.Msg = "请登录"
  1560. br.ErrMsg = "请登录,SysUser Is Empty"
  1561. br.Ret = 408
  1562. return
  1563. }
  1564. var req *data_manage.BatchAddCheckReq
  1565. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  1566. br.Msg = "参数解析异常!"
  1567. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  1568. return
  1569. }
  1570. codeMax := 30
  1571. codeLen := len(req.IndexCodes)
  1572. // 获取指标库已有指标
  1573. existsEdb, e := data_manage.GetEdbCodesBySource(utils.DATA_SOURCE_MYSTEEL_CHEMICAL)
  1574. if e != nil {
  1575. br.Msg = "获取失败"
  1576. br.ErrMsg = "获取自有数据已添加的指标失败, Err: " + e.Error()
  1577. return
  1578. }
  1579. existMap := make(map[string]*data_manage.EdbInfo)
  1580. for _, v := range existsEdb {
  1581. existMap[v.EdbCode] = v
  1582. }
  1583. obj := data_manage.BaseFromMysteelChemicalIndex{}
  1584. if codeLen == 0 {
  1585. br.Msg = "请选择指标"
  1586. return
  1587. }
  1588. if codeLen > codeMax {
  1589. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  1590. return
  1591. }
  1592. // 查询选中的指标
  1593. cond := fmt.Sprintf(` AND %s IN (%s)`, data_manage.BaseFromMysteelChemicalIndexCols.IndexCode, utils.GetOrmInReplace(codeLen))
  1594. pars := make([]interface{}, 0)
  1595. pars = append(pars, req.IndexCodes)
  1596. list, e := obj.GeItemsByCondition(cond, pars, 0)
  1597. if e != nil {
  1598. br.Msg = "获取失败"
  1599. br.ErrMsg = "获取自有数据原始指标列表失败, Err: " + e.Error()
  1600. return
  1601. }
  1602. if len(list) > codeMax {
  1603. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  1604. return
  1605. }
  1606. resp := make([]*data_manage.BaseFromMysteelChemicalIndexItem, 0)
  1607. for _, v := range list {
  1608. if v.EdbInfoId > 0 {
  1609. v.EdbExist = 1
  1610. }
  1611. resp = append(resp, v)
  1612. }
  1613. br.Data = resp
  1614. br.Msg = "校验成功"
  1615. br.Ret = 200
  1616. br.Success = true
  1617. }
  1618. // MysteelChemicalBatchSearch
  1619. // @Title 上海钢联指标查询
  1620. // @Description 上海钢联指标查询
  1621. // @Param BaseFromMysteelChemicalClassifyIds query string true "分类id"
  1622. // @Param Keyword query string true "名称关键词"
  1623. // @Success 200 {object} data_manage.LzFrequency
  1624. // @router /mysteel_chemical/batch_search [get]
  1625. func (this *EdbInfoController) MysteelChemicalBatchSearch() {
  1626. br := new(models.BaseResponse).Init()
  1627. defer func() {
  1628. this.Data["json"] = br
  1629. this.ServeJSON()
  1630. }()
  1631. sysUser := this.SysUser
  1632. if sysUser == nil {
  1633. br.Msg = "请登录"
  1634. br.ErrMsg = "请登录,SysUser Is Empty"
  1635. br.Ret = 408
  1636. return
  1637. }
  1638. classifyIdStr := this.GetString("BaseFromMysteelChemicalClassifyIds")
  1639. var condition string
  1640. var pars []interface{}
  1641. classifyIds := strings.Split(classifyIdStr, ",")
  1642. if len(classifyIds) > 0 && classifyIds[0] != `` {
  1643. condition += " AND base_from_mysteel_chemical_classify_id IN (" + utils.GetOrmInReplace(len(classifyIds)) + " ) "
  1644. pars = append(pars, classifyIds)
  1645. }
  1646. keyword := this.GetString("Keyword")
  1647. if keyword != `` {
  1648. condition += " AND (index_name like ? OR index_code like ?) "
  1649. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  1650. }
  1651. // 频度(多选)
  1652. frequencies := this.GetString("Frequencies")
  1653. frequencies = strings.TrimSpace(frequencies)
  1654. if frequencies != "" {
  1655. freArr := strings.Split(frequencies, ",")
  1656. if len(freArr) > 0 {
  1657. condition += " AND frequency IN ?"
  1658. pars = append(pars, freArr)
  1659. }
  1660. }
  1661. resp := new(data_manage.MysteelChemicalPageListResp)
  1662. resp.List = make([]*data_manage.MysteelChemicalList, 0)
  1663. //if classifyIdStr == `` && keyword == `` {
  1664. // var list = make([]*data_manage.MysteelChemicalList, 0)
  1665. // br.Ret = 200
  1666. // br.Success = true
  1667. // br.Msg = "获取成功"
  1668. // br.Data = list
  1669. // return
  1670. //}
  1671. // 分页查询
  1672. pageSize, _ := this.GetInt("PageSize")
  1673. currentIndex, _ := this.GetInt("CurrentIndex")
  1674. var startSize int
  1675. if pageSize <= 0 {
  1676. pageSize = utils.PageSize10
  1677. }
  1678. if currentIndex <= 0 {
  1679. currentIndex = 1
  1680. }
  1681. startSize = utils.StartIndex(currentIndex, pageSize)
  1682. // 获取未加入指标库的原始指标
  1683. total, e := data_manage.GetNoEdbMysteelChemicalIndexCount(condition, pars)
  1684. if e != nil {
  1685. br.Msg = "获取失败"
  1686. br.ErrMsg = fmt.Sprintf("获取未加入指标库的钢联指标数失败, %v", e)
  1687. return
  1688. }
  1689. if total <= 0 {
  1690. page := paging.GetPaging(currentIndex, pageSize, 0)
  1691. resp.Paging = page
  1692. br.Data = resp
  1693. br.Ret = 200
  1694. br.Success = true
  1695. br.Msg = "获取成功"
  1696. return
  1697. }
  1698. list, err := data_manage.GetNoEdbMysteelChemicalIndexPageList(condition, pars, startSize, pageSize)
  1699. if err != nil {
  1700. br.Msg = "获取失败"
  1701. br.ErrMsg = "获取失败,Err:" + err.Error()
  1702. return
  1703. }
  1704. classifyIdInts := make([]int, 0)
  1705. for _, v := range list {
  1706. v.UniqueCode = fmt.Sprint(v.BaseFromMysteelChemicalClassifyId, "_", v.Id)
  1707. if v.BaseFromMysteelChemicalClassifyId > 0 {
  1708. classifyIdInts = append(classifyIdInts, v.BaseFromMysteelChemicalClassifyId)
  1709. }
  1710. }
  1711. classifyListMap := make(map[int]int, 0)
  1712. if len(classifyIdInts) > 0 {
  1713. // 查询父级分类信息
  1714. classifyList, e := data_manage.GetBaseFromMysteelChemicalClassifyByIds(classifyIdInts)
  1715. if e != nil {
  1716. br.Msg = "获取目录信息失败"
  1717. br.ErrMsg = "获取失获取目录信息失败,Err:" + e.Error()
  1718. return
  1719. }
  1720. for _, v := range classifyList {
  1721. classifyListMap[v.BaseFromMysteelChemicalClassifyId] = v.ParentId
  1722. }
  1723. }
  1724. for _, v := range list {
  1725. if p, ok := classifyListMap[v.BaseFromMysteelChemicalClassifyId]; ok {
  1726. v.ParentClassifyId = p
  1727. }
  1728. resp.List = append(resp.List, v)
  1729. }
  1730. page := paging.GetPaging(currentIndex, pageSize, total)
  1731. resp.Paging = page
  1732. br.Ret = 200
  1733. br.Success = true
  1734. br.Msg = "获取成功"
  1735. br.Data = resp
  1736. }