message.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. // Code generated by goctl. DO NOT EDIT!
  2. // Source: message.proto
  3. //go:generate mockgen -destination ./message_mock.go -package messageclient -source $GOFILE
  4. package messageclient
  5. import (
  6. "context"
  7. "app.yhyue.com/moapp/MessageCenter/rpc/message"
  8. "github.com/tal-tech/go-zero/zrpc"
  9. )
  10. type (
  11. Request = message.Request
  12. ChangeReadStatusRequest = message.ChangeReadStatusRequest
  13. DeleteSingleMessageRequest = message.DeleteSingleMessageRequest
  14. DeleteMultipleMessageRequest = message.DeleteMultipleMessageRequest
  15. GetUnreadCountRequest = message.GetUnreadCountRequest
  16. Response = message.Response
  17. GetUnreadCountResponse = message.GetUnreadCountResponse
  18. Message interface {
  19. Ping(ctx context.Context, in *Request) (*Response, error)
  20. // 修改消息阅读状态
  21. ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error)
  22. // 删除单一消息
  23. DeleteSingleMessage(ctx context.Context, in *DeleteSingleMessageRequest) (*Response, error)
  24. // 删除多条消息
  25. DeleteMultipleMessage(ctx context.Context, in *DeleteMultipleMessageRequest) (*Response, error)
  26. // 查询指定用户未读消息合计
  27. GetUnreadCount(ctx context.Context, in *DeleteMultipleMessageRequest) (*GetUnreadCountResponse, error)
  28. }
  29. defaultMessage struct {
  30. cli zrpc.Client
  31. }
  32. )
  33. func NewMessage(cli zrpc.Client) Message {
  34. return &defaultMessage{
  35. cli: cli,
  36. }
  37. }
  38. func (m *defaultMessage) Ping(ctx context.Context, in *Request) (*Response, error) {
  39. client := message.NewMessageClient(m.cli.Conn())
  40. return client.Ping(ctx, in)
  41. }
  42. // 修改消息阅读状态
  43. func (m *defaultMessage) ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error) {
  44. client := message.NewMessageClient(m.cli.Conn())
  45. return client.ChangeReadStatus(ctx, in)
  46. }
  47. // 删除单一消息
  48. func (m *defaultMessage) DeleteSingleMessage(ctx context.Context, in *DeleteSingleMessageRequest) (*Response, error) {
  49. client := message.NewMessageClient(m.cli.Conn())
  50. return client.DeleteSingleMessage(ctx, in)
  51. }
  52. // 删除多条消息
  53. func (m *defaultMessage) DeleteMultipleMessage(ctx context.Context, in *DeleteMultipleMessageRequest) (*Response, error) {
  54. client := message.NewMessageClient(m.cli.Conn())
  55. return client.DeleteMultipleMessage(ctx, in)
  56. }
  57. // 查询指定用户未读消息合计
  58. func (m *defaultMessage) GetUnreadCount(ctx context.Context, in *DeleteMultipleMessageRequest) (*GetUnreadCountResponse, error) {
  59. client := message.NewMessageClient(m.cli.Conn())
  60. return client.GetUnreadCount(ctx, in)
  61. }