package syncfilemod import ( "context" "fmt" "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 db *mongos.Collection func Init(ctx context.Context) { // 获取数据库表 db = mongos.Shop.Coll(TN.SyncRecord) // 初始化索引 indexes := []mongo.IndexModel{ { Keys: bson.D{ {Key: FN.SyncRecord.CreatedAt, Value: -1}, }, }, } if _, err := db.CreateIndex(ctx, indexes); err != nil { panic(fmt.Sprintf("(SyncRecord:InitSyncRecord) Create SyncRecord Index failed! err:%+v:", err)) } } // CountSyncRecord 统计活动数量 func CountSyncRecord(ctx context.Context, filter bson.M) ( int64, *srvdef.SrvErr) { count, err := db.Count(ctx, filter) if err != nil { log.Warn("SyncRecordmod:CountSyncRecord db.Count failed", log.E(err), log.Any("filter", filter)) return 0, srvdef.New(srvdef.ErrDbQueryError, "", "") } return count, nil } // InsertOneSyncRecord 插入一条活动 func InsertOneSyncRecord(ctx context.Context, SyncRecord *SyncRecord) *srvdef.SrvErr { if _, err := db.InsertOne(ctx, SyncRecord); err != nil { log.Info("SyncRecordmod:InsertOneSyncRecord db.InsertOne failed", log.E(err), log.Any("SyncRecord", SyncRecord)) return srvdef.New(srvdef.ErrDbInsertError, "", "") } return nil } // DelOneSyncRecord 删除一个活动 func DelOneSyncRecord(ctx context.Context, filter bson.M) ( int64, *srvdef.SrvErr) { count, err := db.DeleteOne(ctx, filter) if err != nil { log.Warn("SyncRecordmod:DelOneSyncRecord db.DeleteOne failed", log.E(err), log.Any("filter", filter)) return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "") } return count, nil } // UpdateOneSyncRecord 更新一个SyncRecord func UpdateOneSyncRecord(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr { result, err := db.UpdateOne(ctx, filter, update) if err != nil { log.Warn("SyncRecordmod:UpdateOneSyncRecord db.UpdateOne failed", log.E(err), log.Any("filter", filter)) return srvdef.New(srvdef.ErrDbUpdateError, "", "") } if result.MatchedCount != 1 { log.Warn("SyncRecordmod:UpdateOneSyncRecord db.UpdateOne failed", log.E(err), log.Any("filter", filter)) return srvdef.New(srvdef.ObjectNotExist, "", "") } return nil } // FindManySyncRecord 查找一批活动数据 func FindManySyncRecord(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (SyncRecordList, *srvdef.SrvErr) { data := make(SyncRecordList, 0) if err := db.Find(ctx, &data, filter, opts...); err != nil { log.Warn("SyncRecordmod:FindManyAcivities db.Find failed", log.E(err), log.Any("filter", filter)) return nil, srvdef.New(srvdef.ErrDbQueryError, "", "") } return data, nil } // FindOneSyncRecord 查找一个媒体 func FindOneSyncRecord(ctx context.Context, filter bson.M) ( *SyncRecord, *srvdef.SrvErr) { var SyncRecord SyncRecord if err := db.FindOneEx(ctx, &SyncRecord, filter); err != nil { if err != mongo.ErrNoDocuments { log.Warn("SyncRecord:FindOneSyncRecord db.FindOne failed", log.E(err), log.Any("filter", filter)) return nil, srvdef.New(srvdef.ErrDbQueryError, "", "") } return nil, srvdef.New(srvdef.ObjectDayStatDoesNotExist, "", "") } return &SyncRecord, nil }