[#1237] feat(rust): support populating args by clap

rust/experimental/server/Cargo.toml

@@ -93,6 +93,7 @@ tokio-console = "0.1.8"
 console-subscriber = "0.1.9"
 pin-project-lite = "0.2.8"
 signal-hook = "0.3.17"
+clap = "3.0.14"
 
 [dependencies.hdrs]
 version = "0.3.0"

rust/experimental/server/README.md

@@ -141,7 +141,7 @@ cargo build --features hdfs --release
     ```
 2. worker run with tokio-console. the log level of `trace` must be enabled
     ```shell
-    WORKER_IP={ip} RUST_LOG=trace WORKER_CONFIG_PATH=./config.toml ./uniffle-worker
+    WORKER_IP={ip} RUST_LOG=trace ./uniffle-worker -c ./config.toml 
     ```
 3. tokio-console client side connect
     ```shell

rust/experimental/server/src/config.rs

@@ -232,6 +232,15 @@ impl StorageType {
 const CONFIG_FILE_PATH_KEY: &str = "WORKER_CONFIG_PATH";
 
 impl Config {
+    pub fn from(cfg_path: &str) -> Self {
+        let path = Path::new(cfg_path);
+
+        // Read the file content as a string
+        let file_content = fs::read_to_string(path).expect("Failed to read file");
+
+        toml::from_str(&file_content).unwrap()
+    }
+
     pub fn create_from_env() -> Config {
         let path = match std::env::var(CONFIG_FILE_PATH_KEY) {
             Ok(val) => val,
@@ -241,12 +250,7 @@ impl Config {
             ),
         };
 
-        let path = Path::new(&path);
-
-        // Read the file content as a string
-        let file_content = fs::read_to_string(path).expect("Failed to read file");
-
-        toml::from_str(&file_content).unwrap()
+        Config::from(&path)
     }
 }
 

rust/experimental/server/src/main.rs

@@ -32,6 +32,7 @@ use crate::signal::details::wait_for_signal;
 use crate::util::{gen_worker_uid, get_local_ip};
 
 use anyhow::Result;
+use clap::{App, Arg};
 use log::info;
 use std::net::{IpAddr, Ipv4Addr, SocketAddr};
 use std::time::Duration;
@@ -166,7 +167,22 @@ fn init_log(log: &LogConfig) -> WorkerGuard {
 }
 
 fn main() -> Result<()> {
-    let config = Config::create_from_env();
+    let args_match = App::new("Uniffle Worker")
+        .version("0.9.0-SNAPSHOT")
+        .about("Rust based shuffle server for Apache Uniffle")
+        .arg(
+            Arg::with_name("config")
+                .short('c')
+                .long("config")
+                .value_name("FILE")
+                .default_value("./config.toml")
+                .help("Sets a custom config file")
+                .takes_value(true),
+        )
+        .get_matches();
+
+    let config_path = args_match.value_of("config").unwrap_or("./config.toml");
+    let config = Config::from(config_path);
 
     let runtime_manager = RuntimeManager::from(config.runtime_config.clone());