manual_edb.go 36 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472
  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. "github.com/rdlucklib/rdluck_tools/paging"
  13. "github.com/shopspring/decimal"
  14. "os"
  15. "strconv"
  16. "strings"
  17. "time"
  18. )
  19. // ManualEdbController 手工指标服务(鉴权)
  20. type ManualEdbController struct {
  21. controllers.BaseAuthController
  22. }
  23. // EdbDetail
  24. // @Title 指标列表
  25. // @Description 指标列表
  26. // @Param TradeCode query string true "指标编码"
  27. // @Success 200 {object} models.TargetDetailResp
  28. // @router /target/edb/detail [get]
  29. func (c *ManualEdbController) EdbDetail() {
  30. br := new(models.BaseResponse).Init()
  31. defer func() {
  32. c.Data["json"] = br
  33. c.ServeJSON()
  34. }()
  35. sysUser := c.SysUser
  36. if sysUser == nil {
  37. br.Msg = "请重新登录"
  38. return
  39. }
  40. // 指标编码
  41. tradeCode := c.GetString("TradeCode")
  42. if tradeCode == `` {
  43. br.Msg = "请输入指标编码"
  44. br.IsSendEmail = false
  45. return
  46. }
  47. manualEdbInfo, err := models.GetTargetByTradeCode(tradeCode)
  48. if err != nil {
  49. br.Msg = "获取指标失败"
  50. br.ErrMsg = "获取指标失败,err:" + err.Error()
  51. return
  52. }
  53. // 如果不是超管账号,那么得校验下当前用户是否有该指标的权限
  54. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  55. //获取账户所拥有权限的分类id集合
  56. count, err := models.GetCountManualUserClassify(sysUser.AdminId, manualEdbInfo.ClassifyId)
  57. if err != nil {
  58. br.Msg = "获取分类数据失败"
  59. br.ErrMsg = "获取分类数据失败,err:" + err.Error()
  60. return
  61. }
  62. if count <= 0 {
  63. br.Msg = "无权访问"
  64. br.ErrMsg = "无权访问"
  65. br.IsSendEmail = false
  66. return
  67. }
  68. }
  69. classifyList := make([]*models.EdbdataClassify, 0)
  70. {
  71. classify, err := models.GetManualClassifyByClassifyId(manualEdbInfo.ClassifyId)
  72. if err != nil {
  73. br.Msg = "获取分类信息失败"
  74. br.ErrMsg = "获取分类信息失败,err:" + err.Error()
  75. return
  76. }
  77. parentClassify, err := models.GetManualClassifyByClassifyId(classify.ParentId)
  78. if err != nil {
  79. br.Msg = "获取分类信息失败"
  80. br.ErrMsg = "获取父级分类信息失败,err:" + err.Error()
  81. return
  82. }
  83. classifyList = []*models.EdbdataClassify{
  84. parentClassify, classify,
  85. }
  86. }
  87. // 明细数据
  88. {
  89. dataList, err := models.GetEdbDataListByCode(manualEdbInfo.TradeCode)
  90. if err != nil {
  91. br.Msg = "获取明细数据失败"
  92. br.ErrMsg = "获取明细数据失败,err:" + err.Error()
  93. return
  94. }
  95. manualEdbInfo.DataList = dataList
  96. }
  97. resp := models.TargetDetailResp{
  98. Detail: manualEdbInfo,
  99. ClassifyList: classifyList,
  100. }
  101. br.Ret = 200
  102. br.Success = true
  103. br.Msg = "获取成功"
  104. //br.Data = resp
  105. br.Data = resp
  106. }
  107. // ClassifyEdbList
  108. // @Title 分类指标列表
  109. // @Description 指标列表
  110. // @Param ClassifyId query string true "分类id"
  111. // @Success 200 {object} models.TargetDetailResp
  112. // @router /target/classify/edb/list [get]
  113. func (c *ManualEdbController) ClassifyEdbList() {
  114. br := new(models.BaseResponse).Init()
  115. defer func() {
  116. c.Data["json"] = br
  117. c.ServeJSON()
  118. }()
  119. sysUser := c.SysUser
  120. if sysUser == nil {
  121. br.Msg = "请登录"
  122. br.Ret = 408
  123. return
  124. }
  125. resp := new(models.EdbdataClassifyResp)
  126. classifyId, _ := c.GetInt("ClassifyId") //分类
  127. if classifyId <= 0 {
  128. br.Msg = "请传入分类"
  129. br.ErrMsg = "请传入分类"
  130. br.IsSendEmail = false
  131. }
  132. var condition string
  133. var pars []interface{}
  134. // 如果不是超管账号,那么得校验下当前用户是否有该指标的权限
  135. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  136. //获取账户所拥有权限的分类id集合
  137. count, err := models.GetCountManualUserClassify(sysUser.AdminId, classifyId)
  138. if err != nil {
  139. br.Msg = "获取分类数据失败"
  140. br.ErrMsg = "获取分类数据失败,err:" + err.Error()
  141. return
  142. }
  143. if count <= 0 {
  144. br.Msg = "无权访问"
  145. br.ErrMsg = "无权访问"
  146. br.IsSendEmail = false
  147. return
  148. }
  149. }
  150. list := make([]*models.EdbdataClassifyList, 0)
  151. condition += ` AND a.classify_id = ? `
  152. pars = append(pars, classifyId)
  153. tmpList, err := models.GetEdbInfoList(condition, pars, 0, 0)
  154. if err != nil {
  155. br.Msg = "获取失败"
  156. br.ErrMsg = "获取失败,Err:" + err.Error()
  157. return
  158. }
  159. for _, v := range tmpList {
  160. list = append(list, &models.EdbdataClassifyList{
  161. ClassifyId: 0,
  162. ClassifyName: v.SecName,
  163. ParentId: v.ClassifyId,
  164. Child: nil,
  165. TradeCode: v.TradeCode,
  166. UniqueCode: utils.MD5(v.TradeCode),
  167. })
  168. }
  169. resp.List = list
  170. br.Ret = 200
  171. br.Success = true
  172. br.Msg = "获取成功"
  173. br.Data = resp
  174. }
  175. // EdbSearch
  176. // @Title 指标搜索
  177. // @Description 指标搜索
  178. // @Param Keyword query string false "关键字搜索"
  179. // @Success 200 {object} models.TargetItemListResp
  180. // @router /target/edb/search [get]
  181. func (c *ManualEdbController) EdbSearch() {
  182. br := new(models.BaseResponse).Init()
  183. defer func() {
  184. c.Data["json"] = br
  185. c.ServeJSON()
  186. }()
  187. sysUser := c.SysUser
  188. if sysUser == nil {
  189. br.Msg = "请重新登录"
  190. return
  191. }
  192. pageSize, _ := c.GetInt("PageSize")
  193. currentIndex, _ := c.GetInt("CurrentIndex")
  194. var startSize int
  195. if pageSize <= 0 {
  196. pageSize = utils.PageSize20
  197. }
  198. if currentIndex <= 0 {
  199. currentIndex = 1
  200. }
  201. startSize = utils.StartIndex(currentIndex, pageSize)
  202. //关键字
  203. keyword := c.GetString("Keyword")
  204. var condition string
  205. var pars []interface{}
  206. userId := sysUser.AdminId
  207. //超管账号可以查看分类下的所有频度数据
  208. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  209. classifyIdList, err := data.GetUserManualClassifyIdList(userId)
  210. if err != nil {
  211. br.Msg = "获取分类数据失败"
  212. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  213. return
  214. }
  215. num := len(classifyIdList)
  216. if num > 0 {
  217. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  218. pars = append(pars, classifyIdList)
  219. }
  220. }
  221. if keyword != "" {
  222. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  223. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  224. }
  225. total, err := models.GetCountEdbInfoList(condition, pars)
  226. if err != nil {
  227. br.Msg = "获取失败"
  228. br.ErrMsg = "获取失败,Err:" + err.Error()
  229. return
  230. }
  231. list, err := models.GetEdbInfoList(condition, pars, startSize, pageSize)
  232. if err != nil {
  233. br.Msg = "获取失败"
  234. br.ErrMsg = "获取失败,Err:" + err.Error()
  235. return
  236. }
  237. for _, v := range list {
  238. v.UniqueCode = utils.MD5(v.TradeCode)
  239. }
  240. resp := models.EdbListResp{
  241. List: list,
  242. Paging: paging.GetPaging(currentIndex, pageSize, total),
  243. }
  244. br.Ret = 200
  245. br.Success = true
  246. br.Msg = "获取成功"
  247. br.Data = resp
  248. }
  249. // EdbList
  250. // @Title 获取研究员指标
  251. // @Description 获取研究员指标
  252. // @Param Frequency query string false "频度;枚举值:日度、周度、月度、季度、半年度、年度"
  253. // @Param Keyword query string false "关键字搜索"
  254. // @Param TradeCode query string false "指标唯一编码"
  255. // @Param ClassifyId query int false "分类id"
  256. // @Param UserId query int false "用户id"
  257. // @Success 200 {object} models.TargetItemListResp
  258. // @router /target/edb/list [get]
  259. func (c *ManualEdbController) EdbList() {
  260. br := new(models.BaseResponse).Init()
  261. defer func() {
  262. c.Data["json"] = br
  263. c.ServeJSON()
  264. }()
  265. sysUser := c.SysUser
  266. if sysUser == nil {
  267. br.Msg = "请重新登录"
  268. return
  269. }
  270. pageSize, _ := c.GetInt("PageSize")
  271. currentIndex, _ := c.GetInt("CurrentIndex")
  272. var startSize int
  273. if pageSize <= 0 {
  274. pageSize = utils.PageSize20
  275. }
  276. if currentIndex <= 0 {
  277. currentIndex = 1
  278. }
  279. startSize = utils.StartIndex(currentIndex, pageSize)
  280. // 频度
  281. frequency := c.GetString("Frequency")
  282. //关键字
  283. keyword := c.GetString("Keyword")
  284. //关键字
  285. classifyId, _ := c.GetInt("ClassifyId", 0)
  286. //用户ID
  287. userId, _ := c.GetInt("UserId", 0)
  288. var condition string
  289. var pars []interface{}
  290. //超管账号可以查看分类下的所有频度数据
  291. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  292. classifyIdList, err := data.GetUserManualClassifyIdList(userId)
  293. if err != nil {
  294. br.Msg = "获取分类数据失败"
  295. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  296. return
  297. }
  298. num := len(classifyIdList)
  299. if num > 0 {
  300. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  301. pars = append(pars, classifyIdList)
  302. }
  303. }
  304. // 关键词
  305. if keyword != "" {
  306. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  307. pars = utils.GetLikeKeywordPars(pars, keyword, 2)
  308. }
  309. if frequency != "" {
  310. condition += ` AND a.frequency = ?`
  311. pars = append(pars, frequency)
  312. }
  313. // 所属分类
  314. if classifyId > 0 {
  315. classifyIdList, err := data.GetUserManualClassifyIdList(userId)
  316. if err != nil {
  317. br.Msg = "获取分类数据失败"
  318. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  319. return
  320. }
  321. num := len(classifyIdList)
  322. if num > 0 {
  323. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  324. pars = append(pars, classifyIdList)
  325. } else {
  326. condition += ` AND a.classify_id = ? `
  327. pars = append(pars, classifyId)
  328. }
  329. }
  330. // 所属用户
  331. if userId > 0 {
  332. condition += ` AND a.user_id = ? `
  333. pars = append(pars, userId)
  334. }
  335. total, err := models.GetCountEdbInfoList(condition, pars)
  336. if err != nil {
  337. br.Msg = "获取失败"
  338. br.ErrMsg = "获取失败,Err:" + err.Error()
  339. return
  340. }
  341. list, err := models.GetEdbInfoList(condition, pars, startSize, pageSize)
  342. if err != nil {
  343. br.Msg = "获取失败"
  344. br.ErrMsg = "获取失败,Err:" + err.Error()
  345. return
  346. }
  347. resp := models.EdbListResp{
  348. List: list,
  349. Paging: paging.GetPaging(currentIndex, pageSize, total),
  350. }
  351. br.Ret = 200
  352. br.Success = true
  353. br.Msg = "获取成功"
  354. br.Data = resp
  355. }
  356. // EditExcelData
  357. // @Title 根据excel的样式去编辑指标
  358. // @Description 根据excel的样式去编辑指标
  359. // @Param request body models.ManualEdbExcelStyleEditReq true "type json string"
  360. // @Success 200
  361. // @router /target/edb/excel_style/edit [post]
  362. func (c *ManualEdbController) EditExcelData() {
  363. br := new(models.BaseResponse).Init()
  364. var err error
  365. errs := make([]string, 0)
  366. defer func() {
  367. if len(errs) > 0 {
  368. utils.FileLog.Info("编辑EXCEL数据 新增或修改数据失败,Err:" + strings.Join(errs, "\n"))
  369. }
  370. c.Data["json"] = br
  371. c.ServeJSON()
  372. }()
  373. sysUser := c.SysUser
  374. if sysUser == nil {
  375. br.Msg = "请重新登录"
  376. return
  377. }
  378. var req data_manage.ManualEdbExcelStyleEditReq
  379. err = json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  380. if err != nil {
  381. br.Msg = "参数解析异常!"
  382. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  383. return
  384. }
  385. if req.TradeCode == `` {
  386. br.Msg = "指标编码异常!"
  387. br.ErrMsg = "指标编码异常!"
  388. return
  389. }
  390. if req.Unit == `` {
  391. br.Msg = "单位填写异常!"
  392. br.ErrMsg = "单位填写异常!"
  393. return
  394. }
  395. if req.Frequency == `` {
  396. br.Msg = "频度填写异常!"
  397. br.ErrMsg = "频度填写异常!"
  398. return
  399. }
  400. if req.ClassifyId <= 0 {
  401. br.Msg = "品种填写异常!"
  402. br.ErrMsg = "品种填写异常!"
  403. return
  404. }
  405. // 获取指标信息
  406. manualEdbInfo, err := models.GetEdbinfoByTradeCode(req.TradeCode)
  407. if err != nil {
  408. br.Msg = `找不到该指标`
  409. br.Msg = `找不到该指标,ERR:` + err.Error()
  410. return
  411. }
  412. // 如果不是超管账号,那么得校验下当前用户是否有该指标的权限
  413. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  414. //获取账户所拥有权限的分类id集合
  415. count, err := models.GetCountManualUserClassify(sysUser.AdminId, manualEdbInfo.ClassifyId)
  416. if err != nil {
  417. br.Msg = "获取分类数据失败"
  418. br.ErrMsg = "获取分类数据失败,err:" + err.Error()
  419. return
  420. }
  421. if count <= 0 {
  422. br.Msg = "无权访问"
  423. br.ErrMsg = "无权访问"
  424. br.IsSendEmail = false
  425. return
  426. }
  427. }
  428. dateValueMap := make(map[string]string)
  429. //取到所有数据
  430. for _, dateValue := range req.Data {
  431. dateValueMap[dateValue.Date] = strconv.FormatFloat(dateValue.Value, 'f', -1, 64)
  432. }
  433. //操作指标,新增指标及数据等
  434. {
  435. var isUpdate bool
  436. secName := strings.TrimSpace(req.EdbName)
  437. frequency := strings.TrimSpace(req.Frequency)
  438. unit := strings.TrimSpace(req.Unit)
  439. // 指标基础信息变更
  440. {
  441. updateCols := make([]string, 0)
  442. if manualEdbInfo.SecName != secName {
  443. updateCols = append(updateCols, "SecName")
  444. manualEdbInfo.SecName = secName
  445. }
  446. if manualEdbInfo.ClassifyId != req.ClassifyId {
  447. updateCols = append(updateCols, "ClassifyId")
  448. manualEdbInfo.ClassifyId = req.ClassifyId
  449. }
  450. if manualEdbInfo.Frequency != frequency {
  451. updateCols = append(updateCols, "Frequency")
  452. manualEdbInfo.Frequency = frequency
  453. }
  454. if manualEdbInfo.Unit != unit {
  455. updateCols = append(updateCols, "Unit")
  456. manualEdbInfo.Unit = unit
  457. }
  458. if len(updateCols) > 0 {
  459. isUpdate = true
  460. manualEdbInfo.ModifyTime = time.Now().Format(utils.FormatDateTime)
  461. updateCols = append(updateCols, "ModifyTime")
  462. err = manualEdbInfo.Update(updateCols)
  463. if err != nil {
  464. br.Msg = `指标保存失败`
  465. br.Msg = `指标保存失败,ERR:` + err.Error()
  466. return
  467. }
  468. }
  469. }
  470. // 当前已经存在的指标明细数据
  471. targetDataList, tmpErr := models.GetTargetsDataList(manualEdbInfo.TradeCode)
  472. if tmpErr != nil {
  473. err = tmpErr
  474. }
  475. existDataMap := make(map[string]string)
  476. deleteDataMap := make(map[string]string)
  477. for _, tmpData := range targetDataList {
  478. existDataMap[tmpData.Dt] = tmpData.Close
  479. deleteDataMap[tmpData.Dt] = tmpData.Dt
  480. }
  481. addDataList := make([]*models.Edbdata, 0)
  482. for createDate, closeVal := range dateValueMap {
  483. if createDate == "" || closeVal == "" {
  484. continue
  485. }
  486. //判断数据是否已经存在
  487. tmpVal, ok3 := existDataMap[createDate]
  488. // 不存在,那么后面进行插入
  489. if !ok3 {
  490. addDataList = append(addDataList, &models.Edbdata{
  491. TradeCode: manualEdbInfo.TradeCode,
  492. Dt: createDate,
  493. Close: closeVal,
  494. ModifyTime: time.Now(),
  495. })
  496. continue
  497. }
  498. delete(deleteDataMap, createDate)
  499. // 库里面的数据
  500. tmpValDecimal, tmpErr := decimal.NewFromString(tmpVal)
  501. if tmpErr != nil {
  502. fmt.Println("tmpVal Parse err:", tmpErr.Error())
  503. err = tmpErr
  504. continue
  505. }
  506. // 用户填写的数据
  507. closeValDecimal, tmpErr := decimal.NewFromString(closeVal)
  508. if tmpErr != nil {
  509. fmt.Println("closeVal Parse err:", tmpErr.Error())
  510. err = tmpErr
  511. continue
  512. }
  513. if !tmpValDecimal.Equal(closeValDecimal) {
  514. isUpdate = true
  515. fmt.Println("更新数值")
  516. err = models.ModifyTargetsDataByImport(manualEdbInfo.TradeCode, createDate, closeVal)
  517. if err != nil {
  518. fmt.Println("ModifyTargetsDataByImport err:", err.Error())
  519. errs = append(errs, err.Error())
  520. //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 修改数据失败:Err:"+err.Error(), utils.EmailSendToUsers)
  521. }
  522. }
  523. }
  524. // 新增指标
  525. if len(addDataList) > 0 {
  526. isUpdate = true
  527. err = models.OnlyMultiAddEdbdata(addDataList)
  528. if err != nil {
  529. fmt.Println("MultiAddEdbdata err:", err.Error())
  530. errs = append(errs, err.Error())
  531. }
  532. }
  533. // 删除不需要的日期数据
  534. num := len(deleteDataMap)
  535. if num > 0 {
  536. isUpdate = true
  537. deleteDateList := make([]string, 0)
  538. for date := range deleteDataMap {
  539. deleteDateList = append(deleteDateList, date)
  540. }
  541. err = models.DelEdbdataByCodeAndDateList(manualEdbInfo.TradeCode, deleteDateList)
  542. if err != nil {
  543. fmt.Println("DelEdbdataByCodeAndDateList err:", err.Error())
  544. errs = append(errs, err.Error())
  545. }
  546. }
  547. // 添加记录
  548. if isUpdate {
  549. record := &models.EdbinfoOpRecord{
  550. TradeCode: manualEdbInfo.TradeCode,
  551. Remark: "编辑指标",
  552. UserId: sysUser.AdminId,
  553. UserName: sysUser.RealName,
  554. CreateTime: time.Now(),
  555. }
  556. go func() {
  557. data.ModifyManualEdbMaxMinDate(manualEdbInfo.TradeCode)
  558. _ = record.Create()
  559. }()
  560. }
  561. }
  562. br.Ret = 200
  563. br.Success = true
  564. br.Msg = "保存成功"
  565. }
  566. // BatchEdbList
  567. // @Title 获取研究员指标
  568. // @Description 获取研究员指标
  569. // @Param request body models.BatchAddEdbReq true "type json string"
  570. // @Success 200 {object} models.TargetItemListResp
  571. // @router /target/edb/batch/list [post]
  572. func (c *ManualEdbController) BatchEdbList() {
  573. br := new(models.BaseResponse).Init()
  574. defer func() {
  575. c.Data["json"] = br
  576. c.ServeJSON()
  577. }()
  578. sysUser := c.SysUser
  579. if sysUser == nil {
  580. br.Msg = "请重新登录"
  581. return
  582. }
  583. var req models.BatchManualEdbReq
  584. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  585. pageSize := req.PageSize
  586. currentIndex := req.CurrentIndex
  587. var startSize int
  588. if pageSize <= 0 {
  589. pageSize = utils.PageSize20
  590. }
  591. if currentIndex <= 0 {
  592. currentIndex = 1
  593. }
  594. startSize = utils.StartIndex(currentIndex, pageSize)
  595. req.Keyword = strings.TrimSpace(req.Keyword)
  596. var condition string
  597. var pars []interface{}
  598. // 是否加到指标库
  599. condition += ` AND a.is_join_edb = ? `
  600. pars = append(pars, 0)
  601. //超管账号可以查看分类下的所有频度数据
  602. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  603. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  604. if err != nil {
  605. br.Msg = "获取分类数据失败"
  606. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  607. return
  608. }
  609. num := len(classifyIdList)
  610. if num > 0 {
  611. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  612. pars = append(pars, classifyIdList)
  613. }
  614. }
  615. // 关键词
  616. if req.Keyword != "" {
  617. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  618. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  619. }
  620. // 所属分类
  621. lenFrequency := len(req.FrequencyList)
  622. if lenFrequency > 0 {
  623. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  624. pars = append(pars, req.FrequencyList)
  625. }
  626. // 所属分类
  627. lenClassify := len(req.ClassifyIdList)
  628. if lenClassify > 0 {
  629. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  630. pars = append(pars, req.ClassifyIdList)
  631. }
  632. // 所属用户
  633. lenUser := len(req.UserIdList)
  634. if lenUser > 0 {
  635. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  636. pars = append(pars, req.UserIdList)
  637. }
  638. total, err := models.GetCountEdbInfoList(condition, pars)
  639. if err != nil {
  640. br.Msg = "获取失败"
  641. br.ErrMsg = "获取失败,Err:" + err.Error()
  642. return
  643. }
  644. list, err := models.GetEdbInfoList(condition, pars, startSize, pageSize)
  645. if err != nil {
  646. br.Msg = "获取失败"
  647. br.ErrMsg = "获取失败,Err:" + err.Error()
  648. return
  649. }
  650. resp := models.EdbListResp{
  651. List: list,
  652. Paging: paging.GetPaging(currentIndex, pageSize, total),
  653. }
  654. br.Ret = 200
  655. br.Success = true
  656. br.Msg = "获取成功"
  657. br.Data = resp
  658. br.IsAddLog = false
  659. }
  660. // BatchAddEdbCheck
  661. // @Title 新增校验
  662. // @Description 新增校验
  663. // @Param request body data_manage.BatchManualEdbReq true "type json string"
  664. // @Success 200 string "操作成功"
  665. // @router /target/edb/batch/add/check [post]
  666. func (c *ManualEdbController) BatchAddEdbCheck() {
  667. br := new(models.BaseResponse).Init()
  668. defer func() {
  669. if br.ErrMsg == "" {
  670. br.IsSendEmail = false
  671. }
  672. c.Data["json"] = br
  673. c.ServeJSON()
  674. }()
  675. sysUser := c.SysUser
  676. if sysUser == nil {
  677. br.Msg = "请登录"
  678. br.ErrMsg = "请登录,SysUser Is Empty"
  679. br.Ret = 408
  680. return
  681. }
  682. // 最大批量添加的数量
  683. codeMax := 30
  684. var req models.BatchManualEdbReq
  685. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  686. req.Keyword = strings.TrimSpace(req.Keyword)
  687. var condition string
  688. var pars []interface{}
  689. //
  690. condition += ` AND a.is_join_edb = 0 `
  691. //超管账号可以查看分类下的所有频度数据
  692. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  693. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  694. if err != nil {
  695. br.Msg = "获取分类数据失败"
  696. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  697. return
  698. }
  699. num := len(classifyIdList)
  700. if num > 0 {
  701. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  702. pars = append(pars, classifyIdList)
  703. }
  704. }
  705. if req.ListAll {
  706. // 关键词
  707. if req.Keyword != "" {
  708. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  709. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  710. }
  711. // 所属分类
  712. lenFrequency := len(req.FrequencyList)
  713. if lenFrequency > 0 {
  714. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  715. pars = append(pars, req.FrequencyList)
  716. }
  717. // 所属分类
  718. lenClassify := len(req.ClassifyIdList)
  719. if lenClassify > 0 {
  720. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  721. pars = append(pars, req.ClassifyIdList)
  722. }
  723. // 所属用户
  724. lenUser := len(req.UserIdList)
  725. if lenUser > 0 {
  726. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  727. pars = append(pars, req.UserIdList)
  728. }
  729. lenTradeList := len(req.TradeCodeList)
  730. // 指标
  731. if lenTradeList > 0 {
  732. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  733. pars = append(pars, req.TradeCodeList)
  734. }
  735. } else {
  736. lenTradeList := len(req.TradeCodeList)
  737. if lenTradeList <= 0 {
  738. br.Msg = "请选择指标"
  739. br.ErrMsg = "请选择指标"
  740. return
  741. }
  742. // 指标
  743. if lenTradeList > 0 {
  744. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  745. pars = append(pars, req.TradeCodeList)
  746. }
  747. }
  748. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  749. if err != nil {
  750. br.Msg = "获取失败"
  751. br.ErrMsg = "获取失败,Err:" + err.Error()
  752. return
  753. }
  754. if len(list) > codeMax {
  755. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  756. return
  757. }
  758. br.Data = list
  759. br.Msg = "校验成功"
  760. br.Ret = 200
  761. br.Success = true
  762. }
  763. // NameCheck
  764. // @Title 重名校验
  765. // @Description 批量新增
  766. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  767. // @Success 200 string "操作成功"
  768. // @router /target/edb/batch/add/name_check [post]
  769. func (c *ManualEdbController) NameCheck() {
  770. br := new(models.BaseResponse).Init()
  771. defer func() {
  772. if br.ErrMsg == "" {
  773. br.IsSendEmail = false
  774. }
  775. c.Data["json"] = br
  776. c.ServeJSON()
  777. }()
  778. sysUser := c.SysUser
  779. if sysUser == nil {
  780. br.Msg = "请登录"
  781. br.ErrMsg = "请登录,SysUser Is Empty"
  782. br.Ret = 408
  783. return
  784. }
  785. var req []*data_manage.AddEdbInfoReq
  786. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  787. br.Msg = "参数解析异常!"
  788. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  789. return
  790. }
  791. if len(req) == 0 {
  792. br.Msg = "请选择指标"
  793. return
  794. }
  795. type NameCheckResult struct {
  796. EdbCode string
  797. EdbName string
  798. Exist bool
  799. }
  800. indexNames := make([]string, 0)
  801. resp := make([]*NameCheckResult, 0)
  802. for _, v := range req {
  803. v.EdbCode = strings.TrimSpace(v.EdbCode)
  804. if v.EdbCode == "" {
  805. br.Msg = "指标ID不可为空"
  806. return
  807. }
  808. v.EdbName = strings.TrimSpace(v.EdbName)
  809. if v.EdbName == "" {
  810. br.Msg = "请输入指标名称"
  811. return
  812. }
  813. indexNames = append(indexNames, v.EdbName)
  814. resp = append(resp, &NameCheckResult{
  815. EdbCode: v.EdbCode,
  816. EdbName: v.EdbName,
  817. })
  818. }
  819. // 重名校验
  820. edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
  821. if e != nil {
  822. br.Msg = "操作失败"
  823. br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
  824. return
  825. }
  826. nameExists := make(map[string]bool)
  827. for _, v := range edbList {
  828. nameExists[v.EdbName] = true
  829. }
  830. if len(nameExists) > 0 {
  831. for _, v := range resp {
  832. v.Exist = nameExists[v.EdbName]
  833. }
  834. }
  835. br.Data = resp
  836. br.Msg = "操作成功"
  837. br.Ret = 200
  838. br.Success = true
  839. }
  840. // BatchAdd2Edb
  841. // @Title 批量新增到指标库
  842. // @Description 批量新增到指标库
  843. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  844. // @Success 200 string "操作成功"
  845. // @router /target/edb/batch/add [post]
  846. func (c *ManualEdbController) BatchAdd2Edb() {
  847. br := new(models.BaseResponse).Init()
  848. defer func() {
  849. if br.ErrMsg == "" {
  850. br.IsSendEmail = false
  851. }
  852. c.Data["json"] = br
  853. c.ServeJSON()
  854. }()
  855. sysUser := c.SysUser
  856. if sysUser == nil {
  857. br.Msg = "请登录"
  858. br.ErrMsg = "请登录,SysUser Is Empty"
  859. br.Ret = 408
  860. return
  861. }
  862. // 最大批量添加的数量
  863. codeMax := 30
  864. deleteCache := true
  865. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_MANUAL_" + strconv.Itoa(sysUser.AdminId)
  866. defer func() {
  867. if deleteCache {
  868. _ = utils.Rc.Delete(cacheKey)
  869. }
  870. }()
  871. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  872. deleteCache = false
  873. br.Msg = "系统处理中,请稍后重试!"
  874. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(c.Ctx.Input.RequestBody)
  875. return
  876. }
  877. var req []*data_manage.AddEdbInfoReq
  878. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  879. br.Msg = "参数解析异常!"
  880. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  881. return
  882. }
  883. if len(req) == 0 {
  884. br.Msg = "请选择指标"
  885. return
  886. }
  887. if len(req) > codeMax {
  888. br.Msg = "批量添加指标数量不得超过30个"
  889. return
  890. }
  891. indexNames := make([]string, 0)
  892. for _, v := range req {
  893. v.EdbCode = strings.TrimSpace(v.EdbCode)
  894. if v.EdbCode == "" {
  895. br.Msg = "指标ID不可为空"
  896. return
  897. }
  898. v.EdbName = strings.TrimSpace(v.EdbName)
  899. if v.EdbName == "" {
  900. br.Msg = "请输入指标名称"
  901. return
  902. }
  903. indexNames = append(indexNames, v.EdbName)
  904. v.Frequency = strings.TrimSpace(v.Frequency)
  905. if v.Frequency == "" {
  906. br.Msg = "请选择频度"
  907. return
  908. }
  909. v.Unit = strings.TrimSpace(v.Unit)
  910. if v.Unit == "" {
  911. br.Msg = "请输入单位"
  912. return
  913. }
  914. if v.ClassifyId <= 0 {
  915. br.Msg = "请选择分类"
  916. return
  917. }
  918. }
  919. // 限定同一时间最多批量新增30个指标
  920. for _, v := range req {
  921. var r data.ManualIndexSource2EdbReq
  922. r.EdbCode = v.EdbCode
  923. r.EdbName = v.EdbName
  924. r.Frequency = v.Frequency
  925. r.Unit = v.Unit
  926. r.ClassifyId = v.ClassifyId
  927. r.AdminId = sysUser.AdminId
  928. r.AdminRealName = sysUser.RealName
  929. edbInfo, e, errMsg, skip := data.ManualIndexSource2Edb(r, c.Lang)
  930. if e != nil {
  931. br.Msg = "操作失败"
  932. if errMsg != "" {
  933. br.Msg = errMsg
  934. }
  935. br.ErrMsg = e.Error()
  936. return
  937. }
  938. if skip {
  939. continue
  940. }
  941. // 试用平台更新用户累计新增指标数
  942. if utils.BusinessCode == utils.BusinessCodeSandbox {
  943. go func() {
  944. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  945. if e != nil {
  946. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  947. utils.FileLog.Info(tips)
  948. return
  949. }
  950. if adminItem.DepartmentName != "ETA试用客户" {
  951. return
  952. }
  953. var ur etaTrialService.EtaTrialUserReq
  954. ur.Mobile = adminItem.Mobile
  955. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  956. }()
  957. }
  958. // 新增操作日志
  959. {
  960. edbLog := new(data_manage.EdbInfoLog)
  961. edbLog.EdbInfoId = edbInfo.EdbInfoId
  962. edbLog.SourceName = edbInfo.SourceName
  963. edbLog.Source = edbInfo.Source
  964. edbLog.EdbCode = edbInfo.EdbCode
  965. edbLog.EdbName = edbInfo.EdbName
  966. edbLog.ClassifyId = edbInfo.ClassifyId
  967. edbLog.SysUserId = sysUser.AdminId
  968. edbLog.SysUserRealName = sysUser.RealName
  969. edbLog.CreateTime = time.Now()
  970. edbLog.Content = string(c.Ctx.Input.RequestBody)
  971. edbLog.Status = "新增指标"
  972. edbLog.Method = c.Ctx.Input.URI()
  973. go data_manage.AddEdbInfoLog(edbLog)
  974. }
  975. }
  976. br.Msg = "操作成功"
  977. br.Ret = 200
  978. br.Success = true
  979. br.IsAddLog = true
  980. }
  981. // BatchDelEdbCheck
  982. // @Title 新增校验
  983. // @Description 新增校验
  984. // @Param request body data_manage.BatchManualEdbReq true "type json string"
  985. // @Success 200 string "操作成功"
  986. // @router /target/edb/batch/del/check [post]
  987. func (c *ManualEdbController) BatchDelEdbCheck() {
  988. br := new(models.BaseResponse).Init()
  989. defer func() {
  990. if br.ErrMsg == "" {
  991. br.IsSendEmail = false
  992. }
  993. c.Data["json"] = br
  994. c.ServeJSON()
  995. }()
  996. sysUser := c.SysUser
  997. if sysUser == nil {
  998. br.Msg = "请登录"
  999. br.ErrMsg = "请登录,SysUser Is Empty"
  1000. br.Ret = 408
  1001. return
  1002. }
  1003. var req models.BatchManualEdbReq
  1004. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  1005. req.Keyword = strings.TrimSpace(req.Keyword)
  1006. var condition string
  1007. var pars []interface{}
  1008. // 就是为了找出已经加入指标库的指标
  1009. condition += ` AND a.is_join_edb = ? `
  1010. pars = append(pars, 1)
  1011. //超管账号可以查看分类下的所有频度数据
  1012. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  1013. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  1014. if err != nil {
  1015. br.Msg = "获取分类数据失败"
  1016. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  1017. return
  1018. }
  1019. num := len(classifyIdList)
  1020. if num > 0 {
  1021. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  1022. pars = append(pars, classifyIdList)
  1023. }
  1024. }
  1025. if req.ListAll {
  1026. // 关键词
  1027. if req.Keyword != "" {
  1028. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  1029. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  1030. }
  1031. // 所属分类
  1032. lenFrequency := len(req.FrequencyList)
  1033. if lenFrequency > 0 {
  1034. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  1035. pars = append(pars, req.FrequencyList)
  1036. }
  1037. // 所属分类
  1038. lenClassify := len(req.ClassifyIdList)
  1039. if lenClassify > 0 {
  1040. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  1041. pars = append(pars, req.ClassifyIdList)
  1042. }
  1043. // 所属用户
  1044. lenUser := len(req.UserIdList)
  1045. if lenUser > 0 {
  1046. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  1047. pars = append(pars, req.UserIdList)
  1048. }
  1049. lenTradeList := len(req.TradeCodeList)
  1050. // 指标
  1051. if lenTradeList > 0 {
  1052. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1053. pars = append(pars, req.TradeCodeList)
  1054. }
  1055. } else {
  1056. lenTradeList := len(req.TradeCodeList)
  1057. if lenTradeList <= 0 {
  1058. br.Msg = "请选择指标"
  1059. br.ErrMsg = "请选择指标"
  1060. return
  1061. }
  1062. // 指标
  1063. if lenTradeList > 0 {
  1064. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1065. pars = append(pars, req.TradeCodeList)
  1066. }
  1067. }
  1068. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  1069. if err != nil {
  1070. br.Msg = "获取失败"
  1071. br.ErrMsg = "获取失败,Err:" + err.Error()
  1072. return
  1073. }
  1074. br.Data = list
  1075. br.Msg = "校验成功"
  1076. br.Ret = 200
  1077. br.Success = true
  1078. }
  1079. // BatchDel
  1080. // @Title 批量删除
  1081. // @Description 批量新增
  1082. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  1083. // @Success 200 string "操作成功"
  1084. // @router /target/edb/batch/del [post]
  1085. func (c *ManualEdbController) BatchDel() {
  1086. br := new(models.BaseResponse).Init()
  1087. defer func() {
  1088. if br.ErrMsg == "" {
  1089. br.IsSendEmail = false
  1090. }
  1091. c.Data["json"] = br
  1092. c.ServeJSON()
  1093. }()
  1094. sysUser := c.SysUser
  1095. if sysUser == nil {
  1096. br.Msg = "请登录"
  1097. br.ErrMsg = "请登录,SysUser Is Empty"
  1098. br.Ret = 408
  1099. return
  1100. }
  1101. deleteCache := true
  1102. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_MANUAL_" + strconv.Itoa(sysUser.AdminId)
  1103. defer func() {
  1104. if deleteCache {
  1105. _ = utils.Rc.Delete(cacheKey)
  1106. }
  1107. }()
  1108. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  1109. deleteCache = false
  1110. br.Msg = "系统处理中,请稍后重试!"
  1111. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(c.Ctx.Input.RequestBody)
  1112. return
  1113. }
  1114. var req models.BatchManualEdbReq
  1115. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  1116. req.Keyword = strings.TrimSpace(req.Keyword)
  1117. var condition string
  1118. var pars []interface{}
  1119. // 就是为了找出已经加入指标库的指标
  1120. condition += ` AND a.is_join_edb = ? `
  1121. pars = append(pars, 0)
  1122. //超管账号可以查看分类下的所有频度数据
  1123. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  1124. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  1125. if err != nil {
  1126. br.Msg = "获取分类数据失败"
  1127. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  1128. return
  1129. }
  1130. num := len(classifyIdList)
  1131. if num > 0 {
  1132. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  1133. pars = append(pars, classifyIdList)
  1134. }
  1135. }
  1136. if req.ListAll {
  1137. // 关键词
  1138. if req.Keyword != "" {
  1139. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  1140. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  1141. }
  1142. // 所属分类
  1143. lenFrequency := len(req.FrequencyList)
  1144. if lenFrequency > 0 {
  1145. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  1146. pars = append(pars, req.FrequencyList)
  1147. }
  1148. // 所属分类
  1149. lenClassify := len(req.ClassifyIdList)
  1150. if lenClassify > 0 {
  1151. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  1152. pars = append(pars, req.ClassifyIdList)
  1153. }
  1154. // 所属用户
  1155. lenUser := len(req.UserIdList)
  1156. if lenUser > 0 {
  1157. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  1158. pars = append(pars, req.UserIdList)
  1159. }
  1160. lenTradeList := len(req.TradeCodeList)
  1161. // 指标
  1162. if lenTradeList > 0 {
  1163. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1164. pars = append(pars, req.TradeCodeList)
  1165. }
  1166. } else {
  1167. lenTradeList := len(req.TradeCodeList)
  1168. if lenTradeList <= 0 {
  1169. br.Msg = "请选择指标"
  1170. br.ErrMsg = "请选择指标"
  1171. return
  1172. }
  1173. // 指标
  1174. if lenTradeList > 0 {
  1175. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1176. pars = append(pars, req.TradeCodeList)
  1177. }
  1178. }
  1179. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  1180. if err != nil {
  1181. br.Msg = "删除失败"
  1182. br.ErrMsg = "删除失败,Err:" + err.Error()
  1183. return
  1184. }
  1185. // TODO 二次校验指标是否已经加到指标库
  1186. {
  1187. codeList := make([]string, 0)
  1188. // 限定同一时间最多批量新增30个指标
  1189. for _, v := range list {
  1190. codeList = append(codeList, v.TradeCode)
  1191. }
  1192. err = models.DelManualIndexByCodeList(codeList)
  1193. if err != nil {
  1194. br.Msg = "删除失败"
  1195. br.ErrMsg = "删除失败,Err:" + err.Error()
  1196. return
  1197. }
  1198. }
  1199. br.Msg = "删除成功"
  1200. br.Ret = 200
  1201. br.Success = true
  1202. br.IsAddLog = true
  1203. }
  1204. // ImportData
  1205. // @Title 导入数据
  1206. // @Description 导入数据
  1207. // @Param EntryFile query file true "文件"
  1208. // @Success 200 Ret=200 录入成功
  1209. // @router /import/data [post]
  1210. func (c *ManualEdbController) ImportData() {
  1211. br := new(models.BaseResponse).Init()
  1212. defer func() {
  1213. c.Data["json"] = br
  1214. c.ServeJSON()
  1215. }()
  1216. sysUser := c.SysUser
  1217. if sysUser == nil {
  1218. br.Msg = "请重新登录"
  1219. return
  1220. }
  1221. file, _, err := c.GetFile("EntryFile")
  1222. if err != nil {
  1223. br.Msg = "获取文件失败"
  1224. br.ErrMsg = "获取文件失败,Err:" + err.Error()
  1225. return
  1226. }
  1227. path := "./static/数据导入_" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
  1228. defer file.Close()
  1229. err = c.SaveToFile("EntryFile", path)
  1230. if err != nil {
  1231. br.Msg = "文件保存失败"
  1232. br.ErrMsg = "文件保存失败,Err:" + err.Error()
  1233. return
  1234. }
  1235. if utils.RunMode == "debug" {
  1236. defer os.Remove(path)
  1237. }
  1238. successCount, failCount, err, errMsg := data.ImportManualData(path, sysUser)
  1239. if err != nil {
  1240. br.Msg = errMsg
  1241. br.ErrMsg = err.Error()
  1242. return
  1243. }
  1244. resp := models.EdbdataImportResp{
  1245. SuccessCount: successCount,
  1246. FailCount: failCount,
  1247. }
  1248. if failCount > 0 {
  1249. if successCount == 0 {
  1250. resp.Status = -1
  1251. resp.Msg = "导入失败"
  1252. } else {
  1253. resp.Status = 1
  1254. resp.Msg = "存在部分导入失败"
  1255. }
  1256. } else {
  1257. resp.Status = 0
  1258. }
  1259. br.Msg = "导入成功"
  1260. br.Ret = 200
  1261. br.Success = true
  1262. br.Data = resp
  1263. }
  1264. // RecordList
  1265. // @Title 获取指标操作记录列表
  1266. // @Description 获取指标操作记录列表
  1267. // @Param TradeCode query string false "指标编码"
  1268. // @Success 200 {object} models.EdbinfoOpRecordListResp
  1269. // @router /target/edb/op/record/list [get]
  1270. func (c *ManualEdbController) RecordList() {
  1271. br := new(models.BaseResponse).Init()
  1272. defer func() {
  1273. c.Data["json"] = br
  1274. c.ServeJSON()
  1275. }()
  1276. sysUser := c.SysUser
  1277. if sysUser == nil {
  1278. br.Msg = "请重新登录"
  1279. return
  1280. }
  1281. pageSize, _ := c.GetInt("PageSize")
  1282. currentIndex, _ := c.GetInt("CurrentIndex")
  1283. var startSize int
  1284. if pageSize <= 0 {
  1285. pageSize = utils.PageSize20
  1286. }
  1287. if currentIndex <= 0 {
  1288. currentIndex = 1
  1289. }
  1290. startSize = utils.StartIndex(currentIndex, pageSize)
  1291. // 频度
  1292. tradeCode := c.GetString("TradeCode")
  1293. if tradeCode == `` {
  1294. br.Msg = `请选择指标`
  1295. br.ErrMsg = `请选择指标`
  1296. br.IsSendEmail = false
  1297. return
  1298. }
  1299. total, list, err := models.GetEdbinfoOpRecordPageList(tradeCode, startSize, pageSize)
  1300. if err != nil {
  1301. br.Msg = "获取失败"
  1302. br.ErrMsg = "获取失败,Err:" + err.Error()
  1303. return
  1304. }
  1305. resp := models.EdbinfoOpRecordListResp{
  1306. List: list,
  1307. Paging: paging.GetPaging(currentIndex, pageSize, total),
  1308. }
  1309. br.Ret = 200
  1310. br.Success = true
  1311. br.Msg = "获取成功"
  1312. br.Data = resp
  1313. }