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

package logic

import (
	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/api/messagecenter/util"
	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/rpc/messagecenter/messagecenter"
	"context"
	"net/http"

	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/api/messagecenter/internal/svc"
	"bp.jydev.jianyu360.cn/SocialPlatform/messageCenter/api/messagecenter/internal/types"

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

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

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

func (l *MessageCountLogic) MessageCount(req *types.CountReq) (resp *types.CommonRes, err error) {
	if req.NewUserId == 0 {
		r, err := util.GetTouristInfo(l.r, l.svcCtx.Social, l.ctx)
		if err != nil || r.BaseUserId == 0 {
			return nil, err
		}
		req.NewUserId = r.BaseUserId
	}
	res, err := l.svcCtx.Message.Count(l.ctx, &messagecenter.CountReq{
		UserType:   req.UserType,
		EntUserId:  req.EntUserId,
		NewUserId:  req.NewUserId,
		PositionId: req.PositionId,
	})
	if err != nil {
		return nil, err
	}
	resp = &types.CommonRes{
		Error_msg:  res.ErrorMsg,
		Error_code: int(res.ErrorCode),
		Count:      res.Count,
		Data:       res.LastMessage,
	}
	return resp, nil
}