report.go 29 KB

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