messageCenter/rpc/messagecenterclient/messagecenter.go

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

//go:generate mockgen -destination ./messagecenter_mock.go -package messagecenter -source $GOFILE

package messagecenterclient

import (
	"context"

	"SocialPlatform.messageCenter/rpc/messageCenter"
	"github.com/zeromicro/go-zero/zrpc"
)

type (
	GetResp   = messageCenter.GetResp
	CountReq  = messageCenter.CountReq
	CountResp = messageCenter.CountResp
	GetReq    = messageCenter.GetReq

	MessageCenter interface {
		//  查询数量
		Count(ctx context.Context, in *CountReq) (*CountResp, error)
		//  获取消息
		GetLast(ctx context.Context, in *GetReq) (*GetResp, error)
	}

	defaultMessageCenter struct {
		cli zrpc.Client
	}
)

func NewMessageCenter(cli zrpc.Client) MessageCenter {
	return &defaultMessageCenter{
		cli: cli,
	}
}

//  查询数量
func (m *defaultMessageCenter) Count(ctx context.Context, in *CountReq) (*CountResp, error) {
	client := messageCenter.NewMessageCenterClient(m.cli.Conn())
	return client.Count(ctx, in)
}

//  获取消息
func (m *defaultMessageCenter) GetLast(ctx context.Context, in *GetReq) (*GetResp, error) {
	client := messageCenter.NewMessageCenterClient(m.cli.Conn())
	return client.GetLast(ctx, in)
}