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
use crate::{Dialect, ToSql};
use crate::util::{quote, table_name};
#[derive(Debug, Copy, Clone, PartialEq)]
pub enum OnConflict {
Ignore,
Abort,
}
impl Default for OnConflict {
fn default() -> Self {
OnConflict::Abort
}
}
pub struct Insert {
pub schema: Option<String>,
pub table: String,
pub columns: Vec<String>,
pub values: Vec<Vec<String>>,
pub on_conflict: OnConflict,
pub returning: Vec<String>,
}
fn query_start(dialect: Dialect, on_conflict: OnConflict) -> String {
use Dialect::*;
use OnConflict::*;
if dialect == Sqlite {
match on_conflict {
Ignore => "INSERT OR IGNORE INTO ".to_string(),
Abort => "INSERT OR ABORT INTO ".to_string(),
}
} else {
"INSERT INTO ".to_string()
}
}
impl ToSql for Insert {
fn to_sql(&self, dialect: Dialect) -> String {
let mut q = query_start(dialect, self.on_conflict);
q.push_str(&table_name(self.schema.as_ref(), &self.table, None));
q.push_str(" (");
let mut first = true;
for column in &self.columns {
if !first {
q.push_str(", ");
}
q.push_str(quote(column).as_str());
first = false;
}
q.push_str(") VALUES ");
let mut first = true;
for row in &self.values {
if !first {
q.push_str(", ");
}
q.push_str("(");
let mut first = true;
for value in row {
if !first {
q.push_str(", ");
}
q.push_str(value.as_str());
first = false;
}
q.push_str(")");
first = false;
}
if !self.returning.is_empty() {
q.push_str(" RETURNING ");
let mut first = true;
for column in &self.returning {
if !first {
q.push_str(", ");
}
q.push_str(quote(column).as_str());
first = false;
}
}
q
}
}