54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
use crate::database::builder::merge_query_with_params;
|
|
use crate::database::prelude::*;
|
|
use crate::error::StdResult;
|
|
use postgres::{Client, NoTls};
|
|
|
|
pub struct PostgresConnection {
|
|
client: Client,
|
|
}
|
|
|
|
impl OpenDatabaseConnection for PostgresConnection {
|
|
fn open(connection_string: &str) -> StdResult<Self> {
|
|
let client = Client::connect(connection_string, NoTls)?;
|
|
Ok(PostgresConnection { client })
|
|
}
|
|
}
|
|
|
|
impl DatabaseConnection for PostgresConnection {
|
|
fn migration_table_stmt(&self) -> String {
|
|
r#"CREATE TABLE IF NOT EXISTS migrations (
|
|
id serial PRIMARY KEY,
|
|
name text NOT NULL UNIQUE
|
|
)"#
|
|
.to_string()
|
|
}
|
|
|
|
fn batch_execute(&mut self, query: &str) -> StdResult<()> {
|
|
self.client.batch_execute(query)?;
|
|
Ok(())
|
|
}
|
|
|
|
fn execute<'b>(&mut self, query: &str, params: ToSqlParams<'b>) -> StdResult<u64> {
|
|
let stmt = merge_query_with_params(query, params);
|
|
|
|
let res = self.client.execute(stmt.as_str(), &[])?;
|
|
Ok(res)
|
|
}
|
|
|
|
fn query<'b>(&mut self, query: &str, params: ToSqlParams<'b>) -> StdResult<Vec<Vec<String>>> {
|
|
let stmt = merge_query_with_params(query, params);
|
|
|
|
let res = self.client.query(stmt.as_str(), &[])?;
|
|
|
|
let res = res
|
|
.into_iter()
|
|
.map(|row| {
|
|
let column: String = row.get(0);
|
|
vec![column]
|
|
})
|
|
.collect::<Vec<_>>();
|
|
|
|
Ok(res)
|
|
}
|
|
}
|