init: web_hgdjc_server

2026-11-07 10:44:09 +08:00
commit 71a4f4e175
713 changed files with 147749 additions and 0 deletions
@@ -0,0 +1,100 @@
+package srv_aicomment
+
+import (
+	"context"
+	"time"
+
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/models/aicommentrecord"
+)
+
+const (
+	staleSubmittedTimeout = 60 * time.Minute
+	staleSubmittedDesc    = "自动评论任务提交超过60分钟未返回，已释放并等待补交"
+)
+
+type aiCommentContentKey struct {
+	objType int64
+	vid     string
+}
+
+func preparePendingAICommentVideos(ctx context.Context, videos []VideoItem) ([]VideoItem, *srvdef.SrvErr) {
+	if serr := expireStaleSubmittedTasks(ctx); serr != nil {
+		return nil, serr
+	}
+	vidsByType := make(map[int64][]string)
+	for _, video := range videos {
+		if video.VideoId != "" {
+			vidsByType[video.ObjType] = append(vidsByType[video.ObjType], video.VideoId)
+		}
+	}
+
+	recordsByContent := make(map[aiCommentContentKey]aicommentrecord.AiCommentRecordList)
+	for objType, vids := range vidsByType {
+		records, serr := aicommentrecord.FindByVIDsAndObjType(ctx, objType, vids)
+		if serr != nil {
+			return nil, serr
+		}
+		for _, record := range records {
+			if record == nil {
+				continue
+			}
+			key := aiCommentContentKey{objType: objType, vid: record.Vid}
+			recordsByContent[key] = append(recordsByContent[key], record)
+		}
+	}
+
+	pending := make([]VideoItem, 0, len(videos))
+	for _, video := range videos {
+		key := aiCommentContentKey{objType: video.ObjType, vid: video.VideoId}
+		if resumed, ok := resumeAICommentVideo(video, recordsByContent[key]); ok {
+			pending = append(pending, resumed)
+		}
+	}
+	return pending, nil
+}
+
+func expireStaleSubmittedTasks(ctx context.Context) *srvdef.SrvErr {
+	deadline := time.Now().Add(-staleSubmittedTimeout)
+	expired, serr := aicommentrecord.ExpireSubmittedBefore(ctx, deadline, staleSubmittedDesc)
+	if serr != nil {
+		return serr
+	}
+	if expired > 0 {
+		log.Warn("自动评论释放超时提交任务", log.Any("count", expired), log.Any("deadline", deadline))
+	}
+	return nil
+}
+
+func resumeAICommentVideo(video VideoItem, records aicommentrecord.AiCommentRecordList) (VideoItem, bool) {
+	targetCount := aiCommentPlayletTotal
+	if video.CurrentCount >= int64(targetCount) {
+		return VideoItem{}, false
+	}
+
+	accountedCount := 0
+	for _, record := range records {
+		if record == nil {
+			continue
+		}
+		if record.Status != aicommentrecord.StatusSubmitted && record.Status != aicommentrecord.StatusSuccess {
+			continue
+		}
+		retainCount := record.RetainCount
+		if retainCount <= 0 {
+			retainCount = commentsPerGenerationTask
+		}
+		accountedCount += retainCount
+	}
+	if video.CurrentCount > 0 && accountedCount < int(video.CurrentCount) {
+		accountedCount = int(video.CurrentCount)
+	}
+	if accountedCount >= targetCount {
+		return VideoItem{}, false
+	}
+
+	video.TargetCount = targetCount
+	video.CommentTotal = targetCount - accountedCount
+	return video, true
+}