1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
//! Retrieves multiple documents from a collection by their id field.
//!
//! Retrieves documents from collection if they match the ids specified in the input.
//! Returns a Vec of the type that the collection is assigned to.
//!
//! # Examples
//!
//! ```
//! use mungos::{Mungos}
//! use serde::{Serialize, Deserialize}
//!
//! #[derive(Debug, Serialize, Deserialize)]
//! struct Item {
//! field: String
//! }
//!
//! let db = Mungos::new("uri", "app name", timeout).await;
//! let collection = db.connection::<Item>("db name", "collection name");
//!
//! let ids = Vec::from(["...", "...", "..."]);
//! collection.get_many_by_id(ids).await.unwrap();
//!
//! ```
//!
use crate::Collection;
use futures::stream::TryStreamExt;
use mongodb::{
bson::{doc, oid::ObjectId},
error::Result,
};
use serde::de::DeserializeOwned;
use std::str::FromStr;
impl<T: DeserializeOwned + Unpin + Send + Sync> Collection<T> {
pub async fn get_many_by_id(&self, ids: &Vec<String>) -> Result<Vec<T>> {
let ids: Vec<ObjectId> = ids
.iter()
.map(|id| ObjectId::from_str(id).unwrap())
.collect();
let mut cursor = self
.collection
.find(
doc! {
"_id": {
"$in": ids
}
},
None,
)
.await?;
let mut items = Vec::new();
while let Some(item) = cursor.try_next().await? {
items.push(item);
}
Ok(items)
}
}