saola-quaint 0.2.0-alpha.14

An abstraction layer for SQL databases (PostgreSQL, MySQL, SQLite, MSSQL)
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

use std::borrow::Cow;

use crate::Value;
use query_template::{Fragment, QueryTemplate};

pub(crate) trait QueryWriter {
    fn write_string_chunk(&mut self, value: String);
    fn write_parameter(&mut self);
    fn write_parameter_tuple(&mut self);
    fn write_parameter_tuple_list(
        &mut self,
        item_prefix: impl Into<Cow<'static, str>>,
        item_separator: impl Into<Cow<'static, str>>,
        item_suffix: impl Into<Cow<'static, str>>,
        group_separator: impl Into<Cow<'static, str>>,
    );
}

impl QueryWriter for QueryTemplate<Value<'_>> {
    fn write_string_chunk(&mut self, value: String) {
        match self.fragments.last_mut() {
            Some(Fragment::StringChunk { chunk }) => {
                chunk.push_str(value.as_str());
            }
            _ => {
                self.fragments.push(Fragment::StringChunk { chunk: value });
            }
        }
    }

    fn write_parameter(&mut self) {
        self.fragments.push(Fragment::Parameter);
    }

    fn write_parameter_tuple(&mut self) {
        self.fragments.push(Fragment::ParameterTuple);
    }

    fn write_parameter_tuple_list(
        &mut self,
        item_prefix: impl Into<Cow<'static, str>>,
        item_separator: impl Into<Cow<'static, str>>,
        item_suffix: impl Into<Cow<'static, str>>,
        group_separator: impl Into<Cow<'static, str>>,
    ) {
        self.fragments.push(Fragment::ParameterTupleList {
            item_prefix: item_prefix.into(),
            item_separator: item_separator.into(),
            item_suffix: item_suffix.into(),
            group_separator: group_separator.into(),
        });
    }
}