permission_service.go 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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. // 获取一级品种(海通的板块)
  36. func GetReportPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  37. switch source {
  38. case SourceETA:
  39. //展示一级品种
  40. return getETAReportFirstPermissions(id)
  41. case SourceHT:
  42. return []PermissionDTO{}
  43. }
  44. return
  45. }
  46. func GetReportSecondPermissionsById(id int, source string) (permissionIds []PermissionDTO) {
  47. switch source {
  48. case SourceETA:
  49. //展示一级品种
  50. return getETAReportSecondPermissions(id)
  51. case SourceHT:
  52. return []PermissionDTO{}
  53. }
  54. return
  55. }
  56. func convertPermissionDTO(node etaDao.ChartPermission) PermissionDTO {
  57. return PermissionDTO{
  58. ID: node.ChartPermissionID,
  59. Name: node.PermissionName,
  60. ParentID: node.ParentID,
  61. }
  62. }