slow_wild_backend/internal/logic/postcollectionlogic.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 PostCollectionLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

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

// 收藏帖子
func (l *PostCollectionLogic) PostCollection(in *slowwildserver.PostCollectionReq) (*slowwildserver.PostCollectionRsp, error) {
	// 从context中获取userId
	userId, err := cast.ToInt64E(l.ctx.Value(constants.UserIDKey))
	if err != nil {
		return nil, errorx.ErrInvalidParam
	}

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

	// 开启事务
	err = l.svcCtx.UserModel.Transaction(l.ctx, func(tx *gorm.DB) error {
		postModel := model.NewPostModel(tx, l.svcCtx.Redis)
		userModel := model.NewUserModel(tx)
		postActionModel := model.NewPostActionModel(tx, l.svcCtx.Redis)

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

		// 检查是否已经收藏
		isCollected, err := userModel.IsPostCollectedByUser(l.ctx, userId, in.PostId)
		if err != nil {
			return err
		}

		// 创建或更新收藏记录
		postAction := &model.PostAction{
			Model: &model.Model{
				CreatedOn:  time.Now().Unix(),
				ModifiedOn: time.Now().Unix(),
			},
			PostId:     in.PostId,
			UserId:     userId,
			ActionType: 1, // 1表示收藏
		}

		if !isCollected {
			// 收藏操作
			if err := postActionModel.CreatePostAction(l.ctx, postAction); err != nil {
				return err
			}
			// 增加收藏数
			if err := postModel.IncrementCollectionCount(l.ctx, in.PostId, 1); err != nil {
				return err
			}

			// 如果不是自己的帖子，发送消息通知
			if post.UserId != userId {
				messageModel := model.NewMessageModel(tx)
				err = messageModel.SendNotification(l.ctx, userId, post.UserId, 2,
					"收藏通知",
					"有人收藏了你的动态",
					in.PostId, 0, 0)
				if err != nil {
					return err
				}
			}
		} else {
			// 取消收藏操作
			postAction.IsDel = 1
			postAction.DeletedOn = time.Now().Unix()
			if err := postActionModel.DeletePostAction(l.ctx, postAction); err != nil {
				return err
			}
			// 减少收藏数
			if err := postModel.IncrementCollectionCount(l.ctx, in.PostId, -1); err != nil {
				return err
			}
		}

		return nil
	})

	if err != nil {
		if err == errorx.ErrInvalidOperation {
			return nil, err
		}
		l.Logger.Errorf("收藏帖子失败: %v", err)
		return nil, errorx.NewCodeError(20005, "收藏操作失败")
	}

	return &slowwildserver.PostCollectionRsp{}, nil
}