计算文档

在此页面上

Overview

准确计数
聚合(Aggregation)
预计数量
更多信息

Overview

本指南中，您可以了解如何获取集合中文档数量 accurate 和估计计数。

样本数据

本部分的示例使用以下Tea 结构体作为tea 集合中文档的模型：

type Tea struct {
	Type   string
	Rating int32
}

要运行本指南中的示例，请使用以下代码段将样本数据加载到 db 数据库中的 tea 集合中：

coll := client.Database("db").Collection("tea")
docs := []interface{}{
	Tea{Type: "Masala", Rating: 10},
	Tea{Type: "Matcha", Rating: 7},
	Tea{Type: "Assam", Rating: 4},
	Tea{Type: "Oolong", Rating: 9},
	Tea{Type: "Chrysanthemum", Rating: 5},
	Tea{Type: "Earl Grey", Rating: 8},
	Tea{Type: "Jasmine", Rating: 3},
	Tea{Type: "English Breakfast", Rating: 6},
	Tea{Type: "White Peony", Rating: 4},
}
result, err := coll.InsertMany(context.TODO(), docs)

提示

不存在的数据库和集合

如果执行写操作时不存在必要的数据库和集合，服务器会隐式创建这些数据库和集合。

每个文档都描述了一种茶叶类型及其评级。这些项目对应于 type 和 rating 字段。

准确计数

要计算与查询筛选器匹配的文档数量，请使用 CountDocuments() 方法。如果传递空查询筛选器，则该方法将返回集合中的文档总数。

提示

当你使用 CountDocuments() 返回集合中的文档总数时，MongoDB 会执行集合扫描。您可以通过使用提示来利用 _id 字段的内置索引，从而避免集合扫描并改进此方法的性能。仅当使用空查询参数调用 CountDocuments() 时才使用此技术。

opts := options.Count().SetHint("_id_")
count, err := coll.CountDocuments(context.TODO(), bson.D{}, opts)
if err != nil {
     panic(err)
}

修改行为

可以通过传入 CountOptions 类型来修改 CountDocuments() 的行为。如果未指定任何选项，驱动程序将使用默认值。

CountOptions 类型允许您使用以下方法配置选项：

方法	说明
`SetCollation()`	The type of language collation to use when sorting results. Default: `nil`
`SetHint()`	The index to use to scan for documents to count. Default: `nil`
`SetLimit()`	The maximum number of documents to count. Default: `0`
`SetMaxTime()`	The maximum amount of time that the query can run on the server. Default: `nil`
`SetSkip()`	The number of documents to skip before counting. Default: `0`

例子

以下示例计算其中的 rating 小于 6 的文档数：

filter := bson.D{{"rating", bson.D{{"$lt", 6}}}}
count, err := coll.CountDocuments(context.TODO(), filter)
if err != nil {
    panic(err)
}
fmt.Printf("Number of documents with a rating less than six: %d\n", count)

Number of documents with a rating less than six: 4

聚合(Aggregation)

还可以使用 $count 阶段计算聚合管道中的文档数量。

例子

以下示例执行以下动作：

计算 rating 字段的值大于 5 的文档数
将该计数赋值给 counted_documents 字段

matchStage := bson.D{{"$match", bson.D{{"rating", bson.D{{"$gt", 5}}}}}}
countStage := bson.D{{"$count", "counted_documents"}}
cursor, err := coll.Aggregate(context.TODO(), mongo.Pipeline{matchStage, countStage})
if err != nil {
   panic(err)
}
var results []bson.D
if err = cursor.All(context.TODO(), &results); err != nil {
   panic(err)
}
for _, result := range results {
   fmt.Println(result)
}

[{counted_documents 5}]

预计数量

要估计集合中的文档数，请使用 EstimatedDocumentCount() 方法。

注意

EstimatedDocumentCount() 方法比 CountDocuments() 方法更快，因为它使用集合的元数据而不是扫描整个集合。

修改行为

您可以通过传入 EstimatedDocumentCountOptions 类型来修改 EstimatedDocumentCount() 的行为。如果不指定任何选项，驱动程序将使用默认值。

EstimatedDocumentCountOptions 类型允许您使用以下方法配置选项：

方法	说明
`SetMaxTime()`	The maximum amount of time that the query can run on the server. Default: `nil`

例子

以下示例估计了 tea 集合中的文档数量：

count, err := coll.EstimatedDocumentCount(context.TODO())
if err != nil {
   panic(err)
}
fmt.Printf("Estimated number of documents in the tea collection: %d\n", count)

Estimated number of documents in the tea collection: 9

Overview

样本数据

提示

不存在的数据库和集合

准确计数

提示

修改行为

例子

聚合(Aggregation)

例子

预计数量

注意

修改行为

例子

更多信息

API 文档