permission_service.go 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package report
  2. import (
  3. logger "eta/eta_mini_ht_api/common/component/log"
  4. etaDao "eta/eta_mini_ht_api/models/eta"
  5. )
  6. func GetClassifyIdsByPermissionIds(permissionIds []int) (classifyIds []int, err error) {
  7. return etaDao.GetClassifyIdsByPermissionIds(permissionIds)
  8. }
  9. func GetPermissionList() (dtoList []PermissionDTO, err error) {
  10. list, err := etaDao.GetChartPermissionList()
  11. if err != nil {
  12. logger.Error("获取研报列表失败")
  13. return
  14. }
  15. dtoList = make([]PermissionDTO, 0)
  16. for _, node := range list {
  17. dto := convertPermissionDTO(node)
  18. dtoList = append(dtoList, dto)
  19. }
  20. return
  21. }
  22. func GetFirstPermissionsByIds(ids []int) (permissionDTOS []PermissionDTO, err error) {
  23. list, err := etaDao.GetFirstPermissionsByIds(ids)
  24. if err != nil {
  25. logger.Error("查询品种信息失败:%v", err)
  26. }
  27. for _, node := range list {
  28. permissionDTOS = append(permissionDTOS, convertPermissionDTO(node))
  29. }
  30. return
  31. }
  32. // 获取一级品种(海通的板块)
  33. func GetReportPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  34. switch source {
  35. case SourceETA:
  36. //展示一级品种
  37. return getETAReportFirstPermissions(id)
  38. case SourceHT:
  39. return []PermissionDTO{}
  40. }
  41. return
  42. }
  43. func GetReportSecondPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  44. switch source {
  45. case SourceETA:
  46. //展示一级品种
  47. return getETAReportSecondPermissions(id)
  48. case SourceHT:
  49. return []PermissionDTO{}
  50. }
  51. return
  52. }
  53. func convertPermissionDTO(node etaDao.ChartPermission) PermissionDTO {
  54. return PermissionDTO{
  55. ID: node.ChartPermissionID,
  56. Name: node.PermissionName,
  57. ParentID: node.ParentID,
  58. }
  59. }