feat(worker): use hostname as ws arg

AOSC-Dev · Jun 19, 2024 · 8b75f5b · 8b75f5b
1 parent 807728e
commit 8b75f5b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/worker/src/build.rs b/worker/src/build.rs
@@ -363,16 +363,18 @@ async fn build_worker_inner(args: &Args) -> anyhow::Result<()> {
         .timeout(Duration::from_secs(30))
         .build()
         .unwrap();
+
+    let hostname = gethostname::gethostname().to_string_lossy().to_string();
     let req = WorkerPollRequest {
-        hostname: gethostname::gethostname().to_string_lossy().to_string(),
+        hostname: hostname.clone(),
         arch: args.arch.clone(),
         worker_secret: args.worker_secret.clone(),
         memory_bytes: get_memory_bytes(),
         disk_free_space_bytes: fs2::free_space(std::env::current_dir()?)? as i64,
         logical_cores: num_cpus::get() as i32,
     };
 
-    let ws = Url::parse(&args.websocket)?.join(&args.port.to_string())?;
+    let ws = Url::parse(&args.websocket)?.join(&hostname)?;
 
     let (tx, rx) = unbounded();
 

diff --git a/worker/src/lib.rs b/worker/src/lib.rs
@@ -53,10 +53,6 @@ pub struct Args {
     /// Websocket uri
     #[arg(short = 'w', long, env = "BUILDIT_WS")]
     pub websocket: String,
-
-    /// Worker machine relay port
-    #[arg(long, env = "BUILDIT_RELAY_PORT")]
-    pub port: u16,
 }
 
 pub fn get_memory_bytes() -> i64 {