report.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077
  1. package controllers
  2. import (
  3. "eta/eta_mini_bridge/models"
  4. "eta/eta_mini_bridge/models/response"
  5. "eta/eta_mini_bridge/services"
  6. "eta/eta_mini_bridge/utils"
  7. "html"
  8. "strconv"
  9. "time"
  10. "github.com/rdlucklib/rdluck_tools/paging"
  11. )
  12. type ReportController struct {
  13. BaseAuthController
  14. }
  15. type ReportNoAuthController struct {
  16. BaseCommonController
  17. }
  18. // @Title List
  19. // @Description create users
  20. // @Param ChartPermissionId query int true "品种ID"
  21. // @Param Level query int true "品种层级"
  22. // @Param PageSize query int true "每页数据条数"
  23. // @Param CurrentIndex query int true "当前页页码,从1开始"
  24. // @Param RangeType query string true "范围类型,1-一天内,2-一周内,3-半年内"
  25. // @Param ClassifyId query int true "分类id"
  26. // @Success 200 {object} response.ReportListResp
  27. // @router /list [get]
  28. func (this *ReportController) List() {
  29. br := new(models.BaseResponse).Init()
  30. defer func() {
  31. this.Data["json"] = br
  32. this.ServeJSON()
  33. }()
  34. pageSize, _ := this.GetInt("PageSize")
  35. currentIndex, _ := this.GetInt("CurrentIndex")
  36. chartPermissionId, _ := this.GetInt("ChartPermissionId")
  37. level, _ := this.GetInt("Level")
  38. rangeType, _ := this.GetInt("RangeType")
  39. classifyId, _ := this.GetInt("ClassifyId")
  40. if chartPermissionId <= 0 {
  41. br.Msg = "品种参数错误"
  42. return
  43. }
  44. var condition string
  45. switch rangeType {
  46. case 1:
  47. condition += ` AND DATE(a.modify_time)=DATE(NOW()) `
  48. case 2:
  49. condition += ` AND DATE(a.modify_time) BETWEEN DATE_SUB(NOW(),INTERVAL 1 WEEK) AND NOW() `
  50. case 3:
  51. condition += ` AND DATE(a.modify_time) BETWEEN DATE_SUB(NOW(),INTERVAL 6 MONTH) AND NOW() `
  52. }
  53. var startSize int
  54. if pageSize <= 0 {
  55. pageSize = utils.PageSize20
  56. }
  57. if currentIndex <= 0 {
  58. currentIndex = 1
  59. }
  60. startSize = utils.StartIndex(currentIndex, pageSize)
  61. classifyList, err := models.GetClassifyList()
  62. if err != nil {
  63. br.Msg = "获取数据失败"
  64. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  65. return
  66. }
  67. classifyMap := make(map[int]*models.ClassifyView)
  68. isHas := false
  69. for _, v := range classifyList {
  70. if v.Id == classifyId && classifyId != 0 {
  71. isHas = true
  72. }
  73. classifyMap[v.Id] = v
  74. }
  75. if !isHas && classifyId != 0 {
  76. br.Msg = "分类不存在"
  77. return
  78. }
  79. classifyIdMap := make(map[int]struct{})
  80. if classifyId > 0 {
  81. leafClassifyIds := getLeafClassifyIds(classifyMap, classifyId)
  82. for _, v := range leafClassifyIds {
  83. classifyIdMap[v] = struct{}{}
  84. }
  85. }
  86. var total int
  87. var reportList []*models.ReportList
  88. classifyFirstIds := make([]int, 0)
  89. classifySecondIds := make([]int, 0)
  90. classifyThirdIds := make([]int, 0)
  91. switch level {
  92. case 2:
  93. classifyIds, err := models.GetClassifyIdsListById(chartPermissionId)
  94. if err != nil {
  95. br.Msg = "获取报告列表失败"
  96. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  97. return
  98. }
  99. if len(classifyIds) == 0 {
  100. br.Msg = "该品种下没有绑定分类"
  101. br.ErrMsg = "获取数据失败,品种id:" + strconv.Itoa(chartPermissionId)
  102. return
  103. }
  104. classifyIds = utils.Unique(classifyIds)
  105. // 根据分类筛选报告
  106. var selectedClassifyId []int
  107. if classifyId > 0 {
  108. for _, v := range classifyIds {
  109. if _, ok := classifyIdMap[v]; ok {
  110. selectedClassifyId = append(selectedClassifyId, v)
  111. }
  112. }
  113. } else {
  114. selectedClassifyId = classifyIds
  115. }
  116. classifyList, err := models.GetClassifyListByIds(selectedClassifyId)
  117. if err != nil {
  118. br.Msg = "获取报告列表失败"
  119. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  120. return
  121. }
  122. for _, v := range classifyList {
  123. if v.HasChild == 1 {
  124. continue
  125. }
  126. switch v.Level {
  127. case 1:
  128. classifyFirstIds = append(classifyFirstIds, v.Id)
  129. case 2:
  130. classifySecondIds = append(classifySecondIds, v.Id)
  131. case 3:
  132. classifyThirdIds = append(classifyThirdIds, v.Id)
  133. }
  134. }
  135. tmptotal, err := models.GetReportCountByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition)
  136. if err != nil {
  137. br.Msg = "获取数据失败"
  138. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  139. return
  140. }
  141. total = tmptotal
  142. tmpReportList, err := models.GetReportListByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition, startSize, pageSize)
  143. if err != nil {
  144. br.Msg = "获取报告列表失败"
  145. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  146. return
  147. }
  148. reportList = tmpReportList
  149. case 1:
  150. chartPermissionIds, err := models.GetChildChartPermissionIdsById(chartPermissionId)
  151. if err != nil {
  152. br.Msg = "获取报告列表失败"
  153. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  154. return
  155. }
  156. if len(chartPermissionIds) == 0 {
  157. br.Msg = "该品种下没有绑定分类"
  158. br.ErrMsg = "获取数据失败,品种id:" + strconv.Itoa(chartPermissionId)
  159. return
  160. }
  161. classifyIds, err := models.GetClassifyIdsListByIds(chartPermissionIds)
  162. if err != nil {
  163. br.Msg = "获取报告列表失败"
  164. br.ErrMsg = "获取报告列表失败,Err:" + err.Error()
  165. return
  166. }
  167. if len(classifyIds) == 0 {
  168. br.Msg = "该品种下没有绑定分类"
  169. return
  170. }
  171. classifyIds = utils.Unique(classifyIds)
  172. // 根据分类筛选报告
  173. var selectedClassifyId []int
  174. if classifyId > 0 {
  175. for _, v := range classifyIds {
  176. if _, ok := classifyIdMap[v]; ok {
  177. selectedClassifyId = append(selectedClassifyId, v)
  178. }
  179. }
  180. } else {
  181. selectedClassifyId = classifyIds
  182. }
  183. classifyList, err := models.GetClassifyListByIds(selectedClassifyId)
  184. if err != nil {
  185. br.Msg = "获取报告列表失败"
  186. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  187. return
  188. }
  189. for _, v := range classifyList {
  190. if v.HasChild == 1 {
  191. continue
  192. }
  193. switch v.Level {
  194. case 1:
  195. classifyFirstIds = append(classifyFirstIds, v.Id)
  196. case 2:
  197. classifySecondIds = append(classifySecondIds, v.Id)
  198. case 3:
  199. classifyThirdIds = append(classifyThirdIds, v.Id)
  200. }
  201. }
  202. tmptotal, err := models.GetReportCountByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition)
  203. if err != nil {
  204. br.Msg = "获取数据失败"
  205. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  206. return
  207. }
  208. total = tmptotal
  209. tmpReportList, err := models.GetReportListByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition, startSize, pageSize)
  210. if err != nil {
  211. br.Msg = "获取报告列表失败"
  212. br.ErrMsg = "获取报告列表失败,Err:" + err.Error()
  213. return
  214. }
  215. reportList = tmpReportList
  216. default:
  217. br.Msg = "层级参数错误"
  218. return
  219. }
  220. page := paging.GetPaging(currentIndex, pageSize, total)
  221. resp := new(response.ReportListResp)
  222. resp.Paging = page
  223. resp.List = reportList
  224. br.Ret = 200
  225. br.Success = true
  226. br.Msg = "获取成功"
  227. br.Data = resp
  228. }
  229. func getLeafClassifyIds(classifyMap map[int]*models.ClassifyView, keyId int) []int {
  230. var leafClassifyIds []int
  231. curClassify := classifyMap[keyId]
  232. if curClassify.HasChild == 0 {
  233. leafClassifyIds = append(leafClassifyIds, curClassify.Id)
  234. return leafClassifyIds
  235. }
  236. for _, v := range classifyMap {
  237. if v.ParentId == curClassify.Id {
  238. if v.HasChild == 0 {
  239. leafClassifyIds = append(leafClassifyIds, v.Id)
  240. } else {
  241. leafClassifyIds = append(leafClassifyIds, getLeafClassifyIds(classifyMap, v.Id)...)
  242. }
  243. }
  244. }
  245. return leafClassifyIds
  246. }
  247. // @Title 研报详情
  248. // @Description 研报详情接口
  249. // @Param ReportId query int true "报告id"
  250. // @Param UserId query int true "用户id"
  251. // @Success 200 {object} response.ReportDetailResp
  252. // @router /detail [get]
  253. func (this *ReportController) Detail() {
  254. br := new(models.BaseResponse).Init()
  255. defer func() {
  256. this.Data["json"] = br
  257. this.ServeJSON()
  258. }()
  259. reportId, err := this.GetInt("ReportId")
  260. userId, err := this.GetInt("UserId")
  261. if err != nil {
  262. br.Msg = "参数获取失败"
  263. br.ErrMsg = "参数获取失败,Err:" + err.Error()
  264. return
  265. }
  266. if reportId <= 0 {
  267. br.Msg = "参数错误"
  268. br.ErrMsg = "参数错误,报告id小于等于0"
  269. return
  270. }
  271. if userId <= 0 {
  272. br.Msg = "参数错误"
  273. br.ErrMsg = "参数错误,用户id小于等于0"
  274. return
  275. }
  276. user, err := models.GetUserById(userId)
  277. if err != nil {
  278. br.Msg = "用户不存在"
  279. br.ErrMsg = "用户不存在,系统异常,Err:" + err.Error()
  280. return
  281. }
  282. // 有效期是否到期
  283. var vailStatus int
  284. if user.Status == utils.UserStatusFormal {
  285. if user.ValidEndTime.After(time.Now()) && user.ValidStartTime.Before(time.Now()) {
  286. vailStatus = 2
  287. } else {
  288. // 不在有效期时间则设置为过期, 将用户状态禁用
  289. vailStatus = 1
  290. user.Status = utils.UserStatusNo
  291. }
  292. }
  293. report, err := models.GetReportById(reportId)
  294. if err != nil {
  295. if err.Error() == utils.ErrNoRow() {
  296. br.Ret = 200
  297. br.Data = new(response.ReportDetailResp)
  298. br.Success = true
  299. br.Msg = "该报告已删除或不存在"
  300. return
  301. }
  302. br.Msg = "该报告已删除"
  303. br.ErrMsg = "获取报告详情失败,Err:" + err.Error()
  304. return
  305. }
  306. if report.HeadResourceId > 0 || report.EndResourceId > 0 {
  307. headImg, err := models.GetSmartReportResourceById(report.HeadResourceId)
  308. if err != nil && err.Error() != utils.ErrNoRow() {
  309. utils.FileLog.Warn("版头数据获取失败,Err:" + err.Error())
  310. }
  311. endImg, err := models.GetSmartReportResourceById(report.EndResourceId)
  312. if err != nil && err.Error() != utils.ErrNoRow() {
  313. utils.FileLog.Warn("版尾数据获取失败,Err:" + err.Error())
  314. }
  315. if headImg != nil {
  316. report.HeadResource = headImg
  317. }
  318. if endImg != nil {
  319. report.EndResource = endImg
  320. }
  321. }
  322. if report.HasChapter == 1 {
  323. chapterList, err := models.GetReportChapterList(report.Id)
  324. if err != nil {
  325. br.Msg = "该报告已删除"
  326. br.ErrMsg = "获取章节列表失败,Err:" + err.Error()
  327. return
  328. }
  329. for _, v := range chapterList {
  330. v.Content = html.UnescapeString(v.Content)
  331. }
  332. report.ChapterContent = chapterList
  333. }
  334. report.ContentSub = html.UnescapeString(report.ContentSub)
  335. report.Content = html.UnescapeString(report.Content)
  336. if report == nil {
  337. br.Msg = "报告不存在"
  338. return
  339. }
  340. reportChartPermissionIds, err := models.GetChartPermissionIdsListByClassifyId(report.ClassifyIdSecond)
  341. if err != nil {
  342. br.Msg = "获取研报权限失败"
  343. br.ErrMsg = "获取研报权限失败,Err:" + err.Error()
  344. return
  345. }
  346. chartPermissionList, err := models.GetChartPermissionListByStrIds(reportChartPermissionIds)
  347. if err != nil {
  348. br.Msg = "获取研报权限失败"
  349. br.ErrMsg = "获取研报权限失败,Err:" + err.Error()
  350. return
  351. }
  352. var IsHas bool
  353. var IsPublic bool
  354. for _, v := range chartPermissionList {
  355. if v.IsPublic == 1 {
  356. IsPublic = true
  357. break
  358. }
  359. }
  360. resp := new(response.ReportDetailResp)
  361. if !IsPublic {
  362. // 私有报告
  363. // 用户权限过期或处于禁用状态,直接返回有效期已过
  364. if vailStatus == 1 || user.Status == utils.UserStatusNo {
  365. resp.Report = report
  366. resp.Status = utils.ReportPermissionStatusExpired
  367. report.Content = ""
  368. br.Ret = 200
  369. br.Data = resp
  370. br.Msg = "用户权限不足"
  371. return
  372. }
  373. // 潜在用户,直接返回无阅读报告权限
  374. if user.Status == utils.UserStatusPotential {
  375. resp.Report = report
  376. resp.Status = utils.ReportPermissionStatusNo
  377. report.Content = ""
  378. br.Ret = 200
  379. br.Data = resp
  380. br.Msg = "用户权限不足"
  381. return
  382. }
  383. // 正式用户,判断是否有权限查看
  384. chartPermissionIds, err := models.GetChartPermissionIdByUserId(userId)
  385. if err != nil {
  386. br.Msg = "获取用户权限失败"
  387. br.ErrMsg = "获取用户权限失败,Err:" + err.Error()
  388. return
  389. }
  390. if len(chartPermissionIds) <= 0 {
  391. resp.Report = report
  392. resp.Status = utils.ReportPermissionStatusNoPermission
  393. report.Content = ""
  394. br.Ret = 200
  395. br.Data = resp
  396. br.Msg = "用户权限不足"
  397. return
  398. }
  399. classifyIds, err := models.GetClassifyIdsListByIds(chartPermissionIds)
  400. if err != nil {
  401. br.Msg = "获取用户权限失败"
  402. br.ErrMsg = "获取用户分类权限失败,Err:" + err.Error()
  403. return
  404. }
  405. if len(classifyIds) <= 0 {
  406. resp.Report = report
  407. resp.Status = utils.ReportPermissionStatusNoPermission
  408. report.Content = ""
  409. br.Ret = 200
  410. br.Data = resp
  411. br.Msg = "用户权限不足"
  412. return
  413. }
  414. var reportClassifyId int
  415. if report.ClassifyIdFirst > 0 {
  416. reportClassifyId = report.ClassifyIdFirst
  417. }
  418. if report.ClassifyIdSecond > 0 {
  419. reportClassifyId = report.ClassifyIdSecond
  420. }
  421. if report.ClassifyIdThird > 0 {
  422. reportClassifyId = report.ClassifyIdThird
  423. }
  424. for _, v := range classifyIds {
  425. if v == reportClassifyId {
  426. IsHas = true
  427. break
  428. }
  429. }
  430. if !IsHas {
  431. resp.Report = report
  432. resp.Status = utils.ReportPermissionStatusNoPermission
  433. report.Content = ""
  434. br.Ret = 200
  435. br.Data = resp
  436. br.Msg = "用户权限不足"
  437. return
  438. }
  439. } else {
  440. report.IsPublic = IsPublic
  441. }
  442. resp.Report = report
  443. resp.Status = utils.ReportPermissionStatusHas
  444. br.Data = resp
  445. br.Ret = 200
  446. br.Success = true
  447. br.Msg = "获取成功"
  448. }
  449. // @Title 研报详情
  450. // @Description 研报详情接口
  451. // @Param ReportId query int true "报告id"
  452. // @Success 200 {object} response.ReportDetailResp
  453. // @router /detail/noUser [get]
  454. func (this *ReportController) DetailNoUser() {
  455. br := new(models.BaseResponse).Init()
  456. defer func() {
  457. this.Data["json"] = br
  458. this.ServeJSON()
  459. }()
  460. reportId, err := this.GetInt("ReportId")
  461. if err != nil {
  462. br.Msg = "参数获取失败"
  463. br.ErrMsg = "参数获取失败,Err:" + err.Error()
  464. return
  465. }
  466. if reportId <= 0 {
  467. br.Msg = "参数错误"
  468. br.ErrMsg = "参数错误,报告id小于等于0"
  469. return
  470. }
  471. report, err := models.GetReportById(reportId)
  472. if err != nil {
  473. if err.Error() == utils.ErrNoRow() {
  474. br.Ret = 200
  475. br.Data = new(response.ReportDetailResp)
  476. br.Success = true
  477. br.Msg = "该报告已删除或不存在"
  478. return
  479. }
  480. br.Msg = "该报告已删除"
  481. br.ErrMsg = "获取报告详情失败,Err:" + err.Error()
  482. return
  483. }
  484. report.ContentSub = html.UnescapeString(report.ContentSub)
  485. report.Content = ""
  486. if report == nil {
  487. br.Msg = "报告不存在"
  488. return
  489. }
  490. resp := new(response.ReportDetailResp)
  491. resp.Report = report
  492. resp.Status = utils.ReportPermissionStatusNoUser
  493. br.Data = resp
  494. br.Ret = 200
  495. br.Success = true
  496. br.Msg = "获取成功"
  497. }
  498. // @Title 今日研报列表
  499. // @Description 今日研报列表
  500. // @Success 200 {object} response.ReportListResp
  501. // @router /daily/list [get]
  502. func (this *ReportController) Today() {
  503. br := new(models.BaseResponse).Init()
  504. defer func() {
  505. this.Data["json"] = br
  506. this.ServeJSON()
  507. }()
  508. total, err := models.GetReportDailyListCount()
  509. if err != nil {
  510. br.Msg = "获取数据失败"
  511. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  512. return
  513. }
  514. list, err := models.GetReportDailyList(0, total)
  515. if err != nil {
  516. br.Msg = "获取失败"
  517. br.ErrMsg = "获取失败,Err:" + err.Error()
  518. return
  519. }
  520. classifyIds := make([]string, 0)
  521. for _, v := range list {
  522. classifyIds = append(classifyIds, strconv.Itoa(v.ClassifyIdSecond))
  523. }
  524. classifyIds = utils.Unique(classifyIds)
  525. // 获取二级分类和二级品种权限的映射
  526. chartPermissionMapping, err := models.GetChartPermissionListByClassifyStrIds(classifyIds)
  527. if err != nil {
  528. br.Msg = "获取研报权限失败"
  529. br.ErrMsg = "获取研报权限失败,Err:" + err.Error()
  530. return
  531. }
  532. classifyToPermissionMap2 := make(map[int][]int)
  533. chartPermissionIds := make([]string, 0)
  534. for _, v := range chartPermissionMapping {
  535. classifyToPermissionMap2[v.ClassifyId] = append(classifyToPermissionMap2[v.ClassifyId], v.ChartPermissionId)
  536. chartPermissionIds = append(chartPermissionIds, strconv.Itoa(v.ChartPermissionId))
  537. }
  538. // 获取二级品种的权限,并建立映射
  539. chartPermissionList2, err := models.GetChartPermissionListByStrIds(chartPermissionIds)
  540. if err != nil {
  541. br.Msg = "获取研报二级品种权限失败"
  542. br.ErrMsg = "获取研报二级品种权限失败,Err:" + err.Error()
  543. return
  544. }
  545. chartPermissionViewMap2 := make(map[int]*models.ChartPermission)
  546. for _, v := range chartPermissionList2 {
  547. chartPermissionViewMap2[v.ChartPermissionId] = v
  548. }
  549. // 获取一级品种的权限,并建立映射
  550. chartPermissionList1, err := models.GetChildChartPermissionListById(0)
  551. if err != nil {
  552. br.Msg = "获取研报一级品种权限失败"
  553. br.ErrMsg = "获取研报一级品种权限失败,Err:" + err.Error()
  554. return
  555. }
  556. chartPermissionMap1 := make(map[int]*models.ChartPermission)
  557. for _, v := range chartPermissionList1 {
  558. chartPermissionMap1[v.ChartPermissionId] = v
  559. }
  560. // 组合数据
  561. finalList := make([]*models.ReportList, 0)
  562. for _, v := range list {
  563. var permissionNames []string
  564. for _, vv := range classifyToPermissionMap2[v.ClassifyIdSecond] {
  565. if item, ok := chartPermissionViewMap2[vv]; ok {
  566. parent2 := item.ParentId
  567. permissionNames = append(permissionNames, chartPermissionMap1[parent2].PermissionName)
  568. }
  569. }
  570. v.PermissionNames = utils.Unique(permissionNames)
  571. if len(v.PermissionNames) != 0 {
  572. finalList = append(finalList, v)
  573. }
  574. }
  575. resp := new(response.ReportListResp)
  576. resp.List = finalList
  577. br.Ret = 200
  578. br.Success = true
  579. br.Msg = "获取成功"
  580. br.Data = resp
  581. }
  582. // @Title 最新研报列表
  583. // @Description 最新研报列表
  584. // @Param PageSize query int true "每页数据条数"
  585. // @Param CurrentIndex query int true "当前页页码,从1开始"
  586. // @Success 200 {object} response.ReportListResp
  587. // @router /recent/list [get]
  588. func (this *ReportController) RecentList() {
  589. br := new(models.BaseResponse).Init()
  590. defer func() {
  591. this.Data["json"] = br
  592. this.ServeJSON()
  593. }()
  594. pageSize, _ := this.GetInt("PageSize")
  595. currentIndex, _ := this.GetInt("CurrentIndex")
  596. var startSize int
  597. if pageSize <= 0 {
  598. pageSize = utils.PageSize30
  599. }
  600. if currentIndex <= 0 {
  601. currentIndex = 1
  602. }
  603. startSize = utils.StartIndex(currentIndex, pageSize)
  604. total, err := models.GetReportRecentListCount()
  605. if err != nil {
  606. br.Msg = "获取数据失败"
  607. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  608. return
  609. }
  610. list, err := models.GetReportRecentList(startSize, pageSize)
  611. if err != nil {
  612. br.Msg = "获取失败"
  613. br.ErrMsg = "获取失败,Err:" + err.Error()
  614. return
  615. }
  616. classifyIds := make([]string, 0)
  617. for _, v := range list {
  618. classifyIds = append(classifyIds, strconv.Itoa(v.ClassifyIdSecond))
  619. }
  620. classifyIds = utils.Unique(classifyIds)
  621. // 获取二级分类和二级品种权限的映射
  622. chartPermissionMapping, err := models.GetChartPermissionListByClassifyStrIds(classifyIds)
  623. if err != nil {
  624. br.Msg = "获取研报权限失败"
  625. br.ErrMsg = "获取研报权限失败,Err:" + err.Error()
  626. return
  627. }
  628. classifyToPermissionMap2 := make(map[int][]int)
  629. chartPermissionIds := make([]string, 0)
  630. for _, v := range chartPermissionMapping {
  631. classifyToPermissionMap2[v.ClassifyId] = append(classifyToPermissionMap2[v.ClassifyId], v.ChartPermissionId)
  632. chartPermissionIds = append(chartPermissionIds, strconv.Itoa(v.ChartPermissionId))
  633. }
  634. // 获取二级品种的权限,并建立映射
  635. chartPermissionList2, err := models.GetChartPermissionListByStrIds(chartPermissionIds)
  636. if err != nil {
  637. br.Msg = "获取研报二级品种权限失败"
  638. br.ErrMsg = "获取研报二级品种权限失败,Err:" + err.Error()
  639. return
  640. }
  641. chartPermissionViewMap2 := make(map[int]*models.ChartPermission)
  642. for _, v := range chartPermissionList2 {
  643. chartPermissionViewMap2[v.ChartPermissionId] = v
  644. }
  645. // 获取一级品种的权限,并建立映射
  646. chartPermissionList1, err := models.GetChildChartPermissionListById(0)
  647. if err != nil {
  648. br.Msg = "获取研报一级品种权限失败"
  649. br.ErrMsg = "获取研报一级品种权限失败,Err:" + err.Error()
  650. return
  651. }
  652. chartPermissionMap1 := make(map[int]*models.ChartPermission)
  653. for _, v := range chartPermissionList1 {
  654. chartPermissionMap1[v.ChartPermissionId] = v
  655. }
  656. // 组合数据
  657. finalList := make([]*models.ReportList, 0)
  658. for _, v := range list {
  659. var permissionNames []string
  660. for _, vv := range classifyToPermissionMap2[v.ClassifyIdSecond] {
  661. if item, ok := chartPermissionViewMap2[vv]; ok {
  662. parent2 := item.ParentId
  663. permissionNames = append(permissionNames, chartPermissionMap1[parent2].PermissionName)
  664. }
  665. }
  666. v.PermissionNames = utils.Unique(permissionNames)
  667. if len(v.PermissionNames) != 0 {
  668. finalList = append(finalList, v)
  669. }
  670. }
  671. page := paging.GetPaging(currentIndex, pageSize, total)
  672. resp := new(response.ReportListResp)
  673. resp.Paging = page
  674. resp.List = finalList
  675. br.Ret = 200
  676. br.Success = true
  677. br.Msg = "获取成功"
  678. br.Data = resp
  679. }
  680. // @Title 研报搜索
  681. // @Description 研报搜索
  682. // @Param PageSize query int true "每页数据条数"
  683. // @Param CurrentIndex query int true "当前页页码,从1开始"
  684. // @Param KeyWord query string true "关键字"
  685. // @Success 200 {object} response.ReportSearchResp
  686. // @router /search [get]
  687. func (this *ReportController) Search() {
  688. br := new(models.BaseResponse).Init()
  689. defer func() {
  690. this.Data["json"] = br
  691. this.ServeJSON()
  692. }()
  693. keyWord := this.GetString("KeyWord")
  694. pageSize, _ := this.GetInt("PageSize")
  695. currentIndex, _ := this.GetInt("CurrentIndex")
  696. if pageSize <= 0 {
  697. pageSize = utils.PageSize30
  698. }
  699. if currentIndex <= 0 {
  700. currentIndex = 1
  701. }
  702. if keyWord == "" {
  703. br.Msg = "请输入关键字"
  704. return
  705. }
  706. reportList, err, errMsg := services.SearchReport(keyWord, currentIndex, pageSize)
  707. if err != nil {
  708. br.Msg = "研报搜索失败"
  709. br.ErrMsg = errMsg + ",Err:" + err.Error()
  710. return
  711. }
  712. br.Data = reportList
  713. br.Msg = "查询成功"
  714. br.Ret = 200
  715. br.Success = true
  716. }
  717. // @Title List
  718. // @Description create users
  719. // @Param ChartPermissionId query int true "品种ID"
  720. // @Param Level query int true "品种层级"
  721. // @Param PageSize query int true "每页数据条数"
  722. // @Param CurrentIndex query int true "当前页页码,从1开始"
  723. // @Param RangeType query string true "范围类型,1-一天内,2-一周内,3-半年内"
  724. // @Param ClassifyId query int true "分类id"
  725. // @Success 200 {object} response.ReportListResp
  726. // @router /list [get]
  727. func (this *ReportNoAuthController) List() {
  728. br := new(models.BaseResponse).Init()
  729. defer func() {
  730. this.Data["json"] = br
  731. this.ServeJSON()
  732. }()
  733. pageSize, _ := this.GetInt("PageSize")
  734. currentIndex, _ := this.GetInt("CurrentIndex")
  735. chartPermissionId, _ := this.GetInt("ChartPermissionId")
  736. level, _ := this.GetInt("Level")
  737. rangeType, _ := this.GetInt("RangeType")
  738. classifyId, _ := this.GetInt("ClassifyId")
  739. if chartPermissionId < 0 {
  740. br.Msg = "品种参数错误"
  741. return
  742. }
  743. var condition string
  744. switch rangeType {
  745. case 1:
  746. condition += ` AND DATE(modify_time)=DATE(NOW()) `
  747. case 2:
  748. condition += ` AND DATE(modify_time) BETWEEN DATE_SUB(NOW(),INTERVAL 1 WEEK) AND NOW() `
  749. case 3:
  750. condition += ` AND DATE(modify_time) BETWEEN DATE_SUB(NOW(),INTERVAL 6 MONTH) AND NOW() `
  751. }
  752. var startSize int
  753. if pageSize <= 0 {
  754. pageSize = utils.PageSize20
  755. }
  756. if currentIndex <= 0 {
  757. currentIndex = 1
  758. }
  759. startSize = utils.StartIndex(currentIndex, pageSize)
  760. classifyList, err := models.GetClassifyList()
  761. if err != nil {
  762. br.Msg = "获取数据失败"
  763. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  764. return
  765. }
  766. classifyMap := make(map[int]*models.ClassifyView)
  767. isHas := false
  768. for _, v := range classifyList {
  769. if v.Id == classifyId && classifyId != 0 {
  770. isHas = true
  771. }
  772. classifyMap[v.Id] = v
  773. }
  774. if !isHas && classifyId != 0 {
  775. br.Msg = "分类不存在"
  776. return
  777. }
  778. classifyIdMap := make(map[int]struct{})
  779. if classifyId > 0 {
  780. leafClassifyIds := getLeafClassifyIds(classifyMap, classifyId)
  781. for _, v := range leafClassifyIds {
  782. classifyIdMap[v] = struct{}{}
  783. }
  784. }
  785. var total int
  786. var reportList []*models.ReportPushView
  787. classifyFirstIds := make([]int, 0)
  788. classifySecondIds := make([]int, 0)
  789. classifyThirdIds := make([]int, 0)
  790. switch level {
  791. case 2:
  792. classifyIds, err := models.GetClassifyIdsListById(chartPermissionId)
  793. if err != nil {
  794. br.Msg = "获取报告列表失败"
  795. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  796. return
  797. }
  798. if len(classifyIds) == 0 {
  799. br.Msg = "该品种下没有绑定分类"
  800. br.ErrMsg = "获取数据失败,品种id:" + strconv.Itoa(chartPermissionId)
  801. return
  802. }
  803. classifyIds = utils.Unique(classifyIds)
  804. // 根据分类筛选报告
  805. var selectedClassifyId []int
  806. if classifyId > 0 {
  807. for _, v := range classifyIds {
  808. if _, ok := classifyIdMap[v]; ok {
  809. selectedClassifyId = append(selectedClassifyId, v)
  810. }
  811. }
  812. } else {
  813. selectedClassifyId = classifyIds
  814. }
  815. classifyList, err := models.GetClassifyListByIds(selectedClassifyId)
  816. if err != nil {
  817. br.Msg = "获取报告列表失败"
  818. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  819. return
  820. }
  821. for _, v := range classifyList {
  822. if v.HasChild == 1 {
  823. continue
  824. }
  825. switch v.Level {
  826. case 1:
  827. classifyFirstIds = append(classifyFirstIds, v.Id)
  828. case 2:
  829. classifySecondIds = append(classifySecondIds, v.Id)
  830. case 3:
  831. classifyThirdIds = append(classifyThirdIds, v.Id)
  832. }
  833. }
  834. tmptotal, err := models.GetPushReportCountByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition)
  835. if err != nil {
  836. br.Msg = "获取数据失败"
  837. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  838. return
  839. }
  840. total = tmptotal
  841. tmpReportList, err := models.GetPushReportListByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition, startSize, pageSize)
  842. if err != nil {
  843. br.Msg = "获取报告列表失败"
  844. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  845. return
  846. }
  847. reportList = tmpReportList
  848. case 1:
  849. chartPermissionIds, err := models.GetChildChartPermissionIdsById(chartPermissionId)
  850. if err != nil {
  851. br.Msg = "获取报告列表失败"
  852. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  853. return
  854. }
  855. if len(chartPermissionIds) == 0 {
  856. br.Msg = "该品种下没有绑定分类"
  857. br.ErrMsg = "获取数据失败,品种id:" + strconv.Itoa(chartPermissionId)
  858. return
  859. }
  860. classifyIds, err := models.GetClassifyIdsListByIds(chartPermissionIds)
  861. if err != nil {
  862. br.Msg = "获取报告列表失败"
  863. br.ErrMsg = "获取报告列表失败,Err:" + err.Error()
  864. return
  865. }
  866. if len(classifyIds) == 0 {
  867. br.Msg = "该品种下没有绑定分类"
  868. return
  869. }
  870. classifyIds = utils.Unique(classifyIds)
  871. // 根据分类筛选报告
  872. var selectedClassifyId []int
  873. if classifyId > 0 {
  874. for _, v := range classifyIds {
  875. if _, ok := classifyIdMap[v]; ok {
  876. selectedClassifyId = append(selectedClassifyId, v)
  877. }
  878. }
  879. } else {
  880. selectedClassifyId = classifyIds
  881. }
  882. classifyList, err := models.GetClassifyListByIds(selectedClassifyId)
  883. if err != nil {
  884. br.Msg = "获取报告列表失败"
  885. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  886. return
  887. }
  888. for _, v := range classifyList {
  889. if v.HasChild == 1 {
  890. continue
  891. }
  892. switch v.Level {
  893. case 1:
  894. classifyFirstIds = append(classifyFirstIds, v.Id)
  895. case 2:
  896. classifySecondIds = append(classifySecondIds, v.Id)
  897. case 3:
  898. classifyThirdIds = append(classifyThirdIds, v.Id)
  899. }
  900. }
  901. tmptotal, err := models.GetPushReportCountByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition)
  902. if err != nil {
  903. br.Msg = "获取数据失败"
  904. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  905. return
  906. }
  907. total = tmptotal
  908. tmpReportList, err := models.GetPushReportListByClassifyIds(classifyFirstIds, classifySecondIds, classifyThirdIds, condition, startSize, pageSize)
  909. if err != nil {
  910. br.Msg = "获取报告列表失败"
  911. br.ErrMsg = "获取报告列表失败,Err:" + err.Error()
  912. return
  913. }
  914. reportList = tmpReportList
  915. case 0:
  916. var selectedClassifyIds []int
  917. for k := range classifyIdMap {
  918. selectedClassifyIds = append(selectedClassifyIds, k)
  919. }
  920. tmptotal, err := models.GetPushReportCount(selectedClassifyIds, condition)
  921. if err != nil {
  922. br.Msg = "获取数据失败"
  923. br.ErrMsg = "获取数据失败,Err:" + err.Error()
  924. return
  925. }
  926. total = tmptotal
  927. tmpReportList, err := models.GetPushReportListByPage(selectedClassifyIds, condition, startSize, pageSize)
  928. if err != nil {
  929. br.Msg = "获取报告列表失败"
  930. br.ErrMsg = "获取报告列表失败,Err:" + err.Error()
  931. return
  932. }
  933. reportList = tmpReportList
  934. default:
  935. br.Msg = "层级参数错误"
  936. return
  937. }
  938. page := paging.GetPaging(currentIndex, pageSize, total)
  939. resp := new(response.ReportPushListResp)
  940. resp.Paging = page
  941. resp.List = reportList
  942. br.Ret = 200
  943. br.Success = true
  944. br.Msg = "获取成功"
  945. br.Data = resp
  946. }
  947. // @Title 研报详情
  948. // @Description 研报详情接口
  949. // @Param ReportId query int true "报告id"
  950. // @Param UserId query int true "用户id"
  951. // @Success 200 {object} response.ReportDetailResp
  952. // @router /detail [get]
  953. func (this *ReportNoAuthController) Detail() {
  954. br := new(models.BaseResponse).Init()
  955. defer func() {
  956. this.Data["json"] = br
  957. this.ServeJSON()
  958. }()
  959. reportId, err := this.GetInt("ReportId")
  960. if err != nil {
  961. br.Msg = "参数获取失败"
  962. br.ErrMsg = "参数获取失败,Err:" + err.Error()
  963. return
  964. }
  965. if reportId <= 0 {
  966. br.Msg = "参数错误"
  967. br.ErrMsg = "参数错误,报告id小于等于0"
  968. return
  969. }
  970. report, err := models.GetReportById(reportId)
  971. if err != nil {
  972. if err.Error() == utils.ErrNoRow() {
  973. br.Ret = 200
  974. br.Data = new(response.ReportDetailResp)
  975. br.Success = true
  976. br.Msg = "该报告已删除或不存在"
  977. return
  978. }
  979. br.Msg = "该报告已删除"
  980. br.ErrMsg = "获取报告详情失败,Err:" + err.Error()
  981. return
  982. }
  983. if report.HeadResourceId > 0 || report.EndResourceId > 0 {
  984. headImg, err := models.GetSmartReportResourceById(report.HeadResourceId)
  985. if err != nil && err.Error() != utils.ErrNoRow() {
  986. utils.FileLog.Warn("版头数据获取失败,Err:" + err.Error())
  987. }
  988. endImg, err := models.GetSmartReportResourceById(report.EndResourceId)
  989. if err != nil && err.Error() != utils.ErrNoRow() {
  990. utils.FileLog.Warn("版尾数据获取失败,Err:" + err.Error())
  991. }
  992. if headImg != nil {
  993. report.HeadResource = headImg
  994. }
  995. if endImg != nil {
  996. report.EndResource = endImg
  997. }
  998. }
  999. if report.HasChapter == 1 {
  1000. chapterList, err := models.GetReportChapterList(report.Id)
  1001. if err != nil {
  1002. br.Msg = "该报告已删除"
  1003. br.ErrMsg = "获取章节列表失败,Err:" + err.Error()
  1004. return
  1005. }
  1006. for _, v := range chapterList {
  1007. v.Content = html.UnescapeString(v.Content)
  1008. }
  1009. report.ChapterContent = chapterList
  1010. }
  1011. report.ContentSub = html.UnescapeString(report.ContentSub)
  1012. report.Content = html.UnescapeString(report.Content)
  1013. if report == nil {
  1014. br.Msg = "报告不存在"
  1015. return
  1016. }
  1017. resp := new(response.ReportDetailResp)
  1018. resp.Report = report
  1019. br.Data = resp
  1020. br.Ret = 200
  1021. br.Success = true
  1022. br.Msg = "获取成功"
  1023. }