From 8fcd764a8c6bf4aa2bfd5f9f7b7b4ee72a807821 Mon Sep 17 00:00:00 2001 From: Nikita Tokarchuk Date: Wed, 22 Jul 2020 04:45:32 +0200 Subject: [PATCH] Add context query support to a single deleting --- mongox/database/deleteone.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mongox/database/deleteone.go b/mongox/database/deleteone.go index dc9a3ae..dfac03d 100644 --- a/mongox/database/deleteone.go +++ b/mongox/database/deleteone.go @@ -19,6 +19,7 @@ func (d *Database) DeleteOne(target interface{}, filters ...interface{}) (err er opts := &options.FindOneAndDeleteOptions{} composed := query.Compose(filters...) protected := base.GetProtection(target) + ctx := query.WithContext(d.Context(), composed) opts.Sort = composed.Sorter() @@ -32,7 +33,7 @@ func (d *Database) DeleteOne(target interface{}, filters ...interface{}) (err er protected.V = time.Now().Unix() } - result := collection.FindOneAndDelete(d.Context(), composed.M(), opts) + result := collection.FindOneAndDelete(ctx, composed.M(), opts) if result.Err() != nil { return fmt.Errorf("can't create find one and delete result: %w", result.Err()) }