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