report_service.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736
  1. package report
  2. import (
  3. "encoding/json"
  4. "errors"
  5. logger "eta/eta_mini_ht_api/common/component/log"
  6. "eta/eta_mini_ht_api/common/exception"
  7. "eta/eta_mini_ht_api/common/utils/date"
  8. "eta/eta_mini_ht_api/common/utils/page"
  9. permissionService "eta/eta_mini_ht_api/domian/config"
  10. mediaService "eta/eta_mini_ht_api/domian/media"
  11. productService "eta/eta_mini_ht_api/domian/merchant"
  12. reportService "eta/eta_mini_ht_api/domian/report"
  13. userService "eta/eta_mini_ht_api/domian/user"
  14. productDao "eta/eta_mini_ht_api/models/merchant"
  15. "eta/eta_mini_ht_api/service/config"
  16. user "eta/eta_mini_ht_api/service/user"
  17. "fmt"
  18. "gorm.io/gorm"
  19. "strconv"
  20. "strings"
  21. "sync"
  22. "time"
  23. )
  24. const (
  25. SourceETA = "ETA"
  26. SourceHT = "HT"
  27. RiskLevelUnMatch = "unMatch"
  28. RiskLevelUnTest = "unTest"
  29. RiskLevelExpired = "expired"
  30. RiskLevelMatch = "match"
  31. defaultProductPrice = "0"
  32. )
  33. type PublishRankedReport struct {
  34. Id int `json:"reportId"`
  35. OrgId int `json:"orgId"`
  36. Title string `json:"title"`
  37. Abstract string `json:"abstract"`
  38. SecondPermissions map[int]string `json:"-"`
  39. Permissions map[int]string `json:"-"`
  40. PermissionNames interface{} `json:"permissionNames,omitempty"`
  41. PublishedTime string `json:"publishedTime"`
  42. CoverUrl string `json:"coverUrl"`
  43. RiskLevel string `json:"riskLevel"`
  44. IsFree bool `json:"isFree"`
  45. Price string `json:"price"`
  46. IsSubscribe bool `json:"isSubscribe"`
  47. Login bool `json:"login"`
  48. ProductId int `json:"productId"`
  49. }
  50. type HotRankedReport struct {
  51. Id int `json:"reportId"`
  52. OrgId int `json:"orgId"`
  53. Abstract string `json:"abstract"`
  54. Count int `json:"count"`
  55. Title string `json:"title"`
  56. PublishedTime string `json:"publishedTime"`
  57. SecondPermissions map[int]string `json:"-"`
  58. Permissions map[int]string `json:"-"`
  59. PermissionNames interface{} `json:"permissionNames,omitempty"`
  60. CoverUrl string `json:"coverUrl"`
  61. RiskLevel string `json:"riskLevel"`
  62. IsFree bool `json:"isFree"`
  63. Price string `json:"price"`
  64. ProductId int `json:"productId"`
  65. IsSubscribe bool `json:"isSubscribe"`
  66. Login bool `json:"login"`
  67. }
  68. type RecordCount struct {
  69. UserId int
  70. TraceId string
  71. Mobile string
  72. ReportId int
  73. IpAddress string
  74. Location string
  75. Referer string
  76. Additional string
  77. }
  78. func matchRiskLevel(userId int, report reportService.ReportDTO) (riskLevelMatch string, riskLevel string, err error) {
  79. userProfile, userErr := user.GetUserProfile(userId)
  80. if userErr != nil {
  81. if errors.Is(userErr, gorm.ErrRecordNotFound) {
  82. logger.Error("用户信息不存在,mobile:%d", userProfile.Mobile)
  83. err = exception.New(exception.TemplateUserNotFound)
  84. return
  85. } else {
  86. logger.Error("获取用户信息失败:%v", userErr)
  87. err = exception.New(exception.TemplateUserFoundFailed)
  88. return
  89. }
  90. }
  91. //比较风险等级
  92. if userProfile.RiskLevelStatus == user.RiskUnTest {
  93. logger.Info("客户风险等级未测试,mobile:%d", userProfile.Mobile)
  94. riskLevelMatch = RiskLevelUnTest
  95. return
  96. }
  97. if userProfile.RiskLevelStatus == user.RiskExpired {
  98. logger.Info("客户风险等级已过期,mobile:%v", userProfile.Mobile)
  99. riskLevelMatch = RiskLevelExpired
  100. return
  101. }
  102. level, err := permissionService.GetRiskMappingByCustomerRiskLevel(userProfile.RiskLevel)
  103. if err != nil {
  104. logger.Error("获取eta报告风险等级失败:%v", err)
  105. return
  106. }
  107. permissions := reportService.GetReportSecondPermissionsById(report.OrgId, report.Source)
  108. if len(permissions) == 0 {
  109. logger.Error("获取eta报告分类失败:%v", err)
  110. riskLevelMatch = RiskLevelUnMatch
  111. return
  112. }
  113. var permissionIds []int
  114. for _, permission := range permissions {
  115. permissionIds = append(permissionIds, permission.PermissionId)
  116. }
  117. permissionDTOs, err := permissionService.GetPermissionListByIds(permissionIds)
  118. if err != nil {
  119. logger.Error("获取品种风险等级失败:%v", err)
  120. return
  121. }
  122. //能够查看最高等级
  123. matchNum, err := config.ParseRiskLevel(level.ProductRiskLevel)
  124. if err != nil {
  125. logger.Error("解析风险等级失败:%v", err)
  126. return
  127. }
  128. if len(permissionDTOs) == 0 {
  129. logger.Error("当前报告对应品种未设置风险等级")
  130. err = exception.New(exception.ReportRiskLevelUnSet)
  131. return
  132. }
  133. //能够查看需要的最小等级
  134. //num := getLowestRiskLevel(permissionDTOs)
  135. num := config.GetHighestRiskLevel(permissionDTOs)
  136. riskLevel = fmt.Sprintf("R%d", num)
  137. if num > matchNum {
  138. riskLevelMatch = RiskLevelUnMatch
  139. return
  140. } else {
  141. riskLevelMatch = RiskLevelMatch
  142. return
  143. }
  144. }
  145. func GetReportById(reportId int, login bool, userId int) (report reportService.ReportDTO, err error) {
  146. report, err = reportService.GetReportById(reportId)
  147. if err != nil {
  148. logger.Error("获取研报失败:%v", err)
  149. err = exception.New(exception.GetReportFailed)
  150. return
  151. }
  152. var status string
  153. status, report.RiskLevel, err = matchRiskLevel(userId, report)
  154. if err != nil {
  155. logger.Error("匹配风险等级失败:%v", err)
  156. err = exception.New(exception.ReportRiskLevelUnSet)
  157. return
  158. }
  159. var pdfUrl string
  160. switch report.Source {
  161. case SourceETA:
  162. var detail reportService.ETAReportDTO
  163. detail, err = getETAReportDetail(&report)
  164. if err != nil {
  165. logger.Error("获取研报详情失败失败:%v", err)
  166. err = exception.New(exception.GetReportFailed)
  167. return
  168. }
  169. if !login {
  170. detail.Content = ""
  171. report.RiskLevelStatus = RiskLevelUnMatch
  172. report.Login = false
  173. } else {
  174. if status != RiskLevelMatch {
  175. detail.Content = ""
  176. }
  177. report.RiskLevelStatus = status
  178. report.Login = true
  179. }
  180. var jsonStr []byte
  181. jsonStr, err = json.Marshal(detail)
  182. if err != nil {
  183. logger.Error("生成研报详情失败:%v", err)
  184. err = exception.New(exception.GetReportFailed)
  185. }
  186. report.Detail = jsonStr
  187. return
  188. case SourceHT:
  189. pdfUrl, err = getHTReportDetail(&report)
  190. if err != nil {
  191. logger.Error("获取研报详情失败失败:%v")
  192. err = exception.New(exception.GetReportFailed)
  193. return
  194. }
  195. if !login {
  196. report.PdfUrl = ""
  197. report.RiskLevelStatus = RiskLevelUnMatch
  198. report.Login = false
  199. } else {
  200. if status == RiskLevelMatch {
  201. report.PdfUrl = pdfUrl
  202. }
  203. report.RiskLevelStatus = status
  204. report.Login = true
  205. }
  206. return
  207. default:
  208. logger.Error("不支持的研报来演:%v")
  209. err = exception.New(exception.GetReportFailed)
  210. return
  211. }
  212. }
  213. func getETAReportDetail(report *reportService.ReportDTO) (etaReport reportService.ETAReportDTO, err error) {
  214. return reportService.GetETAReport(report.OrgId)
  215. }
  216. func getHTReportDetail(report *reportService.ReportDTO) (url string, err error) {
  217. return reportService.GetHtReport(report.OrgId)
  218. }
  219. func GetTotalPageCountByPermissionIds(permissionIds []int, isLogin bool, userId int) (total int64, latestId int64, ids map[string][]int) {
  220. return getCount(permissionIds, isLogin, userId)
  221. }
  222. func filterPermissionsByRisk(permissionList []permissionService.PermissionDTO, riskLevel string) (resultList []permissionService.PermissionDTO) {
  223. if riskLevel != "" {
  224. riskLevelNum, err := config.ParseRiskLevel(riskLevel)
  225. if err != nil {
  226. logger.Error("风险等级解析失败:%v", err)
  227. return
  228. }
  229. for _, permission := range permissionList {
  230. pRiskNum, riskErr := config.ParseRiskLevel(permission.RiskLevel)
  231. if riskErr != nil {
  232. logger.Error("解析品种风险等级失败 permission:%d,risk:%v", permission.PermissionId, permission.RiskLevel)
  233. continue
  234. }
  235. if pRiskNum <= riskLevelNum {
  236. resultList = append(resultList, permission)
  237. }
  238. }
  239. } else {
  240. resultList = permissionList
  241. }
  242. return
  243. }
  244. // ParseRiskLevel 解析风险等级字符串,并返回数字部分
  245. func SearchReportList(key string, Ids []int, pageInfo page.PageInfo, isLogin bool, userId int) (list []reportService.ReportDTO, err error) {
  246. offset := page.StartIndex(pageInfo.Current, pageInfo.PageSize)
  247. var reports []reportService.ReportDTO
  248. reports, err = reportService.SearchReportList(key, Ids, offset, pageInfo.PageSize, pageInfo.LatestId)
  249. list, err = dealReportInfo(reports, isLogin, userId)
  250. if err != nil {
  251. err = exception.New(exception.SearchReportPageFailed)
  252. }
  253. return
  254. }
  255. func SearchReportProduct(key string, docIds []int) (list []reportService.ReportDTO, err error) {
  256. list, err = reportService.SearchReportProduct(key, 100, 0, docIds)
  257. if err != nil {
  258. err = exception.New(exception.SearchReportPageFailed)
  259. }
  260. return
  261. }
  262. func RangeSearchByAnalyst(analystName string, userId int) (total int64, latestId int64, ids []int) {
  263. return getCountByAnalyst(nil, true, userId, analystName)
  264. }
  265. func RangeSearch(key string, isLogin bool, userId int) (total int64, latestId int64, reportIds []int, err error) {
  266. var orgIds map[string][]int
  267. _, latestId, orgIds = getCount(nil, isLogin, userId)
  268. reportIds, err = GetReportByIdListByOrgIds(orgIds)
  269. if err != nil {
  270. logger.Error("获取报告ID列表失败:%v", err)
  271. err = exception.NewWithException(exception.GetReportSearchRangeFailed, err.Error())
  272. return
  273. }
  274. total = reportService.SearchMaxReportIdWithRange(key, reportIds)
  275. return
  276. }
  277. func dealReportInfo(list []reportService.ReportDTO, isLogin bool, userId int) (resultList []reportService.ReportDTO, err error) {
  278. var wg sync.WaitGroup
  279. wg.Add(len(list))
  280. for i := 0; i < len(list); i++ {
  281. go func(report *reportService.ReportDTO) {
  282. defer wg.Done()
  283. report, err = DealReportInfo(report, isLogin, userId)
  284. if err != nil {
  285. logger.Error("处理报告信息失败:%v", err)
  286. }
  287. }(&list[i])
  288. }
  289. wg.Wait()
  290. resultList = list
  291. return
  292. }
  293. func DealReportInfo(report *reportService.ReportDTO, isLogin bool, userId int) (resultReport *reportService.ReportDTO, err error) {
  294. report.Login = isLogin
  295. report.Permissions, report.PermissionNames = GetReportPermissionNames(report.OrgId, report.Source)
  296. var permissions []permissionService.PermissionDTO
  297. permissions, report.SecondPermission = getReportSecondPermissions(report.OrgId, report.Source)
  298. if len(permissions) == 0 {
  299. resultReport = report
  300. return
  301. }
  302. riskNum := config.GetHighestRiskLevel(permissions)
  303. report.RiskLevel = strings.Join([]string{"R", strconv.Itoa(riskNum)}, "")
  304. var src string
  305. src, err = mediaService.GetImageSrc(report.CoverSrc)
  306. if err != nil {
  307. logger.Error("获取图片地址失败:%v", err)
  308. src = ""
  309. } else {
  310. report.CoverUrl = src
  311. }
  312. //套餐信息
  313. var packageList []productService.MerchantProductDTO
  314. //查询产品信息
  315. product, pdErr := productService.GetProductBySourceId(report.ReportID, productDao.Report)
  316. if pdErr != nil {
  317. var permissionIds []int
  318. if len(permissions) > 0 {
  319. for _, permission := range permissions {
  320. permissionIds = append(permissionIds, permission.PermissionId)
  321. }
  322. //单品不存在的话查套餐
  323. packageList, err = productService.GetProductListBySourceIds(permissionIds, "package")
  324. if err != nil || len(packageList) == 0 {
  325. logger.Error("获取套餐列表失败:%v", err)
  326. report.Price = defaultProductPrice
  327. report.IsFree = true
  328. report.IsSubscribe = false
  329. report.IsPackage = false
  330. } else {
  331. report.Price = defaultProductPrice
  332. report.IsFree = false
  333. report.IsSubscribe = false
  334. report.IsPackage = true
  335. report.ProductId = packageList[0].Id
  336. }
  337. } else {
  338. report.Price = defaultProductPrice
  339. report.IsFree = true
  340. report.IsSubscribe = false
  341. report.IsPackage = false
  342. }
  343. } else {
  344. report.Price = product.Price.String()
  345. report.IsFree = false
  346. report.ProductId = product.Id
  347. report.IsPackage = false
  348. }
  349. if isLogin {
  350. var productIds []int
  351. if len(packageList) > 0 {
  352. for _, packageItem := range packageList {
  353. productIds = append(productIds, packageItem.Id)
  354. }
  355. }
  356. if product.Id > 0 {
  357. productIds = append(productIds, product.Id)
  358. }
  359. subscribeList, subscribeErr := userService.GetUserSubscribe(productIds, userId)
  360. if subscribeErr != nil {
  361. report.IsSubscribe = false
  362. } else {
  363. for _, subscribe := range subscribeList {
  364. if subscribe.Status == productDao.SubscribeValid {
  365. report.IsSubscribe = true
  366. break
  367. }
  368. }
  369. }
  370. }
  371. resultReport = report
  372. return
  373. }
  374. // GetReportPage 分页获取报告列表
  375. func GetReportPage(pageInfo page.PageInfo, orgIds map[string][]int, searchAll bool, isLogin bool, userId int) (reports []reportService.ReportDTO, err error) {
  376. var list []reportService.ReportDTO
  377. list, err = reportService.GetReportPageByOrgIds(pageInfo, orgIds, searchAll)
  378. reports, err = dealReportInfo(list, isLogin, userId)
  379. if err != nil {
  380. err = exception.New(exception.QueryReportPageFailed)
  381. }
  382. return
  383. }
  384. func GetReportPageByAnalyst(pageInfo page.PageInfo, analyst string, reportIds []int) (list []reportService.ReportDTO, err error) {
  385. list, err = reportService.GetReportPageByAnalyst(pageInfo, analyst, reportIds)
  386. //并发获取研报的标签
  387. var wg sync.WaitGroup
  388. wg.Add(len(list))
  389. for i := 0; i < len(list); i++ {
  390. go func(report *reportService.ReportDTO) {
  391. defer wg.Done()
  392. report.Permissions, report.PermissionNames = GetReportPermissionNames(report.OrgId, report.Source)
  393. }(&list[i])
  394. }
  395. wg.Wait()
  396. if err != nil {
  397. err = exception.New(exception.QueryReportPageFailed)
  398. }
  399. return
  400. }
  401. func CountReport(count RecordCount) (traceId string, err error) {
  402. report, err := reportService.GetReportById(count.ReportId)
  403. if err != nil {
  404. err = exception.New(exception.GetReportFailed)
  405. return
  406. }
  407. dto := convertToRecordCountDTO(count)
  408. dto.SourceTitle = report.Title
  409. return userService.CountReport(dto)
  410. }
  411. func GetRandedReportByWeeklyHot(limit int, isLogin bool, userId int, pdRiskLevel string) (reports []HotRankedReport, err error) {
  412. end := time.Now()
  413. begin := date.GetBeginOfTheWeek(end, time.Monday)
  414. hotReports := userService.GetHotReports(begin.Format(time.DateOnly), end.Format(time.DateOnly), limit)
  415. if len(hotReports) > 0 {
  416. var dtoList []reportService.ReportDTO
  417. var ids []int
  418. for i := 0; i < len(hotReports); i++ {
  419. ids = append(ids, hotReports[i].ReportId)
  420. }
  421. dtoList, err = reportService.GetListByCondition("id", ids)
  422. if err != nil {
  423. logger.Error("获取本周最热研报列表失败:%v", err)
  424. err = exception.New(exception.GetHotRandListFailed)
  425. return
  426. }
  427. dtoList, err = dealReportInfo(dtoList, isLogin, userId)
  428. if err != nil {
  429. logger.Error("获取本周最热研报列表失败:%v", err)
  430. err = exception.New(exception.GetHotRandListFailed)
  431. return
  432. }
  433. var filterList []reportService.ReportDTO
  434. if pdRiskLevel != "" {
  435. for _, report := range dtoList {
  436. pdRiskNum, paresErr := config.ParseRiskLevel(report.RiskLevel)
  437. if paresErr != nil {
  438. logger.Error("解析风险等级失败:%v", err)
  439. continue
  440. }
  441. reRiskNum, paresErr := config.ParseRiskLevel(pdRiskLevel)
  442. if paresErr != nil {
  443. logger.Error("解析风险等级失败:%v", err)
  444. continue
  445. }
  446. if pdRiskNum <= reRiskNum {
  447. filterList = append(filterList, report)
  448. }
  449. }
  450. } else {
  451. filterList = dtoList
  452. }
  453. reports = make([]HotRankedReport, len(ids))
  454. for i := 0; i < len(filterList); i++ {
  455. report := convertToHotRankedReport(filterList[i])
  456. for j := 0; j < len(hotReports); j++ {
  457. if hotReports[j].ReportId == report.Id {
  458. report.Count = hotReports[j].Count
  459. reports[j] = report
  460. break
  461. }
  462. }
  463. }
  464. } else {
  465. reports = []HotRankedReport{}
  466. }
  467. return
  468. }
  469. func GetRandedReportByPublishTimeWeekly(limit int, week bool, isLogin bool, userId int, pdRiskLevel string) (reports []PublishRankedReport, err error) {
  470. dtoList, err := reportService.GetListOrderByConditionWeekly(week, "published_time", limit, reportService.DESC)
  471. if err != nil {
  472. logger.Error("获取最新发布的研报列表失败:%v", err)
  473. err = exception.New(exception.GetPublishedRandListFailed)
  474. return
  475. }
  476. var filterList []reportService.ReportDTO
  477. dtoList, err = dealReportInfo(dtoList, isLogin, userId)
  478. if err != nil {
  479. logger.Error("获取最新发布的研报列表失败:%v", err)
  480. err = exception.New(exception.GetPublishedRandListFailed)
  481. return
  482. }
  483. if pdRiskLevel != "" {
  484. for _, report := range dtoList {
  485. pdRiskNum, paresErr := config.ParseRiskLevel(report.RiskLevel)
  486. if paresErr != nil {
  487. logger.Error("解析风险等级失败:%v", err)
  488. continue
  489. }
  490. reRiskNum, paresErr := config.ParseRiskLevel(pdRiskLevel)
  491. if paresErr != nil {
  492. logger.Error("解析风险等级失败:%v", err)
  493. continue
  494. }
  495. if pdRiskNum <= reRiskNum {
  496. filterList = append(filterList, report)
  497. }
  498. }
  499. } else {
  500. filterList = dtoList
  501. }
  502. reports = convertToPublishRankedReportList(filterList)
  503. return
  504. }
  505. func GetReportPermissionNames(id int, source string) (permissionMap map[int]string, labels []string) {
  506. permissions := reportService.GetReportPermissionsById(id, source)
  507. permissionMap = make(map[int]string, len(permissions))
  508. if len(permissions) > 0 {
  509. for _, permission := range permissions {
  510. labels = append(labels, permission.PermissionName)
  511. permissionMap[permission.PermissionId] = permission.PermissionName
  512. }
  513. }
  514. return
  515. }
  516. func GetReportSecondPermissionsMap(id int, source string) (permissionMap map[int]string) {
  517. permissionMap = make(map[int]string)
  518. permissions := reportService.GetReportSecondPermissionsById(id, source)
  519. for _, permission := range permissions {
  520. permissionMap[permission.PermissionId] = permission.PermissionName
  521. }
  522. return
  523. }
  524. func getReportSecondPermissions(id int, source string) (permissionList []permissionService.PermissionDTO, secondPermissionMap map[int]string) {
  525. permissionList = reportService.GetReportSecondPermissionsById(id, source)
  526. if len(permissionList) > 0 {
  527. secondPermissionMap = make(map[int]string, len(permissionList))
  528. for _, permission := range permissionList {
  529. secondPermissionMap[permission.PermissionId] = permission.PermissionName
  530. }
  531. }
  532. return
  533. }
  534. func getReportPermissionsMap(id int, source string) (permissionMap map[int]string) {
  535. permissionMap = make(map[int]string)
  536. permissions := reportService.GetReportPermissionsById(id, source)
  537. for _, permission := range permissions {
  538. permissionMap[permission.PermissionId] = permission.PermissionName
  539. }
  540. return
  541. }
  542. func GetPermissionList() (root *permissionService.PermissionNode, err error) {
  543. return permissionService.GetPermissionList()
  544. }
  545. func convertToHotRankedReport(dto reportService.ReportDTO) (report HotRankedReport) {
  546. report = HotRankedReport{
  547. Id: dto.ReportID,
  548. OrgId: dto.OrgId,
  549. Abstract: dto.Abstract,
  550. PublishedTime: dto.PublishedTime,
  551. Title: dto.Title,
  552. SecondPermissions: dto.SecondPermission,
  553. Permissions: dto.Permissions,
  554. PermissionNames: dto.PermissionNames,
  555. CoverUrl: dto.CoverUrl,
  556. IsSubscribe: dto.IsSubscribe,
  557. IsFree: dto.IsFree,
  558. Price: dto.Price,
  559. RiskLevel: dto.RiskLevel,
  560. Login: dto.Login,
  561. ProductId: dto.ProductId,
  562. }
  563. return
  564. }
  565. func convertToPublishRankedReportList(dtoList []reportService.ReportDTO) (reports []PublishRankedReport) {
  566. reports = []PublishRankedReport{}
  567. for _, dto := range dtoList {
  568. risk, err := config.ParseRiskLevel(dto.RiskLevel)
  569. if err != nil || risk == 0 {
  570. continue
  571. }
  572. src, err := mediaService.GetImageSrc(dto.CoverSrc)
  573. if err != nil {
  574. logger.Error("获取封面图片失败:%v", err)
  575. src = ""
  576. }
  577. report := PublishRankedReport{
  578. Id: dto.ReportID,
  579. OrgId: dto.OrgId,
  580. PublishedTime: dto.PublishedTime,
  581. Abstract: dto.Abstract,
  582. Title: dto.Title,
  583. Permissions: dto.Permissions,
  584. SecondPermissions: dto.SecondPermission,
  585. PermissionNames: dto.PermissionNames,
  586. CoverUrl: src,
  587. IsSubscribe: dto.IsSubscribe,
  588. IsFree: dto.IsFree,
  589. Price: dto.Price,
  590. RiskLevel: dto.RiskLevel,
  591. Login: dto.Login,
  592. ProductId: dto.ProductId,
  593. }
  594. reports = append(reports, report)
  595. }
  596. return
  597. }
  598. func convertToRecordCountDTO(record RecordCount) (dto userService.RecordCountDTO) {
  599. return userService.RecordCountDTO{
  600. UserId: record.UserId,
  601. TraceId: record.TraceId,
  602. Mobile: record.Mobile,
  603. SourceId: record.ReportId,
  604. IpAddress: record.IpAddress,
  605. Location: record.Location,
  606. Referer: record.Referer,
  607. Additional: record.Additional,
  608. }
  609. }
  610. func GetReportByIdListByOrgIds(orgIds map[string][]int) (ids []int, err error) {
  611. ids, err = reportService.GetReportByIdListByOrgIds(orgIds)
  612. if err != nil {
  613. logger.Error("获取报告ID列表失败:%v", err)
  614. err = exception.New(exception.GetReportSearchRangeFailed)
  615. }
  616. return
  617. }
  618. func RangePermissionIds(isLogin bool, userId int) (filterPermissionIds []int, riskLevel string, err error) {
  619. return CheckUserRisk(nil, isLogin, userId)
  620. }
  621. func CheckUserRisk(permissionIds []int, isLogin bool, userId int) (filterPermissionIds []int, riskLevel string, err error) {
  622. if isLogin {
  623. userProfile, userErr := user.GetUserProfile(userId)
  624. if userErr != nil {
  625. if errors.Is(userErr, gorm.ErrRecordNotFound) {
  626. err = exception.New(exception.TemplateUserNotFound)
  627. } else {
  628. err = exception.New(exception.TemplateUserFoundFailed)
  629. }
  630. logger.Error("分页查询列表失败:%v", err)
  631. return
  632. }
  633. //获取产品风险等级
  634. if userProfile.RiskLevelStatus == user.RiskUnTest {
  635. logger.Warn("客户未做风险等级测评,mobile:%v", userProfile.Mobile)
  636. }
  637. if userProfile.RiskLevelStatus == user.RiskExpired {
  638. logger.Warn("客户风险等级已过期,mobile:%v", userProfile.Mobile)
  639. }
  640. var mapping permissionService.CustomerProductRiskMappingDTO
  641. if userProfile.RiskLevel != "" {
  642. mapping, err = permissionService.GetRiskMappingByCustomerRiskLevel(userProfile.RiskLevel)
  643. if err != nil {
  644. logger.Error("查询产品风险等级映射失败:%v", err)
  645. return
  646. }
  647. }
  648. var permissionList []permissionService.PermissionDTO
  649. if len(permissionIds) == 0 {
  650. //获取所有设置风险等级的品种
  651. permissionList, err = permissionService.GetPermissionListWithRisk()
  652. } else {
  653. //更具id过滤设置了风险等级的品种
  654. permissionList, err = permissionService.GetPermissionListByIds(permissionIds)
  655. }
  656. permissionList = filterPermissionsByRisk(permissionList, mapping.ProductRiskLevel)
  657. riskLevel = mapping.ProductRiskLevel
  658. if len(permissionList) == 0 {
  659. return
  660. }
  661. for _, permission := range permissionList {
  662. filterPermissionIds = append(filterPermissionIds, permission.PermissionId)
  663. }
  664. return
  665. } else { //没有登录的时候展示所有设置了风险等级的品种报告,筛选的时候过滤传入ID中没有设置风险等级的品种
  666. var permissionList []permissionService.PermissionDTO
  667. if len(permissionIds) == 0 {
  668. //获取所有设置风险等级的品种
  669. permissionList, err = permissionService.GetPermissionListWithRisk()
  670. } else {
  671. //更具id过滤设置了风险等级的品种
  672. permissionList, err = permissionService.GetPermissionListByIds(permissionIds)
  673. }
  674. if err != nil {
  675. logger.Error("根据ID查询品种列表失败:%v", err)
  676. }
  677. for _, permission := range permissionList {
  678. filterPermissionIds = append(filterPermissionIds, permission.PermissionId)
  679. }
  680. //查询品种
  681. return
  682. }
  683. }
  684. func getCount(permissionIds []int, isLogin bool, userId int) (total int64, latestId int64, ids map[string][]int) {
  685. filterPermissionIds, riskLevel, err := CheckUserRisk(permissionIds, isLogin, userId)
  686. if err != nil {
  687. logger.Error("校验用户风险等级失败:%v", err)
  688. return
  689. }
  690. return reportService.GetTotalPageCountByPermissionIds(filterPermissionIds, riskLevel)
  691. }
  692. func getCountByAnalyst(permissionIds []int, isLogin bool, userId int, analystName string) (total int64, latestId int64, ids []int) {
  693. filterPermissionIds, riskLevel, err := CheckUserRisk(permissionIds, isLogin, userId)
  694. if err != nil {
  695. logger.Error("校验用户风险等级失败:%v", err)
  696. return
  697. }
  698. return reportService.GetTotalPageCountByAnalyst(analystName, filterPermissionIds, riskLevel)
  699. }
  700. func CountPermissionWeight(ids []int) (permissionMap map[int]int) {
  701. list, err := reportService.CountPermissionWeight(ids)
  702. permissionMap = make(map[int]int)
  703. if err != nil {
  704. return
  705. }
  706. for _, item := range list {
  707. permissionMap[item.PermissionId] = item.Weight
  708. }
  709. return
  710. }