welds 0.4.22

An async ORM for (postgres, mssql, mysql, sqlite)
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

use crate::Syntax;

pub struct CountWriter {
    syntax: Syntax,
}

impl CountWriter {
    pub fn new(syntax: Syntax) -> Self {
        Self { syntax }
    }
    pub fn count(&self, prefix: Option<&str>, x: Option<&str>) -> String {
        match self.syntax {
            Syntax::Mysql => MySql::count(prefix, x),
            Syntax::Postgres => Postgres::count(prefix, x),
            Syntax::Sqlite => Sqlite::count(prefix, x),
            Syntax::Mssql => Mssql::count(prefix, x),
        }
    }
}

struct Postgres;
impl Postgres {
    fn count(prefix: Option<&str>, x: Option<&str>) -> String {
        let mut x = x.unwrap_or("*").to_owned();
        if let Some(prefix) = prefix {
            x = format!("{}.{}", prefix, x);
        }
        format!("CAST( COUNT({}) as BIGINT )", x)
    }
}

struct Sqlite;
impl Sqlite {
    fn count(_prefix: Option<&str>, x: Option<&str>) -> String {
        let x = x.unwrap_or("*");
        format!("CAST( COUNT({}) as BIGINT )", x)
    }
}

struct Mssql;
impl Mssql {
    fn count(_prefix: Option<&str>, x: Option<&str>) -> String {
        let x = x.unwrap_or("*");
        format!("CAST( COUNT({}) as BIGINT )", x)
    }
}

struct MySql;
impl MySql {
    fn count(_prefix: Option<&str>, x: Option<&str>) -> String {
        let x = x.unwrap_or("*");
        format!("COUNT({})", x)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn pg_should_counts() {
        let w = CountWriter::new(Syntax::Postgres);
        assert_eq!(w.count(None, None), "CAST( COUNT(*) as BIGINT )");
        assert_eq!(
            w.count(None, Some("sheep")),
            "CAST( COUNT(sheep) as BIGINT )"
        );
        assert_eq!(
            w.count(Some("t1"), Some("sheep")),
            "CAST( COUNT(t1.sheep) as BIGINT )"
        );
    }
}