bloomberg_data.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  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. "strconv"
  13. "strings"
  14. "time"
  15. "github.com/rdlucklib/rdluck_tools/paging"
  16. )
  17. // BloombergDataController 彭博数据源
  18. type BloombergDataController struct {
  19. controllers.BaseAuthController
  20. }
  21. // List
  22. // @Title 指标列表
  23. // @Description 指标列表
  24. // @Param PageSize query int false "每页数据量"
  25. // @Param CurrentIndex query int false "页码"
  26. // @Param Frequency query string false "频度"
  27. // @Param Keywords query string false "指标ID/指标名称"
  28. // @Param ListAll query bool false "列表全选"
  29. // @Param SortField query int false "排序字段: 0-默认; 1-开始时间; 2-最新时间; 3-更新时间"
  30. // @Param SortRule query int false "排序方式: 0-默认; 1-正序; 2-倒序"
  31. // @Success 200 {object} data_manage.BloombergSourceListResp
  32. // @router /bloomberg_source/list [get]
  33. func (this *BloombergDataController) List() {
  34. br := new(models.BaseResponse).Init()
  35. defer func() {
  36. if br.ErrMsg == "" {
  37. br.IsSendEmail = false
  38. }
  39. this.Data["json"] = br
  40. this.ServeJSON()
  41. }()
  42. sysUser := this.SysUser
  43. if sysUser == nil {
  44. br.Msg = "请登录"
  45. br.ErrMsg = "请登录,SysUser Is Empty"
  46. br.Ret = 408
  47. return
  48. }
  49. params := new(data_manage.BloombergSourceListReq)
  50. if e := this.ParseForm(params); e != nil {
  51. br.Msg = "获取失败"
  52. br.ErrMsg = "参数解析失败, Err: " + e.Error()
  53. return
  54. }
  55. var startSize int
  56. if params.PageSize <= 0 {
  57. params.PageSize = utils.PageSize20
  58. }
  59. if params.CurrentIndex <= 0 {
  60. params.CurrentIndex = 1
  61. }
  62. startSize = utils.StartIndex(params.CurrentIndex, params.PageSize)
  63. dataResp := new(data_manage.BloombergSourceListResp)
  64. cond := ``
  65. pars := make([]interface{}, 0)
  66. // 筛选项
  67. {
  68. params.Keywords = strings.TrimSpace(params.Keywords)
  69. if params.Keywords != "" {
  70. // 空格分词搜
  71. indexCodeCol := data_manage.BaseFromBloombergIndexCols.IndexCode
  72. indexNameCol := data_manage.BaseFromBloombergIndexCols.IndexName
  73. //keywordArr := strings.Split(params.Keywords, " ")
  74. //if len(keywordArr) > 1 {
  75. // sliceArr := make([]string, 0)
  76. // sliceArr = append(sliceArr, fmt.Sprintf(` %s LIKE ? OR %s LIKE ? `, indexCodeCol, indexNameCol))
  77. // pars = utils.GetLikeKeywordPars(pars, params.Keywords, 2)
  78. //
  79. // for _, v := range keywordArr {
  80. // if v == ` ` || v == `` {
  81. // continue
  82. // }
  83. // sliceArr = append(sliceArr, fmt.Sprintf(` %s LIKE ? OR %s LIKE ? `, indexCodeCol, indexNameCol))
  84. // pars = utils.GetLikeKeywordPars(pars, v, 2)
  85. // }
  86. // cond += ` AND (` + strings.Join(sliceArr, " OR ") + `)`
  87. //} else {
  88. // cond += fmt.Sprintf(` AND (%s LIKE ? OR %s LIKE ?)`, indexCodeCol, indexNameCol)
  89. // pars = utils.GetLikeKeywordPars(pars, params.Keywords, 2)
  90. //}
  91. cond += fmt.Sprintf(` AND (%s LIKE ? OR %s LIKE ?)`, indexCodeCol, indexNameCol)
  92. pars = utils.GetLikeKeywordPars(pars, params.Keywords, 2)
  93. // ES搜
  94. //_, list, e := elastic.SearchDataSourceIndex(utils.EsDataSourceIndexName, params.Keywords, utils.DATA_SOURCE_BLOOMBERG, 0, []int{}, []int{}, []string{}, startSize, params.PageSize)
  95. //if e != nil {
  96. // br.Msg = "获取失败"
  97. // br.ErrMsg = fmt.Sprintf("ES-搜索Bloomberg指标失败, %v", e)
  98. // return
  99. //}
  100. //if len(list) == 0 {
  101. // dataResp.Paging = paging.GetPaging(params.CurrentIndex, params.PageSize, 0)
  102. // dataResp.List = make([]*data_manage.BaseFromBloombergIndexItem, 0)
  103. // br.Data = dataResp
  104. // br.Ret = 200
  105. // br.Success = true
  106. // br.Msg = "获取成功"
  107. // return
  108. //}
  109. //var indexIds []int
  110. //for _, v := range list {
  111. // indexIds = append(indexIds, v.PrimaryId)
  112. //}
  113. //cond += fmt.Sprintf(" AND %s IN (%s)", data_manage.BaseFromBloombergIndexCols.BaseFromBloombergIndexId, utils.GetOrmInReplace(len(indexIds)))
  114. //pars = append(pars, indexIds)
  115. }
  116. if params.Frequency != "" {
  117. cond += fmt.Sprintf(` AND %s = ?`, data_manage.BaseFromBloombergIndexCols.Frequency)
  118. pars = append(pars, params.Frequency)
  119. }
  120. }
  121. // 分页列表
  122. bloombergOb := new(data_manage.BaseFromBloombergIndex)
  123. total, e := bloombergOb.GetCountByCondition(cond, pars)
  124. if e != nil {
  125. br.Msg = "获取失败"
  126. br.ErrMsg = "获取Bloomberg原始指标列表总数失败, Err: " + e.Error()
  127. return
  128. }
  129. // 排序, 默认创建时间倒序
  130. orderFields := map[int]string{
  131. 0: data_manage.BaseFromBloombergIndexCols.CreateTime, 1: data_manage.BaseFromBloombergIndexCols.StartDate,
  132. 2: data_manage.BaseFromBloombergIndexCols.EndDate, 3: data_manage.BaseFromBloombergIndexCols.ModifyTime,
  133. }
  134. orderType := map[int]string{0: "DESC", 1: "ASC", 2: "DESC"}
  135. orderRule := fmt.Sprintf("%s %s", orderFields[params.SortField], orderType[params.SortRule])
  136. // 列表
  137. edbList := make([]*data_manage.BaseFromBloombergIndex, 0)
  138. if params.ListAll {
  139. list, e := bloombergOb.GetItemsByCondition(cond, pars, []string{}, orderRule, 0)
  140. if e != nil {
  141. br.Msg = "获取失败"
  142. br.ErrMsg = "获取Bloomberg原始指标列表失败, Err: " + e.Error()
  143. return
  144. }
  145. edbList = list
  146. } else {
  147. list, e := bloombergOb.GetPageItemsByCondition(cond, pars, []string{}, orderRule, startSize, params.PageSize)
  148. if e != nil {
  149. br.Msg = "获取失败"
  150. br.ErrMsg = "获取Bloomberg原始指标分页列表失败, Err: " + e.Error()
  151. return
  152. }
  153. edbList = list
  154. }
  155. // 获取指标库已有指标
  156. existsEdb, e := data_manage.GetEdbCodesBySource(utils.DATA_SOURCE_BLOOMBERG)
  157. if e != nil {
  158. br.Msg = "获取失败"
  159. br.ErrMsg = "获取Bloomberg已添加的指标失败, Err: " + e.Error()
  160. return
  161. }
  162. existMap := make(map[string]*data_manage.EdbInfo)
  163. for _, v := range existsEdb {
  164. existMap[v.EdbCode] = v
  165. }
  166. respList := make([]*data_manage.BaseFromBloombergIndexItem, 0)
  167. for _, v := range edbList {
  168. t := data_manage.FormatBaseFromBloombergIndex2Item(v)
  169. ed := existMap[v.IndexCode]
  170. if ed != nil {
  171. t.EdbExist = 1
  172. t.EdbInfoId = ed.EdbInfoId
  173. t.EdbUniqueCode = ed.UniqueCode
  174. t.EdbClassifyId = ed.ClassifyId
  175. }
  176. respList = append(respList, t)
  177. }
  178. page := paging.GetPaging(params.CurrentIndex, params.PageSize, total)
  179. dataResp.Paging = page
  180. dataResp.List = respList
  181. br.Data = dataResp
  182. br.Ret = 200
  183. br.Success = true
  184. br.Msg = "获取成功"
  185. }
  186. // BatchAdd
  187. // @Title 批量新增
  188. // @Description 批量新增
  189. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  190. // @Success 200 string "操作成功"
  191. // @router /bloomberg_source/batch_add [post]
  192. func (this *BloombergDataController) BatchAdd() {
  193. br := new(models.BaseResponse).Init()
  194. defer func() {
  195. if br.ErrMsg == "" {
  196. br.IsSendEmail = false
  197. }
  198. this.Data["json"] = br
  199. this.ServeJSON()
  200. }()
  201. sysUser := this.SysUser
  202. if sysUser == nil {
  203. br.Msg = "请登录"
  204. br.ErrMsg = "请登录,SysUser Is Empty"
  205. br.Ret = 408
  206. return
  207. }
  208. deleteCache := true
  209. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_BLOOMBERG_" + strconv.Itoa(sysUser.AdminId)
  210. defer func() {
  211. if deleteCache {
  212. _ = utils.Rc.Delete(cacheKey)
  213. }
  214. }()
  215. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  216. deleteCache = false
  217. br.Msg = "系统处理中,请稍后重试!"
  218. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  219. return
  220. }
  221. var req []*data_manage.AddEdbInfoReq
  222. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  223. br.Msg = "参数解析异常!"
  224. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  225. return
  226. }
  227. if len(req) == 0 {
  228. br.Msg = "请选择指标"
  229. return
  230. }
  231. if len(req) > 30 {
  232. br.Msg = "批量添加指标数量不得超过30个"
  233. return
  234. }
  235. indexNames := make([]string, 0)
  236. for _, v := range req {
  237. v.EdbCode = strings.TrimSpace(v.EdbCode)
  238. if v.EdbCode == "" {
  239. br.Msg = "指标ID不可为空"
  240. return
  241. }
  242. v.EdbName = strings.TrimSpace(v.EdbName)
  243. if v.EdbName == "" {
  244. br.Msg = "请输入指标名称"
  245. return
  246. }
  247. indexNames = append(indexNames, v.EdbName)
  248. v.Frequency = strings.TrimSpace(v.Frequency)
  249. if v.Frequency == "" {
  250. br.Msg = "请选择频度"
  251. return
  252. }
  253. v.Unit = strings.TrimSpace(v.Unit)
  254. if v.Unit == "" {
  255. br.Msg = "请输入单位"
  256. return
  257. }
  258. if v.ClassifyId <= 0 {
  259. br.Msg = "请选择分类"
  260. return
  261. }
  262. }
  263. // 限定同一时间最多批量新增30个指标
  264. for _, v := range req {
  265. var r data.BloombergIndexSource2EdbReq
  266. r.EdbCode = v.EdbCode
  267. r.EdbName = v.EdbName
  268. r.Frequency = v.Frequency
  269. r.Unit = v.Unit
  270. r.ClassifyId = v.ClassifyId
  271. r.AdminId = sysUser.AdminId
  272. r.AdminRealName = sysUser.RealName
  273. edbInfo, e, errMsg, skip := data.BloombergIndexSource2Edb(r, this.Lang)
  274. if e != nil {
  275. br.Msg = "操作失败"
  276. if errMsg != "" {
  277. br.Msg = errMsg
  278. }
  279. br.ErrMsg = e.Error()
  280. return
  281. }
  282. if skip {
  283. continue
  284. }
  285. // 试用平台更新用户累计新增指标数
  286. if utils.BusinessCode == utils.BusinessCodeSandbox {
  287. go func() {
  288. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  289. if e != nil {
  290. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  291. utils.FileLog.Info(tips)
  292. return
  293. }
  294. if adminItem.DepartmentName != "ETA试用客户" {
  295. return
  296. }
  297. var ur etaTrialService.EtaTrialUserReq
  298. ur.Mobile = adminItem.Mobile
  299. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  300. }()
  301. }
  302. // 新增操作日志
  303. {
  304. edbLog := new(data_manage.EdbInfoLog)
  305. edbLog.EdbInfoId = edbInfo.EdbInfoId
  306. edbLog.SourceName = edbInfo.SourceName
  307. edbLog.Source = edbInfo.Source
  308. edbLog.EdbCode = edbInfo.EdbCode
  309. edbLog.EdbName = edbInfo.EdbName
  310. edbLog.ClassifyId = edbInfo.ClassifyId
  311. edbLog.SysUserId = sysUser.AdminId
  312. edbLog.SysUserRealName = sysUser.RealName
  313. edbLog.CreateTime = time.Now()
  314. edbLog.Content = string(this.Ctx.Input.RequestBody)
  315. edbLog.Status = "新增指标"
  316. edbLog.Method = this.Ctx.Input.URI()
  317. go data_manage.AddEdbInfoLog(edbLog)
  318. }
  319. }
  320. br.Msg = "操作成功"
  321. br.Ret = 200
  322. br.Success = true
  323. br.IsAddLog = true
  324. }
  325. // NameCheck
  326. // @Title 重名校验
  327. // @Description 批量新增
  328. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  329. // @Success 200 string "操作成功"
  330. // @router /bloomberg_source/name_check [post]
  331. func (this *BloombergDataController) NameCheck() {
  332. br := new(models.BaseResponse).Init()
  333. defer func() {
  334. if br.ErrMsg == "" {
  335. br.IsSendEmail = false
  336. }
  337. this.Data["json"] = br
  338. this.ServeJSON()
  339. }()
  340. sysUser := this.SysUser
  341. if sysUser == nil {
  342. br.Msg = "请登录"
  343. br.ErrMsg = "请登录,SysUser Is Empty"
  344. br.Ret = 408
  345. return
  346. }
  347. var req []*data_manage.AddEdbInfoReq
  348. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  349. br.Msg = "参数解析异常!"
  350. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  351. return
  352. }
  353. if len(req) == 0 {
  354. br.Msg = "请选择指标"
  355. return
  356. }
  357. type NameCheckResult struct {
  358. EdbCode string
  359. EdbName string
  360. Exist bool
  361. }
  362. indexNames := make([]string, 0)
  363. resp := make([]*NameCheckResult, 0)
  364. for _, v := range req {
  365. v.EdbCode = strings.TrimSpace(v.EdbCode)
  366. if v.EdbCode == "" {
  367. br.Msg = "指标ID不可为空"
  368. return
  369. }
  370. v.EdbName = strings.TrimSpace(v.EdbName)
  371. if v.EdbName == "" {
  372. br.Msg = "请输入指标名称"
  373. return
  374. }
  375. indexNames = append(indexNames, v.EdbName)
  376. resp = append(resp, &NameCheckResult{
  377. EdbCode: v.EdbCode,
  378. EdbName: v.EdbName,
  379. })
  380. }
  381. // 重名校验
  382. edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
  383. if e != nil {
  384. br.Msg = "操作失败"
  385. br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
  386. return
  387. }
  388. nameExists := make(map[string]bool)
  389. for _, v := range edbList {
  390. nameExists[v.EdbName] = true
  391. }
  392. if len(nameExists) > 0 {
  393. for _, v := range resp {
  394. v.Exist = nameExists[v.EdbName]
  395. }
  396. }
  397. br.Data = resp
  398. br.Msg = "操作成功"
  399. br.Ret = 200
  400. br.Success = true
  401. }
  402. // AddCheck
  403. // @Title 新增校验
  404. // @Description 新增校验
  405. // @Param request body data_manage.BloombergSourceBatchAddCheckReq true "type json string"
  406. // @Success 200 string "操作成功"
  407. // @router /bloomberg_source/add_check [post]
  408. func (this *BloombergDataController) AddCheck() {
  409. br := new(models.BaseResponse).Init()
  410. defer func() {
  411. if br.ErrMsg == "" {
  412. br.IsSendEmail = false
  413. }
  414. this.Data["json"] = br
  415. this.ServeJSON()
  416. }()
  417. sysUser := this.SysUser
  418. if sysUser == nil {
  419. br.Msg = "请登录"
  420. br.ErrMsg = "请登录,SysUser Is Empty"
  421. br.Ret = 408
  422. return
  423. }
  424. var req *data_manage.BloombergSourceBatchAddCheckReq
  425. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  426. br.Msg = "参数解析异常!"
  427. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  428. return
  429. }
  430. codeMax := 30
  431. codeLen := len(req.IndexCodes)
  432. // 排序, 默认创建时间倒序
  433. orderFields := map[int]string{
  434. 0: data_manage.BaseFromBloombergIndexCols.CreateTime, 1: data_manage.BaseFromBloombergIndexCols.StartDate,
  435. 2: data_manage.BaseFromBloombergIndexCols.EndDate, 3: data_manage.BaseFromBloombergIndexCols.ModifyTime,
  436. }
  437. orderType := map[int]string{0: "DESC", 1: "ASC", 2: "DESC"}
  438. orderRule := fmt.Sprintf("%s %s", orderFields[req.SortField], orderType[req.SortRule])
  439. // 获取指标库已有指标
  440. existsEdb, e := data_manage.GetEdbCodesBySource(utils.DATA_SOURCE_BLOOMBERG)
  441. if e != nil {
  442. br.Msg = "获取失败"
  443. br.ErrMsg = "获取Bloomberg已添加的指标失败, Err: " + e.Error()
  444. return
  445. }
  446. existMap := make(map[string]*data_manage.EdbInfo)
  447. for _, v := range existsEdb {
  448. existMap[v.EdbCode] = v
  449. }
  450. // 非全选-不需要频率等筛选条件
  451. bloombergOb := new(data_manage.BaseFromBloombergIndex)
  452. if !req.ListAll {
  453. if codeLen == 0 {
  454. br.Msg = "请选择指标"
  455. return
  456. }
  457. if codeLen > codeMax {
  458. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  459. return
  460. }
  461. // 查询选中的指标
  462. cond := fmt.Sprintf(` AND %s IN (%s)`, data_manage.BaseFromBloombergIndexCols.IndexCode, utils.GetOrmInReplace(codeLen))
  463. pars := make([]interface{}, 0)
  464. pars = append(pars, req.IndexCodes)
  465. list, e := bloombergOb.GetItemsByCondition(cond, pars, []string{}, orderRule, 0)
  466. if e != nil {
  467. br.Msg = "获取失败"
  468. br.ErrMsg = "获取Bloomberg原始指标列表失败, Err: " + e.Error()
  469. return
  470. }
  471. resp := make([]*data_manage.BaseFromBloombergIndexItem, 0)
  472. for _, v := range list {
  473. t := data_manage.FormatBaseFromBloombergIndex2Item(v)
  474. ed := existMap[v.IndexCode]
  475. if ed != nil {
  476. t.EdbExist = 1
  477. t.EdbInfoId = ed.EdbInfoId
  478. t.EdbUniqueCode = ed.UniqueCode
  479. t.EdbClassifyId = ed.ClassifyId
  480. }
  481. resp = append(resp, t)
  482. }
  483. br.Data = resp
  484. br.Msg = "校验成功"
  485. br.Ret = 200
  486. br.Success = true
  487. return
  488. }
  489. // 全选-需要频率等筛选条件
  490. cond := ``
  491. pars := make([]interface{}, 0)
  492. // 筛选项
  493. req.Keywords = strings.TrimSpace(req.Keywords)
  494. if req.Keywords != "" {
  495. // 空格分词搜
  496. indexCodeCol := data_manage.BaseFromBloombergIndexCols.IndexCode
  497. indexNameCol := data_manage.BaseFromBloombergIndexCols.IndexName
  498. keywordArr := strings.Split(req.Keywords, " ")
  499. if len(keywordArr) > 1 {
  500. sliceArr := make([]string, 0)
  501. sliceArr = append(sliceArr, fmt.Sprintf(` %s LIKE ? OR %s LIKE ? `, indexCodeCol, indexNameCol))
  502. pars = utils.GetLikeKeywordPars(pars, req.Keywords, 2)
  503. for _, v := range keywordArr {
  504. if v == ` ` || v == `` {
  505. continue
  506. }
  507. sliceArr = append(sliceArr, fmt.Sprintf(` %s LIKE ? OR %s LIKE ? `, indexCodeCol, indexNameCol))
  508. pars = utils.GetLikeKeywordPars(pars, v, 2)
  509. }
  510. cond += ` AND (` + strings.Join(sliceArr, " OR ") + `)`
  511. } else {
  512. cond += fmt.Sprintf(` AND (%s LIKE ? OR %s LIKE ?)`, indexCodeCol, indexNameCol)
  513. pars = utils.GetLikeKeywordPars(pars, req.Keywords, 2)
  514. }
  515. }
  516. if req.Frequency != "" {
  517. cond += fmt.Sprintf(` AND %s = ?`, data_manage.BaseFromBloombergIndexCols.Frequency)
  518. pars = append(pars, req.Frequency)
  519. }
  520. // 排除对应指标
  521. if codeLen > 0 {
  522. cond += fmt.Sprintf(` AND %s NOT IN (%s)`, data_manage.BaseFromBloombergIndexCols.IndexCode, utils.GetOrmInReplace(codeLen))
  523. pars = append(pars, req.IndexCodes)
  524. }
  525. // 查询max+1个指标
  526. list, e := bloombergOb.GetItemsByCondition(cond, pars, []string{}, orderRule, codeMax+1)
  527. if e != nil {
  528. br.Msg = "获取失败"
  529. br.ErrMsg = "获取Bloomberg原始指标列表失败, Err: " + e.Error()
  530. return
  531. }
  532. if len(list) > codeMax {
  533. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  534. return
  535. }
  536. resp := make([]*data_manage.BaseFromBloombergIndexItem, 0)
  537. for _, v := range list {
  538. t := data_manage.FormatBaseFromBloombergIndex2Item(v)
  539. ed := existMap[v.IndexCode]
  540. if ed != nil {
  541. t.EdbExist = 1
  542. t.EdbInfoId = ed.EdbInfoId
  543. t.EdbUniqueCode = ed.UniqueCode
  544. t.EdbClassifyId = ed.ClassifyId
  545. }
  546. resp = append(resp, t)
  547. }
  548. br.Data = resp
  549. br.Msg = "校验成功"
  550. br.Ret = 200
  551. br.Success = true
  552. }
  553. // EdbInfoAdd
  554. // @Title 加入到指标库
  555. // @Description 加入到指标库
  556. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  557. // @Success 200 string "操作成功"
  558. // @router /bloomberg_source/edb_add [post]
  559. func (this *BloombergDataController) EdbInfoAdd() {
  560. br := new(models.BaseResponse).Init()
  561. defer func() {
  562. this.Data["json"] = br
  563. this.ServeJSON()
  564. }()
  565. sysUser := this.SysUser
  566. if sysUser == nil {
  567. br.Msg = "请登录"
  568. br.ErrMsg = "请登录,SysUser Is Empty"
  569. br.Ret = 408
  570. return
  571. }
  572. deleteCache := true
  573. cacheKey := "CACHE_EDB_INFO_SINGLE_ADD_BLOOMBERG_" + strconv.Itoa(sysUser.AdminId)
  574. defer func() {
  575. if deleteCache {
  576. _ = utils.Rc.Delete(cacheKey)
  577. }
  578. }()
  579. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  580. deleteCache = false
  581. br.Msg = "系统处理中,请稍后重试!"
  582. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(this.Ctx.Input.RequestBody)
  583. return
  584. }
  585. req := new(data_manage.AddEdbInfoReq)
  586. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  587. br.Msg = "参数解析异常!"
  588. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  589. return
  590. }
  591. if req.EdbCode == "" {
  592. br.Msg = "指标ID不可为空"
  593. return
  594. }
  595. req.EdbName = strings.TrimSpace(req.EdbName)
  596. if req.EdbName == "" {
  597. br.Msg = "请输入指标名称"
  598. return
  599. }
  600. req.Frequency = strings.TrimSpace(req.Frequency)
  601. if req.Frequency == "" {
  602. br.Msg = "请选择频度"
  603. return
  604. }
  605. req.Unit = strings.TrimSpace(req.Unit)
  606. if req.Unit == "" {
  607. br.Msg = "请输入单位"
  608. return
  609. }
  610. if req.ClassifyId <= 0 {
  611. br.Msg = "请选择分类"
  612. return
  613. }
  614. // 新增指标
  615. var r data.BloombergIndexSource2EdbReq
  616. r.EdbCode = req.EdbCode
  617. r.EdbName = req.EdbName
  618. r.Frequency = req.Frequency
  619. r.Unit = req.Unit
  620. r.ClassifyId = req.ClassifyId
  621. r.AdminId = this.SysUser.AdminId
  622. r.AdminRealName = this.SysUser.RealName
  623. edbInfo, e, errMsg, skip := data.BloombergIndexSource2Edb(r, this.Lang)
  624. if e != nil {
  625. br.Msg = "操作失败"
  626. if errMsg != "" {
  627. br.Msg = errMsg
  628. }
  629. br.ErrMsg = e.Error()
  630. return
  631. }
  632. if skip {
  633. br.Msg = "指标已存在"
  634. return
  635. }
  636. // 试用平台更新用户累计新增指标数
  637. if utils.BusinessCode == utils.BusinessCodeSandbox {
  638. go func() {
  639. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  640. if e != nil {
  641. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  642. utils.FileLog.Info(tips)
  643. return
  644. }
  645. if adminItem.DepartmentName != "ETA试用客户" {
  646. return
  647. }
  648. var ur etaTrialService.EtaTrialUserReq
  649. ur.Mobile = adminItem.Mobile
  650. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  651. }()
  652. }
  653. // 新增操作日志
  654. {
  655. edbLog := new(data_manage.EdbInfoLog)
  656. edbLog.EdbInfoId = edbInfo.EdbInfoId
  657. edbLog.SourceName = edbInfo.SourceName
  658. edbLog.Source = edbInfo.Source
  659. edbLog.EdbCode = edbInfo.EdbCode
  660. edbLog.EdbName = edbInfo.EdbName
  661. edbLog.ClassifyId = edbInfo.ClassifyId
  662. edbLog.SysUserId = sysUser.AdminId
  663. edbLog.SysUserRealName = sysUser.RealName
  664. edbLog.CreateTime = time.Now()
  665. edbLog.Content = string(this.Ctx.Input.RequestBody)
  666. edbLog.Status = "新增指标"
  667. edbLog.Method = this.Ctx.Input.URI()
  668. go data_manage.AddEdbInfoLog(edbLog)
  669. }
  670. br.Msg = "操作成功"
  671. br.Ret = 200
  672. br.Success = true
  673. br.IsAddLog = true
  674. }