vantage-mongodb 0.4.10

MongoDB persistence backend for Vantage framework
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//! Builder methods for MongoSelect — produce native MongoDB types
//! (filter Document, projection Document, sort Document, FindOptions).

use bson::Document;

use super::MongoSelect;

impl MongoSelect {
    /// Build the filter `Document` by resolving all conditions.
    pub async fn build_filter(&self) -> vantage_core::Result<Document> {
        crate::condition::resolve_conditions(self.conditions.iter()).await
    }

    /// Build the projection `Document`. Returns `None` if all fields requested.
    pub fn build_projection(&self) -> Option<Document> {
        if self.fields.is_empty() {
            return None;
        }
        let mut proj = Document::new();
        for f in &self.fields {
            proj.insert(f.as_str(), 1);
        }
        Some(proj)
    }

    /// Build the sort `Document`. Returns `None` if no ordering.
    pub fn build_sort(&self) -> Option<Document> {
        if self.sort.is_empty() {
            return None;
        }
        let mut doc = Document::new();
        for (field, dir) in &self.sort {
            doc.insert(field.as_str(), *dir);
        }
        Some(doc)
    }

    /// Build `mongodb::options::FindOptions` from the current state.
    pub fn build_find_options(&self) -> mongodb::options::FindOptions {
        let mut opts = mongodb::options::FindOptions::default();
        opts.projection = self.build_projection();
        opts.sort = self.build_sort();
        opts.limit = self.limit;
        opts.skip = self.skip.map(|s| s as u64);
        opts
    }
}

#[cfg(test)]
mod tests {
    use bson::doc;

    use super::*;
    use vantage_expressions::Selectable;

    #[test]
    fn test_build_projection() {
        let s = MongoSelect::new().with_field("name").with_field("price");
        let proj = s.build_projection().unwrap();
        assert_eq!(proj, doc! { "name": 1, "price": 1 });
    }

    #[test]
    fn test_build_projection_empty() {
        let s = MongoSelect::new();
        assert!(s.build_projection().is_none());
    }

    #[test]
    fn test_build_sort() {
        let mut s = MongoSelect::new();
        s.sort.push(("price".into(), 1));
        s.sort.push(("name".into(), -1));
        let sort = s.build_sort().unwrap();
        assert_eq!(sort, doc! { "price": 1, "name": -1 });
    }

    #[test]
    fn test_build_find_options() {
        let s = MongoSelect::new()
            .with_field("name")
            .with_limit(Some(5), Some(10));
        let opts = s.build_find_options();
        assert!(opts.projection.is_some());
        assert_eq!(opts.limit, Some(5));
        assert_eq!(opts.skip, Some(10));
    }

    #[tokio::test]
    async fn test_build_filter_empty() {
        let s = MongoSelect::new();
        let filter = s.build_filter().await.unwrap();
        assert_eq!(filter, doc! {});
    }

    #[tokio::test]
    async fn test_build_filter_single() {
        let s = MongoSelect::new().with_condition(doc! { "active": true });
        let filter = s.build_filter().await.unwrap();
        assert_eq!(filter, doc! { "active": true });
    }

    #[tokio::test]
    async fn test_build_filter_multiple() {
        let s = MongoSelect::new()
            .with_condition(doc! { "active": true })
            .with_condition(doc! { "price": { "$gt": 100 } });
        let filter = s.build_filter().await.unwrap();
        assert_eq!(
            filter,
            doc! { "$and": [{ "active": true }, { "price": { "$gt": 100 } }] }
        );
    }
}