base_from_ths_hf.go 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575
  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/services/data"
  8. "eta/eta_api/utils"
  9. "fmt"
  10. "github.com/rdlucklib/rdluck_tools/paging"
  11. "sort"
  12. "strconv"
  13. "strings"
  14. "time"
  15. )
  16. // BaseFromThsHfController 同花顺高频数据
  17. type BaseFromThsHfController struct {
  18. controllers.BaseAuthController
  19. }
  20. // Search
  21. // @Title 新增指标-查询指标信息接口
  22. // @Description 新增指标-查询指标信息接口
  23. // @Param request body data_manage.ThsHfSearchEdbReq true "type json string"
  24. // @Success 200 {object} data_manage.ThsHfSearchEdbResp
  25. // @router /ths_hf/index/search [get]
  26. func (this *BaseFromThsHfController) Search() {
  27. br := new(models.BaseResponse).Init()
  28. defer func() {
  29. if br.ErrMsg == "" {
  30. br.IsSendEmail = false
  31. }
  32. this.Data["json"] = br
  33. this.ServeJSON()
  34. }()
  35. sysUser := this.SysUser
  36. if sysUser == nil {
  37. br.Msg = "请登录"
  38. br.ErrMsg = "请登录,SysUser Is Empty"
  39. br.Ret = 408
  40. return
  41. }
  42. var params data_manage.ThsHfSearchEdbReq
  43. if e := this.ParseForm(&params); e != nil {
  44. br.Msg = "参数解析失败"
  45. br.ErrMsg = fmt.Sprintf("参数解析失败, err: %v", e)
  46. return
  47. }
  48. params.StockCode = strings.TrimSpace(params.StockCode)
  49. if params.StockCode == "" {
  50. br.Msg = "请输入证券代码"
  51. return
  52. }
  53. stockCodes := strings.Split(params.StockCode, ",")
  54. if len(stockCodes) == 0 {
  55. br.Msg = "请输入证券代码"
  56. return
  57. }
  58. if len(stockCodes) > 10 {
  59. br.Msg = "最多输入10个证券代码"
  60. return
  61. }
  62. params.EdbCode = strings.TrimSpace(params.EdbCode)
  63. if params.EdbCode == "" {
  64. br.Msg = "请输入指标代码"
  65. return
  66. }
  67. edbCodes := strings.Split(params.EdbCode, ",")
  68. if len(edbCodes) == 0 {
  69. br.Msg = "请输入指标代码"
  70. return
  71. }
  72. if len(edbCodes) > 20 {
  73. br.Msg = "最多选择/输入20个指标代码"
  74. return
  75. }
  76. if params.StartTime == "" {
  77. br.Msg = "请选择起始时间"
  78. return
  79. }
  80. _, e := time.ParseInLocation(utils.FormatDateTime, params.StartTime, time.Local)
  81. if e != nil {
  82. br.Msg = "起始时间格式有误"
  83. br.ErrMsg = fmt.Sprintf("起始时间格式有误, %v", e)
  84. return
  85. }
  86. // 结束时间选填, 不填则为当前时间
  87. if params.EndTime != "" {
  88. _, e := time.ParseInLocation(utils.FormatDateTime, params.EndTime, time.Local)
  89. if e != nil {
  90. br.Msg = "截止时间格式有误"
  91. br.ErrMsg = fmt.Sprintf("截止时间格式有误, %v", e)
  92. return
  93. }
  94. }
  95. if params.EndTime == "" {
  96. params.EndTime = time.Now().Local().Format(utils.FormatDateTime)
  97. }
  98. if !utils.InArrayByInt(data_manage.ThsHfPeriodArr, params.Interval) {
  99. br.Msg = "时间周期有误"
  100. br.ErrMsg = fmt.Sprintf("时间周期有误, Interval: %d", params.Interval)
  101. return
  102. }
  103. if params.CPS != "" && !utils.InArrayByStr(data_manage.ThsHfCPSArr, params.CPS) {
  104. br.Msg = "复权方式有误"
  105. br.ErrMsg = fmt.Sprintf("复权方式有误, CPS: %s", params.CPS)
  106. return
  107. }
  108. if params.BaseDate != "" {
  109. _, e = time.ParseInLocation(utils.FormatDate, params.BaseDate, time.Local)
  110. if e != nil {
  111. br.Msg = "复权基点格式有误"
  112. br.ErrMsg = fmt.Sprintf("复权基点格式有误, %v", e)
  113. return
  114. }
  115. }
  116. if params.Fill != "" && !utils.InArrayByStr(data_manage.ThsHfFillArr, params.Fill) {
  117. br.Msg = "非交易间隔处理有误"
  118. br.ErrMsg = fmt.Sprintf("非交易间隔处理有误, Fill: %s", params.Fill)
  119. return
  120. }
  121. // 校验已入库的指标
  122. checkResp, indexExists, e := data.CheckExistThsHfEdb(stockCodes, edbCodes)
  123. if e != nil {
  124. br.Msg = "获取失败"
  125. br.ErrMsg = fmt.Sprintf("校验已存在的高频指标失败, %v", e)
  126. return
  127. }
  128. if len(checkResp.ExistIndex) > 0 && checkResp.ExistAll {
  129. br.Msg = "指标均已存在"
  130. return
  131. }
  132. resp := make([]*data_manage.ThsHfSearchEdbResp, 0)
  133. // 请求API获取数据
  134. indexes, e := data.GetEdbDataThsHf(params)
  135. if e != nil {
  136. br.Msg = "未搜索到指标"
  137. br.ErrMsg = fmt.Sprintf("获取高频指标失败, %v", e)
  138. return
  139. }
  140. if len(indexes) == 0 {
  141. br.Msg = "未搜索到指标"
  142. return
  143. }
  144. for _, v := range indexes {
  145. // 忽略掉校验出来的已入库指标
  146. k := fmt.Sprintf("%s-%s", v.StockCode, v.EdbCode)
  147. if indexExists[k] {
  148. continue
  149. }
  150. // 默认指标名称
  151. suffix := data_manage.ThsHfEdbCodeCn[v.EdbCode]
  152. if suffix == "" {
  153. suffix = v.EdbCode
  154. }
  155. indexName := fmt.Sprintf("%s%dm%s", v.StockCode, params.Interval, suffix)
  156. item := new(data_manage.ThsHfSearchEdbResp)
  157. item.StockCode = v.StockCode
  158. item.EdbCode = v.EdbCode
  159. item.IndexName = indexName
  160. item.Frequency = params.Interval
  161. // 搜索只展示100条数据
  162. var limit int
  163. sort.Slice(v.IndexData, func(i, j int) bool {
  164. return v.IndexData[i].DataTime.After(v.IndexData[j].DataTime)
  165. })
  166. for _, d := range v.IndexData {
  167. if limit > 100 {
  168. break
  169. }
  170. limit += 1
  171. item.IndexData = append(item.IndexData, data_manage.ThsHfSearchEdbData{
  172. DataTime: d.DataTime.Format(utils.FormatDateTime),
  173. Value: fmt.Sprintf("%.f", d.Value),
  174. })
  175. }
  176. resp = append(resp, item)
  177. }
  178. br.Data = resp
  179. br.Ret = 200
  180. br.Success = true
  181. br.Msg = "获取成功"
  182. }
  183. // ExistCheck
  184. // @Title 新增指标-指标存在校验
  185. // @Description 新增指标-指标存在校验
  186. // @Param request body data_manage.ThsHfSearchEdbReq true "type json string"
  187. // @Success 200 {object} data_manage.ThsHfExistCheckResp
  188. // @router /ths_hf/index/exist_check [get]
  189. func (this *BaseFromThsHfController) ExistCheck() {
  190. br := new(models.BaseResponse).Init()
  191. defer func() {
  192. if br.ErrMsg == "" {
  193. br.IsSendEmail = false
  194. }
  195. this.Data["json"] = br
  196. this.ServeJSON()
  197. }()
  198. sysUser := this.SysUser
  199. if sysUser == nil {
  200. br.Msg = "请登录"
  201. br.ErrMsg = "请登录,SysUser Is Empty"
  202. br.Ret = 408
  203. return
  204. }
  205. var params data_manage.ThsHfSearchEdbReq
  206. if e := this.ParseForm(&params); e != nil {
  207. br.Msg = "参数解析失败"
  208. br.ErrMsg = fmt.Sprintf("参数解析失败, err: %v", e)
  209. return
  210. }
  211. params.StockCode = strings.TrimSpace(params.StockCode)
  212. if params.StockCode == "" {
  213. br.Msg = "请输入证券代码"
  214. return
  215. }
  216. stockCodes := strings.Split(params.StockCode, ",")
  217. if len(stockCodes) == 0 {
  218. br.Msg = "请输入证券代码"
  219. return
  220. }
  221. params.EdbCode = strings.TrimSpace(params.EdbCode)
  222. if params.EdbCode == "" {
  223. br.Msg = "请输入指标代码"
  224. return
  225. }
  226. edbCodes := strings.Split(params.EdbCode, ",")
  227. if len(edbCodes) == 0 {
  228. br.Msg = "请输入指标代码"
  229. return
  230. }
  231. // 校验已存在的高频指标
  232. resp, _, e := data.CheckExistThsHfEdb(stockCodes, edbCodes)
  233. if e != nil {
  234. br.Msg = "获取失败"
  235. br.ErrMsg = fmt.Sprintf("校验已存在的高频指标失败, %v", e)
  236. return
  237. }
  238. br.Data = resp
  239. br.Ret = 200
  240. br.Success = true
  241. br.Msg = "获取成功"
  242. }
  243. // List
  244. // @Title 列表
  245. // @Description 列表
  246. // @Param request body data_manage.ThsHfIndexListForm true "type json string"
  247. // @Success 200 {object} data_manage.ThsHfIndexPageListResp
  248. // @router /ths_hf/index/list [get]
  249. func (this *BaseFromThsHfController) List() {
  250. br := new(models.BaseResponse).Init()
  251. defer func() {
  252. if br.ErrMsg == "" {
  253. br.IsSendEmail = false
  254. }
  255. this.Data["json"] = br
  256. this.ServeJSON()
  257. }()
  258. sysUser := this.SysUser
  259. if sysUser == nil {
  260. br.Msg = "请登录"
  261. br.ErrMsg = "请登录,SysUser Is Empty"
  262. br.Ret = 408
  263. return
  264. }
  265. var params data_manage.ThsHfIndexListForm
  266. if e := this.ParseForm(&params); e != nil {
  267. br.Msg = "参数解析失败"
  268. br.ErrMsg = fmt.Sprintf("参数解析失败, err: %v", e)
  269. return
  270. }
  271. if params.SortField > 0 && !utils.InArrayByInt([]int{1, 2, 3, 4}, params.SortField) {
  272. br.Msg = "参数有误"
  273. br.ErrMsg = fmt.Sprintf("参数有误, SortField: %d", params.SortField)
  274. return
  275. }
  276. if params.SortType > 0 && !utils.InArrayByInt([]int{1, 2}, params.SortType) {
  277. br.Msg = "参数有误"
  278. br.ErrMsg = fmt.Sprintf("参数有误, SortType: %d", params.SortType)
  279. return
  280. }
  281. resp := new(data_manage.ThsHfIndexPageListResp)
  282. resp.List = make([]*data_manage.BaseFromThsHfIndexItem, 0)
  283. // 查询所有分类-分类查询/分类完整路径用
  284. classifyOb := new(data_manage.BaseFromThsHfClassify)
  285. classifies := make([]*data_manage.BaseFromThsHfClassify, 0)
  286. classifyIdItem := make(map[int]*data_manage.BaseFromThsHfClassify)
  287. {
  288. list, e := classifyOb.GetItemsByCondition(``, make([]interface{}, 0), []string{classifyOb.Cols().PrimaryId, classifyOb.Cols().ClassifyName, classifyOb.Cols().ClassifyNameEn, classifyOb.Cols().LevelPath}, "")
  289. if e != nil {
  290. br.Msg = "获取失败"
  291. br.ErrMsg = fmt.Sprintf("获取分类列表失败, %v", e)
  292. return
  293. }
  294. for _, v := range list {
  295. classifyIdItem[v.BaseFromThsHfClassifyId] = v
  296. }
  297. classifies = list
  298. }
  299. // 筛选项
  300. var (
  301. cond string
  302. pars []interface{}
  303. listOrder string
  304. )
  305. indexOb := new(data_manage.BaseFromThsHfIndex)
  306. {
  307. // 分类
  308. if params.ClassifyId != "" {
  309. classifyIdArr := strings.Split(params.ClassifyId, ",")
  310. classifyIds := make([]int, 0)
  311. for _, v := range classifyIdArr {
  312. t, _ := strconv.Atoi(v)
  313. if t > 0 {
  314. classifyIds = append(classifyIds, t)
  315. }
  316. }
  317. // 不包含子分类
  318. if len(classifyIds) > 0 && !params.IncludeChild {
  319. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().BaseFromThsHfClassifyId, utils.GetOrmInReplace(len(classifyIds)))
  320. pars = append(pars, classifyIds)
  321. }
  322. // 包含子分类
  323. if len(classifyIds) > 0 && params.IncludeChild {
  324. queryClassifyIds := make([]int, 0)
  325. queryClassifyExist := make(map[int]bool)
  326. for _, v := range classifyIds {
  327. // 遍历所有分类从LevelPath中找含有查询分类ID的...=_=!
  328. for _, cv := range classifies {
  329. if queryClassifyExist[cv.BaseFromThsHfClassifyId] {
  330. continue
  331. }
  332. if cv.LevelPath == "" {
  333. continue
  334. }
  335. strArr := strings.Split(cv.LevelPath, ",")
  336. if len(strArr) == 0 {
  337. continue
  338. }
  339. for _, sv := range strArr {
  340. tv, _ := strconv.Atoi(sv)
  341. if tv == v {
  342. queryClassifyIds = append(queryClassifyIds, cv.BaseFromThsHfClassifyId)
  343. queryClassifyExist[cv.BaseFromThsHfClassifyId] = true
  344. break
  345. }
  346. }
  347. }
  348. }
  349. if len(queryClassifyIds) == 0 {
  350. page := paging.GetPaging(params.CurrentIndex, params.PageSize, 0)
  351. resp.Paging = page
  352. br.Ret = 200
  353. br.Success = true
  354. br.Msg = "获取成功"
  355. return
  356. }
  357. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().BaseFromThsHfClassifyId, utils.GetOrmInReplace(len(queryClassifyIds)))
  358. pars = append(pars, queryClassifyIds)
  359. }
  360. }
  361. if params.Frequency != "" {
  362. frequencyArr := strings.Split(params.Frequency, ",")
  363. if len(frequencyArr) > 0 {
  364. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().Frequency, utils.GetOrmInReplace(len(frequencyArr)))
  365. pars = append(pars, frequencyArr)
  366. }
  367. }
  368. if params.SysAdminId != "" {
  369. adminIdArr := strings.Split(params.SysAdminId, ",")
  370. adminIds := make([]int, 0)
  371. for _, v := range adminIdArr {
  372. t, _ := strconv.Atoi(v)
  373. if t > 0 {
  374. adminIds = append(adminIds, t)
  375. }
  376. }
  377. if len(adminIds) > 0 {
  378. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().SysUserId, utils.GetOrmInReplace(len(adminIds)))
  379. pars = append(pars, adminIds)
  380. }
  381. }
  382. params.Keywords = strings.TrimSpace(params.Keywords)
  383. if params.Keywords != "" {
  384. cond += fmt.Sprintf(" AND (%s LIKE ? OR %s LIKE ?)", indexOb.Cols().IndexCode, indexOb.Cols().IndexName)
  385. kw := fmt.Sprint("%", params.Keywords, "%")
  386. pars = append(pars, kw, kw)
  387. }
  388. // 排序
  389. if params.SortField > 0 && params.SortType > 0 {
  390. fieldMap := map[int]string{1: indexOb.Cols().StartDate, 2: indexOb.Cols().EndDate, 3: indexOb.Cols().ModifyTime, 4: indexOb.Cols().LatestValue}
  391. typeMap := map[int]string{1: "ASC", 2: "DESC"}
  392. listOrder = fmt.Sprintf("%s %s", fieldMap[params.SortField], typeMap[params.SortType])
  393. }
  394. }
  395. // 列表总计
  396. total, e := indexOb.GetCountByCondition(cond, pars)
  397. if e != nil {
  398. br.Msg = "获取失败"
  399. br.ErrMsg = fmt.Sprintf("获取指标总数失败, %v", e)
  400. return
  401. }
  402. if total <= 0 {
  403. page := paging.GetPaging(params.CurrentIndex, params.PageSize, 0)
  404. resp.Paging = page
  405. br.Data = resp
  406. br.Ret = 200
  407. br.Success = true
  408. br.Msg = "获取成功"
  409. return
  410. }
  411. // 分页查询
  412. var startSize int
  413. if params.PageSize <= 0 {
  414. params.PageSize = utils.PageSize20
  415. }
  416. if params.CurrentIndex <= 0 {
  417. params.CurrentIndex = 1
  418. }
  419. startSize = utils.StartIndex(params.CurrentIndex, params.PageSize)
  420. items, e := indexOb.GetPageItemsByCondition(cond, pars, []string{}, listOrder, startSize, params.PageSize)
  421. if e != nil {
  422. br.Msg = "获取失败"
  423. br.ErrMsg = fmt.Sprintf("获取指标列表失败, %v", e)
  424. return
  425. }
  426. for _, v := range items {
  427. t := v.Format2Item()
  428. // 分类完整路径
  429. classify := classifyIdItem[v.BaseFromThsHfClassifyId]
  430. levelArr := make([]string, 0)
  431. if classify != nil && classify.LevelPath != "" {
  432. arr := strings.Split(classify.LevelPath, ",")
  433. for _, a := range arr {
  434. i, _ := strconv.Atoi(a)
  435. if classifyIdItem[i] != nil {
  436. if this.Lang == utils.EnLangVersion {
  437. levelArr = append(levelArr, classifyIdItem[i].ClassifyNameEn)
  438. } else {
  439. levelArr = append(levelArr, classifyIdItem[i].ClassifyName)
  440. }
  441. }
  442. }
  443. }
  444. t.ClassifyPath = strings.Join(levelArr, "/")
  445. resp.List = append(resp.List, t)
  446. }
  447. page := paging.GetPaging(params.CurrentIndex, params.PageSize, total)
  448. resp.Paging = page
  449. br.Data = resp
  450. br.Ret = 200
  451. br.Success = true
  452. br.Msg = "获取成功"
  453. }
  454. // Add
  455. // @Title 新增指标
  456. // @Description 新增指标
  457. // @Param request body data_manage.ThsHfAddEdbReq true "type json string"
  458. // @Success 200 string "操作成功"
  459. // @router /ths_hf/index/add [post]
  460. func (this *BaseFromThsHfController) Add() {
  461. br := new(models.BaseResponse).Init()
  462. defer func() {
  463. if br.ErrMsg == "" {
  464. br.IsSendEmail = false
  465. }
  466. this.Data["json"] = br
  467. this.ServeJSON()
  468. }()
  469. sysUser := this.SysUser
  470. if sysUser == nil {
  471. br.Msg = "请登录"
  472. br.ErrMsg = "请登录,SysUser Is Empty"
  473. br.Ret = 408
  474. return
  475. }
  476. var params data_manage.ThsHfAddEdbReq
  477. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  478. br.Msg = "参数解析异常"
  479. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  480. return
  481. }
  482. if params.StartTime == "" {
  483. br.Msg = "请选择起始时间"
  484. return
  485. }
  486. _, e := time.ParseInLocation(utils.FormatDateTime, params.StartTime, time.Local)
  487. if e != nil {
  488. br.Msg = "起始时间格式有误"
  489. br.ErrMsg = fmt.Sprintf("起始时间格式有误, %v", e)
  490. return
  491. }
  492. if params.EndTime != "" {
  493. _, e := time.ParseInLocation(utils.FormatDateTime, params.EndTime, time.Local)
  494. if e != nil {
  495. br.Msg = "截止时间格式有误"
  496. br.ErrMsg = fmt.Sprintf("截止时间格式有误, %v", e)
  497. return
  498. }
  499. }
  500. if params.EndTime == "" {
  501. params.EndTime = time.Now().Local().Format(utils.FormatDateTime)
  502. }
  503. if !utils.InArrayByInt(data_manage.ThsHfPeriodArr, params.Interval) {
  504. br.Msg = "时间周期有误"
  505. br.ErrMsg = fmt.Sprintf("时间周期有误, Interval: %d", params.Interval)
  506. return
  507. }
  508. if params.CPS != "" && !utils.InArrayByStr(data_manage.ThsHfCPSArr, params.CPS) {
  509. br.Msg = "复权方式有误"
  510. br.ErrMsg = fmt.Sprintf("复权方式有误, CPS: %s", params.CPS)
  511. return
  512. }
  513. if params.BaseDate != "" {
  514. _, e = time.ParseInLocation(utils.FormatDate, params.BaseDate, time.Local)
  515. if e != nil {
  516. br.Msg = "复权基点格式有误"
  517. br.ErrMsg = fmt.Sprintf("复权基点格式有误, %v", e)
  518. return
  519. }
  520. }
  521. if params.Fill != "" && !utils.InArrayByStr(data_manage.ThsHfFillArr, params.Fill) {
  522. br.Msg = "非交易间隔处理有误"
  523. br.ErrMsg = fmt.Sprintf("非交易间隔处理有误, Fill: %s", params.Fill)
  524. return
  525. }
  526. if len(params.IndexList) == 0 {
  527. br.Msg = "请选择指标"
  528. return
  529. }
  530. formExist := make(map[string]bool)
  531. indexNames := make([]string, 0)
  532. classifyIds := make([]int, 0)
  533. for _, v := range params.IndexList {
  534. if v.ClassifyId <= 0 {
  535. br.Msg = "请选择分类"
  536. return
  537. }
  538. if !utils.InArrayByInt(classifyIds, v.ClassifyId) {
  539. classifyIds = append(classifyIds, v.ClassifyId)
  540. }
  541. if v.Unit == "" {
  542. br.Msg = "请输入单位"
  543. return
  544. }
  545. v.IndexName = strings.TrimSpace(v.IndexName)
  546. if v.IndexName == "" {
  547. br.Msg = "请输入指标名称"
  548. return
  549. }
  550. if formExist[v.IndexName] {
  551. br.Msg = "指标名称重复, 请重新输入"
  552. return
  553. }
  554. formExist[v.IndexName] = true
  555. indexNames = append(indexNames, v.IndexName)
  556. }
  557. // 校验指标分类
  558. classifyOb := new(data_manage.BaseFromThsHfClassify)
  559. {
  560. cond := fmt.Sprintf(` AND %s IN (%s)`, classifyOb.Cols().PrimaryId, utils.GetOrmInReplace(len(classifyIds)))
  561. pars := make([]interface{}, 0)
  562. pars = append(pars, classifyIds)
  563. list, e := classifyOb.GetItemsByCondition(cond, pars, []string{classifyOb.Cols().PrimaryId, classifyOb.Cols().ClassifyName}, "")
  564. if e != nil {
  565. br.Msg = "操作失败"
  566. br.ErrMsg = fmt.Sprintf("查询分类失败, %v", e)
  567. return
  568. }
  569. classifyMap := make(map[int]*data_manage.BaseFromThsHfClassify)
  570. for _, v := range list {
  571. classifyMap[v.BaseFromThsHfClassifyId] = v
  572. }
  573. for _, v := range params.IndexList {
  574. t := classifyMap[v.ClassifyId]
  575. if t == nil {
  576. br.Msg = fmt.Sprintf("%s分类不存在, 请重新选择或刷新页面", t.ClassifyName)
  577. return
  578. }
  579. }
  580. }
  581. // 校验指标名称
  582. indexOb := new(data_manage.BaseFromThsHfIndex)
  583. {
  584. cond := fmt.Sprintf(` AND %s IN (%s)`, indexOb.Cols().IndexName, utils.GetOrmInReplace(len(indexNames)))
  585. pars := make([]interface{}, 0)
  586. pars = append(pars, indexNames)
  587. list, e := indexOb.GetItemsByCondition(cond, pars, []string{indexOb.Cols().IndexName}, "")
  588. if e != nil {
  589. br.Msg = "操作失败"
  590. br.ErrMsg = fmt.Sprintf("查询指标名称失败, %v", e)
  591. return
  592. }
  593. if len(list) > 0 {
  594. tips := "以下指标名称重复, 请重新输入: \n"
  595. for _, v := range list {
  596. tips += fmt.Sprintf("%s\n", v.IndexName)
  597. }
  598. br.Msg = tips
  599. return
  600. }
  601. }
  602. // 新增指标
  603. for _, v := range params.IndexList {
  604. var req data_manage.ThsHfBaseAddReq
  605. req.StartTime = params.StartTime
  606. req.EndTime = params.EndTime
  607. req.Interval = params.Interval
  608. req.Fill = params.Fill
  609. req.CPS = params.CPS
  610. req.BaseDate = params.BaseDate
  611. req.SysAdminId = sysUser.AdminId
  612. req.SysAdminName = sysUser.RealName
  613. req.ClassifyId = v.ClassifyId
  614. req.Unit = v.Unit
  615. req.IndexName = v.IndexName
  616. req.Frequency = v.Frequency
  617. req.StockCode = v.StockCode
  618. req.EdbCode = v.EdbCode
  619. _, e = data.BaseAddThsHf(req)
  620. if e != nil {
  621. utils.FileLog.Info(fmt.Sprintf("BaseAddThsHf err: %v", e))
  622. continue
  623. }
  624. }
  625. br.Ret = 200
  626. br.Success = true
  627. br.Msg = "操作成功"
  628. }
  629. // Edit
  630. // @Title 编辑指标
  631. // @Description 编辑指标
  632. // @Param request body data_manage.ThsHfIndexEditReq true "type json string"
  633. // @Success 200 string "操作成功"
  634. // @router /ths_hf/index/edit [post]
  635. func (this *BaseFromThsHfController) Edit() {
  636. br := new(models.BaseResponse).Init()
  637. defer func() {
  638. if br.ErrMsg == "" {
  639. br.IsSendEmail = false
  640. }
  641. this.Data["json"] = br
  642. this.ServeJSON()
  643. }()
  644. sysUser := this.SysUser
  645. if sysUser == nil {
  646. br.Msg = "请登录"
  647. br.ErrMsg = "请登录,SysUser Is Empty"
  648. br.Ret = 408
  649. return
  650. }
  651. var params data_manage.ThsHfIndexEditReq
  652. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  653. br.Msg = "参数解析异常"
  654. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  655. return
  656. }
  657. if params.IndexId <= 0 {
  658. br.Msg = "参数有误"
  659. br.ErrMsg = fmt.Sprintf("参数有误, IndexId: %d", params.IndexId)
  660. return
  661. }
  662. params.IndexName = strings.TrimSpace(params.IndexName)
  663. if params.IndexName == "" {
  664. br.Msg = "请输入指标名称"
  665. return
  666. }
  667. if params.ClassifyId <= 0 {
  668. br.Msg = "请选择分类"
  669. return
  670. }
  671. params.Unit = strings.TrimSpace(params.Unit)
  672. if params.Unit == "" {
  673. br.Msg = "请输入单位"
  674. return
  675. }
  676. indexOb := new(data_manage.BaseFromThsHfIndex)
  677. item, e := indexOb.GetItemById(params.IndexId)
  678. if e != nil {
  679. if e.Error() == utils.ErrNoRow() {
  680. br.Msg = "指标不存在, 请刷新页面"
  681. return
  682. }
  683. br.Msg = "操作失败"
  684. br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  685. return
  686. }
  687. classifyOb := new(data_manage.BaseFromThsHfClassify)
  688. _, e = classifyOb.GetItemById(params.ClassifyId)
  689. if e != nil {
  690. br.Msg = "分类有误"
  691. br.ErrMsg = fmt.Sprintf("分类信息有误, %v", e)
  692. return
  693. }
  694. // 重名验证
  695. {
  696. cond := fmt.Sprintf(" AND %s = ? AND %s <> ?", indexOb.Cols().IndexName, indexOb.Cols().PrimaryId)
  697. pars := make([]interface{}, 0)
  698. pars = append(pars, params.IndexName, params.IndexId)
  699. count, e := indexOb.GetCountByCondition(cond, pars)
  700. if e != nil {
  701. br.Msg = "操作失败"
  702. br.ErrMsg = fmt.Sprintf("获取重名指标失败, %v", e)
  703. return
  704. }
  705. if count > 0 {
  706. br.Msg = "指标名称已存在, 请重新输入"
  707. return
  708. }
  709. }
  710. item.IndexName = params.IndexName
  711. item.BaseFromThsHfClassifyId = params.ClassifyId
  712. item.Unit = params.Unit
  713. item.ModifyTime = time.Now().Local()
  714. updateCols := []string{item.Cols().IndexName, item.Cols().BaseFromThsHfClassifyId, item.Cols().Unit, item.Cols().ModifyTime}
  715. if e = item.Update(updateCols); e != nil {
  716. br.Msg = "操作失败"
  717. br.ErrMsg = fmt.Sprintf("更新指标信息失败, %v", e)
  718. return
  719. }
  720. br.Ret = 200
  721. br.Success = true
  722. br.Msg = "操作成功"
  723. }
  724. // Detail
  725. // @Title 详情
  726. // @Description 详情
  727. // @Param IndexId query int true "指标ID"
  728. // @Param DataDate query string false "数据日期"
  729. // @Success 200 {object} data_manage.ThsHfSearchEdbResp
  730. // @router /ths_hf/index/detail [get]
  731. func (this *BaseFromThsHfController) Detail() {
  732. br := new(models.BaseResponse).Init()
  733. defer func() {
  734. if br.ErrMsg == "" {
  735. br.IsSendEmail = false
  736. }
  737. this.Data["json"] = br
  738. this.ServeJSON()
  739. }()
  740. sysUser := this.SysUser
  741. if sysUser == nil {
  742. br.Msg = "请登录"
  743. br.ErrMsg = "请登录,SysUser Is Empty"
  744. br.Ret = 408
  745. return
  746. }
  747. indexId, _ := this.GetInt("IndexId")
  748. if indexId <= 0 {
  749. br.Msg = "参数有误"
  750. br.ErrMsg = fmt.Sprintf("参数有误, IndexId: %d", indexId)
  751. return
  752. }
  753. dataDate := this.GetString("DataDate")
  754. if dataDate != "" {
  755. _, e := time.Parse(utils.FormatDate, dataDate)
  756. if e != nil {
  757. br.Msg = "数据日期格式有误"
  758. br.ErrMsg = fmt.Sprintf("数据日期格式有误, DataDate: %s", dataDate)
  759. return
  760. }
  761. }
  762. indexOb := new(data_manage.BaseFromThsHfIndex)
  763. item, e := indexOb.GetItemById(indexId)
  764. if e != nil {
  765. if e.Error() == utils.ErrNoRow() {
  766. br.Msg = "指标不存在, 请刷新页面"
  767. return
  768. }
  769. br.Msg = "获取失败"
  770. br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  771. return
  772. }
  773. indexItem := item.Format2Item()
  774. if item.EndDate.IsZero() {
  775. item.EndDate = time.Now()
  776. }
  777. // 默认取有数据的最新日期
  778. if dataDate == "" {
  779. dataDate = item.EndDate.Format(utils.FormatDate)
  780. }
  781. dataList := make([]*data_manage.BaseFromThsHfDataItem, 0)
  782. {
  783. dataOb := new(data_manage.BaseFromThsHfData)
  784. cond := fmt.Sprintf(" AND %s = ? AND (%s BETWEEN ? AND ?)", dataOb.Cols().IndexCode, dataOb.Cols().DataTime)
  785. pars := make([]interface{}, 0)
  786. pars = append(pars, indexItem.IndexCode, fmt.Sprintf("%s 00:00:00", dataDate), fmt.Sprintf("%s 23:59:59", dataDate))
  787. list, e := dataOb.GetItemsByCondition(cond, pars, []string{}, fmt.Sprintf("%s DESC", dataOb.Cols().DataTime))
  788. if e != nil {
  789. br.Msg = "获取失败"
  790. br.ErrMsg = fmt.Sprintf("获取指标数据失败, %v", e)
  791. return
  792. }
  793. for _, v := range list {
  794. dataList = append(dataList, v.Format2Item())
  795. }
  796. }
  797. type DetailResp struct {
  798. Index *data_manage.BaseFromThsHfIndexItem
  799. DataDate string `description:"数据日期"`
  800. DataList []*data_manage.BaseFromThsHfDataItem
  801. }
  802. resp := new(DetailResp)
  803. resp.Index = indexItem
  804. resp.DataDate = dataDate
  805. resp.DataList = dataList
  806. br.Data = resp
  807. br.Ret = 200
  808. br.Success = true
  809. br.Msg = "获取成功"
  810. }
  811. // Refresh
  812. // @Title 刷新指标
  813. // @Description 刷新指标
  814. // @Param request body data_manage.ThsHfIndexOptReq true "type json string"
  815. // @Success 200 string "操作成功"
  816. // @router /ths_hf/index/refresh [post]
  817. func (this *BaseFromThsHfController) Refresh() {
  818. br := new(models.BaseResponse).Init()
  819. defer func() {
  820. if br.ErrMsg == "" {
  821. br.IsSendEmail = false
  822. }
  823. this.Data["json"] = br
  824. this.ServeJSON()
  825. }()
  826. sysUser := this.SysUser
  827. if sysUser == nil {
  828. br.Msg = "请登录"
  829. br.ErrMsg = "请登录,SysUser Is Empty"
  830. br.Ret = 408
  831. return
  832. }
  833. var params data_manage.ThsHfIndexOptReq
  834. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  835. br.Msg = "参数解析异常"
  836. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  837. return
  838. }
  839. if params.IndexId <= 0 {
  840. br.Msg = "参数有误"
  841. br.ErrMsg = fmt.Sprintf("参数有误, IndexId: %d", params.IndexId)
  842. return
  843. }
  844. //indexOb := new(data_manage.BaseFromThsHfIndex)
  845. //item, e := indexOb.GetItemById(params.IndexId)
  846. //if e != nil {
  847. // if e.Error() == utils.ErrNoRow() {
  848. // br.Msg = "指标不存在, 请刷新页面"
  849. // return
  850. // }
  851. // br.Msg = "操作失败"
  852. // br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  853. // return
  854. //}
  855. // TODO:指标刷新逻辑
  856. br.Ret = 200
  857. br.Success = true
  858. br.Msg = "操作成功"
  859. }
  860. // Remove
  861. // @Title 删除指标
  862. // @Description 删除指标
  863. // @Param request body data_manage.ThsHfIndexOptReq true "type json string"
  864. // @Success 200 string "操作成功"
  865. // @router /ths_hf/index/remove [post]
  866. func (this *BaseFromThsHfController) Remove() {
  867. br := new(models.BaseResponse).Init()
  868. defer func() {
  869. if br.ErrMsg == "" {
  870. br.IsSendEmail = false
  871. }
  872. this.Data["json"] = br
  873. this.ServeJSON()
  874. }()
  875. sysUser := this.SysUser
  876. if sysUser == nil {
  877. br.Msg = "请登录"
  878. br.ErrMsg = "请登录,SysUser Is Empty"
  879. br.Ret = 408
  880. return
  881. }
  882. var params data_manage.ThsHfIndexOptReq
  883. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  884. br.Msg = "参数解析异常"
  885. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  886. return
  887. }
  888. if params.IndexId <= 0 {
  889. br.Msg = "参数有误"
  890. br.ErrMsg = fmt.Sprintf("参数有误, IndexId: %d", params.IndexId)
  891. return
  892. }
  893. indexOb := new(data_manage.BaseFromThsHfIndex)
  894. item, e := indexOb.GetItemById(params.IndexId)
  895. if e != nil {
  896. if e.Error() == utils.ErrNoRow() {
  897. br.Ret = 200
  898. br.Success = true
  899. br.Msg = "操作成功"
  900. return
  901. }
  902. br.Msg = "操作失败"
  903. br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  904. return
  905. }
  906. // TODO:删除校验, 生成了指标就不可删除
  907. // TODO:同步Mongo
  908. if e = item.Remove(); e != nil {
  909. br.Msg = "操作失败"
  910. br.ErrMsg = fmt.Sprintf("删除指标失败, %v", e)
  911. return
  912. }
  913. dataOb := new(data_manage.BaseFromThsHfData)
  914. cond := fmt.Sprintf(" %s = ?", dataOb.Cols().BaseFromThsHfIndexId)
  915. pars := make([]interface{}, 0)
  916. pars = append(pars, params.IndexId)
  917. if e = dataOb.RemoveByCondition(cond, pars); e != nil {
  918. br.Msg = "操作失败"
  919. br.ErrMsg = fmt.Sprintf("删除指标数据失败, %v", e)
  920. return
  921. }
  922. br.Ret = 200
  923. br.Success = true
  924. br.Msg = "操作成功"
  925. }
  926. // ListChoice
  927. // @Title 列表选择
  928. // @Description 列表选择
  929. // @Param request body data_manage.ThsHfIndexMultiOptReq true "type json string"
  930. // @Success 200 string "获取成功"
  931. // @router /ths_hf/index/list_choice [get]
  932. func (this *BaseFromThsHfController) ListChoice() {
  933. br := new(models.BaseResponse).Init()
  934. defer func() {
  935. if br.ErrMsg == "" {
  936. br.IsSendEmail = false
  937. }
  938. this.Data["json"] = br
  939. this.ServeJSON()
  940. }()
  941. sysUser := this.SysUser
  942. if sysUser == nil {
  943. br.Msg = "请登录"
  944. br.ErrMsg = "请登录,SysUser Is Empty"
  945. br.Ret = 408
  946. return
  947. }
  948. var params data_manage.ThsHfIndexListChoiceReq
  949. if e := this.ParseForm(&params); e != nil {
  950. br.Msg = "参数解析失败"
  951. br.ErrMsg = fmt.Sprintf("参数解析失败, err: %v", e)
  952. return
  953. }
  954. var listIds []int
  955. if params.ListIds != "" {
  956. strArr := strings.Split(params.ListIds, ",")
  957. for _, v := range strArr {
  958. t, _ := strconv.Atoi(v)
  959. if t > 0 {
  960. listIds = append(listIds, t)
  961. }
  962. }
  963. }
  964. var (
  965. cond string
  966. pars []interface{}
  967. )
  968. indexOb := new(data_manage.BaseFromThsHfIndex)
  969. resp := make([]*data_manage.ThsHfIndexListChoiceItem, 0)
  970. // 非列表全选
  971. if !params.SelectAll {
  972. if len(listIds) > 0 {
  973. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().PrimaryId, utils.GetOrmInReplace(len(listIds)))
  974. pars = append(pars, listIds)
  975. }
  976. }
  977. // 列表全选, 根据条件筛选列表并过滤params.ListIds中选择的ID
  978. if params.SelectAll {
  979. // 查询所有分类-分类查询用
  980. classifyOb := new(data_manage.BaseFromThsHfClassify)
  981. classifies := make([]*data_manage.BaseFromThsHfClassify, 0)
  982. classifyIdItem := make(map[int]*data_manage.BaseFromThsHfClassify)
  983. {
  984. list, e := classifyOb.GetItemsByCondition(``, make([]interface{}, 0), []string{classifyOb.Cols().PrimaryId, classifyOb.Cols().ClassifyName, classifyOb.Cols().ClassifyNameEn, classifyOb.Cols().LevelPath}, "")
  985. if e != nil {
  986. br.Msg = "获取失败"
  987. br.ErrMsg = fmt.Sprintf("获取分类列表失败, %v", e)
  988. return
  989. }
  990. for _, v := range list {
  991. classifyIdItem[v.BaseFromThsHfClassifyId] = v
  992. }
  993. classifies = list
  994. }
  995. // 筛选项
  996. if params.ClassifyId != "" {
  997. classifyIdArr := strings.Split(params.ClassifyId, ",")
  998. classifyIds := make([]int, 0)
  999. for _, v := range classifyIdArr {
  1000. t, _ := strconv.Atoi(v)
  1001. if t > 0 {
  1002. classifyIds = append(classifyIds, t)
  1003. }
  1004. }
  1005. // 不包含子分类
  1006. if len(classifyIds) > 0 && !params.IncludeChild {
  1007. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().BaseFromThsHfClassifyId, utils.GetOrmInReplace(len(classifyIds)))
  1008. pars = append(pars, classifyIds)
  1009. }
  1010. // 包含子分类
  1011. if len(classifyIds) > 0 && params.IncludeChild {
  1012. queryClassifyIds := make([]int, 0)
  1013. queryClassifyExist := make(map[int]bool)
  1014. for _, v := range classifyIds {
  1015. // 遍历所有分类从LevelPath中找含有查询分类ID的...=_=!
  1016. for _, cv := range classifies {
  1017. if queryClassifyExist[cv.BaseFromThsHfClassifyId] {
  1018. continue
  1019. }
  1020. if cv.LevelPath == "" {
  1021. continue
  1022. }
  1023. strArr := strings.Split(cv.LevelPath, ",")
  1024. if len(strArr) == 0 {
  1025. continue
  1026. }
  1027. for _, sv := range strArr {
  1028. tv, _ := strconv.Atoi(sv)
  1029. if tv == v {
  1030. queryClassifyIds = append(queryClassifyIds, cv.BaseFromThsHfClassifyId)
  1031. queryClassifyExist[cv.BaseFromThsHfClassifyId] = true
  1032. break
  1033. }
  1034. }
  1035. }
  1036. }
  1037. if len(queryClassifyIds) == 0 {
  1038. br.Data = resp
  1039. br.Ret = 200
  1040. br.Success = true
  1041. br.Msg = "获取成功"
  1042. return
  1043. }
  1044. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().BaseFromThsHfClassifyId, utils.GetOrmInReplace(len(queryClassifyIds)))
  1045. pars = append(pars, queryClassifyIds)
  1046. }
  1047. }
  1048. if params.Frequency != "" {
  1049. frequencyArr := strings.Split(params.Frequency, ",")
  1050. if len(frequencyArr) > 0 {
  1051. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().Frequency, utils.GetOrmInReplace(len(frequencyArr)))
  1052. pars = append(pars, frequencyArr)
  1053. }
  1054. }
  1055. if params.SysAdminId != "" {
  1056. adminIdArr := strings.Split(params.SysAdminId, ",")
  1057. adminIds := make([]int, 0)
  1058. for _, v := range adminIdArr {
  1059. t, _ := strconv.Atoi(v)
  1060. if t > 0 {
  1061. adminIds = append(adminIds, t)
  1062. }
  1063. }
  1064. if len(adminIds) > 0 {
  1065. cond += fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().SysUserId, utils.GetOrmInReplace(len(adminIds)))
  1066. pars = append(pars, adminIds)
  1067. }
  1068. }
  1069. params.Keywords = strings.TrimSpace(params.Keywords)
  1070. if params.Keywords != "" {
  1071. cond += fmt.Sprintf(" AND (%s LIKE ? OR %s LIKE ?)", indexOb.Cols().IndexCode, indexOb.Cols().IndexName)
  1072. kw := fmt.Sprint("%", params.Keywords, "%")
  1073. pars = append(pars, kw, kw)
  1074. }
  1075. // 过滤掉选择的指标
  1076. if len(listIds) > 0 {
  1077. cond += fmt.Sprintf(" AND %s NOT IN (%s)", indexOb.Cols().PrimaryId, utils.GetOrmInReplace(len(listIds)))
  1078. pars = append(pars, listIds)
  1079. }
  1080. }
  1081. fields := []string{indexOb.Cols().PrimaryId, indexOb.Cols().IndexCode, indexOb.Cols().IndexName}
  1082. list, e := indexOb.GetItemsByCondition(cond, pars, fields, "")
  1083. if e != nil {
  1084. br.Msg = "获取失败"
  1085. br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  1086. return
  1087. }
  1088. if len(list) > 500 {
  1089. br.Msg = "选择指标超过500个, 请重新选择"
  1090. return
  1091. }
  1092. for _, v := range list {
  1093. resp = append(resp, &data_manage.ThsHfIndexListChoiceItem{
  1094. IndexId: v.BaseFromThsHfIndexId,
  1095. IndexCode: v.IndexCode,
  1096. IndexName: v.IndexName,
  1097. })
  1098. }
  1099. br.Data = resp
  1100. br.Ret = 200
  1101. br.Success = true
  1102. br.Msg = "获取成功"
  1103. }
  1104. // MultiOpt
  1105. // @Title 批量操作-移动分类/删除/刷新
  1106. // @Description 批量操作-移动分类/删除/刷新
  1107. // @Param request body data_manage.ThsHfIndexMultiOptReq true "type json string"
  1108. // @Success 200 string "操作成功"
  1109. // @router /ths_hf/index/multi_opt [post]
  1110. func (this *BaseFromThsHfController) MultiOpt() {
  1111. br := new(models.BaseResponse).Init()
  1112. defer func() {
  1113. if br.ErrMsg == "" {
  1114. br.IsSendEmail = false
  1115. }
  1116. this.Data["json"] = br
  1117. this.ServeJSON()
  1118. }()
  1119. sysUser := this.SysUser
  1120. if sysUser == nil {
  1121. br.Msg = "请登录"
  1122. br.ErrMsg = "请登录,SysUser Is Empty"
  1123. br.Ret = 408
  1124. return
  1125. }
  1126. var params data_manage.ThsHfIndexMultiOptReq
  1127. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  1128. br.Msg = "参数解析异常"
  1129. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  1130. return
  1131. }
  1132. if !utils.InArrayByInt([]int{1, 2, 3}, params.OptType) {
  1133. br.Msg = "请选择操作类型"
  1134. return
  1135. }
  1136. if len(params.IndexIds) == 0 {
  1137. br.Msg = "请选择指标"
  1138. return
  1139. }
  1140. if params.OptType == 1 && params.MoveClassifyId <= 0 {
  1141. br.Msg = "请选择分类"
  1142. return
  1143. }
  1144. if params.OptType == 3 && !utils.InArrayByInt([]int{1, 2}, params.RefreshType) {
  1145. br.Msg = "请选择刷新方式"
  1146. return
  1147. }
  1148. // 批量移动
  1149. indexOb := new(data_manage.BaseFromThsHfIndex)
  1150. if params.OptType == 1 {
  1151. if e := indexOb.UpdateClassifyMulti(params.IndexIds, params.MoveClassifyId); e != nil {
  1152. br.Msg = "移动失败"
  1153. br.ErrMsg = fmt.Sprintf("批量移动分类失败, %v", e)
  1154. return
  1155. }
  1156. }
  1157. // 批量删除
  1158. dataOb := new(data_manage.BaseFromThsHfData)
  1159. if params.OptType == 2 {
  1160. // TODO: 删除前的校验
  1161. if e := indexOb.MultiRemove(params.IndexIds); e != nil {
  1162. br.Msg = "删除失败"
  1163. br.ErrMsg = fmt.Sprintf("批量删除指标失败, %v", e)
  1164. return
  1165. }
  1166. cond := fmt.Sprintf(" %s IN (%s)", dataOb.Cols().BaseFromThsHfIndexId, utils.GetOrmInReplace(len(params.IndexIds)))
  1167. pars := make([]interface{}, 0)
  1168. pars = append(pars, params.IndexIds)
  1169. if e := dataOb.RemoveByCondition(cond, pars); e != nil {
  1170. br.Msg = "删除失败"
  1171. br.ErrMsg = fmt.Sprintf("批量删除指标数据失败, %v", e)
  1172. return
  1173. }
  1174. }
  1175. // TODO:批量刷新
  1176. if params.OptType == 3 {
  1177. }
  1178. br.Ret = 200
  1179. br.Success = true
  1180. br.Msg = "操作成功"
  1181. }
  1182. // Save2EdbPre
  1183. // @Title 批量新增指标库-前置
  1184. // @Description 批量新增指标库-前置
  1185. // @Param request body data_manage.ThsHfIndexMultiSave2EdbReq true "type json string"
  1186. // @Success 200 string "操作成功"
  1187. // @router /ths_hf/index/save2edb_pre [post]
  1188. func (this *BaseFromThsHfController) Save2EdbPre() {
  1189. br := new(models.BaseResponse).Init()
  1190. defer func() {
  1191. if br.ErrMsg == "" {
  1192. br.IsSendEmail = false
  1193. }
  1194. this.Data["json"] = br
  1195. this.ServeJSON()
  1196. }()
  1197. sysUser := this.SysUser
  1198. if sysUser == nil {
  1199. br.Msg = "请登录"
  1200. br.ErrMsg = "请登录,SysUser Is Empty"
  1201. br.Ret = 408
  1202. return
  1203. }
  1204. var params data_manage.ThsHfIndexMultiSave2EdbPreReq
  1205. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  1206. br.Msg = "参数解析异常"
  1207. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  1208. return
  1209. }
  1210. if len(params.IndexIds) == 0 {
  1211. br.Msg = "请选择指标"
  1212. return
  1213. }
  1214. convertRule := params.ConvertRule
  1215. if convertRule.ConvertType != 1 && convertRule.ConvertType != 2 {
  1216. br.Msg = "请选择数据转换方式"
  1217. return
  1218. }
  1219. // 生成的指标名称后缀
  1220. var (
  1221. suffixName string
  1222. startTimeCn string
  1223. endTimeCn string
  1224. calculateCn string
  1225. )
  1226. dayMap := map[int]string{1: "当日", 2: "前日"}
  1227. calculateMap := map[int]string{1: "均值", 2: "最大值", 3: "最小值"}
  1228. if convertRule.ConvertType == 1 {
  1229. if convertRule.ConvertFixed.FixedDay != 1 && convertRule.ConvertFixed.FixedDay != 2 {
  1230. br.Msg = "请选择指定时间"
  1231. return
  1232. }
  1233. if convertRule.ConvertFixed.FixedTime == "" {
  1234. br.Msg = "请选择指定时间"
  1235. return
  1236. }
  1237. timePrefix := time.Now().Local().Format(utils.FormatDate)
  1238. st := fmt.Sprintf("%s %s", timePrefix, convertRule.ConvertFixed.FixedTime)
  1239. startTime, e := time.Parse(utils.FormatDateTime, st)
  1240. if e != nil {
  1241. br.Msg = "指定时间格式有误"
  1242. return
  1243. }
  1244. startTimeCn = fmt.Sprintf("%s%s", dayMap[convertRule.ConvertFixed.FixedDay], startTime.Format("15点04"))
  1245. }
  1246. if convertRule.ConvertType == 2 {
  1247. if convertRule.ConvertArea.StartDay != 1 && convertRule.ConvertArea.StartDay != 2 {
  1248. br.Msg = "请选择起始时间"
  1249. return
  1250. }
  1251. if convertRule.ConvertArea.StartTime == "" {
  1252. br.Msg = "请选择起始时间"
  1253. return
  1254. }
  1255. var startTimePre string
  1256. if convertRule.ConvertArea.StartDay == 1 {
  1257. startTimePre = time.Now().Local().Format(utils.FormatDate)
  1258. }
  1259. if convertRule.ConvertArea.StartDay == 2 {
  1260. startTimePre = time.Now().Local().AddDate(0, 0, -1).Format(utils.FormatDate)
  1261. }
  1262. st := fmt.Sprintf("%s %s", startTimePre, convertRule.ConvertArea.StartTime)
  1263. startTime, e := time.Parse(utils.FormatDateTime, st)
  1264. if e != nil {
  1265. br.Msg = "起始时间格式有误"
  1266. return
  1267. }
  1268. startTimeCn = fmt.Sprintf("%s%s", dayMap[convertRule.ConvertArea.StartDay], startTime.Format("15点04"))
  1269. if convertRule.ConvertArea.EndDay != 1 && convertRule.ConvertArea.EndDay != 2 {
  1270. br.Msg = "请选择截止时间"
  1271. return
  1272. }
  1273. if convertRule.ConvertArea.EndTime == "" {
  1274. br.Msg = "请选择截止时间"
  1275. return
  1276. }
  1277. var endTimePre string
  1278. if convertRule.ConvertArea.EndDay == 1 {
  1279. endTimePre = time.Now().Local().Format(utils.FormatDate)
  1280. }
  1281. if convertRule.ConvertArea.EndDay == 2 {
  1282. endTimePre = time.Now().Local().AddDate(0, 0, -1).Format(utils.FormatDate)
  1283. }
  1284. ed := fmt.Sprintf("%s %s", endTimePre, convertRule.ConvertArea.EndTime)
  1285. endTime, e := time.Parse(utils.FormatDateTime, ed)
  1286. if e != nil {
  1287. br.Msg = "截止时间格式有误"
  1288. return
  1289. }
  1290. endTimeCn = fmt.Sprintf("至%s%s", dayMap[convertRule.ConvertArea.EndDay], endTime.Format("15点04"))
  1291. if startTime.After(endTime) {
  1292. br.Msg = "起始日期不可早于截止日期"
  1293. return
  1294. }
  1295. calculateCn = calculateMap[convertRule.ConvertArea.CalculateType]
  1296. }
  1297. suffixName = fmt.Sprint(startTimeCn, endTimeCn, calculateCn)
  1298. // 查询指标
  1299. indexOb := new(data_manage.BaseFromThsHfIndex)
  1300. cond := fmt.Sprintf(" AND %s IN (%s)", indexOb.Cols().PrimaryId, utils.GetOrmInReplace(len(params.IndexIds)))
  1301. pars := make([]interface{}, 0)
  1302. pars = append(pars, params.IndexIds)
  1303. list, e := indexOb.GetItemsByCondition(cond, pars, []string{}, "")
  1304. if e != nil {
  1305. br.Msg = "获取失败"
  1306. br.ErrMsg = fmt.Sprintf("获取指标失败, %v", e)
  1307. return
  1308. }
  1309. resp := make([]*data_manage.ThsHfIndexMultiSave2EdbPreItem, 0)
  1310. for _, v := range list {
  1311. t := new(data_manage.ThsHfIndexMultiSave2EdbPreItem)
  1312. t.IndexId = v.BaseFromThsHfIndexId
  1313. t.IndexCode = v.IndexCode
  1314. t.IndexName = v.IndexName
  1315. edbCn := data_manage.ThsHfEdbCodeCn[v.Indicator]
  1316. if edbCn == "" {
  1317. edbCn = v.Indicator
  1318. }
  1319. t.NewIndexName = fmt.Sprint(v.StockCode, v.Frequency, edbCn, suffixName)
  1320. t.StockCode = v.StockCode
  1321. t.EdbCode = v.Indicator
  1322. t.Unit = v.Unit
  1323. t.Frequency = v.Frequency
  1324. t.NewFrequency = "日度"
  1325. resp = append(resp, t)
  1326. }
  1327. br.Data = resp
  1328. br.Ret = 200
  1329. br.Success = true
  1330. br.Msg = "获取成功"
  1331. }
  1332. // Save2Edb
  1333. // @Title 批量新增指标库
  1334. // @Description 批量新增指标库
  1335. // @Param request body data_manage.ThsHfIndexMultiSave2EdbReq true "type json string"
  1336. // @Success 200 string "操作成功"
  1337. // @router /ths_hf/index/save2edb [post]
  1338. func (this *BaseFromThsHfController) Save2Edb() {
  1339. br := new(models.BaseResponse).Init()
  1340. defer func() {
  1341. if br.ErrMsg == "" {
  1342. br.IsSendEmail = false
  1343. }
  1344. this.Data["json"] = br
  1345. this.ServeJSON()
  1346. }()
  1347. sysUser := this.SysUser
  1348. if sysUser == nil {
  1349. br.Msg = "请登录"
  1350. br.ErrMsg = "请登录,SysUser Is Empty"
  1351. br.Ret = 408
  1352. return
  1353. }
  1354. var params data_manage.ThsHfIndexMultiSave2EdbReq
  1355. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &params); e != nil {
  1356. br.Msg = "参数解析异常"
  1357. br.ErrMsg = fmt.Sprintf("参数解析异常, %v", e)
  1358. return
  1359. }
  1360. if len(params.NewIndexes) == 0 {
  1361. br.Msg = "请选择指标"
  1362. return
  1363. }
  1364. convertRule := params.ConvertRule
  1365. if convertRule.ConvertType != 1 && convertRule.ConvertType != 2 {
  1366. br.Msg = "请选择数据转换方式"
  1367. return
  1368. }
  1369. if convertRule.ConvertType == 1 {
  1370. if convertRule.ConvertFixed.FixedDay != 1 && convertRule.ConvertFixed.FixedDay != 2 {
  1371. br.Msg = "请选择指定时间"
  1372. return
  1373. }
  1374. if convertRule.ConvertFixed.FixedTime == "" {
  1375. br.Msg = "请选择指定时间"
  1376. return
  1377. }
  1378. timePrefix := time.Now().Local().Format(utils.FormatDate)
  1379. st := fmt.Sprintf("%s %s", timePrefix, convertRule.ConvertFixed.FixedTime)
  1380. _, e := time.Parse(utils.FormatDateTime, st)
  1381. if e != nil {
  1382. br.Msg = "指定时间格式有误"
  1383. return
  1384. }
  1385. }
  1386. if convertRule.ConvertType == 2 {
  1387. if convertRule.ConvertArea.StartDay != 1 && convertRule.ConvertArea.StartDay != 2 {
  1388. br.Msg = "请选择起始时间"
  1389. return
  1390. }
  1391. if convertRule.ConvertArea.StartTime == "" {
  1392. br.Msg = "请选择起始时间"
  1393. return
  1394. }
  1395. var startTimePre string
  1396. if convertRule.ConvertArea.StartDay == 1 {
  1397. startTimePre = time.Now().Local().Format(utils.FormatDate)
  1398. }
  1399. if convertRule.ConvertArea.StartDay == 2 {
  1400. startTimePre = time.Now().Local().AddDate(0, 0, -1).Format(utils.FormatDate)
  1401. }
  1402. st := fmt.Sprintf("%s %s", startTimePre, convertRule.ConvertArea.StartTime)
  1403. startTime, e := time.Parse(utils.FormatDateTime, st)
  1404. if e != nil {
  1405. br.Msg = "起始时间格式有误"
  1406. return
  1407. }
  1408. if convertRule.ConvertArea.EndDay != 1 && convertRule.ConvertArea.EndDay != 2 {
  1409. br.Msg = "请选择截止时间"
  1410. return
  1411. }
  1412. if convertRule.ConvertArea.EndTime == "" {
  1413. br.Msg = "请选择截止时间"
  1414. return
  1415. }
  1416. var endTimePre string
  1417. if convertRule.ConvertArea.EndDay == 1 {
  1418. endTimePre = time.Now().Local().Format(utils.FormatDate)
  1419. }
  1420. if convertRule.ConvertArea.EndDay == 2 {
  1421. endTimePre = time.Now().Local().AddDate(0, 0, -1).Format(utils.FormatDate)
  1422. }
  1423. ed := fmt.Sprintf("%s %s", endTimePre, convertRule.ConvertArea.EndTime)
  1424. endTime, e := time.Parse(utils.FormatDateTime, ed)
  1425. if e != nil {
  1426. br.Msg = "截止时间格式有误"
  1427. return
  1428. }
  1429. if startTime.After(endTime) {
  1430. br.Msg = "起始日期不可早于截止日期"
  1431. return
  1432. }
  1433. }
  1434. resp := new(data_manage.ThsHfIndexMultiSave2EdbResp)
  1435. // 判断指标名称是否重复
  1436. edbNameExist := make(map[string]bool)
  1437. {
  1438. items, e := data_manage.GetEdbInfoFieldList(``, make([]interface{}, 0), []string{"edb_name"})
  1439. if e != nil {
  1440. br.Msg = "操作失败"
  1441. br.ErrMsg = fmt.Sprintf("获取指标名称列表失败, %v", e)
  1442. return
  1443. }
  1444. for _, v := range items {
  1445. edbNameExist[v.EdbName] = true
  1446. }
  1447. }
  1448. for _, v := range params.NewIndexes {
  1449. if edbNameExist[v.NewIndexName] {
  1450. resp.Exist = append(resp.Exist, v)
  1451. }
  1452. }
  1453. if len(resp.Exist) > 0 {
  1454. br.Data = resp
  1455. br.Ret = 200
  1456. br.Success = true
  1457. br.Msg = "含重复指标名称"
  1458. return
  1459. }
  1460. // 请求指标库新增
  1461. for _, v := range params.NewIndexes {
  1462. var r data_manage.ThsHfIndexMultiSave2EdbLibReq
  1463. r.ConvertRule = convertRule
  1464. v.SysAdminId = sysUser.AdminId
  1465. v.SysAdminName = sysUser.RealName
  1466. r.NewIndex = v
  1467. b, e := json.Marshal(r)
  1468. if e != nil {
  1469. v.Tips = "新增失败"
  1470. v.ErrMsg = e.Error()
  1471. resp.Fail = append(resp.Fail, v)
  1472. continue
  1473. }
  1474. res, e := data.AddBaseEdbInfo(string(b), utils.DATA_SOURCE_THS, utils.DATA_SUB_SOURCE_HIGH_FREQUENCY, this.Lang)
  1475. if e != nil {
  1476. v.Tips = res.Msg
  1477. v.ErrMsg = e.Error()
  1478. resp.Fail = append(resp.Fail, v)
  1479. continue
  1480. }
  1481. if res.Ret != 200 {
  1482. v.Tips = res.Msg
  1483. v.ErrMsg = fmt.Sprintf("新增失败, ErrMsg: %s", res.ErrMsg)
  1484. resp.Fail = append(resp.Fail, v)
  1485. continue
  1486. }
  1487. v.Tips = "新增成功"
  1488. resp.Success = append(resp.Success, v)
  1489. }
  1490. br.Data = resp
  1491. br.Ret = 200
  1492. br.Success = true
  1493. br.Msg = "操作成功"
  1494. }