Browse Source

Small adjustment

Signed-off-by: BlackDex <black.dex@gmail.com>
pull/6279/head
BlackDex 6 days ago
parent
commit
65b119f178
No known key found for this signature in database GPG Key ID: 58C80A2AA6C765E1
  1. 10
      src/db/mod.rs

10
src/db/mod.rs

@ -67,22 +67,20 @@ impl DbConnManager {
} }
fn establish_connection(&self) -> Result<DbConnInner, diesel::r2d2::Error> { fn establish_connection(&self) -> Result<DbConnInner, diesel::r2d2::Error> {
let url = &self.database_url; match DbConnType::from_url(&self.database_url) {
match DbConnType::from_url(url) {
#[cfg(mysql)] #[cfg(mysql)]
Ok(DbConnType::Mysql) => { Ok(DbConnType::Mysql) => {
let conn = diesel::mysql::MysqlConnection::establish(url)?; let conn = diesel::mysql::MysqlConnection::establish(&self.database_url)?;
Ok(DbConnInner::Mysql(conn)) Ok(DbConnInner::Mysql(conn))
} }
#[cfg(postgresql)] #[cfg(postgresql)]
Ok(DbConnType::Postgresql) => { Ok(DbConnType::Postgresql) => {
let conn = diesel::pg::PgConnection::establish(url)?; let conn = diesel::pg::PgConnection::establish(&self.database_url)?;
Ok(DbConnInner::Postgresql(conn)) Ok(DbConnInner::Postgresql(conn))
} }
#[cfg(sqlite)] #[cfg(sqlite)]
Ok(DbConnType::Sqlite) => { Ok(DbConnType::Sqlite) => {
let conn = diesel::sqlite::SqliteConnection::establish(url)?; let conn = diesel::sqlite::SqliteConnection::establish(&self.database_url)?;
Ok(DbConnInner::Sqlite(conn)) Ok(DbConnInner::Sqlite(conn))
} }

Loading…
Cancel
Save