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

	Message interface {
		Ping(ctx context.Context, in *Request) (*Response, error)
		ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error)
	}

	defaultMessage struct {
		cli zrpc.Client
	}
)

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

func (m *defaultMessage) Ping(ctx context.Context, in *Request) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.Ping(ctx, in)
}

func (m *defaultMessage) ChangeReadStatus(ctx context.Context, in *ChangeReadStatusRequest) (*Response, error) {
	client := message.NewMessageClient(m.cli.Conn())
	return client.ChangeReadStatus(ctx, in)
}