MessageCenter/api/internal/logic/messageunreadclasscountlogic.go

package logic

import (
	"app.yhyue.com/moapp/MessageCenter/api/internal/svc"
	"app.yhyue.com/moapp/MessageCenter/api/internal/types"
	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
	"context"

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

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

func NewMessageUnreadClassCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) MessageUnreadClassCountLogic {
	return MessageUnreadClassCountLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

// 获取用户收到的分类及未读消息数量
func (l *MessageUnreadClassCountLogic) MessageUnreadClassCount(req types.MessageUnreadClassCountReq) (*types.MessageUnreadClassCountRes, error) {
	result := &types.MessageUnreadClassCountRes{}
	lsi := l.svcCtx.MessageCenter
	resp, err := lsi.GetUnreadClassCount(l.ctx, &messageclient.GetUnreadClassCountReq{
		Appid:  req.AppId,
		UserId: req.UserId,
	})
	if err != nil {
		return nil, err
	}

	result.Code = resp.Code
	result.Message = resp.Message
	for _, v := range resp.Data {
		result.Data = append(result.Data, map[string]interface{}{
			"type":  v.MsgType,
			"count": v.Count,
		})

	}
	return result, nil
}