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 115dae5334
commit 9a5babd603
2 changed files with 42 additions and 34 deletions
Show Stats Download Patch File Download Diff File

38
utils/dorm/mongo_curd.go
Unescape View File

@ -15,7 +15,8 @@ type InsertOneResult struct {
func (c *MongoClient) InsertOne(document interface{}) (result *InsertOneResult, err error) { func (c *MongoClient) InsertOne(document interface{}) (result *InsertOneResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
res, err := collection.InsertOne(context.TODO(), document) res, err := collection.InsertOne(context.TODO(), document)
return &InsertOneResult{InsertedID: res.InsertedID}, err result = &InsertOneResult{InsertedID: res.InsertedID}
return
} }
type InsertManyResult struct { type InsertManyResult struct {
@ -26,21 +27,22 @@ type InsertManyResult struct {
func (c *MongoClient) InsertMany(documents []interface{}) (result *InsertManyResult, err error) { func (c *MongoClient) InsertMany(documents []interface{}) (result *InsertManyResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
res, err := collection.InsertMany(context.TODO(), documents) res, err := collection.InsertMany(context.TODO(), documents)
return &InsertManyResult{InsertedIDs: res.InsertedIDs}, err result = &InsertManyResult{InsertedIDs: res.InsertedIDs}
return
} }
// Delete 删除文档 // Delete 删除文档
func (c *MongoClient) Delete(filter interface{}) error { func (c *MongoClient) Delete(filter interface{}) (err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
_, err := collection.DeleteOne(context.TODO(), filter) _, err = collection.DeleteOne(context.TODO(), filter)
return err return
} }
// DeleteId 删除文档 // DeleteId 删除文档
func (c *MongoClient) DeleteId(id interface{}) error { func (c *MongoClient) DeleteId(id interface{}) (err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
_, err := collection.DeleteOne(context.TODO(), bson.M{"_id": id}) _, err = collection.DeleteOne(context.TODO(), bson.M{"_id": id})
return err return
} }
type DeleteResult struct { type DeleteResult struct {
@ -51,7 +53,8 @@ type DeleteResult struct {
func (c *MongoClient) DeleteMany(filter interface{}) (result *DeleteResult, err error) { func (c *MongoClient) DeleteMany(filter interface{}) (result *DeleteResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
res, err := collection.DeleteMany(context.TODO(), filter) res, err := collection.DeleteMany(context.TODO(), filter)
return &DeleteResult{DeletedCount: res.DeletedCount}, err result = &DeleteResult{DeletedCount: res.DeletedCount}
return
} }
// UpdateOne 更新单个文档 // UpdateOne 更新单个文档
@ -59,10 +62,10 @@ func (c *MongoClient) DeleteMany(filter interface{}) (result *DeleteResult, err
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (c *MongoClient) UpdateOne(filter interface{}, update interface{}) error { func (c *MongoClient) UpdateOne(filter interface{}, update interface{}) (err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
_, err := collection.UpdateOne(context.TODO(), filter, update) _, err = collection.UpdateOne(context.TODO(), filter, update)
return err return
} }
// UpdateId 更新单个文档 // UpdateId 更新单个文档
@ -70,10 +73,10 @@ func (c *MongoClient) UpdateOne(filter interface{}, update interface{}) error {
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (c *MongoClient) UpdateId(id interface{}, update interface{}) error { func (c *MongoClient) UpdateId(id interface{}, update interface{}) (err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
_, err := collection.UpdateOne(context.TODO(), bson.M{"_id": id}, update) _, err = collection.UpdateOne(context.TODO(), bson.M{"_id": id}, update)
return err return
} }
type UpdateResult struct { type UpdateResult struct {
@ -91,12 +94,13 @@ type UpdateResult struct {
func (c *MongoClient) UpdateMany(filter interface{}, update interface{}) (result *UpdateResult, err error) { func (c *MongoClient) UpdateMany(filter interface{}, update interface{}) (result *UpdateResult, err error) {
collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName) collection := c.Db.Database(c.getDatabaseName()).Collection(c.collectionName)
res, err := collection.UpdateMany(context.TODO(), filter, update) res, err := collection.UpdateMany(context.TODO(), filter, update)
return &UpdateResult{ result = &UpdateResult{
MatchedCount: res.MatchedCount, MatchedCount: res.MatchedCount,
ModifiedCount: res.ModifiedCount, ModifiedCount: res.ModifiedCount,
UpsertedCount: res.UpsertedCount, UpsertedCount: res.UpsertedCount,
UpsertedID: res.UpsertedID, UpsertedID: res.UpsertedID,
}, err }
return
} }
type FindResultI interface { type FindResultI interface {

38
utils/dorm/mongo_transaction_curd.go
Unescape View File

@ -11,28 +11,30 @@ import (
func (ms *MongoTransaction) InsertOne(document interface{}) (result *InsertOneResult, err error) { func (ms *MongoTransaction) InsertOne(document interface{}) (result *InsertOneResult, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
res, err := collection.InsertOne(ms.Session, document) res, err := collection.InsertOne(ms.Session, document)
return &InsertOneResult{InsertedID: res.InsertedID}, err result = &InsertOneResult{InsertedID: res.InsertedID}
return
} }
// InsertMany 插入多个文档 // InsertMany 插入多个文档
func (ms *MongoTransaction) InsertMany(documents []interface{}) (result *InsertManyResult, err error) { func (ms *MongoTransaction) InsertMany(documents []interface{}) (result *InsertManyResult, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
res, err := collection.InsertMany(ms.Session, documents) res, err := collection.InsertMany(ms.Session, documents)
return &InsertManyResult{InsertedIDs: res.InsertedIDs}, err result = &InsertManyResult{InsertedIDs: res.InsertedIDs}
return
} }
// Delete 删除文档 // Delete 删除文档
func (ms *MongoTransaction) Delete(filter interface{}) error { func (ms *MongoTransaction) Delete(filter interface{}) (err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
_, err := collection.DeleteOne(ms.Session, filter) _, err = collection.DeleteOne(ms.Session, filter)
return err return
} }
// DeleteId 删除文档 // DeleteId 删除文档
func (ms *MongoTransaction) DeleteId(id interface{}) error { func (ms *MongoTransaction) DeleteId(id interface{}) (err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
_, err := collection.DeleteOne(ms.Session, bson.M{"_id": id}) _, err = collection.DeleteOne(ms.Session, bson.M{"_id": id})
return err return
} }
// DeleteMany 删除多个文档 // DeleteMany 删除多个文档
@ -40,7 +42,8 @@ func (ms *MongoTransaction) DeleteMany(key string, value interface{}) (result *D
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
filter := bson.D{{key, value}} filter := bson.D{{key, value}}
res, err := collection.DeleteMany(ms.Session, filter) res, err := collection.DeleteMany(ms.Session, filter)
return &DeleteResult{DeletedCount: res.DeletedCount}, err result = &DeleteResult{DeletedCount: res.DeletedCount}
return
} }
// UpdateOne 更新单个文档 // UpdateOne 更新单个文档
@ -48,10 +51,10 @@ func (ms *MongoTransaction) DeleteMany(key string, value interface{}) (result *D
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (ms *MongoTransaction) UpdateOne(filter interface{}, update interface{}) error { func (ms *MongoTransaction) UpdateOne(filter interface{}, update interface{}) (err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
_, err := collection.UpdateOne(ms.Session, filter, update) _, err = collection.UpdateOne(ms.Session, filter, update)
return err return
} }
// UpdateId 更新单个文档 // UpdateId 更新单个文档
@ -59,10 +62,10 @@ func (ms *MongoTransaction) UpdateOne(filter interface{}, update interface{}) er
// 字段增加值 inc($inc) // 字段增加值 inc($inc)
// 从数组中增加一个元素 push($push) // 从数组中增加一个元素 push($push)
// 从数组中删除一个元素 pull($pull) // 从数组中删除一个元素 pull($pull)
func (ms *MongoTransaction) UpdateId(id interface{}, update interface{}) error { func (ms *MongoTransaction) UpdateId(id interface{}, update interface{}) (err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
_, err := collection.UpdateOne(context.TODO(), bson.M{"_id": id}, update) _, err = collection.UpdateOne(context.TODO(), bson.M{"_id": id}, update)
return err return
} }
// UpdateMany 更新多个文档 // UpdateMany 更新多个文档
@ -73,12 +76,13 @@ func (ms *MongoTransaction) UpdateId(id interface{}, update interface{}) error {
func (ms *MongoTransaction) UpdateMany(filter interface{}, update interface{}) (result *UpdateResult, err error) { func (ms *MongoTransaction) UpdateMany(filter interface{}, update interface{}) (result *UpdateResult, err error) {
collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName) collection := ms.db.Database(ms.getDatabaseName()).Collection(ms.collectionName)
res, err := collection.UpdateMany(ms.Session, filter, update) res, err := collection.UpdateMany(ms.Session, filter, update)
return &UpdateResult{ result = &UpdateResult{
MatchedCount: res.MatchedCount, MatchedCount: res.MatchedCount,
ModifiedCount: res.ModifiedCount, ModifiedCount: res.ModifiedCount,
UpsertedCount: res.UpsertedCount, UpsertedCount: res.UpsertedCount,
UpsertedID: res.UpsertedID, UpsertedID: res.UpsertedID,
}, err }
return
} }
// Find 查询 // Find 查询

Write Preview
Loading…
Cancel
Save