get_whatsapp_channel_search_logic.go 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. package whatsapp_channel
  2. import (
  3. "context"
  4. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  5. "github.com/suyuan32/simple-admin-common/utils/pointy"
  6. "github.com/suyuan32/simple-admin-core/rpc/types/core"
  7. "wechat-api/ent/predicate"
  8. "wechat-api/ent/whatsappchannel"
  9. "wechat-api/internal/utils/dberrorhandler"
  10. "wechat-api/internal/svc"
  11. "wechat-api/internal/types"
  12. "github.com/zeromicro/go-zero/core/logx"
  13. )
  14. type GetWhatsappChannelSearchLogic struct {
  15. logx.Logger
  16. ctx context.Context
  17. svcCtx *svc.ServiceContext
  18. }
  19. func NewGetWhatsappChannelSearchLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWhatsappChannelSearchLogic {
  20. return &GetWhatsappChannelSearchLogic{
  21. Logger: logx.WithContext(ctx),
  22. ctx: ctx,
  23. svcCtx: svcCtx}
  24. }
  25. func (l *GetWhatsappChannelSearchLogic) GetWhatsappChannelSearch(req *types.WhatsappChannelListReq) (*types.WhatsappChannelListResp, error) {
  26. isAdmin := l.ctx.Value("isAdmin").(bool)
  27. organizationId := l.ctx.Value("organizationId").(uint64)
  28. var predicates []predicate.WhatsappChannel
  29. if isAdmin && req.OrganizationId != nil {
  30. predicates = append(predicates, whatsappchannel.OrganizationID(*req.OrganizationId))
  31. } else {
  32. predicates = append(predicates, whatsappchannel.OrganizationID(organizationId))
  33. }
  34. data, err := l.svcCtx.DB.WhatsappChannel.Query().Where(predicates...).Page(l.ctx, req.Page, req.PageSize)
  35. if err != nil {
  36. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  37. }
  38. resp := &types.WhatsappChannelListResp{}
  39. resp.Msg = errormsg.Success
  40. resp.Data.Total = data.PageDetails.Total
  41. for _, v := range data.List {
  42. departmentInfo, err := l.svcCtx.CoreRpc.GetDepartmentById(l.ctx, &core.IDReq{Id: v.OrganizationID})
  43. if err != nil {
  44. l.Error("获取部门信息失败", err)
  45. }
  46. resp.Data.Data = append(resp.Data.Data,
  47. types.WhatsappChannelInfo{
  48. BaseIDInfo: types.BaseIDInfo{
  49. Id: &v.ID,
  50. CreatedAt: pointy.GetPointer(v.CreatedAt.UnixMilli()),
  51. UpdatedAt: pointy.GetPointer(v.UpdatedAt.UnixMilli()),
  52. },
  53. Status: &v.Status,
  54. Ak: &v.Ak,
  55. Sk: &v.Sk,
  56. WaId: &v.WaID,
  57. WaName: &v.WaName,
  58. WabaId: &v.WabaID,
  59. BusinessId: &v.BusinessID,
  60. OrganizationId: &v.OrganizationID,
  61. OrganizationName: departmentInfo.Name,
  62. VerifyAccount: &v.VerifyAccount,
  63. })
  64. }
  65. return resp, nil
  66. }