toasty 0.10.0

An async ORM for Rust supporting SQL and NoSQL databases
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

mod paginate;
mod upsert;

#[cfg(test)]
mod tests;

use toasty_core::{
    Error, Result, Schema,
    driver::Capability,
    stmt::{self, VisitMut},
};

use super::Engine;

/// Normalizes application-level statements in one recursive traversal.
///
/// Node-specific helpers may inspect or mutate the node they receive, but they
/// must not recurse. The [`VisitMut`] implementation owns traversal so adding a
/// normalization rule does not add another full AST walk.
struct Normalize<'a> {
    schema: &'a Schema,
    capability: &'a Capability,
    error: Option<Error>,
}

impl Engine {
    pub(super) fn normalize_stmt(&self, stmt: &mut stmt::Statement) -> Result<()> {
        let mut normalize = Normalize {
            schema: &self.schema,
            capability: self.capability,
            error: None,
        };
        normalize.visit_stmt_mut(stmt);

        match normalize.error {
            Some(error) => Err(error),
            None => Ok(()),
        }
    }
}

impl Normalize<'_> {
    fn record(&mut self, error: Error) {
        if self.error.is_none() {
            self.error = Some(error);
        }
    }
}

impl VisitMut for Normalize<'_> {
    fn visit_stmt_insert_mut(&mut self, insert: &mut stmt::Insert) {
        self.normalize_upsert_defaults(insert);
        if self.error.is_none() {
            stmt::visit_mut::visit_stmt_insert_mut(self, insert);
        }
    }

    fn visit_stmt_query_mut(&mut self, query: &mut stmt::Query) {
        self.normalize_cursor_order(query);
        if self.error.is_none() {
            stmt::visit_mut::visit_stmt_query_mut(self, query);
        }
    }
}