sql >> Base de Datos >  >> NoSQL >> MongoDB

¿Cómo obtener la colección de documentos del cursor mongodb?

Un mongodb Cursor implementa Stream de futures caja . Esto se menciona en los docs :

De hecho, recomendaría usar try_collect() función de TryStreamExt rasgo para obtener un Result<Vec<Document>> en cambio. Entonces puedes usar unwrap_or_else() para devolver la lista. También debe usar collection_with_type() método para obtener la colección para que sus resultados se deserialicen automáticamente al tipo adecuado en lugar de solo Document (solo asegúrese de que implemente Debug , Serialize y Deserialize ).

Aquí hay una muestra de trabajo

use futures::TryStreamExt;
use mongodb::Client;
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize)]
struct Vehicle {
    id: String,
    name: String,
}

async fn list_all() -> Vec<Vehicle> {
    let client = Client::with_uri_str("mongodb://example.com").await.unwrap();
    let database = client.database("test");
    let collection = database.collection_with_type::<Vehicle>("vehicles");
    let cursor = match collection.find(None, None).await {
        Ok(cursor) => cursor,
        Err(_) => return vec![],
    };

    cursor.try_collect().await.unwrap_or_else(|_| vec![])
}