analyst_controller.go 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167
  1. package user
  2. import (
  3. "eta/eta_mini_ht_api/common/exception"
  4. "eta/eta_mini_ht_api/common/utils/page"
  5. "eta/eta_mini_ht_api/controllers"
  6. userDao "eta/eta_mini_ht_api/models/user"
  7. "eta/eta_mini_ht_api/service/analyst"
  8. "eta/eta_mini_ht_api/service/media"
  9. "eta/eta_mini_ht_api/service/report"
  10. "eta/eta_mini_ht_api/service/user"
  11. )
  12. type AnalystController struct {
  13. controllers.ListController
  14. }
  15. // AnalystList 研究员列表
  16. // @Summary 研究员列表
  17. // @Description 研究员列表
  18. // @Success 200 {object} controllers.BaseResponse
  19. // @router /list [get]
  20. func (an *AnalystController) AnalystList() {
  21. controllers.Wrap(&an.BaseController, func() (result *controllers.WrapData, err error) {
  22. result = an.InitWrapData("获取研究员列表失败")
  23. var userInfo user.User
  24. userInfo = an.Data["user"].(user.User)
  25. pageRes := page.Page{
  26. Current: an.PageInfo.Current,
  27. PageSize: an.PageInfo.PageSize,
  28. }
  29. if an.PageInfo.LatestId == 0 {
  30. pageRes.Total, pageRes.LatestId = analyst.GetAnalystCount()
  31. an.PageInfo.LatestId = pageRes.LatestId
  32. an.PageInfo.Total = pageRes.Total
  33. } else {
  34. pageRes.LatestId = an.PageInfo.LatestId
  35. pageRes.Total = an.PageInfo.Total
  36. }
  37. an.PageInfo.Total = pageRes.Total
  38. pageRes.TotalPage = page.TotalPages(pageRes.Total, pageRes.PageSize)
  39. list, err := analyst.GetAnalystList(an.PageInfo, userInfo.Id)
  40. if err != nil {
  41. an.FailedResult("获取研究员列表失败", result)
  42. return
  43. }
  44. analystList := new(page.PageResult)
  45. analystList.Data = list
  46. analystList.Page = pageRes
  47. an.SuccessResult("获取研究员列表成功", analystList, result)
  48. return
  49. })
  50. }
  51. // AnalystDetail 研究员详情
  52. // @Summary 研究员详情
  53. // @Description 研究员详情
  54. // @Success 200 {object} controllers.BaseResponse
  55. // @router /analystDetail [get]
  56. func (an *AnalystController) AnalystDetail(analystId int) {
  57. controllers.Wrap(&an.BaseController, func() (result *controllers.WrapData, err error) {
  58. result = an.InitWrapData("获取研究员详情失败")
  59. userInfo := an.Data["user"].(user.User)
  60. detail, err := user.GetAnalystDetail(userInfo.Id, analystId)
  61. if err != nil {
  62. defaultRes := &user.AnalystDetail{
  63. Followed: string(userDao.Forbidden),
  64. }
  65. an.SuccessResult("获取研究员详情成功", defaultRes, result)
  66. return
  67. }
  68. an.SuccessResult("获取研究员详情成功", detail, result)
  69. return
  70. })
  71. }
  72. // AnalystReportList @Title 获取研究员报告列表
  73. // @Description 获取研究员报告列表
  74. // @Success 200 {object}
  75. // @router /reportList [get]
  76. func (an *AnalystController) AnalystReportList(analystName string) {
  77. controllers.Wrap(&an.BaseController, func() (result *controllers.WrapData, err error) {
  78. result = an.InitWrapData("分页获取研究员报告列表失败")
  79. pageRes := page.Page{
  80. Current: an.PageInfo.Current,
  81. PageSize: an.PageInfo.PageSize,
  82. }
  83. userInfo := an.Data["user"].(user.User)
  84. var reportIds []int
  85. var mappingRiskLevel string
  86. var userRiskStatus string
  87. pageRes.Total, pageRes.LatestId, reportIds, mappingRiskLevel, userRiskStatus = report.RangeSearchByAnalyst(analystName, userInfo.Id)
  88. if len(reportIds) == 0 {
  89. reports := new(page.PageResult)
  90. reports.Data = []interface{}{}
  91. reports.Page = pageRes
  92. an.SuccessResult("分页获取研究员报告列表成功", reports, result)
  93. }
  94. if an.PageInfo.LatestId == 0 {
  95. //pageRes.Total, pageRes.LatestId = report.GetTotalPageCountByAnalyst(analystName)
  96. an.PageInfo.LatestId = pageRes.LatestId
  97. } else {
  98. pageRes.LatestId = an.PageInfo.LatestId
  99. pageRes.Total = an.PageInfo.Total
  100. }
  101. pageRes.TotalPage = page.TotalPages(pageRes.Total, pageRes.PageSize)
  102. list, err := report.GetReportPageByAnalyst(an.PageInfo, analystName, reportIds, userInfo.Id, mappingRiskLevel, userRiskStatus)
  103. if err != nil {
  104. an.FailedResult("分页获取研究员报告列表失败", result)
  105. return
  106. }
  107. reports := new(page.PageResult)
  108. reports.Data = list
  109. reports.Page = pageRes
  110. an.SuccessResult("分页获取研究员报告列表成功", reports, result)
  111. return
  112. })
  113. }
  114. // MediaList 获取研究员媒体列表
  115. // @Description 获取研究员媒体列表
  116. // @Success 200 {object}
  117. // @router /mediaList [get]
  118. func (an *AnalystController) MediaList(mediaType string, analystId int) {
  119. controllers.Wrap(&an.BaseController, func() (result *controllers.WrapData, err error) {
  120. result = an.InitWrapData("分页查询研究员媒体列表失败")
  121. if mediaType == "" || !an.CheckMediaType(mediaType) {
  122. err = exception.New(exception.MediaTypeError)
  123. an.FailedResult("分页查询研究员媒体列表失败", result)
  124. return
  125. }
  126. pageRes := page.Page{
  127. Current: an.PageInfo.Current,
  128. PageSize: an.PageInfo.PageSize,
  129. }
  130. userInfo := an.Data["user"].(user.User)
  131. var mediaIds []int
  132. var mappingRiskLevel string
  133. var userRiskStatus string
  134. pageRes.Total, pageRes.LatestId, mediaIds, mappingRiskLevel, userRiskStatus = media.RangeSearchByAnalyst(mediaType, analystId, userInfo.Id)
  135. if len(mediaIds) == 0 {
  136. mediaList := new(page.PageResult)
  137. mediaList.Data = []interface{}{}
  138. mediaList.Page = pageRes
  139. an.SuccessResult("分页查询研究员媒体列表成功", mediaList, result)
  140. }
  141. if an.PageInfo.LatestId == 0 {
  142. // pageRes.Total, pageRes.LatestId = media.GetTotalPageCountByAnalystId(mediaType, analystId)
  143. an.PageInfo.LatestId = pageRes.LatestId
  144. an.PageInfo.Total = pageRes.Total
  145. } else {
  146. pageRes.LatestId = an.PageInfo.LatestId
  147. pageRes.Total = an.PageInfo.Total
  148. }
  149. pageRes.TotalPage = page.TotalPages(pageRes.Total, pageRes.PageSize)
  150. list, err := media.GetMediaPageByAnalystId(mediaType, an.PageInfo, analystId, mediaIds, userInfo.Id, mappingRiskLevel, userRiskStatus)
  151. if err != nil {
  152. an.FailedResult("分页查询研究员媒体列表失败", result)
  153. return
  154. }
  155. mediaList := new(page.PageResult)
  156. mediaList.Data = list
  157. mediaList.Page = pageRes
  158. an.SuccessResult("分页查询研究员媒体列表成功", mediaList, result)
  159. return
  160. })
  161. }