mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-06-12 16:53:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add context support to a saving

Browse Source
This commit is contained in:
Nikita Tokarchuk
2020-07-22 04:45:32 +02:00
parent 28be1f46b9
commit 06f6f48f11
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/database/saveone.go
+2 -1
View File
@@ -19,6 +19,7 @@ func (d *Database) SaveOne(source interface{}, filters ...interface{}) (err erro
id := base.GetID(source) id := base.GetID(source)
protected := base.GetProtection(source) protected := base.GetProtection(source)
composed := query.Compose(filters...) composed := query.Compose(filters...)
ctx := query.WithContext(d.Context(), composed)
composed.And(bson.M{"_id": id}) composed.And(bson.M{"_id": id})
@@ -31,7 +32,7 @@ func (d *Database) SaveOne(source interface{}, filters ...interface{}) (err erro
protected.V = time.Now().Unix() protected.V = time.Now().Unix()
} }
result := collection.FindOneAndReplace(d.Context(), composed.M(), source, opts) result := collection.FindOneAndReplace(ctx, composed.M(), source, opts)
if result.Err() != nil { if result.Err() != nil {
return result.Err() return result.Err()
} }