修改挂断呼叫逻辑，增添命令行参数以及环境变量适配

src/main.rs

@@ -10,14 +10,22 @@ mod deport;
 
 use crate::events::handle_events;
 use client::handle_client;
-use config::config::STATIC_ADDR as addr;
 use tokio::net::TcpListener;
 use tokio::sync::mpsc;
+use utils::utils::get_addr;
 
 #[tokio::main]
 async fn main() {
 
-    let listener = TcpListener::bind(addr).await.unwrap();
+    let bind_addr = match get_addr() {
+        Ok(other_addr) => other_addr,
+        Err(err) => {
+            eprintln!("Error: {}", err);
+            std::process::exit(1);
+        }
+    };
+
+    let listener = TcpListener::bind(bind_addr).await.unwrap();
     // 创建事件通道
     let (event_sender, event_receiver) = mpsc::unbounded_channel();
     // 启动事件处理任务

src/utils/utils.rs

 use std::collections::HashMap;
+use std::env;
+use std::net::SocketAddr;
+use crate::config::config::STATIC_ADDR;
 
 pub(crate) fn get_connection_params(
     connection_url: String,
@@ -17,3 +20,27 @@ pub(crate) fn get_connection_params(
         Ok(HashMap::new())
     }
 }
+
+// 获取地址的逻辑封装
+pub fn get_addr() -> Result<String, Box<dyn std::error::Error>> {
+    // 1. 处理命令行参数
+    let mut args = env::args().skip(1);
+    if let Some(pos) = args.position(|arg| arg == "--addr") {
+        // 获取下一个参数值
+        return args.nth(pos)
+            .ok_or("--addr requires a value".into())
+            .and_then(|addr| {
+                addr.parse::<SocketAddr>()?;
+                Ok(addr)
+            });
+    }
+
+    // 2. 处理环境变量
+    if let Ok(env_addr) = env::var("RS_WS_ADDR") {
+        env_addr.parse::<SocketAddr>()?;
+        return Ok(env_addr);
+    }
+
+    // 3. 返回默认值
+    Ok(STATIC_ADDR.to_string())
+}
\ No newline at end of file