05c7f8a5eb
Add rich message segment support enabling structured content (text, images, votes, mentions) in posts and comments. Includes segments field in models, DTOs, handlers, and services. Also adds @mention notification processing and migrates vote data to embedded segments.
392 lines
11 KiB
Go
392 lines
11 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"log"
|
||
"strings"
|
||
"time"
|
||
|
||
"with_you/internal/cache"
|
||
"with_you/internal/dto"
|
||
"with_you/internal/model"
|
||
"with_you/internal/pkg/hook"
|
||
"with_you/internal/repository"
|
||
)
|
||
|
||
// VoteService 投票服务
|
||
type VoteService struct {
|
||
voteRepo repository.VoteRepository
|
||
postRepo repository.PostRepository
|
||
cache cache.Cache
|
||
postAIService *PostAIService
|
||
systemMessageService SystemMessageService
|
||
hookManager *hook.Manager
|
||
logService *LogService
|
||
}
|
||
|
||
func NewVoteService(
|
||
voteRepo repository.VoteRepository,
|
||
postRepo repository.PostRepository,
|
||
cache cache.Cache,
|
||
postAIService *PostAIService,
|
||
systemMessageService SystemMessageService,
|
||
hookManager *hook.Manager,
|
||
logService *LogService,
|
||
) *VoteService {
|
||
return &VoteService{
|
||
voteRepo: voteRepo,
|
||
postRepo: postRepo,
|
||
cache: cache,
|
||
postAIService: postAIService,
|
||
systemMessageService: systemMessageService,
|
||
hookManager: hookManager,
|
||
logService: logService,
|
||
}
|
||
}
|
||
|
||
// CreateVotePost 创建投票帖子
|
||
func (s *VoteService) CreateVotePost(ctx context.Context, userID string, req *dto.CreateVotePostRequest) (*dto.PostResponse, error) {
|
||
if len(req.VoteOptions) < 2 {
|
||
return nil, errors.New("投票选项至少需要2个")
|
||
}
|
||
if len(req.VoteOptions) > 10 {
|
||
return nil, errors.New("投票选项最多10个")
|
||
}
|
||
|
||
// 构建 vote segment
|
||
voteOptions := make([]dto.VoteOptionData, len(req.VoteOptions))
|
||
for i, opt := range req.VoteOptions {
|
||
voteOptions[i] = dto.VoteOptionData{
|
||
ID: fmt.Sprintf("opt_%d", i+1),
|
||
Content: opt,
|
||
}
|
||
}
|
||
segments := model.MessageSegments{
|
||
dto.NewTextSegment(req.Content),
|
||
{
|
||
Type: string(dto.SegmentTypeVote),
|
||
Data: map[string]any{
|
||
"options": voteOptions,
|
||
"max_choices": 1,
|
||
"is_multi_choice": false,
|
||
},
|
||
},
|
||
}
|
||
|
||
content := req.Content
|
||
if content == "" {
|
||
content = " "
|
||
}
|
||
|
||
post := &model.Post{
|
||
UserID: userID,
|
||
ChannelID: req.ChannelID,
|
||
Title: req.Title,
|
||
Content: content,
|
||
Segments: segments,
|
||
Status: model.PostStatusPending,
|
||
IsVote: true,
|
||
}
|
||
|
||
err := s.postRepo.Create(post, req.Images)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 异步审核
|
||
go s.reviewVotePostAsync(post.ID, userID, req.Title, content, req.Images)
|
||
|
||
// 重新查询以获取关联的User和Images
|
||
createdPost, err := s.postRepo.GetByID(post.ID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return s.convertToPostResponse(createdPost, userID), nil
|
||
}
|
||
|
||
func (s *VoteService) reviewVotePostAsync(postID, userID, title, content string, images []string) {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
log.Printf("[ERROR] Panic in vote post moderation async flow, fallback publish post=%s panic=%v", postID, r)
|
||
if err := s.updateModerationStatusWithRetry(postID, model.PostStatusPublished, "", "system"); err != nil {
|
||
log.Printf("[WARN] Failed to publish vote post %s after panic recovery: %v", postID, err)
|
||
}
|
||
}
|
||
}()
|
||
|
||
if s.hookManager == nil {
|
||
if err := s.updateModerationStatusWithRetry(postID, model.PostStatusPublished, "", "system"); err != nil {
|
||
log.Printf("[WARN] Failed to publish vote post without hook manager: %v", err)
|
||
} else {
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
cache.InvalidatePostList(s.cache)
|
||
}
|
||
return
|
||
}
|
||
|
||
var authorID uint
|
||
fmt.Sscanf(userID, "%d", &authorID)
|
||
|
||
result := &hook.ModerationResult{}
|
||
s.hookManager.TriggerWithMetadata(context.Background(), hook.HookPostPreModerate, authorID, &hook.PostModerateHookData{
|
||
PostID: postID,
|
||
Title: title,
|
||
Content: content,
|
||
Images: images,
|
||
AuthorID: authorID,
|
||
}, map[string]any{
|
||
"result": result,
|
||
})
|
||
|
||
s.hookManager.Trigger(context.Background(), hook.HookPostModerated, authorID, &hook.PostModeratedHookData{
|
||
PostID: postID,
|
||
AuthorID: authorID,
|
||
Approved: result.Approved,
|
||
RejectReason: result.RejectReason,
|
||
ReviewedBy: result.ReviewedBy,
|
||
})
|
||
|
||
if !result.Approved {
|
||
if result.NeedsReview {
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
cache.InvalidatePostList(s.cache)
|
||
return
|
||
}
|
||
if updateErr := s.updateModerationStatusWithRetry(postID, model.PostStatusRejected, result.RejectReason, result.ReviewedBy); updateErr != nil {
|
||
log.Printf("[WARN] Failed to reject vote post %s: %v", postID, updateErr)
|
||
} else {
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
cache.InvalidatePostList(s.cache)
|
||
}
|
||
s.notifyModerationRejected(userID, result.RejectReason)
|
||
return
|
||
}
|
||
|
||
if err := s.updateModerationStatusWithRetry(postID, model.PostStatusPublished, "", result.ReviewedBy); err != nil {
|
||
log.Printf("[WARN] Failed to publish vote post %s: %v", postID, err)
|
||
return
|
||
}
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
cache.InvalidatePostList(s.cache)
|
||
}
|
||
|
||
func (s *VoteService) updateModerationStatusWithRetry(postID string, status model.PostStatus, rejectReason string, reviewedBy string) error {
|
||
const maxAttempts = 3
|
||
const retryDelay = 200 * time.Millisecond
|
||
|
||
var lastErr error
|
||
for attempt := 1; attempt <= maxAttempts; attempt++ {
|
||
if err := s.postRepo.UpdateModerationStatus(postID, status, rejectReason, reviewedBy); err != nil {
|
||
lastErr = err
|
||
if attempt < maxAttempts {
|
||
log.Printf("[WARN] UpdateModerationStatus for vote post failed post=%s attempt=%d/%d err=%v", postID, attempt, maxAttempts, err)
|
||
time.Sleep(time.Duration(attempt) * retryDelay)
|
||
continue
|
||
}
|
||
} else {
|
||
return nil
|
||
}
|
||
}
|
||
return lastErr
|
||
}
|
||
|
||
func (s *VoteService) notifyModerationRejected(userID, reason string) {
|
||
if s.systemMessageService == nil || strings.TrimSpace(userID) == "" {
|
||
return
|
||
}
|
||
|
||
content := "您发布的投票帖未通过AI审核，请修改后重试。"
|
||
if strings.TrimSpace(reason) != "" {
|
||
content = fmt.Sprintf("您发布的投票帖未通过AI审核，原因：%s。请修改后重试。", reason)
|
||
}
|
||
|
||
go func() {
|
||
_ = s.systemMessageService.SendSystemAnnouncement(
|
||
context.Background(),
|
||
[]string{userID},
|
||
"投票帖审核未通过",
|
||
content,
|
||
)
|
||
}()
|
||
}
|
||
|
||
// resolveVoteOptions 解析投票选项（优先从 segments 读取，兼容旧数据从 vote_options 表读取）
|
||
func (s *VoteService) resolveVoteOptions(postID string, segments model.MessageSegments) ([]dto.VoteOptionDTO, error) {
|
||
if len(segments) > 0 {
|
||
voteData := dto.ExtractVoteSegmentData(segments)
|
||
if voteData != nil {
|
||
result := make([]dto.VoteOptionDTO, 0, len(voteData.Options))
|
||
for _, opt := range voteData.Options {
|
||
count, err := s.voteRepo.CountVotesByOption(postID, opt.ID)
|
||
if err != nil {
|
||
log.Printf("[WARN] Failed to count votes for option %s in post %s: %v", opt.ID, postID, err)
|
||
}
|
||
result = append(result, dto.VoteOptionDTO{
|
||
ID: opt.ID,
|
||
Content: opt.Content,
|
||
VotesCount: count,
|
||
})
|
||
}
|
||
return result, nil
|
||
}
|
||
}
|
||
|
||
options, err := s.voteRepo.GetOptionsByPostID(postID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
result := make([]dto.VoteOptionDTO, 0, len(options))
|
||
for _, option := range options {
|
||
result = append(result, dto.VoteOptionDTO{
|
||
ID: option.ID,
|
||
Content: option.Content,
|
||
VotesCount: option.VotesCount,
|
||
})
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// GetVoteOptions 获取投票选项（优先从 segments 读取，兼容旧数据从 vote_options 表读取）
|
||
func (s *VoteService) GetVoteOptions(postID string) ([]dto.VoteOptionDTO, error) {
|
||
post, err := s.postRepo.GetByID(postID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return s.resolveVoteOptions(postID, post.Segments)
|
||
}
|
||
|
||
// GetVoteResult 获取投票结果（包含用户投票状态）
|
||
func (s *VoteService) GetVoteResult(postID, userID string) (*dto.VoteResultDTO, error) {
|
||
post, err := s.postRepo.GetByID(postID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
optionDTOs, err := s.resolveVoteOptions(postID, post.Segments)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
userVote, err := s.voteRepo.GetUserVote(postID, userID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
result := &dto.VoteResultDTO{
|
||
Options: optionDTOs,
|
||
TotalVotes: 0,
|
||
HasVoted: userVote != nil,
|
||
}
|
||
|
||
if userVote != nil {
|
||
result.VotedOptionID = userVote.OptionID
|
||
}
|
||
|
||
for _, opt := range optionDTOs {
|
||
result.TotalVotes += opt.VotesCount
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// Vote 投票
|
||
func (s *VoteService) Vote(ctx context.Context, postID, userID, optionID string) error {
|
||
// 调用voteRepo.Vote
|
||
err := s.voteRepo.Vote(postID, userID, optionID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 失效帖子详情缓存
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
|
||
return nil
|
||
}
|
||
|
||
// Unvote 取消投票
|
||
func (s *VoteService) Unvote(ctx context.Context, postID, userID string) error {
|
||
// 调用voteRepo.Unvote
|
||
err := s.voteRepo.Unvote(postID, userID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 失效帖子详情缓存
|
||
cache.InvalidatePostDetail(s.cache, postID)
|
||
|
||
return nil
|
||
}
|
||
|
||
// UpdateVoteOption 更新投票选项（作者权限）
|
||
func (s *VoteService) UpdateVoteOption(ctx context.Context, postID, optionID, userID, content string) error {
|
||
// 获取帖子信息
|
||
post, err := s.postRepo.GetByID(postID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 验证用户是否为帖子作者
|
||
if post.UserID != userID {
|
||
return errors.New("只有帖子作者可以更新投票选项")
|
||
}
|
||
|
||
// 调用voteRepo.UpdateOption
|
||
return s.voteRepo.UpdateOption(optionID, content)
|
||
}
|
||
|
||
// convertToPostResponse 将Post模型转换为PostResponse DTO
|
||
func (s *VoteService) convertToPostResponse(post *model.Post, currentUserID string) *dto.PostResponse {
|
||
if post == nil {
|
||
return nil
|
||
}
|
||
|
||
response := &dto.PostResponse{
|
||
ID: post.ID,
|
||
UserID: post.UserID,
|
||
Title: post.Title,
|
||
Content: post.Content,
|
||
Segments: dto.SegmentsOrDefault(post.Segments, post.Content),
|
||
LikesCount: post.LikesCount,
|
||
CommentsCount: post.CommentsCount,
|
||
FavoritesCount: post.FavoritesCount,
|
||
SharesCount: post.SharesCount,
|
||
ViewsCount: post.ViewsCount,
|
||
IsPinned: post.IsPinned,
|
||
IsLocked: post.IsLocked,
|
||
IsVote: post.IsVote,
|
||
CreatedAt: dto.FormatTime(post.CreatedAt),
|
||
UpdatedAt: dto.FormatTime(post.UpdatedAt),
|
||
ContentEditedAt: dto.FormatTimePointer(post.ContentEditedAt),
|
||
Images: make([]dto.PostImageResponse, 0, len(post.Images)),
|
||
}
|
||
|
||
// 转换图片
|
||
for _, img := range post.Images {
|
||
response.Images = append(response.Images, dto.PostImageResponse{
|
||
ID: img.ID,
|
||
URL: img.URL,
|
||
ThumbnailURL: img.ThumbnailURL,
|
||
Width: img.Width,
|
||
Height: img.Height,
|
||
})
|
||
}
|
||
|
||
// 转换作者信息
|
||
if post.User != nil {
|
||
response.Author = &dto.UserResponse{
|
||
ID: post.User.ID,
|
||
Username: post.User.Username,
|
||
Nickname: post.User.Nickname,
|
||
Avatar: post.User.Avatar,
|
||
}
|
||
}
|
||
|
||
return response
|
||
}
|