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 (
	SendMsgRequest          = message.SendMsgRequest
	ChangeReadStatusRequest = message.ChangeReadStatusRequest
	Response                = message.Response

	Message interface {
		// 修改消息阅读状态
		ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error)
		// 指定用户发消息
		SendUserMsg(ctx context.Context, in *SendMsgRequest) (*Response, 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)
}