kpler_data.go 49 KB

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