266 lines
7.7 KiB
Go
266 lines
7.7 KiB
Go
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
|
||
}
|