kienan
/
haunter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Load default driver url from environment variable if it is set

This commit is contained in:
Kienan Stewart 2022-09-25 20:39:06 -04:00
parent cad05a6609
commit 2a25b41baf
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/conf.rs
View File

@ -13,11 +13,15 @@ pub struct Conf {
impl Conf { impl Conf {
pub fn get_default_conf() -> Conf { pub fn get_default_conf() -> Conf {
let default_driver_url = match std::env::var("HAUNTER_DRIVER_URL") {
Err(_) => "http://localhost:4444".to_string(),
Ok(v) => v,
};
return Conf { return Conf {
job_dir: PathBuf::from_str("jobs.d").unwrap(), job_dir: PathBuf::from_str("jobs.d").unwrap(),
output_dir: PathBuf::from_str("results.d").unwrap(), output_dir: PathBuf::from_str("results.d").unwrap(),
check_interval: Duration::new(15*60, 0), check_interval: Duration::new(15*60, 0),
driver_url: String::from_str("http://localhost:4444").unwrap(), driver_url: default_driver_url,
}; };
} }