// -------------------------------------------------------------------------------------------- // 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 PushCoinPoolFilterFunc func(cols *PushCoinPoolColumns) interface{} type PushCoinPoolUpdateFunc func(cols *PushCoinPoolColumns) interface{} type PushCoinPoolPipelineFunc func(cols *PushCoinPoolColumns) interface{} type PushCoinPoolCountOptionsFunc func(cols *PushCoinPoolColumns) *options.CountOptions type PushCoinPoolAggregateOptionsFunc func(cols *PushCoinPoolColumns) *options.AggregateOptions type PushCoinPoolFindOneOptionsFunc func(cols *PushCoinPoolColumns) *options.FindOneOptions type PushCoinPoolFindManyOptionsFunc func(cols *PushCoinPoolColumns) *options.FindOptions type PushCoinPoolUpdateOptionsFunc func(cols *PushCoinPoolColumns) *options.UpdateOptions type PushCoinPoolDeleteOptionsFunc func(cols *PushCoinPoolColumns) *options.DeleteOptions type PushCoinPoolInsertOneOptionsFunc func(cols *PushCoinPoolColumns) *options.InsertOneOptions type PushCoinPoolInsertManyOptionsFunc func(cols *PushCoinPoolColumns) *options.InsertManyOptions type PushCoinPool struct { Columns *PushCoinPoolColumns Database *mongo.Database Collection *mongo.Collection } type PushCoinPoolColumns struct { ID string Pool1 string Pool2 string Pool3 string } var pushCoinPoolColumns = &PushCoinPoolColumns{ ID: "_id", Pool1: "pool1", Pool2: "pool2", Pool3: "pool3", } func NewPushCoinPool() *PushCoinPool { return &PushCoinPool{ Columns: pushCoinPoolColumns, } } // Count returns the number of documents in the collection. func (dao *PushCoinPool) Count(ctx context.Context, filterFunc PushCoinPoolFilterFunc, optionsFunc ...PushCoinPoolCountOptionsFunc) (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 *PushCoinPool) Aggregate(ctx context.Context, pipelineFunc PushCoinPoolPipelineFunc, optionsFunc ...PushCoinPoolAggregateOptionsFunc) (*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 *PushCoinPool) InsertOne(ctx context.Context, model *modelpkg.PushCoinPool, optionsFunc ...PushCoinPoolInsertOneOptionsFunc) (*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 *PushCoinPool) InsertMany(ctx context.Context, models []*modelpkg.PushCoinPool, optionsFunc ...PushCoinPoolInsertManyOptionsFunc) (*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 *PushCoinPool) UpdateOne(ctx context.Context, filterFunc PushCoinPoolFilterFunc, updateFunc PushCoinPoolUpdateFunc, optionsFunc ...PushCoinPoolUpdateOptionsFunc) (*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 *PushCoinPool) UpdateOneByID(ctx context.Context, id string, updateFunc PushCoinPoolUpdateFunc, optionsFunc ...PushCoinPoolUpdateOptionsFunc) (*mongo.UpdateResult, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.UpdateOne(ctx, func(cols *PushCoinPoolColumns) interface{} { return bson.M{"_id": objectID} }, updateFunc, optionsFunc...) } // UpdateMany executes an update command to update documents in the collection. func (dao *PushCoinPool) UpdateMany(ctx context.Context, filterFunc PushCoinPoolFilterFunc, updateFunc PushCoinPoolUpdateFunc, optionsFunc ...PushCoinPoolUpdateOptionsFunc) (*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 *PushCoinPool) FindOne(ctx context.Context, filterFunc PushCoinPoolFilterFunc, optionsFunc ...PushCoinPoolFindOneOptionsFunc) (*modelpkg.PushCoinPool, error) { var ( opts *options.FindOneOptions model = &modelpkg.PushCoinPool{} 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 *PushCoinPool) FindOneByID(ctx context.Context, id string, optionsFunc ...PushCoinPoolFindOneOptionsFunc) (*modelpkg.PushCoinPool, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.FindOne(ctx, func(cols *PushCoinPoolColumns) interface{} { return bson.M{"_id": objectID} }, optionsFunc...) } // FindMany executes a find command and returns many models the matching documents in the collection. func (dao *PushCoinPool) FindMany(ctx context.Context, filterFunc PushCoinPoolFilterFunc, optionsFunc ...PushCoinPoolFindManyOptionsFunc) ([]*modelpkg.PushCoinPool, 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.PushCoinPool, 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 *PushCoinPool) DeleteOne(ctx context.Context, filterFunc PushCoinPoolFilterFunc, optionsFunc ...PushCoinPoolDeleteOptionsFunc) (*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 *PushCoinPool) DeleteOneByID(ctx context.Context, id string, optionsFunc ...PushCoinPoolDeleteOptionsFunc) (*mongo.DeleteResult, error) { objectID, err := primitive.ObjectIDFromHex(id) if err != nil { return nil, err } return dao.DeleteOne(ctx, func(cols *PushCoinPoolColumns) interface{} { return bson.M{"_id": objectID} }, optionsFunc...) } // DeleteMany executes a delete command to delete documents from the collection. func (dao *PushCoinPool) DeleteMany(ctx context.Context, filterFunc PushCoinPoolFilterFunc, optionsFunc ...PushCoinPoolDeleteOptionsFunc) (*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 *PushCoinPool) autofill(ctx context.Context, model *modelpkg.PushCoinPool) error { if model.ID.IsZero() { model.ID = primitive.NewObjectID() } return nil }