From ee4bf466624efa3a5f3f76ac48dc90dc88679165 Mon Sep 17 00:00:00 2001 From: Nikita Tokarchuk Date: Wed, 22 Jul 2020 04:45:32 +0200 Subject: [PATCH] Add callback support for a counting --- mongox/database/count.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mongox/database/count.go b/mongox/database/count.go index 122efda..c2ea792 100644 --- a/mongox/database/count.go +++ b/mongox/database/count.go @@ -13,11 +13,14 @@ func (d *Database) Count(target interface{}, filters ...interface{}) (result int collection := d.GetCollectionOf(target) opts := options.Count() composed := query.Compose(filters...) + ctx := query.WithContext(d.Context(), composed) opts.Limit = composed.Limiter() opts.Skip = composed.Skipper() - result, err = collection.CountDocuments(d.Context(), composed.M(), opts) + result, err = collection.CountDocuments(ctx, composed.M(), opts) + + _ = composed.OnClose().Invoke(ctx, target) return }