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
use std::rc::Rc; use super::backend::Pg; use super::connection::raw::RawConnection; use query_builder::{QueryBuilder, Binds, BuildQueryResult}; use types::HasSqlType; pub struct PgQueryBuilder { conn: Rc<RawConnection>, pub sql: String, pub binds: Binds, pub bind_types: Vec<u32>, bind_idx: u32, } impl PgQueryBuilder { pub fn new(conn: &Rc<RawConnection>) -> Self { PgQueryBuilder { conn: conn.clone(), sql: String::new(), binds: Vec::new(), bind_types: Vec::new(), bind_idx: 0, } } } impl QueryBuilder<Pg> for PgQueryBuilder { fn push_sql(&mut self, sql: &str) { self.sql.push_str(sql); } fn push_identifier(&mut self, identifier: &str) -> BuildQueryResult { let escaped_identifier = try!(self.conn.escape_identifier(identifier)); Ok(self.push_sql(&escaped_identifier)) } fn push_bound_value<T>(&mut self, bind: Option<Vec<u8>>) where Pg: HasSqlType<T>, { self.bind_idx += 1; let sql = format!("${}", self.bind_idx); self.push_sql(&sql); self.binds.push(bind); self.bind_types.push(Pg::metadata().oid); } }