From 06f6f48f11811ce17817db445cd5c3eea3200103 Mon Sep 17 00:00:00 2001 From: Nikita Tokarchuk Date: Wed, 22 Jul 2020 04:45:32 +0200 Subject: [PATCH] Add context support to a saving --- mongox/database/saveone.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mongox/database/saveone.go b/mongox/database/saveone.go index bbd02df..ed70c6a 100644 --- a/mongox/database/saveone.go +++ b/mongox/database/saveone.go @@ -19,6 +19,7 @@ func (d *Database) SaveOne(source interface{}, filters ...interface{}) (err erro id := base.GetID(source) protected := base.GetProtection(source) composed := query.Compose(filters...) + ctx := query.WithContext(d.Context(), composed) 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() } - result := collection.FindOneAndReplace(d.Context(), composed.M(), source, opts) + result := collection.FindOneAndReplace(ctx, composed.M(), source, opts) if result.Err() != nil { return result.Err() }