|
|
@@ -44,6 +44,8 @@ fn log_level_init(log_str: String, port: u32, account_name: String) -> WorkerGua
|
|
|
fn read_params_json() -> Params {
|
|
|
let mut path = "config.json";
|
|
|
let mut call_port = 5555;
|
|
|
+ // 运行模式 0.正常策略运行, 1.清理挂单及仓位
|
|
|
+ let mut run_mode = 0;
|
|
|
|
|
|
let args: Vec<String> = std::env::args().collect();
|
|
|
|
|
|
@@ -66,10 +68,24 @@ fn read_params_json() -> Params {
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+ // Check for the --port argument and assign its value.
|
|
|
+ if arg.starts_with("--run_mode=") {
|
|
|
+ let parts: Vec<&str> = arg.split('=').collect();
|
|
|
+ if parts.len() == 2 {
|
|
|
+ match u32::from_str(parts[1]) {
|
|
|
+ Ok(num) => run_mode = num,
|
|
|
+ Err(_) => eprintln!("Invalid number for run_mode: {}", parts[1]),
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
println!("通讯端口:{}, 配置文件路径:{}", call_port, path);
|
|
|
- let params = Params::new_json(path, call_port).unwrap();
|
|
|
+ let mut params = Params::new_json(path, call_port).unwrap();
|
|
|
+ if run_mode == 1{
|
|
|
+ params.run_mode = 1;
|
|
|
+ }
|
|
|
return params;
|
|
|
}
|
|
|
|