messageCenter/api/messagecenter/internal/logic/messagecountlogic.go

package logic

import (
	"context"
	"messagecenter/rpc/messagecenter/messagecenter"

	"messagecenter/api/messagecenter/internal/svc"
	"messagecenter/api/messagecenter/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type MessageCountLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewMessageCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) *MessageCountLogic {
	return &MessageCountLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *MessageCountLogic) MessageCount(req *types.CountReq) (*types.CommonRes, error) {
	resp, err := l.svcCtx.Message.Count(l.ctx, &messagecenter.CountReq{
		UserType:  req.UserType,
		EntUserId: req.EntUserId,
		NewUserId: req.NewUserId,
	})
	if err != nil {
		return nil, err
	}
	return &types.CommonRes{
		Error_msg:  resp.ErrorMsg,
		Error_code: int(resp.ErrorCode),
		Data:       resp.Count,
	}, nil
}