使用游标访问数据

概述

在本指南中，您可以学习如何使用Rust驱动程序，通过使用游标来访问读取操作或聚合操作返回的数据。游标是一种机制，允许您在给定时间内只将部分文档保留在内存中，同时遍历多个文档。

驱动程序提供了Cursor类型来从游标中检索文档。例如，当您运行可以返回多个文档的查找操作时，驱动程序返回一个Cursor实例，您可以通过该实例访问匹配的文档。

运行读取操作或聚合操作后，返回的Cursor实例包含操作的第一批结果。随着您遍历游标，服务器返回更多的单个结果。如果在达到一批结果结束时还有更多匹配的文档，则Cursor实例将检索下一批文档，直到所有结果返回。

本指南包括以下部分

• 示例数据的样本展示了游标示例所使用的样本数据

• 逐个检索文档描述了如何使用迭代或流逐个访问结果

• 按数组形式检索文档描述了如何通过收集返回的游标结果将所有结果作为单个数组访问

• 指定游标行为描述了如何配置方法返回的游标

• 更多信息提供了到资源链接和本指南中提到的类型和方法的API文档

示例数据

本指南中的示例使用以下存储在结构体中的数据

let docs = vec! [
    Fruit { 
        name: "strawberry".to_string(), 
        color: "red".to_string() 
    },
    Fruit { 
        name: "banana".to_string(), 
        color: "yellow".to_string() 
    },
    Fruit { 
        name: "pomegranate".to_string(), 
        color: "red".to_string() 
    },
    Fruit { 
        name: "pineapple".to_string(), 
        color: "yellow".to_string() 
    }
];

逐个检索文档

驱动程序提供了以下访问模式来遍历由Cursor实例返回的文档

• 内置模式：移动游标，然后检索和反序列化当前文档

• 流实现模式：遍历游标，并调用Stream提供的方法来处理单个或多个文档

以下章节将更详细地描述这些访问模式和相应的方法。

let mut cursor = my_coll.find(doc! { "color": "red" }).await?;
while cursor.advance().await? {
    println!("{:?}", cursor.deserialize_current()?);
}

Fruit { name: "strawberry", color: "red" }
Fruit { name: "pomegranate", color: "red" }

let mut cursor = my_coll.find(doc! { "color": "red" }).await?;
println!("Output from next() iteration:");
while let Some(doc) = cursor.next().await {
    println!("{:?}", doc?);
}
println!();
let mut cursor = my_coll.find(doc! { "color": "yellow" }).await?;
println!("Output from try_next() iteration:");
while let Some(doc) = cursor.try_next().await? {
    println!("{:?}", doc);
}

Output from next() iteration:
Fruit { name: "strawberry", color: "red" }
Fruit { name: "pomegranate", color: "red" }
Output from try_next() iteration:
Fruit { name: "banana", color: "yellow" }
Fruit { name: "pineapple", color: "yellow" }

将文档作为数组检索

因为Cursor类型实现了Stream特质，所以可以将游标的结果收集到一个数组中。

您可以使用以下方法将文档作为数组检索：

• collect()：将游标的结果收集到Vec<Result<T>>类型中

• try_collect()：将游标的结果收集到Result<Vec<T>>类型中

let cursor = my_coll.find(doc! { "color": "red" }).await?;
println!("Output from collect():");
let v: Vec<Result<Fruit>> = cursor.collect().await;
println!("{:?}", v);
println!();
let cursor = my_coll.find(doc! { "color": "yellow" }).await?;
println!("Output from try_collect():");
let v: Vec<Fruit> = cursor.try_collect().await?;
println!("{:?}", v);

Output from collect():
[Ok(Fruit { name: "strawberry", color: "red" }), Ok(Fruit { name: "pomegranate", color: "red" })]
Output from try_collect():
[Fruit { name: "banana", color: "yellow" }, Fruit { name: "pineapple", color: "yellow" }]

指定游标行为

要修改操作返回的游标，请将选项构建器方法链接到返回Cursor实例的方法。例如，可以将与游标相关的选项构建器方法链接到find()方法。

以下表格描述了您可以通过调用其相应的构建器方法设置的与游标相关的选项

以下代码显示了如何通过将选项构建器方法链接到 find() 方法来指定游标相关的设置

let mut cursor = my_coll.find(doc! { "color": "red" })
    .batch_size(5)
    .cursor_type(CursorType::Tailable)
    .no_cursor_timeout(true)
    .await?;

更多信息

有关本指南中操作的更多信息，请参阅以下文档

• 检索数据 指南

• 指定查询 指南

有关更多关于在 Rust 类型和BSON 之间转换的信息，请参阅有关 数据建模和序列化 的指南。