Add dotenv for config

judger/src/client/environment/.env.development

@@ -0,0 +1,2 @@
+BASE_URL="http://localhost:8080/api/v1/judge"
+RUST_LOG=DEBUG

judger/src/client/environment/mod.rs

@@ -0,0 +1,31 @@
+use structopt::StructOpt;
+
+#[derive(StructOpt, Debug, Clone)]
+#[structopt(name = "judge-client")]
+pub struct JudgeClientOpt {
+    #[structopt(long)]
+    pub env_path: Option<String>,
+
+    /// Port to listen to
+    #[structopt(env = "BASE_URL", default_value = "http://localhost:8080/api/v1/judge")]
+    pub base_url: String,
+}
+
+pub fn load_option() -> JudgeClientOpt {
+    // First load env_path from Args
+    let opt = JudgeClientOpt::from_args();
+    if let Some(env_path) = opt.env_path {
+        dotenv::from_path(env_path).ok();
+    } else {
+        dotenv::dotenv().ok();
+    }
+
+    // Load opt again with ENV
+    let opt = JudgeClientOpt::from_args();
+    log::debug!("load opt: {:?}", opt);
+    opt
+}
+
+pub fn setup_logger() {
+    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("debug")).init();
+}

judger/src/client/main.rs

@@ -1,5 +1,7 @@
 mod client;
+mod environment;
 mod error;
+
 use client::HttpClient;
 use error::ClientError;
 use judge_core::judge;
@@ -47,9 +49,10 @@ struct JudgeTask {
 
 #[tokio::main]
 async fn main() {
-    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("debug")).init();
+    let opt = environment::load_option();
+    environment::setup_logger();
     let mut interval = interval(Duration::from_secs(10));
-    let base_url = "http://localhost:8080/api/v1/judge".to_string();
+    let base_url = opt.base_url;
     let client = client::HttpClient::new(base_url);
 
     loop {
@@ -66,7 +69,7 @@ async fn main() {
                             log::debug!("Report failed {:?}", report_response);
                             return;
                         }
-                        log::debug!("Submission {:?} report success", submission_uid);
+                        log::info!("Submission {:?} report success", submission_uid);
                     }
                     Err(e) => log::info!("Error judge task: {:?}", e),
                 }