vantage-mongodb 0.4.2

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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

//! Selectable<AnyMongoType, MongoCondition> implementation for MongoSelect.

use vantage_expressions::traits::selectable::SourceRef;
use vantage_expressions::{Expression, Expressive, ExpressiveEnum, Order, Selectable};

use crate::condition::MongoCondition;
use crate::select::MongoSelect;
use crate::types::AnyMongoType;

impl Selectable<AnyMongoType, MongoCondition> for MongoSelect {
    fn add_source(&mut self, source: impl Into<SourceRef<AnyMongoType>>, _alias: Option<String>) {
        let source_ref = source.into().into_expressive_enum();
        if let ExpressiveEnum::Scalar(val) = source_ref
            && let Some(name) = val.try_get::<String>()
        {
            self.collection = Some(name);
        }
    }

    fn add_field(&mut self, field: impl Into<String>) {
        self.fields.push(field.into());
    }

    fn add_expression(&mut self, _expression: impl Expressive<AnyMongoType>) {
        // MongoDB projections don't support arbitrary expressions in find().
    }

    fn add_where_condition(&mut self, condition: impl Into<MongoCondition>) {
        self.conditions.push(condition.into());
    }

    fn set_distinct(&mut self, distinct: bool) {
        self.distinct = distinct;
    }

    fn add_order_by(&mut self, order: impl Into<MongoCondition>, direction: Order) {
        // Convention: a MongoCondition::Doc with a single key → use that key.
        // The direction from the Order param overrides whatever value is in the doc.
        let cond = order.into();
        if let MongoCondition::Doc(doc) = &cond
            && let Some((key, _)) = doc.iter().next()
        {
            let dir = if direction.ascending { 1 } else { -1 };
            self.sort.push((key.to_string(), dir));
        }
    }

    fn add_group_by(&mut self, expression: impl Expressive<AnyMongoType>) {
        let preview = expression.preview();
        self.group_by.push(preview);
    }

    fn set_limit(&mut self, limit: Option<i64>, skip: Option<i64>) {
        self.limit = limit;
        self.skip = skip;
    }

    fn clear_fields(&mut self) {
        self.fields.clear();
    }

    fn clear_where_conditions(&mut self) {
        self.conditions.clear();
    }

    fn clear_order_by(&mut self) {
        self.sort.clear();
    }

    fn clear_group_by(&mut self) {
        self.group_by.clear();
    }

    fn has_fields(&self) -> bool {
        !self.fields.is_empty()
    }

    fn has_where_conditions(&self) -> bool {
        !self.conditions.is_empty()
    }

    fn has_order_by(&self) -> bool {
        !self.sort.is_empty()
    }

    fn has_group_by(&self) -> bool {
        !self.group_by.is_empty()
    }

    fn is_distinct(&self) -> bool {
        self.distinct
    }

    fn get_limit(&self) -> Option<i64> {
        self.limit
    }

    fn get_skip(&self) -> Option<i64> {
        self.skip
    }

    fn as_field(&self, _field: impl Into<String>) -> Expression<AnyMongoType> {
        todo!("as_field not yet implemented for MongoDB")
    }

    fn as_count(&self) -> Expression<AnyMongoType> {
        let coll = self.collection.as_deref().unwrap_or("?");
        Expression::new(format!("db.{}.countDocuments()", coll), vec![])
    }

    fn as_sum(&self, column: impl Expressive<AnyMongoType>) -> Expression<AnyMongoType> {
        let coll = self.collection.as_deref().unwrap_or("?");
        Expression::new(
            format!("db.{}.aggregate($sum: {})", coll, column.preview()),
            vec![],
        )
    }

    fn as_max(&self, column: impl Expressive<AnyMongoType>) -> Expression<AnyMongoType> {
        let coll = self.collection.as_deref().unwrap_or("?");
        Expression::new(
            format!("db.{}.aggregate($max: {})", coll, column.preview()),
            vec![],
        )
    }

    fn as_min(&self, column: impl Expressive<AnyMongoType>) -> Expression<AnyMongoType> {
        let coll = self.collection.as_deref().unwrap_or("?");
        Expression::new(
            format!("db.{}.aggregate($min: {})", coll, column.preview()),
            vec![],
        )
    }
}