dtapps
/
go-library
1
0
Fork 0
Code Issues Pull Requests Projects Releases 40 Wiki Activity

- update dorm

Browse Source
master
李光春 2 years ago
parent 8ff8e2f9c4
commit 26bb5ccd2f
2 changed files with 231 additions and 39 deletions
Show Stats Download Patch File Download Diff File

72
utils/dorm/mongo_official.go
Unescape View File

@ -8,31 +8,31 @@ import (
) )
// InsertOne 插入单个文档 // InsertOne 插入单个文档
func (c *MongoClient) InsertOne(ctx context.Context, document interface{}, opts ...*options.InsertOneOptions) (result *mongo.InsertOneResult, err error) { func (c *MongoClient) InsertOne(document interface{}) (result *mongo.InsertOneResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err = collection.InsertOne(ctx, document, opts...) result, err = collection.InsertOne(context.TODO(), document)
return result, err return result, err
} }
// InsertMany 插入多个文档 // InsertMany 插入多个文档
func (c *MongoClient) InsertMany(ctx context.Context, documents []interface{}, opts ...*options.InsertManyOptions) (result *mongo.InsertManyResult, err error) { func (c *MongoClient) InsertMany(documents []interface{}) (result *mongo.InsertManyResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err = collection.InsertMany(ctx, documents, opts...) result, err = collection.InsertMany(context.TODO(), documents)
return result, err return result, err
} }
// Delete 删除文档 // Delete 删除文档
func (c *MongoClient) Delete(ctx context.Context, filter interface{}, opts ...*options.DeleteOptions) (int64, error) { func (c *MongoClient) Delete(filter interface{}) (int64, error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
count, err := collection.DeleteOne(ctx, filter, opts...) count, err := collection.DeleteOne(context.TODO(), filter)
return count.DeletedCount, err return count.DeletedCount, err
} }
// DeleteMany 删除多个文档 // DeleteMany 删除多个文档
func (c *MongoClient) DeleteMany(ctx context.Context, key string, value interface{}) (int64, error) { func (c *MongoClient) DeleteMany(key string, value interface{}) (int64, error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
filter := bson.D{{key, value}} filter := bson.D{{key, value}}
count, err := collection.DeleteMany(ctx, filter) count, err := collection.DeleteMany(context.TODO(), filter)
return count.DeletedCount, err return count.DeletedCount, err
} }
@ -41,9 +41,9 @@ func (c *MongoClient) DeleteMany(ctx context.Context, key string, value interfac
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (c *MongoClient) UpdateOne(ctx context.Context, filter interface{}, update interface{}, opts ...*options.UpdateOptions) (int64, error) { func (c *MongoClient) UpdateOne(filter interface{}, update interface{}) (int64, error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err := collection.UpdateOne(ctx, filter, update, opts...) result, err := collection.UpdateOne(context.TODO(), filter, update)
return result.UpsertedCount, err return result.UpsertedCount, err
} }
@ -52,82 +52,82 @@ func (c *MongoClient) UpdateOne(ctx context.Context, filter interface{}, update
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (c *MongoClient) UpdateMany(ctx context.Context, filter interface{}, update interface{}, opts ...*options.UpdateOptions) (int64, error) { func (c *MongoClient) UpdateMany(filter interface{}, update interface{}) (int64, error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err := collection.UpdateMany(ctx, filter, update, opts...) result, err := collection.UpdateMany(context.TODO(), filter, update)
return result.UpsertedCount, err return result.UpsertedCount, err
} }
// Find 查询 // Find 查询
//func (c *MongoClient) Find(filter interface{}, opts ...*options.FindOptions) (result *mongo.Cursor, err error) { //func (c *MongoClient) Find(filter interface{}) (result *mongo.Cursor, err error) {
// collection := c.Db.Database(c.DatabaseName).Collection(c.collectionName) // collection := c.Db.Database(c.DatabaseName).Collection(c.collectionName)
// result, err = collection.Find(context.TODO(), filter, opts...) // result, err = collection.Find(context.TODO(), filter)
// return result, err // return result, err
//} //}
// FindOne 查询单个文档 // FindOne 查询单个文档
func (c *MongoClient) FindOne(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) (result *mongo.SingleResult) { func (c *MongoClient) FindOne(filter interface{}) (result *mongo.SingleResult) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result = collection.FindOne(ctx, filter, opts...) result = collection.FindOne(context.TODO(), filter)
return result return result
} }
// FindMany 查询多个文档 // FindMany 查询多个文档
func (c *MongoClient) FindMany(ctx context.Context, filter interface{}, opts ...*options.FindOptions) (result *mongo.Cursor, err error) { func (c *MongoClient) FindMany(filter interface{}) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err = collection.Find(ctx, filter, opts...) result, err = collection.Find(context.TODO(), filter)
return result, err return result, err
} }
// FindManyByFilters 多条件查询 // FindManyByFilters 多条件查询
func (c *MongoClient) FindManyByFilters(ctx context.Context, filter interface{}) (result *mongo.Cursor, err error) { func (c *MongoClient) FindManyByFilters(filter interface{}) (result *mongo.Cursor, err error) {
collection, err := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName).Clone() collection, err := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName).Clone()
result, err = collection.Find(ctx, bson.M{"$and": filter}) result, err = collection.Find(context.TODO(), bson.M{"$and": filter})
return result, err return result, err
} }
// FindManyByFiltersSort 多条件查询支持排序 // FindManyByFiltersSort 多条件查询支持排序
func (c *MongoClient) FindManyByFiltersSort(ctx context.Context, filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) { func (c *MongoClient) FindManyByFiltersSort(filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) {
collection, err := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName).Clone() collection, err := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName).Clone()
findOptions := options.Find() findOptions := options.Find()
findOptions.SetSort(Sort) findOptions.SetSort(Sort)
result, err = collection.Find(ctx, filter, findOptions) result, err = collection.Find(context.TODO(), filter, findOptions)
return result, err return result, err
} }
// FindCollection 查询集合文档 // FindCollection 查询集合文档
func (c *MongoClient) FindCollection(ctx context.Context, Limit int64) (result *mongo.Cursor, err error) { func (c *MongoClient) FindCollection(Limit int64) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
findOptions := options.Find() findOptions := options.Find()
findOptions.SetLimit(Limit) findOptions.SetLimit(Limit)
result, err = collection.Find(ctx, bson.D{{}}, findOptions) result, err = collection.Find(context.TODO(), bson.D{{}}, findOptions)
return result, err return result, err
} }
// FindCollectionSort 查询集合文档支持排序 // FindCollectionSort 查询集合文档支持排序
func (c *MongoClient) FindCollectionSort(ctx context.Context, Sort interface{}, Limit int64) (result *mongo.Cursor, err error) { func (c *MongoClient) FindCollectionSort(Sort interface{}, Limit int64) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
findOptions := options.Find() findOptions := options.Find()
findOptions.SetSort(Sort) findOptions.SetSort(Sort)
findOptions.SetLimit(Limit) findOptions.SetLimit(Limit)
result, err = collection.Find(ctx, bson.D{{}}, findOptions) result, err = collection.Find(context.TODO(), bson.D{{}}, findOptions)
return result, err return result, err
} }
// FindManyCollectionSort 查询集合文档支持排序支持条件 // FindManyCollectionSort 查询集合文档支持排序支持条件
func (c *MongoClient) FindManyCollectionSort(ctx context.Context, filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) { func (c *MongoClient) FindManyCollectionSort(filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
findOptions := options.Find() findOptions := options.Find()
findOptions.SetSort(Sort) findOptions.SetSort(Sort)
result, err = collection.Find(ctx, filter, findOptions) result, err = collection.Find(context.TODO(), filter, findOptions)
return result, err return result, err
} }
// CollectionCount 查询集合里有多少数据 // CollectionCount 查询集合里有多少数据
func (c *MongoClient) CollectionCount(ctx context.Context) (name string, size int64) { func (c *MongoClient) CollectionCount() (name string, size int64) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
name = collection.Name() name = collection.Name()
size, _ = collection.EstimatedDocumentCount(ctx) size, _ = collection.EstimatedDocumentCount(context.TODO())
return name, size return name, size
} }
@ -135,25 +135,25 @@ func (c *MongoClient) CollectionCount(ctx context.Context) (name string, size in
// Skip 跳过 // Skip 跳过
// Limit 读取数量 // Limit 读取数量
// sort 1 ，-1 . 1 为升序 ， -1 为降序 // sort 1 ，-1 . 1 为升序 ， -1 为降序
func (c *MongoClient) CollectionDocuments(ctx context.Context, Skip, Limit int64, sort int, key string, value interface{}) (result *mongo.Cursor, err error) { func (c *MongoClient) CollectionDocuments(Skip, Limit int64, sort int, key string, value interface{}) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
SORT := bson.D{{"_id", sort}} SORT := bson.D{{"_id", sort}}
filter := bson.D{{key, value}} filter := bson.D{{key, value}}
findOptions := options.Find().SetSort(SORT).SetLimit(Limit).SetSkip(Skip) findOptions := options.Find().SetSort(SORT).SetLimit(Limit).SetSkip(Skip)
result, err = collection.Find(ctx, filter, findOptions) result, err = collection.Find(context.TODO(), filter, findOptions)
return result, err return result, err
} }
// AggregateByFiltersSort 统计分析 // AggregateByFiltersSort 统计分析
func (c *MongoClient) AggregateByFiltersSort(ctx context.Context, pipeline interface{}, opts ...*options.AggregateOptions) (result *mongo.Cursor, err error) { func (c *MongoClient) AggregateByFiltersSort(pipeline interface{}) (result *mongo.Cursor, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
result, err = collection.Aggregate(ctx, pipeline, opts...) result, err = collection.Aggregate(context.TODO(), pipeline)
return result, err return result, err
} }
// CountDocumentsByFilters 统计数量 // CountDocumentsByFilters 统计数量
func (c *MongoClient) CountDocumentsByFilters(ctx context.Context, filter interface{}) (count int64, err error) { func (c *MongoClient) CountDocumentsByFilters(filter interface{}) (count int64, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
count, err = collection.CountDocuments(ctx, filter) count, err = collection.CountDocuments(context.TODO(), filter)
return count, err return count, err
} }

198
utils/dorm/mongo_transaction.go
Unescape View File

@ -2,19 +2,27 @@ package dorm
import ( import (
"context" "context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"reflect"
) )
type MongoTransaction struct { type MongoTransaction struct {
startSession mongo.Session startSession mongo.Session
Session mongo.SessionContext Session mongo.SessionContext
db *mongo.Client // 驱动
databaseName string // 库名
collectionName string // 表名
} }
// Begin 开始事务，会同时创建开始会话需要在退出时关闭会话 // Begin 开始事务，会同时创建开始会话需要在退出时关闭会话
func (c *MongoClient) Begin() (ms MongoTransaction, err error) { func (c *MongoClient) Begin() (ms MongoTransaction, err error) {
ms.db = c.Db
// 开始会话 // 开始会话
ms.startSession, err = c.Db.StartSession() ms.startSession, err = ms.db.StartSession()
if err != nil { if err != nil {
panic(err) panic(err)
} }
@ -41,3 +49,187 @@ func (ms *MongoTransaction) Rollback() error {
func (ms *MongoTransaction) Commit() error { func (ms *MongoTransaction) Commit() error {
return ms.startSession.CommitTransaction(context.Background()) return ms.startSession.CommitTransaction(context.Background())
} }
// Database 设置库名
func (ms *MongoTransaction) Database(databaseName string) *MongoTransaction {
ms.databaseName = databaseName
return ms
}
// Collection 设置表名
func (ms *MongoTransaction) Collection(collectionName string) *MongoTransaction {
ms.collectionName = collectionName
return ms
}
// Model 传入模型自动获取库名和表名
func (ms *MongoTransaction) Model(value interface{}) *MongoTransaction {
// https://studygolang.com/articles/896
val := reflect.ValueOf(value)
if methodValue := val.MethodByName("Database"); methodValue.IsValid() {
ms.databaseName = methodValue.Call(nil)[0].String()
}
if methodValue := val.MethodByName("TableName"); methodValue.IsValid() {
ms.collectionName = methodValue.Call(nil)[0].String()
}
return ms
}
func (ms *MongoTransaction) getDatabaseName() string {
return ms.databaseName
}
func (ms *MongoTransaction) getCollectionName() string {
return ms.collectionName
}
// InsertOne 插入单个文档
func (ms *MongoTransaction) InsertOne(document interface{}) (result *mongo.InsertOneResult, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err = collection.InsertOne(ms.Session, document)
return result, err
}
// InsertMany 插入多个文档
func (ms *MongoTransaction) InsertMany(documents []interface{}) (result *mongo.InsertManyResult, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err = collection.InsertMany(ms.Session, documents)
return result, err
}
// Delete 删除文档
func (ms *MongoTransaction) Delete(filter interface{}) (int64, error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
count, err := collection.DeleteOne(ms.Session, filter)
return count.DeletedCount, err
}
// DeleteMany 删除多个文档
func (ms *MongoTransaction) DeleteMany(key string, value interface{}) (int64, error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
filter := bson.D{{key, value}}
count, err := collection.DeleteMany(ms.Session, filter)
return count.DeletedCount, err
}
// UpdateOne 更新单个文档
// 修改字段的值($set)
// 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull)
func (ms *MongoTransaction) UpdateOne(filter interface{}, update interface{}) (int64, error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err := collection.UpdateOne(ms.Session, filter, update)
return result.UpsertedCount, err
}
// UpdateMany 更新多个文档
// 修改字段的值($set)
// 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull)
func (ms *MongoTransaction) UpdateMany(filter interface{}, update interface{}) (int64, error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err := collection.UpdateMany(ms.Session, filter, update)
return result.UpsertedCount, err
}
// Find 查询
func (ms *MongoTransaction) Find(filter interface{}) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err = collection.Find(ms.Session, filter)
return result, err
}
// FindOne 查询单个文档
func (ms *MongoTransaction) FindOne(filter interface{}) (result *mongo.SingleResult) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result = collection.FindOne(ms.Session, filter)
return result
}
// FindMany 查询多个文档
func (ms *MongoTransaction) FindMany(filter interface{}) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err = collection.Find(ms.Session, filter)
return result, err
}
// FindManyByFilters 多条件查询
func (ms *MongoTransaction) FindManyByFilters(filter interface{}) (result *mongo.Cursor, err error) {
collection, err := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName).Clone()
result, err = collection.Find(ms.Session, bson.M{"$and": filter})
return result, err
}
// FindManyByFiltersSort 多条件查询支持排序
func (ms *MongoTransaction) FindManyByFiltersSort(filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) {
collection, err := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName).Clone()
findOptions := options.Find()
findOptions.SetSort(Sort)
result, err = collection.Find(ms.Session, filter, findOptions)
return result, err
}
// FindCollection 查询集合文档
func (ms *MongoTransaction) FindCollection(Limit int64) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
findOptions := options.Find()
findOptions.SetLimit(Limit)
result, err = collection.Find(ms.Session, bson.D{{}}, findOptions)
return result, err
}
// FindCollectionSort 查询集合文档支持排序
func (ms *MongoTransaction) FindCollectionSort(Sort interface{}, Limit int64) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
findOptions := options.Find()
findOptions.SetSort(Sort)
findOptions.SetLimit(Limit)
result, err = collection.Find(ms.Session, bson.D{{}}, findOptions)
return result, err
}
// FindManyCollectionSort 查询集合文档支持排序支持条件
func (ms *MongoTransaction) FindManyCollectionSort(filter interface{}, Sort interface{}) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
findOptions := options.Find()
findOptions.SetSort(Sort)
result, err = collection.Find(ms.Session, filter, findOptions)
return result, err
}
// CollectionCount 查询集合里有多少数据
func (ms *MongoTransaction) CollectionCount(ctx context.Context) (name string, size int64) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
name = collection.Name()
size, _ = collection.EstimatedDocumentCount(ctx)
return name, size
}
// CollectionDocuments 按选项查询集合
// Skip 跳过
// Limit 读取数量
// sort 1 ，-1 . 1 为升序 ， -1 为降序
func (ms *MongoTransaction) CollectionDocuments(Skip, Limit int64, sort int, key string, value interface{}) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
SORT := bson.D{{"_id", sort}}
filter := bson.D{{key, value}}
findOptions := options.Find().SetSort(SORT).SetLimit(Limit).SetSkip(Skip)
result, err = collection.Find(ms.Session, filter, findOptions)
return result, err
}
// AggregateByFiltersSort 统计分析
func (ms *MongoTransaction) AggregateByFiltersSort(pipeline interface{}) (result *mongo.Cursor, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
result, err = collection.Aggregate(ms.Session, pipeline)
return result, err
}
// CountDocumentsByFilters 统计数量
func (ms *MongoTransaction) CountDocumentsByFilters(filter interface{}) (count int64, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
count, err = collection.CountDocuments(ms.Session, filter)
return count, err
}

Write Preview
Loading…
Cancel
Save