280 lines
9.0 KiB
Go
280 lines
9.0 KiB
Go
// --------------------------------------------------------------------------------------------
|
|
// 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 JybCodeLogFilterFunc func(cols *JybCodeLogColumns) interface{}
|
|
type JybCodeLogUpdateFunc func(cols *JybCodeLogColumns) interface{}
|
|
type JybCodeLogPipelineFunc func(cols *JybCodeLogColumns) interface{}
|
|
type JybCodeLogCountOptionsFunc func(cols *JybCodeLogColumns) *options.CountOptions
|
|
type JybCodeLogAggregateOptionsFunc func(cols *JybCodeLogColumns) *options.AggregateOptions
|
|
type JybCodeLogFindOneOptionsFunc func(cols *JybCodeLogColumns) *options.FindOneOptions
|
|
type JybCodeLogFindManyOptionsFunc func(cols *JybCodeLogColumns) *options.FindOptions
|
|
type JybCodeLogUpdateOptionsFunc func(cols *JybCodeLogColumns) *options.UpdateOptions
|
|
type JybCodeLogDeleteOptionsFunc func(cols *JybCodeLogColumns) *options.DeleteOptions
|
|
type JybCodeLogInsertOneOptionsFunc func(cols *JybCodeLogColumns) *options.InsertOneOptions
|
|
type JybCodeLogInsertManyOptionsFunc func(cols *JybCodeLogColumns) *options.InsertManyOptions
|
|
|
|
type JybCodeLog struct {
|
|
Columns *JybCodeLogColumns
|
|
Database *mongo.Database
|
|
Collection *mongo.Collection
|
|
}
|
|
|
|
type JybCodeLogColumns struct {
|
|
Id string
|
|
JybId string // 礼包id
|
|
Ts string // 兑换时间
|
|
SnId string // 玩家id
|
|
Code string // 兑换码
|
|
}
|
|
|
|
var jybCodeLogColumns = &JybCodeLogColumns{
|
|
Id: "_id",
|
|
JybId: "JybId", // 礼包id
|
|
Ts: "Ts", // 兑换时间
|
|
SnId: "SnId", // 玩家id
|
|
Code: "Code", // 兑换码
|
|
}
|
|
|
|
func NewJybCodeLog() *JybCodeLog {
|
|
return &JybCodeLog{
|
|
Columns: jybCodeLogColumns,
|
|
}
|
|
}
|
|
|
|
// Count returns the number of documents in the collection.
|
|
func (dao *JybCodeLog) Count(ctx context.Context, filterFunc JybCodeLogFilterFunc, optionsFunc ...JybCodeLogCountOptionsFunc) (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 *JybCodeLog) Aggregate(ctx context.Context, pipelineFunc JybCodeLogPipelineFunc, optionsFunc ...JybCodeLogAggregateOptionsFunc) (*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 *JybCodeLog) InsertOne(ctx context.Context, model *modelpkg.JybCodeLog, optionsFunc ...JybCodeLogInsertOneOptionsFunc) (*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 *JybCodeLog) InsertMany(ctx context.Context, models []*modelpkg.JybCodeLog, optionsFunc ...JybCodeLogInsertManyOptionsFunc) (*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 *JybCodeLog) UpdateOne(ctx context.Context, filterFunc JybCodeLogFilterFunc, updateFunc JybCodeLogUpdateFunc, optionsFunc ...JybCodeLogUpdateOptionsFunc) (*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 *JybCodeLog) UpdateOneByID(ctx context.Context, id string, updateFunc JybCodeLogUpdateFunc, optionsFunc ...JybCodeLogUpdateOptionsFunc) (*mongo.UpdateResult, error) {
|
|
objectID, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dao.UpdateOne(ctx, func(cols *JybCodeLogColumns) interface{} {
|
|
return bson.M{"_id": objectID}
|
|
}, updateFunc, optionsFunc...)
|
|
}
|
|
|
|
// UpdateMany executes an update command to update documents in the collection.
|
|
func (dao *JybCodeLog) UpdateMany(ctx context.Context, filterFunc JybCodeLogFilterFunc, updateFunc JybCodeLogUpdateFunc, optionsFunc ...JybCodeLogUpdateOptionsFunc) (*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 *JybCodeLog) FindOne(ctx context.Context, filterFunc JybCodeLogFilterFunc, optionsFunc ...JybCodeLogFindOneOptionsFunc) (*modelpkg.JybCodeLog, error) {
|
|
var (
|
|
opts *options.FindOneOptions
|
|
model = &modelpkg.JybCodeLog{}
|
|
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 *JybCodeLog) FindOneByID(ctx context.Context, id string, optionsFunc ...JybCodeLogFindOneOptionsFunc) (*modelpkg.JybCodeLog, error) {
|
|
objectID, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dao.FindOne(ctx, func(cols *JybCodeLogColumns) interface{} {
|
|
return bson.M{"_id": objectID}
|
|
}, optionsFunc...)
|
|
}
|
|
|
|
// FindMany executes a find command and returns many models the matching documents in the collection.
|
|
func (dao *JybCodeLog) FindMany(ctx context.Context, filterFunc JybCodeLogFilterFunc, optionsFunc ...JybCodeLogFindManyOptionsFunc) ([]*modelpkg.JybCodeLog, 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.JybCodeLog, 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 *JybCodeLog) DeleteOne(ctx context.Context, filterFunc JybCodeLogFilterFunc, optionsFunc ...JybCodeLogDeleteOptionsFunc) (*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 *JybCodeLog) DeleteOneByID(ctx context.Context, id string, optionsFunc ...JybCodeLogDeleteOptionsFunc) (*mongo.DeleteResult, error) {
|
|
objectID, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dao.DeleteOne(ctx, func(cols *JybCodeLogColumns) interface{} {
|
|
return bson.M{"_id": objectID}
|
|
}, optionsFunc...)
|
|
}
|
|
|
|
// DeleteMany executes a delete command to delete documents from the collection.
|
|
func (dao *JybCodeLog) DeleteMany(ctx context.Context, filterFunc JybCodeLogFilterFunc, optionsFunc ...JybCodeLogDeleteOptionsFunc) (*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 *JybCodeLog) autofill(ctx context.Context, model *modelpkg.JybCodeLog) error {
|
|
if model.Id.IsZero() {
|
|
model.Id = primitive.NewObjectID()
|
|
}
|
|
|
|
return nil
|
|
}
|