-
Notifications
You must be signed in to change notification settings - Fork 0
/
countdocs.go
47 lines (37 loc) · 1.35 KB
/
countdocs.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package mongohelper
import (
"context"
"errors"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// CountDocs wraps the mongo.Database.Collection.CountDocuments() method
// It returns the number of documents in the collection and an error
//
// The filter parameter must be a document and can be used to select which documents contribute to the count. It
// cannot be nil. An empty document (e.g. bson.D{}) should be used to count all documents in the collection. This will
// result in a full collection scan.
func (l *Link) CountDocs(database, collection string, filter interface{}) (int64, error) {
if err := l.linkCheck("link.CountDocs"); err != nil {
return 0, err
}
ctx, cancel := context.WithTimeout(context.Background(), l.execTimeout())
defer cancel()
n, err := l.client.Database(database).Collection(collection).CountDocuments(ctx, filter, options.Count())
if err != nil {
// If not connected, try once again
if errors.Is(err, mongo.ErrClientDisconnected) {
if err = l.connect(); err != nil {
return 0, err
}
ctx2, cancel2 := context.WithTimeout(context.Background(), l.execTimeout())
defer cancel2()
if n, err = l.client.Database(database).Collection(collection).CountDocuments(ctx2, filter, options.Count()); err != nil {
return 0, err
}
} else {
return 0, err
}
}
return n, nil
}