wdes/security
1
0
Fork 0
Code Actions Packages Releases Activity

Support ENV WORKER_URL for CLI

Browse Source
This commit is contained in:
William Desportes
2024-09-27 21:02:49 +02:00
parent b731f6dc21
commit 36468e4baf
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
snow-scanner/src/worker/worker.rs
View File

@ -171,9 +171,13 @@ fn main() -> () {
}) })
.start(); .start();
info!("Running the worker"); info!("Running the worker");
let url = "ws://127.0.0.1:8800"; let url = match env::var("WORKER_URL") {
Ok(worker_url) => worker_url,
Err(_) => "ws://127.0.0.1:8800".to_string(),
};
let mut worker = Worker::initial(); let mut worker = Worker::initial();
match ws2::connect(url) { match ws2::connect(&url) {
Ok(mut ws_client) => { Ok(mut ws_client) => {
let connected = ws_client.is_open(); let connected = ws_client.is_open();
if connected { if connected {