// -------------------------------------------------------------------------------------------- // The following code is automatically generated by the mongo-dao-generator tool. // Please do not modify this code manually to avoid being overwritten in the next generation. // For more tool details, please click the link to view https://github.com/dobyte/mongo-dao-generator // -------------------------------------------------------------------------------------------- package internal import ( "context" "errors" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" modelpkg "mongo.games.com/game/model" ) type RedPacketHistoryFilterFunc func(cols *RedPacketHistoryColumns) interface{} type RedPacketHistoryUpdateFunc func(cols *RedPacketHistoryColumns) interface{} type RedPacketHistoryPipelineFunc func(cols *RedPacketHistoryColumns) interface{} type RedPacketHistoryCountOptionsFunc func(cols *RedPacketHistoryColumns) *options.CountOptions type RedPacketHistoryAggregateOptionsFunc func(cols *RedPacketHistoryColumns) *options.AggregateOptions type RedPacketHistoryFindOneOptionsFunc func(cols *RedPacketHistoryColumns) *options.FindOneOptions type RedPacketHistoryFindManyOptionsFunc func(cols *RedPacketHistoryColumns) *options.FindOptions type RedPacketHistoryUpdateOptionsFunc func(cols *RedPacketHistoryColumns) *options.UpdateOptions type RedPacketHistoryDeleteOptionsFunc func(cols *RedPacketHistoryColumns) *options.DeleteOptions type RedPacketHistoryInsertOneOptionsFunc func(cols *RedPacketHistoryColumns) *options.InsertOneOptions type RedPacketHistoryInsertManyOptionsFunc func(cols *RedPacketHistoryColumns) *options.InsertManyOptions type RedPacketHistory struct { Columns *RedPacketHistoryColumns Database *mongo.Database Collection *mongo.Collection } type RedPacketHistoryColumns struct { Platform string // 平台 ID string Cid string // 红包活动id Snid string // 玩家id Ts string // 时间戳 ItemId string // 道具id ItemNum string // 道具数量 } var redPacketHistoryColumns = &RedPacketHistoryColumns{ Platform: "-", // 平台 ID: "_id", Cid: "cid", // 红包活动id Snid: "snid", // 玩家id Ts: "ts", // 时间戳 ItemId: "itemid", // 道具id ItemNum: "itemnum",// 道具数量 } func NewRedPacketHistory() *RedPacketHistory { return &RedPacketHistory{ Columns: redPacketHistoryColumns, } } // Count returns the number of documents in the collection. func (dao *RedPacketHistory) Count(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, optionsFunc ...RedPacketHistoryCountOptionsFunc) (int64, error) { var ( opts *options.CountOptions filter = filterFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.CountDocuments(ctx, filter, opts) } // Aggregate executes an aggregate command against the collection and returns a cursor over the resulting documents. func (dao *RedPacketHistory) Aggregate(ctx context.Context, pipelineFunc RedPacketHistoryPipelineFunc, optionsFunc ...RedPacketHistoryAggregateOptionsFunc) (*mongo.Cursor, error) { var ( opts *options.AggregateOptions pipeline = pipelineFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.Aggregate(ctx, pipeline, opts) } // InsertOne executes an insert command to insert a single document into the collection. func (dao *RedPacketHistory) InsertOne(ctx context.Context, model *modelpkg.RedPacketHistory, optionsFunc ...RedPacketHistoryInsertOneOptionsFunc) (*mongo.InsertOneResult, error) { if model == nil { return nil, errors.New("model is nil") } if err := dao.autofill(ctx, model); err != nil { return nil, err } var opts *options.InsertOneOptions if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.InsertOne(ctx, model, opts) } // InsertMany executes an insert command to insert multiple documents into the collection. func (dao *RedPacketHistory) InsertMany(ctx context.Context, models []*modelpkg.RedPacketHistory, optionsFunc ...RedPacketHistoryInsertManyOptionsFunc) (*mongo.InsertManyResult, error) { if len(models) == 0 { return nil, errors.New("models is empty") } documents := make([]interface{}, 0, len(models)) for i := range models { model := models[i] if err := dao.autofill(ctx, model); err != nil { return nil, err } documents = append(documents, model) } var opts *options.InsertManyOptions if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.InsertMany(ctx, documents, opts) } // UpdateOne executes an update command to update at most one document in the collection. func (dao *RedPacketHistory) UpdateOne(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, updateFunc RedPacketHistoryUpdateFunc, optionsFunc ...RedPacketHistoryUpdateOptionsFunc) (*mongo.UpdateResult, error) { var ( opts *options.UpdateOptions filter = filterFunc(dao.Columns) update = updateFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.UpdateOne(ctx, filter, update, opts) } // UpdateOneByID executes an update command to update at most one document in the collection. func (dao *RedPacketHistory) UpdateOneByID(ctx context.Context, id string, updateFunc RedPacketHistoryUpdateFunc, optionsFunc ...RedPacketHistoryUpdateOptionsFunc) (*mongo.UpdateResult, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.UpdateOne(ctx, func(cols *RedPacketHistoryColumns) interface{} { return bson.M{"_id": objectID} }, updateFunc, optionsFunc...) } // UpdateMany executes an update command to update documents in the collection. func (dao *RedPacketHistory) UpdateMany(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, updateFunc RedPacketHistoryUpdateFunc, optionsFunc ...RedPacketHistoryUpdateOptionsFunc) (*mongo.UpdateResult, error) { var ( opts *options.UpdateOptions filter = filterFunc(dao.Columns) update = updateFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.UpdateMany(ctx, filter, update, opts) } // FindOne executes a find command and returns a model for one document in the collection. func (dao *RedPacketHistory) FindOne(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, optionsFunc ...RedPacketHistoryFindOneOptionsFunc) (*modelpkg.RedPacketHistory, error) { var ( opts *options.FindOneOptions model = &modelpkg.RedPacketHistory{} filter = filterFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } err := dao.Collection.FindOne(ctx, filter, opts).Decode(model) if err != nil { if err == mongo.ErrNoDocuments { return nil, nil } return nil, err } return model, nil } // FindOneByID executes a find command and returns a model for one document in the collection. func (dao *RedPacketHistory) FindOneByID(ctx context.Context, id string, optionsFunc ...RedPacketHistoryFindOneOptionsFunc) (*modelpkg.RedPacketHistory, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.FindOne(ctx, func(cols *RedPacketHistoryColumns) interface{} { return bson.M{"_id": objectID} }, optionsFunc...) } // FindMany executes a find command and returns many models the matching documents in the collection. func (dao *RedPacketHistory) FindMany(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, optionsFunc ...RedPacketHistoryFindManyOptionsFunc) ([]*modelpkg.RedPacketHistory, error) { var ( opts *options.FindOptions filter = filterFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } cur, err := dao.Collection.Find(ctx, filter, opts) if err != nil { return nil, err } models := make([]*modelpkg.RedPacketHistory, 0) if err = cur.All(ctx, &models); err != nil { return nil, err } return models, nil } // DeleteOne executes a delete command to delete at most one document from the collection. func (dao *RedPacketHistory) DeleteOne(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, optionsFunc ...RedPacketHistoryDeleteOptionsFunc) (*mongo.DeleteResult, error) { var ( opts *options.DeleteOptions filter = filterFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.DeleteOne(ctx, filter, opts) } // DeleteOneByID executes a delete command to delete at most one document from the collection. func (dao *RedPacketHistory) DeleteOneByID(ctx context.Context, id string, optionsFunc ...RedPacketHistoryDeleteOptionsFunc) (*mongo.DeleteResult, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.DeleteOne(ctx, func(cols *RedPacketHistoryColumns) interface{} { return bson.M{"_id": objectID} }, optionsFunc...) } // DeleteMany executes a delete command to delete documents from the collection. func (dao *RedPacketHistory) DeleteMany(ctx context.Context, filterFunc RedPacketHistoryFilterFunc, optionsFunc ...RedPacketHistoryDeleteOptionsFunc) (*mongo.DeleteResult, error) { var ( opts *options.DeleteOptions filter = filterFunc(dao.Columns) ) if len(optionsFunc) > 0 { opts = optionsFunc[0](dao.Columns) } return dao.Collection.DeleteMany(ctx, filter, opts) } // autofill when inserting data func (dao *RedPacketHistory) autofill(ctx context.Context, model *modelpkg.RedPacketHistory) error { if model.ID.IsZero() { model.ID = primitive.NewObjectID() } return nil }