db-accelerate 2.0.0

Accelerate your databases back and forth through easy to manage migration files.
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

#[cfg(feature = "driver-test")]
pub mod test;
#[cfg(feature = "driver-postgres")]
pub mod postgres;

use error::Error;

pub trait Driver {
  fn get_records(&self) -> Result<Vec<String>, Error>;
  fn add_record(&mut self, record: &str) -> Result<(), Error>;
  fn sub_record(&mut self, record: &str) -> Result<(), Error>;
  fn execute(&mut self, query: String) -> Result<(), Error>;
}

pub fn get(driver_name: Option<&str>, conn_str: &str) -> Result<Box<Driver>, Error> {
  if let Some(driver_name) = driver_name {
    get_by_name(driver_name, conn_str)
  } else {
    get_by_conn_str(conn_str)
  }
}

fn get_by_name(driver_name: &str, conn_str: &str) -> Result<Box<Driver>, Error> {
  match driver_name {
    #[cfg(feature = "driver-test")]
    "test" => Ok(Box::new(test::TestDriver {
      records: if conn_str == "" { vec![] } else { conn_str.split(',').map(String::from).collect() },
      executions: vec![],
    })),

    #[cfg(feature = "driver-postgres")]
    "postgres" => Ok(Box::new(try!(postgres::PostgresDriver::connect(conn_str)))),

    _ => Err(error!("Driver for name '{}' could not be found.", driver_name)),
  }
}

fn get_by_conn_str(conn_str: &str) -> Result<Box<Driver>, Error> {
  match () {
    #[cfg(feature = "driver-postgres")]
    () if postgres::PostgresDriver::will_accept_connection(conn_str) =>
      Ok(Box::new(try!(postgres::PostgresDriver::connect(conn_str)))),

    _ => Err(error!("No driver will accept connection string '{}'. Try disabling driver inference by defining the driver type.", conn_str)),
  }
}