MessageCenter/rpc/messageclient/message.go

// Code generated by goctl. DO NOT EDIT!
// Source: message.proto

//go:generate mockgen -destination ./message_mock.go -package messageclient -source $GOFILE

package messageclient

import (
	"context"

	"app.yhyue.com/moapp/MessageCenter/rpc/message"

	"github.com/tal-tech/go-zero/zrpc"
)

type (
	GetUnreadCountResponse       = message.GetUnreadCountResponse
	SendMsgRequest               = message.SendMsgRequest
	ChangeReadStatusRequest      = message.ChangeReadStatusRequest
	DeleteSingleMessageRequest   = message.DeleteSingleMessageRequest
	DeleteMultipleMessageRequest = message.DeleteMultipleMessageRequest
	GetUnreadCountRequest        = message.GetUnreadCountRequest
	Response                     = message.Response

	Message interface {
		//  修改消息阅读状态
		ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error)
		// 指定用户发消息
		SendUserMsg(ctx context.Context, in *SendMsgRequest) (*Response, error)
		//  删除单一消息
		DeleteSingleMessage(ctx context.Context, in *DeleteSingleMessageRequest) (*Response, error)
		//  删除多条消息
		DeleteMultipleMessage(ctx context.Context, in *DeleteMultipleMessageRequest) (*Response, error)
		//   查询指定用户未读消息合计
		GetUnreadCount(ctx context.Context, in *DeleteMultipleMessageRequest) (*GetUnreadCountResponse, error)
	}

	defaultMessage struct {
		cli zrpc.Client
	}
)

func NewMessage(cli zrpc.Client) Message {
	return &defaultMessage{
		cli: cli,
	}
}

//  修改消息阅读状态
func (m *defaultMessage) ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.ChangeReadStatus(ctx, in)
}

// 指定用户发消息
func (m *defaultMessage) SendUserMsg(ctx context.Context, in *SendMsgRequest) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.SendUserMsg(ctx, in)
}

//  删除单一消息
func (m *defaultMessage) DeleteSingleMessage(ctx context.Context, in *DeleteSingleMessageRequest) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.DeleteSingleMessage(ctx, in)
}

//  删除多条消息
func (m *defaultMessage) DeleteMultipleMessage(ctx context.Context, in *DeleteMultipleMessageRequest) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.DeleteMultipleMessage(ctx, in)
}

//   查询指定用户未读消息合计
func (m *defaultMessage) GetUnreadCount(ctx context.Context, in *DeleteMultipleMessageRequest) (*GetUnreadCountResponse, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.GetUnreadCount(ctx, in)
}