Modified request-mirror to accept environment variables from docker which would override the .env vars

This commit is contained in:
Camerin Figueroa 2024-05-03 00:41:26 +00:00
parent 151c72d5c1
commit 72c3ae5dbe
1 changed files with 10 additions and 2 deletions

View File

@ -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))
}