slow_wild_backend/internal/logic/postcommentlogic.go

package logic

import (
	"context"
	"slowwild/internal/constants"
	"slowwild/internal/errorx"
	"slowwild/internal/model"
	"slowwild/internal/svc"
	"time"

	"git.banshen.xyz/huangguangrong/slow_wild_protobuff/slowwild/slowwildserver"
	"github.com/spf13/cast"
	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
)

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

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

// 发布评论
func (l *PostCommentLogic) PostComment(in *slowwildserver.PostCommentReq) (*slowwildserver.PostCommentRsp, error) {
	// 从context中获取userId
	userId, err := cast.ToInt64E(l.ctx.Value(constants.UserIDKey))
	if err != nil {
		return nil, errorx.ErrInvalidParam
	}

	if in.PostId <= 0 || in.Content == "" {
		return nil, errorx.ErrInvalidParam
	}

	// 开启事务
	err = l.svcCtx.UserModel.Transaction(l.ctx, func(tx *gorm.DB) error {
		postModel := model.NewPostModel(tx, l.svcCtx.Redis)
		commentModel := model.NewCommentModel(tx, l.svcCtx.Redis)
		messageModel := model.NewMessageModel(tx)

		// 检查帖子是否存在
		post, err := postModel.GetPostDetail(l.ctx, in.PostId)
		if err != nil {
			return err
		}

		// 构建评论数据
		comment := &model.Comment{
			Model: &model.Model{
				CreatedOn:  time.Now().Unix(),
				ModifiedOn: time.Now().Unix(),
			},
			PostId:      in.PostId,
			UserId:      userId,
			Content:     in.Content,
			Image:       in.Image,
			WithUserIds: cast.ToString(in.AtUserId),
			Ip:          in.Ip,
			IpLoc:       in.IpLoc,
		}

		// 创建评论
		if err := commentModel.Create(l.ctx, comment); err != nil {
			return err
		}

		// 增加帖子评论数
		if err := postModel.IncrementCommentCount(l.ctx, in.PostId, 1); err != nil {
			return err
		}

		// 如果不是评论自己的帖子，发送消息通知
		if post.UserId != userId {
			err = messageModel.SendNotification(l.ctx, userId, post.UserId, 3,
				"评论通知",
				"有人评论了你的动态",
				in.PostId, comment.ID, 0)
			if err != nil {
				return err
			}
		}

		// 如果有@用户，给被@的用户发送消息通知
		if in.AtUserId > 0 {
			err = messageModel.SendNotification(l.ctx, userId, in.AtUserId, 4,
				"@通知",
				"有人在评论中@了你",
				in.PostId, comment.ID, 0)
			if err != nil {
				return err
			}
		}

		return nil
	})

	if err != nil {
		l.Logger.Errorf("发布评论失败: %v", err)
		return nil, errorx.NewCodeError(20006, "发布评论失败")
	}

	return &slowwildserver.PostCommentRsp{}, nil
}