permission_service.go 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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 GetPermissionNamesByPermissionIds(permissionIds []int) (permissionNames []string, err error) {
  10. return etaDao.GetPermissionNamesByPermissionIds(permissionIds)
  11. }
  12. func GetPermissionList() (dtoList []PermissionDTO, err error) {
  13. list, err := etaDao.GetChartPermissionList()
  14. if err != nil {
  15. logger.Error("获取研报列表失败")
  16. return
  17. }
  18. dtoList = make([]PermissionDTO, 0)
  19. for _, node := range list {
  20. dto := convertPermissionDTO(node)
  21. dtoList = append(dtoList, dto)
  22. }
  23. return
  24. }
  25. func GetFirstPermissionsByIds(ids []int) (permissionDTOS []PermissionDTO, err error) {
  26. list, err := etaDao.GetFirstPermissionsByIds(ids)
  27. if err != nil {
  28. logger.Error("查询品种信息失败:%v", err)
  29. }
  30. for _, node := range list {
  31. permissionDTOS = append(permissionDTOS, convertPermissionDTO(node))
  32. }
  33. return
  34. }
  35. // 获取二级品种id
  36. func GetPermissionIdByName(name string) (permissionId int, err error) {
  37. return etaDao.GetPermissionIdByName(name)
  38. }
  39. // 获取一级品种(海通的板块)
  40. func GetReportPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  41. switch source {
  42. case SourceETA:
  43. //展示一级品种
  44. return getETAReportFirstPermissions(id)
  45. case SourceHT:
  46. return []PermissionDTO{}
  47. }
  48. return
  49. }
  50. func GetReportSecondPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  51. switch source {
  52. case SourceETA:
  53. //展示一级品种
  54. return getETAReportSecondPermissions(id)
  55. case SourceHT:
  56. return []PermissionDTO{}
  57. }
  58. return
  59. }
  60. func convertPermissionDTO(node etaDao.ChartPermission) PermissionDTO {
  61. return PermissionDTO{
  62. ID: node.ChartPermissionID,
  63. Name: node.PermissionName,
  64. ParentID: node.ParentID,
  65. }
  66. }