mysteel_chemical_data.go 52 KB

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