MongoDB
 sql >> Database >  >> NoSQL >> MongoDB

Come ottenere la raccolta di documenti dal cursore mongodb?

Un Cursor mongodb implementa Stream dal futures cassa . Questo è menzionato nei documenti :

In realtà consiglierei di usare try_collect() funzione da TryStreamExt tratto per ottenere un Result<Vec<Document>> invece. Quindi puoi usare unwrap_or_else() per restituire l'elenco. Dovresti anche usare collection_with_type() metodo per ottenere la raccolta in modo che i risultati vengano automaticamente deserializzati nel tipo corretto anziché solo nel Document (assicurati solo che implementi Debug , Serialize e Deserialize ).

Ecco un esempio funzionante

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![])
}