mongor 0.1.10

Ergonomic MongoDB ODM
Documentation 
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

// Authors: Robert Lopez
// License: MIT (See `LICENSE.md`)

use super::Model;
use crate::error::Error;
use mongodb::{bson::Document, options::*, ClientSession};
use serde::{Deserialize, Serialize};

impl<D> Model<D>
where
    D: Serialize + for<'a> Deserialize<'a> + Send + Sync + Unpin,
{
    /// Returns if a document exists by a filter in `self.collection`.
    ///
    /// ---
    /// Example Usage:
    /// ```
    ///
    /// let model: Model<D> = ...;
    ///
    /// if model.exists(
    ///     doc! { "user": "admin" },
    ///     None::<FindOneOptions>,
    ///     None::<&mut ClientSession>,
    /// ).await? {
    ///     ...
    /// }
    /// ```
    pub async fn exists(
        &self,
        filter: Document,
        options: Option<FindOneOptions>,
        session: Option<&mut ClientSession>,
    ) -> Result<bool, Error> {
        Ok(self.find_one(filter, options, session).await?.is_some())
    }

    /// Returns the document count for `self.collection` by a given filter.
    ///
    /// ---
    /// Example Usage:
    /// ```
    ///
    /// let model: Model<D> = ...;
    ///
    /// let total_documents: usize = model.count_documents(
    ///     doc! { ... },
    ///     None::<CountOptions>,
    ///     None::<&mut ClientSession>,
    /// ).await?;
    /// ```
    pub async fn count_documents(
        &self,
        filter: Document,
        options: Option<CountOptions>,
        session: Option<&mut ClientSession>,
    ) -> Result<usize, Error> {
        if let Some(session) = session {
            self.collection
                .count_documents(filter)
                .with_options(options)
                .session(session)
                .await
                .map(|count| count as usize)
                .map_err(Error::Mongo)
        } else {
            self.collection
                .count_documents(filter)
                .with_options(options)
                .await
                .map(|count| count as usize)
                .map_err(Error::Mongo)
        }
    }
}