extern crate postgres;
use postgres::{Connection, TlsMode};
#[derive(Debug)]
pub struct ConnectionParams {
pub server: String,
pub port: Option<i32>,
pub database: String,
pub user: String,
pub password: Option<String>,
pub tls: bool
}
impl ConnectionParams {
fn url(&self) -> String {
let port = self.port.map(|x| format!(":{}", x)).unwrap_or("".to_string());
let user_pass = self.password.as_ref().map(|p| format!("{}:{}", &self.user, p)).unwrap_or(format!("{}", &self.user));
format!("postgres://{}@{}{}", &user_pass, &self.server, &port)
}
}
impl Default for ConnectionParams {
fn default() -> ConnectionParams {
ConnectionParams {
server: String::from("localhost"),
port: None,
database: String::from("postgres"),
user: String::from("postgres"),
password: None,
tls: false
}
}
}
pub fn open(conn: ConnectionParams) -> Connection {
Connection::connect(conn.url(), TlsMode::None).unwrap()
}