Files
mce/internal/service/reaction_service.go

166 lines
4.5 KiB
Go

package service
import (
"fmt"
"time"
"metazone.cc/mce/internal/model"
)
// ReactionType 反应类型
type ReactionType string
const (
ReactionNone ReactionType = "none"
ReactionLiked ReactionType = "liked"
ReactionDisliked ReactionType = "disliked"
)
// ReactionService 赞/踩业务逻辑
type ReactionService struct {
repo ReactionStore
}
// NewReactionService 构造函数
func NewReactionService(repo ReactionStore) *ReactionService {
return &ReactionService{repo: repo}
}
// ToggleLike 切换点赞(含踩互斥逻辑)
// 返回:当前是否已点赞
func (s *ReactionService) ToggleLike(userID, postID uint) (bool, error) {
var isLiked bool
err := s.repo.Transaction(func(txRepo ReactionStore) error {
// 查当前状态
_, likeErr := txRepo.FindLike(userID, postID)
hasLike := likeErr == nil
_, dislikeErr := txRepo.FindDislike(userID, postID)
hasDislike := dislikeErr == nil
if hasLike {
// 已赞 → 取消赞
if err := txRepo.DeleteLike(userID, postID); err != nil {
return fmt.Errorf("取消点赞失败: %w", err)
}
if err := txRepo.IncrPostLikes(postID, -1); err != nil {
return fmt.Errorf("更新点赞计数失败: %w", err)
}
isLiked = false
} else {
// 无赞 → 加赞(若已有踩,先移除踩)
if hasDislike {
if err := txRepo.DeleteDislike(userID, postID); err != nil {
return fmt.Errorf("移除踩记录失败: %w", err)
}
if err := txRepo.IncrPostDislikes(postID, -1); err != nil {
return fmt.Errorf("更新踩计数失败: %w", err)
}
}
if err := txRepo.CreateLike(&model.PostLike{UserID: userID, PostID: postID}); err != nil {
return fmt.Errorf("创建点赞记录失败: %w", err)
}
if err := txRepo.IncrPostLikes(postID, 1); err != nil {
return fmt.Errorf("更新点赞计数失败: %w", err)
}
isLiked = true
}
return nil
})
if err == nil && isLiked {
// 新赞 → 记录到每日点赞汇总(非关键路径)
s.recordLikeForAggregation(postID, userID)
}
return isLiked, err
}
// recordLikeForAggregation 记录点赞到每日汇总(非关键路径,失败不阻塞)
func (s *ReactionService) recordLikeForAggregation(postID uint, likerID uint) {
authorID, err := s.repo.GetPostAuthorID(postID)
if err != nil {
return
}
// 不记录自己赞自己
if authorID == likerID {
return
}
date := time.Now().Format("2006-01-02")
likerUID := fmt.Sprintf("%d", likerID)
_ = s.repo.UpsertDailyLikeSummary(authorID, date, likerUID)
}
// ToggleDislike 切换踩(含赞互斥逻辑)
// 返回:当前是否已踩
func (s *ReactionService) ToggleDislike(userID, postID uint) (bool, error) {
var isDisliked bool
err := s.repo.Transaction(func(txRepo ReactionStore) error {
// 查当前状态
_, likeErr := txRepo.FindLike(userID, postID)
hasLike := likeErr == nil
_, dislikeErr := txRepo.FindDislike(userID, postID)
hasDislike := dislikeErr == nil
if hasDislike {
// 已踩 → 取消踩
if err := txRepo.DeleteDislike(userID, postID); err != nil {
return fmt.Errorf("取消踩失败: %w", err)
}
if err := txRepo.IncrPostDislikes(postID, -1); err != nil {
return fmt.Errorf("更新踩计数失败: %w", err)
}
isDisliked = false
} else {
// 无踩 → 加踩(若已有赞,先移除赞)
if hasLike {
if err := txRepo.DeleteLike(userID, postID); err != nil {
return fmt.Errorf("移除点赞记录失败: %w", err)
}
if err := txRepo.IncrPostLikes(postID, -1); err != nil {
return fmt.Errorf("更新点赞计数失败: %w", err)
}
}
if err := txRepo.CreateDislike(&model.PostDislike{UserID: userID, PostID: postID}); err != nil {
return fmt.Errorf("创建踩记录失败: %w", err)
}
if err := txRepo.IncrPostDislikes(postID, 1); err != nil {
return fmt.Errorf("更新踩计数失败: %w", err)
}
isDisliked = true
}
return nil
})
return isDisliked, err
}
// GetReaction 查询用户对文章的反应状态
func (s *ReactionService) GetReaction(userID, postID uint) (ReactionType, error) {
_, likeErr := s.repo.FindLike(userID, postID)
if likeErr == nil {
return ReactionLiked, nil
}
_, dislikeErr := s.repo.FindDislike(userID, postID)
if dislikeErr == nil {
return ReactionDisliked, nil
}
return ReactionNone, nil
}
// GetPostLikesCount 查询文章点赞数(公开展示)
func (s *ReactionService) GetPostLikesCount(postID uint) (int, error) {
count, err := s.repo.GetPostLikesCount(postID)
if err != nil {
return 0, fmt.Errorf("查询文章点赞数: %w", err)
}
return count, nil
}