Modified request-mirror to accept environment variables from docker which would override the .env vars
This commit is contained in:
parent
151c72d5c1
commit
72c3ae5dbe
10
src/lib.rs
10
src/lib.rs
|
|
@ -23,9 +23,17 @@ use schema::{
|
||||||
/// ```
|
/// ```
|
||||||
pub fn establish_connection() -> PgConnection {
|
pub fn establish_connection() -> PgConnection {
|
||||||
|
|
||||||
|
let key: &str = "DATABASE_URL";
|
||||||
|
let database_url: String;
|
||||||
|
match env::var(key) {
|
||||||
|
Ok(val) => database_url = val,
|
||||||
|
Err(e) => {
|
||||||
dotenv().ok();
|
dotenv().ok();
|
||||||
|
|
||||||
let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
|
database_url = env::var(key).expect("DATABASE_URL must be set");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
PgConnection::establish(&database_url)
|
PgConnection::establish(&database_url)
|
||||||
.unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
|
.unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue