计算文档

Overview

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

样本数据

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

type Tea struct {	Type   string	Rating int32}

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

coll := client.Database("db").Collection("tea")docs := []any{	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()`	对结果进行排序时要使用的语言排序规则类型。默认：`nil`
`SetHint()`	用于扫描要计数的文档的索引。默认：`nil`
`SetLimit()`	要计数的最大文档数。默认：`0`
`SetSkip()`	计数前要跳过的文档数。默认：`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.Dif err = cursor.All(context.TODO(), &results); err != nil {   panic(err)}for _, result := range results {   fmt.Println(result)}

预计数量

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

注意

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

修改行为

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

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

方法	说明
`SetComment()`	设置要附加到计数操作的注释。默认：`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

计数文档示例：完整文件

注意

设置示例

该示例使用连接 URI 连接到 MongoDB 实例。如需了解有关连接到 MongoDB 实例的更多信息，请参阅创建 MongoClient。本示例还使用了Atlas 示例数据集中包含的 sample_restaurants 数据库中的 restaurants 集合。您可以按照Atlas 入门指南中的步骤，将它们加载到 MongoDB Atlas 免费集群上的数据库中。

此示例对 restaurants集合执行以下操作：

集合中文档的近似数量
计算 cuisine字段的值为 "American" 的文档数量

// Counts documents in a collection by using the Go driverpackage main
import (	"context"	"fmt"	"log"	"os"
	"github.com/joho/godotenv"	"go.mongodb.org/mongo-driver/v2/bson"	"go.mongodb.org/mongo-driver/v2/mongo"	"go.mongodb.org/mongo-driver/v2/mongo/options")
type Restaurant struct {	ID           bson.ObjectID `bson:"_id"`	Name         string	RestaurantId string `bson:"restaurant_id"`	Cuisine      string	Address      interface{}	Borough      string	Grades       interface{}}
func main() {	if err := godotenv.Load(); err != nil {		log.Println("No .env file found")	}
	var uri string	if uri = os.Getenv("MONGODB_URI"); uri == "" {		log.Fatal("You must set your 'MONGODB_URI' environment variable. See\n\t https://www.mongodb.com/zh-cn/docs/drivers/go/current/usage-examples/#environment-variable")	}
	client, err := mongo.Connect(options.Client().ApplyURI(uri))	if err != nil {		panic(err)	}	defer func() {		if err = client.Disconnect(context.TODO()); err != nil {			panic(err)		}	}()
	coll := client.Database("sample_restaurants").Collection("restaurants")
	// Specifies a filter to match documents where the "cuisine" field	// has a value of "American"	filter := bson.D{{"cuisine", "American"}}
	// Retrieves and prints the estimated number of documents in the collection	estCount, estCountErr := coll.EstimatedDocumentCount(context.TODO())	if estCountErr != nil {		panic(estCountErr)	}
	// Retrieves and prints the number of matching documents in the collection	count, err := coll.CountDocuments(context.TODO(), filter)	if err != nil {		panic(err)	}
	// When you run this file, it should print:	// Estimated number of documents in the movies collection: 25359	// Number of restaurants with American cuisine: 6183	fmt.Printf("Estimated number of documents in the restaurants collection: %d\n", estCount)	fmt.Printf("Number of restaurants with American cuisine: %d\n", count)}

Estimated number of documents in the restaurants collection: 25359Number of restaurants with American cuisine: 6183

计算文档

Overview

样本数据

准确计数

修改行为

例子

聚合(Aggregation)

例子

预计数量

修改行为

例子

计数文档示例：完整文件

更多信息

API 文档