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

use crate::types::Field;
use crate::Table;
use std::marker::PhantomData;

pub struct QueryableWriter<Q: ?Sized> {
    queryable: PhantomData<Q>,
}

impl<Q: Queryable + ?Sized> WriteQueryable for QueryableWriter<Q> {
    fn write_query(&self, sql: &mut String) {
        Q::write_queryable(sql);
    }
}

pub trait Queryable {
    fn write_queryable(sql: &mut String);

    fn queryable() -> QueryableWriter<Self> {
        QueryableWriter {
            queryable: PhantomData,
        }
    }
}

pub trait WriteQueryable {
    fn write_query(&self, sql: &mut String);
}

pub struct WildCard;

impl WriteQueryable for WildCard {
    fn write_query(&self, sql: &mut String) {
        sql.push('*');
    }
}

pub struct Count<T> {
    column: T,
}

impl<T> Count<T> {
    pub(crate) fn new(column: T) -> Self {
        Self { column }
    }
}

impl WriteQueryable for Count<()> {
    fn write_query(&self, sql: &mut String) {
        write_wildcard(sql);
    }
}

impl WriteQueryable for Count<WildCard> {
    fn write_query(&self, sql: &mut String) {
        write_wildcard(sql);
    }
}

impl<T: Table, A> WriteQueryable for Count<Field<T, A>> {
    fn write_query(&self, sql: &mut String) {
        sql.push_str("COUNT(");
        self.column.write_field(sql);
        sql.push(')');
    }
}

fn write_wildcard(sql: &mut String) {
    sql.push_str("COUNT(*)");
}