64 lines
2.0 KiB
Rust
64 lines
2.0 KiB
Rust
|
|
||
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
|
||
|
|
pub enum ClusterByoc {
|
||
|
|
http,
|
||
|
|
https,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||
|
|
pub struct Config {
|
||
|
|
pub cluster_port: u32,
|
||
|
|
pub cluster_id: u32,
|
||
|
|
pub cluster_secret: String,
|
||
|
|
/// CLUSTER_BYOC
|
||
|
|
pub cluster_byoc: ClusterByoc,
|
||
|
|
pub no_demaon: bool,
|
||
|
|
/// DISABLE_ACCESS_LOG
|
||
|
|
pub disable_access_log: bool,
|
||
|
|
/// FORCE_NOOPEN
|
||
|
|
pub force_noopen: bool,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Config {
|
||
|
|
pub fn new_from_env() -> Self {
|
||
|
|
let cluster_port = std::env::var("CLUSTER_PORT")
|
||
|
|
.unwrap_or_else(|_| "8080".to_string())
|
||
|
|
.parse::<u32>()
|
||
|
|
.expect("CLUSTER_PORT must be a number");
|
||
|
|
let cluster_id = std::env::var("CLUSTER_ID")
|
||
|
|
.unwrap_or_else(|_| "1".to_string())
|
||
|
|
.parse::<u32>()
|
||
|
|
.expect("CLUSTER_ID must be a number");
|
||
|
|
let cluster_secret = std::env::var("CLUSTER_SECRET").expect("CLUSTER_SECRET must be set");
|
||
|
|
let cluster_byoc = match std::env::var("CLUSTER_BYOC")
|
||
|
|
.unwrap_or_else(|_| "http".to_string())
|
||
|
|
.as_str()
|
||
|
|
{
|
||
|
|
"http" => ClusterByoc::http,
|
||
|
|
"https" => ClusterByoc::https,
|
||
|
|
_ => panic!("CLUSTER_BYOC must be http or https"),
|
||
|
|
};
|
||
|
|
let no_demaon = std::env::var("NO_DAEMON")
|
||
|
|
.unwrap_or_else(|_| "false".to_string())
|
||
|
|
.parse::<bool>()
|
||
|
|
.expect("NO_DAEMON must be true or false");
|
||
|
|
let disable_access_log = std::env::var("DISABLE_ACCESS_LOG")
|
||
|
|
.unwrap_or_else(|_| "false".to_string())
|
||
|
|
.parse::<bool>()
|
||
|
|
.expect("DISABLE_ACCESS_LOG must be true or false");
|
||
|
|
let force_noopen = std::env::var("FORCE_NOOPEN")
|
||
|
|
.unwrap_or_else(|_| "false".to_string())
|
||
|
|
.parse::<bool>()
|
||
|
|
.expect("FORCE_NOOPEN must be true or false");
|
||
|
|
Self {
|
||
|
|
cluster_port,
|
||
|
|
cluster_id,
|
||
|
|
cluster_secret,
|
||
|
|
cluster_byoc,
|
||
|
|
no_demaon,
|
||
|
|
disable_access_log,
|
||
|
|
force_noopen,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|