manual_edb.go 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519
  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(sysUser.AdminId)
  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(sysUser.AdminId)
  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. // 获取有用权限的分类
  316. classifyList, err := models.GetEdbdataClassify(int64(userId))
  317. if err != nil {
  318. return
  319. }
  320. var isParent bool
  321. classifyIdList := make([]int, 0)
  322. for _, v := range classifyList {
  323. if v.ClassifyId == classifyId {
  324. isParent = true
  325. classifyIdList = append(classifyIdList, v.ClassifyId)
  326. if v.Child != nil && len(v.Child) > 0 {
  327. for _, vv := range v.Child {
  328. classifyIdList = append(classifyIdList, vv.ClassifyId)
  329. }
  330. }
  331. break
  332. }
  333. }
  334. num := len(classifyIdList)
  335. if num > 0 && isParent {
  336. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  337. pars = append(pars, classifyIdList)
  338. } else {
  339. condition += ` AND a.classify_id = ? `
  340. pars = append(pars, classifyId)
  341. }
  342. }
  343. // 所属用户
  344. if userId > 0 {
  345. condition += ` AND a.user_id = ? `
  346. pars = append(pars, userId)
  347. }
  348. total, err := models.GetCountEdbInfoList(condition, pars)
  349. if err != nil {
  350. br.Msg = "获取失败"
  351. br.ErrMsg = "获取失败,Err:" + err.Error()
  352. return
  353. }
  354. list, err := models.GetEdbInfoList(condition, pars, startSize, pageSize)
  355. if err != nil {
  356. br.Msg = "获取失败"
  357. br.ErrMsg = "获取失败,Err:" + err.Error()
  358. return
  359. }
  360. resp := models.EdbListResp{
  361. List: list,
  362. Paging: paging.GetPaging(currentIndex, pageSize, total),
  363. }
  364. br.Ret = 200
  365. br.Success = true
  366. br.Msg = "获取成功"
  367. br.Data = resp
  368. }
  369. // EditExcelData
  370. // @Title 根据excel的样式去编辑指标
  371. // @Description 根据excel的样式去编辑指标
  372. // @Param request body models.ManualEdbExcelStyleEditReq true "type json string"
  373. // @Success 200
  374. // @router /target/edb/excel_style/edit [post]
  375. func (c *ManualEdbController) EditExcelData() {
  376. br := new(models.BaseResponse).Init()
  377. var err error
  378. errs := make([]string, 0)
  379. defer func() {
  380. if len(errs) > 0 {
  381. utils.FileLog.Info("编辑EXCEL数据 新增或修改数据失败,Err:" + strings.Join(errs, "\n"))
  382. }
  383. c.Data["json"] = br
  384. c.ServeJSON()
  385. }()
  386. sysUser := c.SysUser
  387. if sysUser == nil {
  388. br.Msg = "请重新登录"
  389. return
  390. }
  391. var req data_manage.ManualEdbExcelStyleEditReq
  392. err = json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  393. if err != nil {
  394. br.Msg = "参数解析异常!"
  395. br.ErrMsg = "参数解析失败,Err:" + err.Error()
  396. return
  397. }
  398. if req.TradeCode == `` {
  399. br.Msg = "指标编码异常!"
  400. br.ErrMsg = "指标编码异常!"
  401. return
  402. }
  403. if req.Unit == `` {
  404. br.Msg = "单位填写异常!"
  405. br.ErrMsg = "单位填写异常!"
  406. return
  407. }
  408. if req.Frequency == `` {
  409. br.Msg = "频度填写异常!"
  410. br.ErrMsg = "频度填写异常!"
  411. return
  412. }
  413. if req.ClassifyId <= 0 {
  414. br.Msg = "品种填写异常!"
  415. br.ErrMsg = "品种填写异常!"
  416. return
  417. }
  418. // 获取指标信息
  419. manualEdbInfo, err := models.GetEdbinfoByTradeCode(req.TradeCode)
  420. if err != nil {
  421. br.Msg = `找不到该指标`
  422. br.Msg = `找不到该指标,ERR:` + err.Error()
  423. return
  424. }
  425. // 如果不是超管账号,那么得校验下当前用户是否有该指标的权限
  426. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  427. //获取账户所拥有权限的分类id集合
  428. count, err := models.GetCountManualUserClassify(sysUser.AdminId, manualEdbInfo.ClassifyId)
  429. if err != nil {
  430. br.Msg = "获取分类数据失败"
  431. br.ErrMsg = "获取分类数据失败,err:" + err.Error()
  432. return
  433. }
  434. if count <= 0 {
  435. br.Msg = "无权访问"
  436. br.ErrMsg = "无权访问"
  437. br.IsSendEmail = false
  438. return
  439. }
  440. }
  441. dateValueMap := make(map[string]string)
  442. //取到所有数据
  443. for _, dateValue := range req.Data {
  444. dateValueMap[dateValue.Date] = strconv.FormatFloat(dateValue.Value, 'f', -1, 64)
  445. }
  446. //操作指标,新增指标及数据等
  447. {
  448. var isUpdateEdb, isUpdateData bool
  449. secName := strings.TrimSpace(req.EdbName)
  450. frequency := strings.TrimSpace(req.Frequency)
  451. unit := strings.TrimSpace(req.Unit)
  452. // 指标基础信息变更
  453. {
  454. updateCols := make([]string, 0)
  455. if manualEdbInfo.SecName != secName {
  456. // 校验是否存在相同指标名称
  457. var condition string
  458. var pars []interface{}
  459. condition += " AND SEC_NAME=? AND left(TRADE_CODE,1)='W' AND REMARK='手动' AND TRADE_CODE != ? "
  460. pars = append(pars, secName, req.TradeCode)
  461. count, err := models.GetManualEdbCountByCondition(condition, pars)
  462. if err != nil {
  463. br.Msg = `指标保存失败`
  464. br.ErrMsg = `获取同名指标信息异常,ERR:` + err.Error()
  465. return
  466. }
  467. if count > 0 {
  468. br.Msg = `已存在同名指标`
  469. br.ErrMsg = `已存在同名指标`
  470. br.IsSendEmail = false
  471. return
  472. }
  473. updateCols = append(updateCols, "SecName")
  474. manualEdbInfo.SecName = secName
  475. }
  476. if manualEdbInfo.ClassifyId != req.ClassifyId {
  477. updateCols = append(updateCols, "ClassifyId")
  478. manualEdbInfo.ClassifyId = req.ClassifyId
  479. }
  480. if manualEdbInfo.Frequency != frequency {
  481. updateCols = append(updateCols, "Frequency")
  482. manualEdbInfo.Frequency = frequency
  483. }
  484. if manualEdbInfo.Unit != unit {
  485. updateCols = append(updateCols, "Unit")
  486. manualEdbInfo.Unit = unit
  487. }
  488. if len(updateCols) > 0 {
  489. isUpdateEdb = true
  490. manualEdbInfo.ModifyTime = time.Now().Format(utils.FormatDateTime)
  491. updateCols = append(updateCols, "ModifyTime")
  492. err = manualEdbInfo.Update(updateCols)
  493. if err != nil {
  494. br.Msg = `指标保存失败`
  495. br.ErrMsg = `指标保存失败,ERR:` + err.Error()
  496. return
  497. }
  498. }
  499. }
  500. // 当前已经存在的指标明细数据
  501. targetDataList, tmpErr := models.GetTargetsDataList(manualEdbInfo.TradeCode)
  502. if tmpErr != nil {
  503. err = tmpErr
  504. }
  505. existDataMap := make(map[string]string)
  506. deleteDataMap := make(map[string]string)
  507. for _, tmpData := range targetDataList {
  508. existDataMap[tmpData.Dt] = tmpData.Close
  509. deleteDataMap[tmpData.Dt] = tmpData.Dt
  510. }
  511. addDataList := make([]*models.Edbdata, 0)
  512. for createDate, closeVal := range dateValueMap {
  513. if createDate == "" || closeVal == "" {
  514. continue
  515. }
  516. //判断数据是否已经存在
  517. tmpVal, ok3 := existDataMap[createDate]
  518. // 不存在,那么后面进行插入
  519. if !ok3 {
  520. addDataList = append(addDataList, &models.Edbdata{
  521. TradeCode: manualEdbInfo.TradeCode,
  522. Dt: createDate,
  523. Close: closeVal,
  524. ModifyTime: time.Now(),
  525. })
  526. continue
  527. }
  528. delete(deleteDataMap, createDate)
  529. // 库里面的数据
  530. tmpValDecimal, tmpErr := decimal.NewFromString(tmpVal)
  531. if tmpErr != nil {
  532. fmt.Println("tmpVal Parse err:", tmpErr.Error())
  533. err = tmpErr
  534. continue
  535. }
  536. // 用户填写的数据
  537. closeValDecimal, tmpErr := decimal.NewFromString(closeVal)
  538. if tmpErr != nil {
  539. fmt.Println("closeVal Parse err:", tmpErr.Error())
  540. err = tmpErr
  541. continue
  542. }
  543. if !tmpValDecimal.Equal(closeValDecimal) {
  544. isUpdateData = true
  545. fmt.Println("更新数值")
  546. err = models.ModifyTargetsDataByImport(manualEdbInfo.TradeCode, createDate, closeVal)
  547. if err != nil {
  548. fmt.Println("ModifyTargetsDataByImport err:", err.Error())
  549. errs = append(errs, err.Error())
  550. //go utils.SendEmail(utils.APPNAME+"失败提醒", "导入数据 修改数据失败:Err:"+err.Error(), utils.EmailSendToUsers)
  551. }
  552. }
  553. }
  554. // 新增指标
  555. if len(addDataList) > 0 {
  556. isUpdateData = true
  557. err = models.OnlyMultiAddEdbdata(addDataList)
  558. if err != nil {
  559. fmt.Println("MultiAddEdbdata err:", err.Error())
  560. errs = append(errs, err.Error())
  561. }
  562. }
  563. // 删除不需要的日期数据
  564. num := len(deleteDataMap)
  565. if num > 0 {
  566. isUpdateData = true
  567. deleteDateList := make([]string, 0)
  568. for date := range deleteDataMap {
  569. deleteDateList = append(deleteDateList, date)
  570. }
  571. err = models.DelEdbdataByCodeAndDateList(manualEdbInfo.TradeCode, deleteDateList)
  572. if err != nil {
  573. fmt.Println("DelEdbdataByCodeAndDateList err:", err.Error())
  574. errs = append(errs, err.Error())
  575. }
  576. }
  577. // 添加记录
  578. if isUpdateEdb {
  579. record := &models.EdbinfoOpRecord{
  580. TradeCode: manualEdbInfo.TradeCode,
  581. Remark: "编辑指标",
  582. UserId: sysUser.AdminId,
  583. UserName: sysUser.RealName,
  584. CreateTime: time.Now(),
  585. }
  586. go func() {
  587. _ = record.Create()
  588. }()
  589. }
  590. // 更新数据
  591. if isUpdateData {
  592. record := &models.EdbinfoOpRecord{
  593. TradeCode: manualEdbInfo.TradeCode,
  594. Remark: "更新数据",
  595. UserId: sysUser.AdminId,
  596. UserName: sysUser.RealName,
  597. CreateTime: time.Now(),
  598. }
  599. go func() {
  600. data.ModifyManualEdbMaxMinDate(manualEdbInfo.TradeCode)
  601. _ = record.Create()
  602. }()
  603. }
  604. }
  605. br.Ret = 200
  606. br.Success = true
  607. br.Msg = "保存成功"
  608. }
  609. // BatchEdbList
  610. // @Title 获取研究员指标
  611. // @Description 获取研究员指标
  612. // @Param request body models.BatchAddEdbReq true "type json string"
  613. // @Success 200 {object} models.TargetItemListResp
  614. // @router /target/edb/batch/list [post]
  615. func (c *ManualEdbController) BatchEdbList() {
  616. br := new(models.BaseResponse).Init()
  617. defer func() {
  618. c.Data["json"] = br
  619. c.ServeJSON()
  620. }()
  621. sysUser := c.SysUser
  622. if sysUser == nil {
  623. br.Msg = "请重新登录"
  624. return
  625. }
  626. var req models.BatchManualEdbReq
  627. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  628. pageSize := req.PageSize
  629. currentIndex := req.CurrentIndex
  630. var startSize int
  631. if pageSize <= 0 {
  632. pageSize = utils.PageSize20
  633. }
  634. if currentIndex <= 0 {
  635. currentIndex = 1
  636. }
  637. startSize = utils.StartIndex(currentIndex, pageSize)
  638. req.Keyword = strings.TrimSpace(req.Keyword)
  639. var condition string
  640. var pars []interface{}
  641. // 是否加到指标库
  642. condition += ` AND a.is_join_edb = ? `
  643. pars = append(pars, 0)
  644. //超管账号可以查看分类下的所有频度数据
  645. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  646. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  647. if err != nil {
  648. br.Msg = "获取分类数据失败"
  649. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  650. return
  651. }
  652. num := len(classifyIdList)
  653. if num > 0 {
  654. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  655. pars = append(pars, classifyIdList)
  656. }
  657. }
  658. // 关键词
  659. if req.Keyword != "" {
  660. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  661. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  662. }
  663. // 所属分类
  664. lenFrequency := len(req.FrequencyList)
  665. if lenFrequency > 0 {
  666. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  667. pars = append(pars, req.FrequencyList)
  668. }
  669. // 所属分类
  670. lenClassify := len(req.ClassifyIdList)
  671. if lenClassify > 0 {
  672. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  673. pars = append(pars, req.ClassifyIdList)
  674. }
  675. // 所属用户
  676. lenUser := len(req.UserIdList)
  677. if lenUser > 0 {
  678. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  679. pars = append(pars, req.UserIdList)
  680. }
  681. total, err := models.GetCountEdbInfoList(condition, pars)
  682. if err != nil {
  683. br.Msg = "获取失败"
  684. br.ErrMsg = "获取失败,Err:" + err.Error()
  685. return
  686. }
  687. list, err := models.GetEdbInfoList(condition, pars, startSize, pageSize)
  688. if err != nil {
  689. br.Msg = "获取失败"
  690. br.ErrMsg = "获取失败,Err:" + err.Error()
  691. return
  692. }
  693. resp := models.EdbListResp{
  694. List: list,
  695. Paging: paging.GetPaging(currentIndex, pageSize, total),
  696. }
  697. br.Ret = 200
  698. br.Success = true
  699. br.Msg = "获取成功"
  700. br.Data = resp
  701. br.IsAddLog = false
  702. }
  703. // BatchAddEdbCheck
  704. // @Title 新增校验
  705. // @Description 新增校验
  706. // @Param request body data_manage.BatchManualEdbReq true "type json string"
  707. // @Success 200 string "操作成功"
  708. // @router /target/edb/batch/add/check [post]
  709. func (c *ManualEdbController) BatchAddEdbCheck() {
  710. br := new(models.BaseResponse).Init()
  711. defer func() {
  712. if br.ErrMsg == "" {
  713. br.IsSendEmail = false
  714. }
  715. c.Data["json"] = br
  716. c.ServeJSON()
  717. }()
  718. sysUser := c.SysUser
  719. if sysUser == nil {
  720. br.Msg = "请登录"
  721. br.ErrMsg = "请登录,SysUser Is Empty"
  722. br.Ret = 408
  723. return
  724. }
  725. // 最大批量添加的数量
  726. codeMax := 30
  727. var req models.BatchManualEdbReq
  728. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  729. req.Keyword = strings.TrimSpace(req.Keyword)
  730. var condition string
  731. var pars []interface{}
  732. //
  733. condition += ` AND a.is_join_edb = 0 `
  734. //超管账号可以查看分类下的所有频度数据
  735. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  736. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  737. if err != nil {
  738. br.Msg = "获取分类数据失败"
  739. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  740. return
  741. }
  742. num := len(classifyIdList)
  743. if num > 0 {
  744. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  745. pars = append(pars, classifyIdList)
  746. }
  747. }
  748. if req.ListAll {
  749. // 关键词
  750. if req.Keyword != "" {
  751. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  752. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  753. }
  754. // 所属分类
  755. lenFrequency := len(req.FrequencyList)
  756. if lenFrequency > 0 {
  757. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  758. pars = append(pars, req.FrequencyList)
  759. }
  760. // 所属分类
  761. lenClassify := len(req.ClassifyIdList)
  762. if lenClassify > 0 {
  763. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  764. pars = append(pars, req.ClassifyIdList)
  765. }
  766. // 所属用户
  767. lenUser := len(req.UserIdList)
  768. if lenUser > 0 {
  769. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  770. pars = append(pars, req.UserIdList)
  771. }
  772. lenTradeList := len(req.TradeCodeList)
  773. // 指标
  774. if lenTradeList > 0 {
  775. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  776. pars = append(pars, req.TradeCodeList)
  777. }
  778. } else {
  779. lenTradeList := len(req.TradeCodeList)
  780. if lenTradeList <= 0 {
  781. br.Msg = "请选择指标"
  782. br.ErrMsg = "请选择指标"
  783. return
  784. }
  785. // 指标
  786. if lenTradeList > 0 {
  787. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  788. pars = append(pars, req.TradeCodeList)
  789. }
  790. }
  791. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  792. if err != nil {
  793. br.Msg = "获取失败"
  794. br.ErrMsg = "获取失败,Err:" + err.Error()
  795. return
  796. }
  797. if len(list) > codeMax {
  798. br.Msg = fmt.Sprintf("最多只能选择%d个指标", codeMax)
  799. return
  800. }
  801. br.Data = list
  802. br.Msg = "校验成功"
  803. br.Ret = 200
  804. br.Success = true
  805. }
  806. // NameCheck
  807. // @Title 重名校验
  808. // @Description 批量新增
  809. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  810. // @Success 200 string "操作成功"
  811. // @router /target/edb/batch/add/name_check [post]
  812. func (c *ManualEdbController) NameCheck() {
  813. br := new(models.BaseResponse).Init()
  814. defer func() {
  815. if br.ErrMsg == "" {
  816. br.IsSendEmail = false
  817. }
  818. c.Data["json"] = br
  819. c.ServeJSON()
  820. }()
  821. sysUser := c.SysUser
  822. if sysUser == nil {
  823. br.Msg = "请登录"
  824. br.ErrMsg = "请登录,SysUser Is Empty"
  825. br.Ret = 408
  826. return
  827. }
  828. var req []*data_manage.AddEdbInfoReq
  829. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  830. br.Msg = "参数解析异常!"
  831. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  832. return
  833. }
  834. if len(req) == 0 {
  835. br.Msg = "请选择指标"
  836. return
  837. }
  838. type NameCheckResult struct {
  839. EdbCode string
  840. EdbName string
  841. Exist bool
  842. }
  843. indexNames := make([]string, 0)
  844. resp := make([]*NameCheckResult, 0)
  845. for _, v := range req {
  846. v.EdbCode = strings.TrimSpace(v.EdbCode)
  847. if v.EdbCode == "" {
  848. br.Msg = "指标ID不可为空"
  849. return
  850. }
  851. v.EdbName = strings.TrimSpace(v.EdbName)
  852. if v.EdbName == "" {
  853. br.Msg = "请输入指标名称"
  854. return
  855. }
  856. indexNames = append(indexNames, v.EdbName)
  857. resp = append(resp, &NameCheckResult{
  858. EdbCode: v.EdbCode,
  859. EdbName: v.EdbName,
  860. })
  861. }
  862. // 重名校验
  863. edbList, e := data_manage.GetEdbInfoByNameArr(indexNames, utils.EDB_INFO_TYPE)
  864. if e != nil {
  865. br.Msg = "操作失败"
  866. br.ErrMsg = "获取重名指标失败, Err: " + e.Error()
  867. return
  868. }
  869. nameExists := make(map[string]bool)
  870. for _, v := range edbList {
  871. nameExists[v.EdbName] = true
  872. }
  873. if len(nameExists) > 0 {
  874. for _, v := range resp {
  875. v.Exist = nameExists[v.EdbName]
  876. }
  877. }
  878. br.Data = resp
  879. br.Msg = "操作成功"
  880. br.Ret = 200
  881. br.Success = true
  882. }
  883. // BatchAdd2Edb
  884. // @Title 批量新增到指标库
  885. // @Description 批量新增到指标库
  886. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  887. // @Success 200 string "操作成功"
  888. // @router /target/edb/batch/add [post]
  889. func (c *ManualEdbController) BatchAdd2Edb() {
  890. br := new(models.BaseResponse).Init()
  891. defer func() {
  892. if br.ErrMsg == "" {
  893. br.IsSendEmail = false
  894. }
  895. c.Data["json"] = br
  896. c.ServeJSON()
  897. }()
  898. sysUser := c.SysUser
  899. if sysUser == nil {
  900. br.Msg = "请登录"
  901. br.ErrMsg = "请登录,SysUser Is Empty"
  902. br.Ret = 408
  903. return
  904. }
  905. // 最大批量添加的数量
  906. codeMax := 30
  907. deleteCache := true
  908. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_MANUAL_" + strconv.Itoa(sysUser.AdminId)
  909. defer func() {
  910. if deleteCache {
  911. _ = utils.Rc.Delete(cacheKey)
  912. }
  913. }()
  914. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  915. deleteCache = false
  916. br.Msg = "系统处理中,请稍后重试!"
  917. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(c.Ctx.Input.RequestBody)
  918. return
  919. }
  920. var req []*data_manage.AddEdbInfoReq
  921. if e := json.Unmarshal(c.Ctx.Input.RequestBody, &req); e != nil {
  922. br.Msg = "参数解析异常!"
  923. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  924. return
  925. }
  926. if len(req) == 0 {
  927. br.Msg = "请选择指标"
  928. return
  929. }
  930. if len(req) > codeMax {
  931. br.Msg = "批量添加指标数量不得超过30个"
  932. return
  933. }
  934. indexNames := make([]string, 0)
  935. for _, v := range req {
  936. v.EdbCode = strings.TrimSpace(v.EdbCode)
  937. if v.EdbCode == "" {
  938. br.Msg = "指标ID不可为空"
  939. return
  940. }
  941. v.EdbName = strings.TrimSpace(v.EdbName)
  942. if v.EdbName == "" {
  943. br.Msg = "请输入指标名称"
  944. return
  945. }
  946. indexNames = append(indexNames, v.EdbName)
  947. v.Frequency = strings.TrimSpace(v.Frequency)
  948. if v.Frequency == "" {
  949. br.Msg = "请选择频度"
  950. return
  951. }
  952. v.Unit = strings.TrimSpace(v.Unit)
  953. if v.Unit == "" {
  954. br.Msg = "请输入单位"
  955. return
  956. }
  957. if v.ClassifyId <= 0 {
  958. br.Msg = "请选择分类"
  959. return
  960. }
  961. }
  962. // 限定同一时间最多批量新增30个指标
  963. for _, v := range req {
  964. var r data.ManualIndexSource2EdbReq
  965. r.EdbCode = v.EdbCode
  966. r.EdbName = v.EdbName
  967. r.Frequency = v.Frequency
  968. r.Unit = v.Unit
  969. r.ClassifyId = v.ClassifyId
  970. r.AdminId = sysUser.AdminId
  971. r.AdminRealName = sysUser.RealName
  972. edbInfo, e, errMsg, skip := data.ManualIndexSource2Edb(r, c.Lang)
  973. if e != nil {
  974. br.Msg = "操作失败"
  975. if errMsg != "" {
  976. br.Msg = errMsg
  977. }
  978. br.ErrMsg = e.Error()
  979. return
  980. }
  981. if skip {
  982. continue
  983. }
  984. // 试用平台更新用户累计新增指标数
  985. if utils.BusinessCode == utils.BusinessCodeSandbox {
  986. go func() {
  987. adminItem, e := system.GetSysAdminById(sysUser.AdminId)
  988. if e != nil {
  989. tips := fmt.Sprintf("试用平台更新用户累计新增指标数-获取用户失败, Err: " + e.Error())
  990. utils.FileLog.Info(tips)
  991. return
  992. }
  993. if adminItem.DepartmentName != "ETA试用客户" {
  994. return
  995. }
  996. var ur etaTrialService.EtaTrialUserReq
  997. ur.Mobile = adminItem.Mobile
  998. _, _ = etaTrialService.UpdateUserIndexNum(ur)
  999. }()
  1000. }
  1001. // 新增操作日志
  1002. {
  1003. edbLog := new(data_manage.EdbInfoLog)
  1004. edbLog.EdbInfoId = edbInfo.EdbInfoId
  1005. edbLog.SourceName = edbInfo.SourceName
  1006. edbLog.Source = edbInfo.Source
  1007. edbLog.EdbCode = edbInfo.EdbCode
  1008. edbLog.EdbName = edbInfo.EdbName
  1009. edbLog.ClassifyId = edbInfo.ClassifyId
  1010. edbLog.SysUserId = sysUser.AdminId
  1011. edbLog.SysUserRealName = sysUser.RealName
  1012. edbLog.CreateTime = time.Now()
  1013. edbLog.Content = string(c.Ctx.Input.RequestBody)
  1014. edbLog.Status = "新增指标"
  1015. edbLog.Method = c.Ctx.Input.URI()
  1016. go data_manage.AddEdbInfoLog(edbLog)
  1017. }
  1018. }
  1019. br.Msg = "操作成功"
  1020. br.Ret = 200
  1021. br.Success = true
  1022. br.IsAddLog = true
  1023. }
  1024. // BatchDelEdbCheck
  1025. // @Title 新增校验
  1026. // @Description 新增校验
  1027. // @Param request body data_manage.BatchManualEdbReq true "type json string"
  1028. // @Success 200 string "操作成功"
  1029. // @router /target/edb/batch/del/check [post]
  1030. func (c *ManualEdbController) BatchDelEdbCheck() {
  1031. br := new(models.BaseResponse).Init()
  1032. defer func() {
  1033. if br.ErrMsg == "" {
  1034. br.IsSendEmail = false
  1035. }
  1036. c.Data["json"] = br
  1037. c.ServeJSON()
  1038. }()
  1039. sysUser := c.SysUser
  1040. if sysUser == nil {
  1041. br.Msg = "请登录"
  1042. br.ErrMsg = "请登录,SysUser Is Empty"
  1043. br.Ret = 408
  1044. return
  1045. }
  1046. var req models.BatchManualEdbReq
  1047. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  1048. req.Keyword = strings.TrimSpace(req.Keyword)
  1049. var condition string
  1050. var pars []interface{}
  1051. // 就是为了找出已经加入指标库的指标
  1052. condition += ` AND a.is_join_edb = ? `
  1053. pars = append(pars, 1)
  1054. //超管账号可以查看分类下的所有频度数据
  1055. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  1056. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  1057. if err != nil {
  1058. br.Msg = "获取分类数据失败"
  1059. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  1060. return
  1061. }
  1062. num := len(classifyIdList)
  1063. if num > 0 {
  1064. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  1065. pars = append(pars, classifyIdList)
  1066. }
  1067. }
  1068. if req.ListAll {
  1069. // 关键词
  1070. if req.Keyword != "" {
  1071. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  1072. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  1073. }
  1074. // 所属分类
  1075. lenFrequency := len(req.FrequencyList)
  1076. if lenFrequency > 0 {
  1077. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  1078. pars = append(pars, req.FrequencyList)
  1079. }
  1080. // 所属分类
  1081. lenClassify := len(req.ClassifyIdList)
  1082. if lenClassify > 0 {
  1083. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  1084. pars = append(pars, req.ClassifyIdList)
  1085. }
  1086. // 所属用户
  1087. lenUser := len(req.UserIdList)
  1088. if lenUser > 0 {
  1089. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  1090. pars = append(pars, req.UserIdList)
  1091. }
  1092. lenTradeList := len(req.TradeCodeList)
  1093. // 指标
  1094. if lenTradeList > 0 {
  1095. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1096. pars = append(pars, req.TradeCodeList)
  1097. }
  1098. } else {
  1099. lenTradeList := len(req.TradeCodeList)
  1100. if lenTradeList <= 0 {
  1101. br.Msg = "请选择指标"
  1102. br.ErrMsg = "请选择指标"
  1103. return
  1104. }
  1105. // 指标
  1106. if lenTradeList > 0 {
  1107. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1108. pars = append(pars, req.TradeCodeList)
  1109. }
  1110. }
  1111. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  1112. if err != nil {
  1113. br.Msg = "获取失败"
  1114. br.ErrMsg = "获取失败,Err:" + err.Error()
  1115. return
  1116. }
  1117. br.Data = list
  1118. br.Msg = "校验成功"
  1119. br.Ret = 200
  1120. br.Success = true
  1121. }
  1122. // BatchDel
  1123. // @Title 批量删除
  1124. // @Description 批量新增
  1125. // @Param request body data_manage.AddEdbInfoReq true "type json string"
  1126. // @Success 200 string "操作成功"
  1127. // @router /target/edb/batch/del [post]
  1128. func (c *ManualEdbController) BatchDel() {
  1129. br := new(models.BaseResponse).Init()
  1130. defer func() {
  1131. if br.ErrMsg == "" {
  1132. br.IsSendEmail = false
  1133. }
  1134. c.Data["json"] = br
  1135. c.ServeJSON()
  1136. }()
  1137. sysUser := c.SysUser
  1138. if sysUser == nil {
  1139. br.Msg = "请登录"
  1140. br.ErrMsg = "请登录,SysUser Is Empty"
  1141. br.Ret = 408
  1142. return
  1143. }
  1144. deleteCache := true
  1145. cacheKey := "CACHE_EDB_INFO_BATCH_ADD_MANUAL_" + strconv.Itoa(sysUser.AdminId)
  1146. defer func() {
  1147. if deleteCache {
  1148. _ = utils.Rc.Delete(cacheKey)
  1149. }
  1150. }()
  1151. if !utils.Rc.SetNX(cacheKey, 1, 30*time.Second) {
  1152. deleteCache = false
  1153. br.Msg = "系统处理中,请稍后重试!"
  1154. br.ErrMsg = "系统处理中,请稍后重试!" + sysUser.RealName + ";data:" + string(c.Ctx.Input.RequestBody)
  1155. return
  1156. }
  1157. var req models.BatchManualEdbReq
  1158. err := json.Unmarshal(c.Ctx.Input.RequestBody, &req)
  1159. req.Keyword = strings.TrimSpace(req.Keyword)
  1160. var condition string
  1161. var pars []interface{}
  1162. // 就是为了找出已经加入指标库的指标
  1163. condition += ` AND a.is_join_edb = ? `
  1164. pars = append(pars, 0)
  1165. //超管账号可以查看分类下的所有频度数据
  1166. if sysUser.RoleTypeCode != utils.ROLE_TYPE_CODE_ADMIN {
  1167. classifyIdList, err := data.GetUserManualClassifyIdList(sysUser.AdminId)
  1168. if err != nil {
  1169. br.Msg = "获取分类数据失败"
  1170. br.Msg = "获取拥有的分类数据失败,Err:" + err.Error()
  1171. return
  1172. }
  1173. num := len(classifyIdList)
  1174. if num > 0 {
  1175. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(num) + `) `
  1176. pars = append(pars, classifyIdList)
  1177. }
  1178. }
  1179. if req.ListAll {
  1180. // 关键词
  1181. if req.Keyword != "" {
  1182. condition += ` AND (a.SEC_NAME like ? or a.TRADE_CODE like ? )`
  1183. pars = utils.GetLikeKeywordPars(pars, req.Keyword, 2)
  1184. }
  1185. // 所属分类
  1186. lenFrequency := len(req.FrequencyList)
  1187. if lenFrequency > 0 {
  1188. condition += ` AND a.frequency in (` + utils.GetOrmInReplace(lenFrequency) + `) `
  1189. pars = append(pars, req.FrequencyList)
  1190. }
  1191. // 所属分类
  1192. lenClassify := len(req.ClassifyIdList)
  1193. if lenClassify > 0 {
  1194. condition += ` AND a.classify_id in (` + utils.GetOrmInReplace(lenClassify) + `) `
  1195. pars = append(pars, req.ClassifyIdList)
  1196. }
  1197. // 所属用户
  1198. lenUser := len(req.UserIdList)
  1199. if lenUser > 0 {
  1200. condition += ` AND a.user_id in (` + utils.GetOrmInReplace(lenUser) + `) `
  1201. pars = append(pars, req.UserIdList)
  1202. }
  1203. lenTradeList := len(req.TradeCodeList)
  1204. // 指标
  1205. if lenTradeList > 0 {
  1206. condition += ` AND a.TRADE_CODE not in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1207. pars = append(pars, req.TradeCodeList)
  1208. }
  1209. } else {
  1210. lenTradeList := len(req.TradeCodeList)
  1211. if lenTradeList <= 0 {
  1212. br.Msg = "请选择指标"
  1213. br.ErrMsg = "请选择指标"
  1214. return
  1215. }
  1216. // 指标
  1217. if lenTradeList > 0 {
  1218. condition += ` AND a.TRADE_CODE in (` + utils.GetOrmInReplace(lenTradeList) + `) `
  1219. pars = append(pars, req.TradeCodeList)
  1220. }
  1221. }
  1222. list, err := models.GetEdbInfoList(condition, pars, 0, 0)
  1223. if err != nil {
  1224. br.Msg = "删除失败"
  1225. br.ErrMsg = "删除失败,Err:" + err.Error()
  1226. return
  1227. }
  1228. // TODO 二次校验指标是否已经加到指标库
  1229. {
  1230. codeList := make([]string, 0)
  1231. // 限定同一时间最多批量新增30个指标
  1232. for _, v := range list {
  1233. codeList = append(codeList, v.TradeCode)
  1234. }
  1235. err = models.DelManualIndexByCodeList(codeList)
  1236. if err != nil {
  1237. br.Msg = "删除失败"
  1238. br.ErrMsg = "删除失败,Err:" + err.Error()
  1239. return
  1240. }
  1241. }
  1242. br.Msg = "删除成功"
  1243. br.Ret = 200
  1244. br.Success = true
  1245. br.IsAddLog = true
  1246. }
  1247. // ImportData
  1248. // @Title 导入数据
  1249. // @Description 导入数据
  1250. // @Param EntryFile query file true "文件"
  1251. // @Success 200 Ret=200 录入成功
  1252. // @router /import/data [post]
  1253. func (c *ManualEdbController) ImportData() {
  1254. br := new(models.BaseResponse).Init()
  1255. defer func() {
  1256. c.Data["json"] = br
  1257. c.ServeJSON()
  1258. }()
  1259. sysUser := c.SysUser
  1260. if sysUser == nil {
  1261. br.Msg = "请重新登录"
  1262. return
  1263. }
  1264. file, _, err := c.GetFile("EntryFile")
  1265. if err != nil {
  1266. br.Msg = "获取文件失败"
  1267. br.ErrMsg = "获取文件失败,Err:" + err.Error()
  1268. return
  1269. }
  1270. path := "./static/数据导入_" + time.Now().Format(utils.FormatDateTimeUnSpace) + ".xlsx"
  1271. defer file.Close()
  1272. err = c.SaveToFile("EntryFile", path)
  1273. if err != nil {
  1274. br.Msg = "文件保存失败"
  1275. br.ErrMsg = "文件保存失败,Err:" + err.Error()
  1276. return
  1277. }
  1278. if utils.RunMode == "debug" {
  1279. defer os.Remove(path)
  1280. }
  1281. successCount, failCount, err, errMsg := data.ImportManualData(path, sysUser)
  1282. if err != nil {
  1283. br.Msg = errMsg
  1284. br.ErrMsg = err.Error()
  1285. return
  1286. }
  1287. resp := models.EdbdataImportResp{
  1288. SuccessCount: successCount,
  1289. FailCount: failCount,
  1290. }
  1291. if failCount > 0 {
  1292. if successCount == 0 {
  1293. resp.Status = -1
  1294. resp.Msg = "导入失败"
  1295. } else {
  1296. resp.Status = 1
  1297. resp.Msg = "存在部分导入失败"
  1298. }
  1299. } else {
  1300. resp.Status = 0
  1301. }
  1302. br.Msg = "导入成功"
  1303. br.Ret = 200
  1304. br.Success = true
  1305. br.Data = resp
  1306. }
  1307. // RecordList
  1308. // @Title 获取指标操作记录列表
  1309. // @Description 获取指标操作记录列表
  1310. // @Param TradeCode query string false "指标编码"
  1311. // @Success 200 {object} models.EdbinfoOpRecordListResp
  1312. // @router /target/edb/op/record/list [get]
  1313. func (c *ManualEdbController) RecordList() {
  1314. br := new(models.BaseResponse).Init()
  1315. defer func() {
  1316. c.Data["json"] = br
  1317. c.ServeJSON()
  1318. }()
  1319. sysUser := c.SysUser
  1320. if sysUser == nil {
  1321. br.Msg = "请重新登录"
  1322. return
  1323. }
  1324. pageSize, _ := c.GetInt("PageSize")
  1325. currentIndex, _ := c.GetInt("CurrentIndex")
  1326. var startSize int
  1327. if pageSize <= 0 {
  1328. pageSize = utils.PageSize20
  1329. }
  1330. if currentIndex <= 0 {
  1331. currentIndex = 1
  1332. }
  1333. startSize = utils.StartIndex(currentIndex, pageSize)
  1334. // 频度
  1335. tradeCode := c.GetString("TradeCode")
  1336. if tradeCode == `` {
  1337. br.Msg = `请选择指标`
  1338. br.ErrMsg = `请选择指标`
  1339. br.IsSendEmail = false
  1340. return
  1341. }
  1342. total, list, err := models.GetEdbinfoOpRecordPageList(tradeCode, startSize, pageSize)
  1343. if err != nil {
  1344. br.Msg = "获取失败"
  1345. br.ErrMsg = "获取失败,Err:" + err.Error()
  1346. return
  1347. }
  1348. resp := models.EdbinfoOpRecordListResp{
  1349. List: list,
  1350. Paging: paging.GetPaging(currentIndex, pageSize, total),
  1351. }
  1352. br.Ret = 200
  1353. br.Success = true
  1354. br.Msg = "获取成功"
  1355. br.Data = resp
  1356. }