service.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. package service
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "gadmin/config"
  6. "gadmin/internal/admin/forms"
  7. "gadmin/internal/gorm/model"
  8. "gadmin/internal/gorm/query"
  9. "gadmin/utility/token"
  10. "github.com/gin-gonic/gin"
  11. "github.com/sirupsen/logrus"
  12. "time"
  13. "github.com/patrickmn/go-cache"
  14. )
  15. var Service = &sService{
  16. SystemCache: cache.New(5*time.Minute, 10*time.Minute),
  17. }
  18. type sService struct {
  19. SystemCache *cache.Cache
  20. }
  21. func (s *sService) ServiceList(c *gin.Context) ([]*forms.Service, error) {
  22. cUser, ok := c.Get("user")
  23. if !ok {
  24. return nil, errors.New("获取用户信息失败")
  25. }
  26. claims := cUser.(*token.UserClaims)
  27. roleId := claims.RoleId
  28. var (
  29. q = query.Use(config.AdminDB).AdminRole
  30. sq = query.Use(config.AdminDB).AdminSystem
  31. )
  32. systemModels := make([]*model.AdminSystem, 0)
  33. if config.IsSuperRole(roleId) {
  34. models, err := sq.Find()
  35. if err != nil {
  36. logrus.Error(err)
  37. return nil, err
  38. }
  39. systemModels = models
  40. } else {
  41. systems := ""
  42. if err := q.Where(q.ID.Eq(roleId)).Pluck(q.Systems, &systems); err != nil {
  43. return nil, err
  44. }
  45. if systems == "" {
  46. return []*forms.Service{}, nil
  47. }
  48. systemIds := make([]int32, 0)
  49. if err := json.Unmarshal([]byte(systems), &systemIds); err != nil {
  50. return nil, err
  51. }
  52. models, err := sq.Where(sq.ID.In(systemIds...)).Order(sq.ID).Find()
  53. if err != nil {
  54. logrus.Error(err)
  55. return nil, err
  56. }
  57. systemModels = models
  58. }
  59. var services []*forms.Service
  60. for _, v := range systemModels {
  61. services = append(services, &forms.Service{
  62. Id: v.ID,
  63. Name: v.Name,
  64. Path: v.URL + "?access-token={access-token}",
  65. })
  66. }
  67. return services, nil
  68. }