derive_sql/proxy/sqlite/
log.rs1use super::*;
2
3pub struct Log<C>
4where C: SqliteTrait
5{
6 conn: Option<C>,
7}
8
9impl<C> std::default::Default for Log<C>
10where C: SqliteTrait
11{
12 fn default() -> Self { Log { conn: None } }
13}
14
15impl<C> Log<C>
16where C: SqliteTrait
17{
18 pub fn with(mut self, conn: C) -> Self { self.conn = Some(conn); self }
19
20 fn log(&self, sql: &str) { ::log::info!("{sql}"); }
21}
22
23impl<C> SqliteTrait for Log<C>
24where C: SqliteTrait
25{
26 fn execute<P>(&self, sql: &str, params: P) -> DeriveSqlResult<usize>
27 where P: rusqlite::Params
28 {
29 self.log(sql);
30 self.conn.as_ref().ok_or(Error::SqliteProxyNoConnectionProvided)?
31 .execute(sql, params)
32 }
33
34 fn query_first<T, P, F>(&self, sql: &str, params: P, f: F) -> DeriveSqlResult<T>
35 where P: rusqlite::Params,
36 F: FnOnce(&rusqlite::Row<'_>) -> rusqlite::Result<T>
37 {
38 self.log(sql);
39 self.conn.as_ref().ok_or(Error::SqliteProxyNoConnectionProvided)?
40 .query_first(sql, params, f)
41 }
42
43 fn query_map<T, P, F>(&self, sql: &str, params: P, f: F) -> DeriveSqlResult<Vec<T>>
44 where P: rusqlite::Params,
45 F: FnMut(&rusqlite::Row<'_>) -> rusqlite::Result<T>
46 {
47 self.log(sql);
48 self.conn.as_ref().ok_or(Error::SqliteProxyNoConnectionProvided)?
49 .query_map(sql, params, f)
50 }
51}
52
53