Initial commit: web_hghl_server

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-10-26 16:28:14 +08:00
commit 65fca60622
653 changed files with 135901 additions and 0 deletions
@@ -0,0 +1,265 @@
+package playletchaptermod
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/mongos"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+var coll *mongos.Collection
+
+// Init 初始化 playlet_chapter 集合和常用查询索引
+func Init(ctx context.Context) {
+	coll = mongos.Shop.Coll(TN.PlayletChapter)
+	initIndex(ctx)
+}
+
+func initIndex(ctx context.Context) {
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.PlayletId, Value: 1},
+				{Key: FN.PlayletChapter.CurrentEpisode, Value: 1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.PlayletId, Value: 1},
+				{Key: FN.PlayletChapter.MediaContentId, Value: 1},
+			},
+			Options: options.Index().
+				SetUnique(true).
+				SetPartialFilterExpression(bson.M{
+					FN.PlayletChapter.MediaContentId: bson.M{"$gt": 0},
+				}),
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.PlayletId, Value: 1},
+				{Key: FN.PlayletChapter.CreatedAt, Value: -1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.CreatedAt, Value: -1},
+				{Key: FN.PlayletChapter.Id, Value: -1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.WatermarkStatus, Value: 1},
+				{Key: FN.PlayletChapter.CreatedAt, Value: -1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.PlayletChapter.MediaSourceId, Value: 1},
+			},
+			Options: options.Index().
+				SetPartialFilterExpression(bson.M{
+					FN.PlayletChapter.MediaSourceId: bson.M{"$gt": ""},
+				}),
+		},
+	}
+	if _, err := coll.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(playletchaptermod:initIndex) Create Index failed! err:%+v:", err))
+	}
+}
+
+// InsertOnePlayletChapter 插入一条章节数据
+func InsertOnePlayletChapter(ctx context.Context, doc *PlayletChapter) *srvdef.SrvErr {
+	if doc.Id == 0 {
+		doc.Id = mongos.Lid.PlayletChapter.Get()
+	}
+	now := time.Now()
+	if doc.CreatedAt.IsZero() {
+		doc.CreatedAt = now
+	}
+	doc.UpdatedAt = now
+	if _, err := coll.InsertOne(ctx, doc); err != nil {
+		log.Warn("playletchaptermod:InsertOnePlayletChapter failed", log.Any("doc", doc), log.E(err))
+		return srvdef.New(srvdef.ErrDbInsertError, "", "")
+	}
+	return nil
+}
+
+// InsertManyPlayletChapter 批量插入章节数据
+func InsertManyPlayletChapter(ctx context.Context, docs PlayletChapterList) *srvdef.SrvErr {
+	data := make(PlayletChapterList, 0, len(docs))
+	now := time.Now()
+	for _, doc := range docs {
+		if doc == nil {
+			continue
+		}
+		if doc.Id == 0 {
+			doc.Id = mongos.Lid.PlayletChapter.Get()
+		}
+		if doc.CreatedAt.IsZero() {
+			doc.CreatedAt = now
+		}
+		doc.UpdatedAt = now
+		data = append(data, doc)
+	}
+	if len(data) == 0 {
+		return nil
+	}
+	if _, err := coll.InsertMany(ctx, &data); err != nil {
+		log.Warn("playletchaptermod:InsertManyPlayletChapter failed", log.Any("count", len(data)), log.E(err))
+		return srvdef.New(srvdef.ErrDbInsertError, "", "")
+	}
+	return nil
+}
+
+// DelOnePlayletChapter 删除一条章节数据
+func DelOnePlayletChapter(ctx context.Context, filter bson.M) (int64, *srvdef.SrvErr) {
+	count, err := coll.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Warn("playletchaptermod:DelOnePlayletChapter failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "")
+	}
+	return count, nil
+}
+
+// DelManyPlayletChapter 按条件批量删除章节数据
+func DelManyPlayletChapter(ctx context.Context, filter bson.M) (int64, *srvdef.SrvErr) {
+	count, err := coll.DeleteMany(ctx, filter)
+	if err != nil {
+		log.Warn("playletchaptermod:DelManyPlayletChapter failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "")
+	}
+	return count, nil
+}
+
+// FindOnePlayletChapter 查询一条章节数据
+func FindOnePlayletChapter(ctx context.Context, filter bson.M, opts ...*options.FindOneOptions) (*PlayletChapter, *srvdef.SrvErr) {
+	var doc PlayletChapter
+	if err := coll.FindOneEx(ctx, &doc, filter, opts...); err != nil {
+		if err != mongo.ErrNoDocuments {
+			log.Warn("playletchaptermod:FindOnePlayletChapter failed", log.E(err), log.Any("filter", filter))
+			return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+		}
+		return nil, srvdef.New(srvdef.ErrNoData, "", "")
+	}
+	return &doc, nil
+}
+
+// FindManyPlayletChapter 按条件查询多条章节数据
+func FindManyPlayletChapter(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (PlayletChapterList, *srvdef.SrvErr) {
+	data := make(PlayletChapterList, 0)
+	if err := coll.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("playletchaptermod:FindManyPlayletChapter failed", log.E(err), log.Any("filter", filter))
+		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return data, nil
+}
+
+// CountPlayletChapter 统计满足条件的章节数量
+func CountPlayletChapter(ctx context.Context, filter bson.M, opts ...*options.CountOptions) (int64, *srvdef.SrvErr) {
+	count, err := coll.Count(ctx, filter, opts...)
+	if err != nil {
+		log.Warn("playletchaptermod:CountPlayletChapter failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return count, nil
+}
+
+// UpdateOnePlayletChapter 更新一条章节数据
+func UpdateOnePlayletChapter(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
+	result, err := coll.UpdateOne(ctx, filter, update)
+	if err != nil {
+		log.Warn("playletchaptermod:UpdateOnePlayletChapter failed", log.E(err), log.Any("filter", filter))
+		return srvdef.New(srvdef.ErrDbUpdateError, "", "")
+	}
+	if result.MatchedCount != 1 {
+		return srvdef.New(srvdef.ErrNoData, "", "")
+	}
+	return nil
+}
+
+// IsLinkedMedia 是否短剧媒资：显式标记或已挂到剧集
+func IsLinkedMedia(ctx context.Context, isPlaylet bool, ids ...string) bool {
+	if isPlaylet {
+		return true
+	}
+	return ExistsByMediaIDs(ctx, ids...)
+}
+
+// ExistsByMediaIDs 判断媒资 ID 是否属于短剧剧集（mediaSourceId / mediaTaskId）
+func ExistsByMediaIDs(ctx context.Context, ids ...string) bool {
+	clean := uniqueNonEmptyStrings(ids)
+	if len(clean) == 0 {
+		return false
+	}
+	item, _ := FindOnePlayletChapter(ctx, bson.M{
+		"$or": []bson.M{
+			{FN.PlayletChapter.MediaSourceId: bson.M{"$in": clean}},
+			{FN.PlayletChapter.MediaTaskId: bson.M{"$in": clean}},
+		},
+	}, options.FindOne().SetProjection(bson.M{FN.PlayletChapter.Id: 1}))
+	return item != nil
+}
+
+// ListLinkedMediaIDs 返回短剧剧集关联的媒资 ID，用于排除短剧视频
+func ListLinkedMediaIDs(ctx context.Context) []string {
+	list, err := FindManyPlayletChapter(ctx, bson.M{
+		"$or": []bson.M{
+			{FN.PlayletChapter.MediaSourceId: bson.M{"$gt": ""}},
+			{FN.PlayletChapter.MediaTaskId: bson.M{"$gt": ""}},
+		},
+	}, options.Find().SetProjection(bson.M{
+		FN.PlayletChapter.MediaSourceId: 1,
+		FN.PlayletChapter.MediaTaskId:   1,
+	}))
+	if err != nil || len(list) == 0 {
+		return nil
+	}
+	ids := make([]string, 0, len(list)*2)
+	for _, item := range list {
+		if item == nil {
+			continue
+		}
+		ids = append(ids, item.MediaSourceId, item.MediaTaskId)
+	}
+	return uniqueNonEmptyStrings(ids)
+}
+
+func uniqueNonEmptyStrings(ids []string) []string {
+	if len(ids) == 0 {
+		return nil
+	}
+	seen := make(map[string]struct{}, len(ids))
+	out := make([]string, 0, len(ids))
+	for _, id := range ids {
+		id = strings.TrimSpace(id)
+		if id == "" {
+			continue
+		}
+		if _, ok := seen[id]; ok {
+			continue
+		}
+		seen[id] = struct{}{}
+		out = append(out, id)
+	}
+	return out
+}
+
+// UpdateManyPlayletChapter 批量更新章节数据
+func UpdateManyPlayletChapter(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
+	result, err := coll.UpdateMany(ctx, filter, update)
+	if err != nil {
+		log.Warn("批量更新短剧章节数据失败", log.E(err), log.Any("filter", filter))
+		return srvdef.New(srvdef.ErrDbUpdateError, "", "")
+	}
+	if result.MatchedCount == 0 {
+		return srvdef.New(srvdef.ErrNoData, "", "")
+	}
+	return nil
+}