package contactusmod 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.ContactUsModel) // 获取数据库表 indexes := []mongo.IndexModel{ { Keys: bson.D{ {Key: FN.ContactUsModel.Status, Value: 1}, }, }, { Keys: bson.D{ {Key: FN.ContactUsModel.CreatedAt, Value: -1}, }, }, { Keys: bson.D{ {Key: FN.ContactUsModel.Name, Value: 1}, }, }, { Keys: bson.D{ {Key: FN.ContactUsModel.Email, Value: 1}, }, }, { Keys: bson.D{ {Key: FN.ContactUsModel.Ip, Value: 1}, }, }, } if _, err := db.CreateIndex(ctx, indexes); err != nil { panic(fmt.Sprintf("(ContactUsModel:initContactUsModelIndex) Create ContactUsModel Index failed! err:%+v:", err)) } } // InsertOneContactUsModel 插入一个评论 func InsertOneContactUsModel(ctx context.Context, ContactUsModel *ContactUsModel) *srvdef.SrvErr { if _, err := db.InsertOne(ctx, ContactUsModel); err != nil { log.Warn("ContactUsModel:InsertOneContactUsModel db.InsertOne failed", log.E(err)) return srvdef.New(srvdef.ErrDbInsertError, "", "") } return nil } // DelOneContactUsModel 删除一个评论 func DelOneContactUsModel(ctx context.Context, filter bson.M) ( int64, *srvdef.SrvErr) { count, err := db.DeleteOne(ctx, filter) if err != nil { log.Warn("ContactUsModel:DelOneContactUsModel db.DeleteOne failed", log.E(err), log.Any("filter", filter)) return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "") } return count, nil } // DelManyContactUsModel 删除一个评论评论 func DelManyContactUsModel(ctx context.Context, filter bson.M) ( int64, *srvdef.SrvErr) { count, err := db.DeleteMany(ctx, filter) if err != nil { log.Warn("ContactUsModel:DelOneContactUsModel db.DeleteOne failed", log.E(err), log.Any("filter", filter)) return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "") } return count, nil } // FindOneContactUsModel 查找一个评论 func FindOneContactUsModel(ctx context.Context, filter bson.M, opts ...*options.FindOneOptions) ( *ContactUsModel, *srvdef.SrvErr) { var ContactUsModel ContactUsModel if err := db.FindOneEx( ctx, &ContactUsModel, filter, opts...); err != nil { if err != mongo.ErrNoDocuments { log.Info("ContactUsModel:FindOneContactUsModel db.FindOne failed", log.E(err), log.Any("filter", filter)) return nil, srvdef.New(srvdef.ErrDbQueryError, "", "") } return nil, srvdef.New(srvdef.ErrNoData, "", "") } return &ContactUsModel, nil } // CountContactUsModels 统计评论视频评论数量 func CountContactUsModels(ctx context.Context, filter bson.M) ( int64, *srvdef.SrvErr) { count, err := db.Count(ctx, filter) if err != nil { log.Warn("ContactUsModel:CountContactUsModels db.Count failed", log.E(err), log.Any("filter", filter)) return 0, srvdef.New(srvdef.ErrDbQueryError, "", "") } return count, nil } // UpdateOneContactUsModel 更新一个评论 func UpdateOneContactUsModel(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr { result, err := db.UpdateOne(ctx, filter, update) if err != nil { log.Warn("ContactUsModel:UpdateOneContactUsModel db.UpdateOne failed", log.E(err), log.Any("filter", filter)) return srvdef.New(srvdef.ErrDbUpdateError, "", "") } if result.MatchedCount != 1 { log.Warn("ContactUsModel:UpdateOneContactUsModel db.UpdateOne failed", log.E(err), log.Any("filter", filter)) return srvdef.New(srvdef.ErrNoData, "", "") } return nil } // UpdateManyContactUsModel 更新一批评论 func UpdateManyContactUsModel(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr { result, err := db.UpdateMMany(ctx, filter, update) if err != nil { log.Warn("ContactUsModel:UpdateManyContactUsModel db.UpdateMMany failed", log.E(err), log.Any("filter", filter)) return srvdef.New(srvdef.ErrDbUpdateError, "", "") } if result.MatchedCount <= 0 { log.Warn("ContactUsModel:UpdateManyContactUsModel db.UpdateMMany failed", log.E(err), log.Any("filter", filter)) //return srvdef.New(srvdef.ErrContactUsModelNotExsit, "", "") } return nil } // FindManyContactUsModel 查找一批评论 func FindManyContactUsModel(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (ContactUsModelList, *srvdef.SrvErr) { data := make(ContactUsModelList, 0) if err := db.Find(ctx, &data, filter, opts...); err != nil { log.Warn("ContactUsModel:FindManyContactUsModel db.Find failed", log.E(err), log.Any("filter", filter)) return nil, srvdef.New(srvdef.ErrDbQueryError, "", "") } return data, nil } // InsertManyContactUsModel 插入一批用户 func InsertManyContactUsModel(ctx context.Context, histrory *ContactUsModelList) *srvdef.SrvErr { if _, err := db.InsertMany(ctx, histrory); err != nil { log.Warn("ContactUsModel:InsertManyContactUsModel db.InsertMany failed", log.E(err)) return srvdef.New(srvdef.ErrDbInsertError, "", "") } return nil } // AggregateContactUsModel 聚合查找一批数据 func AggregateContactUsModel(ctx context.Context, bind interface{}, pipeline []bson.M) *srvdef.SrvErr { if err := db.Aggregate(ctx, bind, pipeline); err != nil { log.Warn("ContactUsModel:AggregateContactUsModel db.Aggregate failed", log.E(err), log.Any("pipeline", pipeline)) return srvdef.New(srvdef.ErrDbQueryError, "", "") } return nil } // ObjectAggregateContactUsModel 聚合查找一批数据 func ObjectAggregateContactUsModel(ctx context.Context, bind interface{}, pipeline []bson.M) *srvdef.SrvErr { if err := db.ObjectAggregate(ctx, bind, pipeline); err != nil { log.Warn("ContactUsModel:AggregateContactUsModel db.Aggregate failed", log.E(err), log.Any("pipeline", pipeline)) return srvdef.New(srvdef.ErrDbQueryError, "", "") } return nil }