get_label_by_id_logic.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package label
  2. import (
  3. "context"
  4. "wechat-api/ent/label"
  5. "wechat-api/internal/svc"
  6. "wechat-api/internal/types"
  7. "wechat-api/internal/utils/dberrorhandler"
  8. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  9. "github.com/suyuan32/simple-admin-common/utils/pointy"
  10. "github.com/zeromicro/go-zero/core/logx"
  11. )
  12. type GetLabelByIdLogic struct {
  13. ctx context.Context
  14. svcCtx *svc.ServiceContext
  15. logx.Logger
  16. }
  17. func NewGetLabelByIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetLabelByIdLogic {
  18. return &GetLabelByIdLogic{
  19. ctx: ctx,
  20. svcCtx: svcCtx,
  21. Logger: logx.WithContext(ctx),
  22. }
  23. }
  24. func (l *GetLabelByIdLogic) GetLabelById(req *types.IDReq) (*types.LabelInfoResp, error) {
  25. organizationId := l.ctx.Value("organizationId").(uint64)
  26. data, err := l.svcCtx.DB.Label.Query().
  27. Where(
  28. label.IDEQ(req.Id), // Filter by ID
  29. label.OrganizationID(organizationId), // Additional filter by organizationId
  30. ).
  31. Only(l.ctx)
  32. if err != nil {
  33. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  34. }
  35. return &types.LabelInfoResp{
  36. BaseDataInfo: types.BaseDataInfo{
  37. Code: 0,
  38. Msg: errormsg.Success,
  39. },
  40. Data: types.LabelInfo{
  41. BaseIDInfo: types.BaseIDInfo{
  42. Id: &data.ID,
  43. CreatedAt: pointy.GetPointer(data.CreatedAt.UnixMilli()),
  44. UpdatedAt: pointy.GetPointer(data.UpdatedAt.UnixMilli()),
  45. },
  46. Status: &data.Status,
  47. Type: &data.Type,
  48. Name: &data.Name,
  49. From: &data.From,
  50. Mode: &data.Mode,
  51. Conditions: &data.Conditions,
  52. },
  53. }, nil
  54. }