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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
use std::collections::HashMap;
pub fn select(
dbconn: &rusqlite::Connection,
table_name: &str,
record_id: &str,
) -> yaml_rust::Yaml {
use crate::yaml::Yaml;
let mut stmt = dbconn.prepare(
&format!(
r#"
SELECT
*
FROM "{}"
WHERE "{}" = :pk;
"#,
table_name,
"id",
),
).unwrap();
let column_count = stmt.column_count();
let column_names: Vec<yaml_rust::Yaml> = stmt.column_names()
.iter()
.map(|col| yaml_rust::Yaml::String((*col).to_owned()))
.collect();
// dbg!(column_names);
let rows = stmt.query_map(
rusqlite::named_params! {
":pk": record_id,
},
|row| {
// let mut data: Vec<Yaml> = Vec::with_capacity(column_count);
//
// for i in 0..column_count {
// data.push(row.get(i)?);
// }
let mut data: HashMap<&yaml_rust::Yaml, Yaml> = HashMap::new();
for (i, column) in column_names.iter().enumerate() {
data.insert(&column, row.get(i)?);
}
Ok(data)
},
).unwrap();
for row_result in rows {
let row = row_result.unwrap();
dbg!(&row);
}
// todo!();
yaml_rust::Yaml::Null
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn select_extracts_a_database_record_as_yaml() {
struct TestRecord {
count: i16,
description: String,
}
let record = TestRecord {
count: 99,
description: "This is a test.
With multiple paragraphs.".to_owned(),
};
let mut yaml_hash = yaml_rust::yaml::Hash::new();
yaml_hash.insert(
yaml_rust::Yaml::String("count".to_owned()),
yaml_rust::Yaml::Integer(record.count.into()),
);
yaml_hash.insert(
yaml_rust::Yaml::String("description".to_owned()),
yaml_rust::Yaml::String(record.description.clone()),
);
let expected = yaml_rust::Yaml::Hash(yaml_hash);
let conn = rusqlite::Connection::open_in_memory().unwrap();
conn.execute(
r#"
CREATE TABLE "test" (
id INTEGER PRIMARY KEY,
count INTEGER,
description TEXT
);
"#,
[]
).unwrap();
{
let mut stmt = conn.prepare(r#"
INSERT INTO "test"
(count, description)
VALUES
(?, ?);
"#).unwrap();
stmt.insert(
rusqlite::params![record.count, record.description],
).unwrap();
let got = select(&conn, "test", "1");
dbg!(&got);
assert_eq!(expected, got);
}
conn.close().unwrap();
}
}
|
