fix ic_tee_logtail

ldclabs · Nov 6, 2024 · 63432a7 · 63432a7
1 parent afb780e
commit 63432a7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/ic_tee_logtail/Cargo.toml b/src/ic_tee_logtail/Cargo.toml
@@ -13,3 +13,4 @@ license.workspace = true
 tokio = { workspace = true }
 anyhow = { workspace = true }
 clap = { workspace = true }
+log = { workspace = true }
diff --git a/src/ic_tee_logtail/src/main.rs b/src/ic_tee_logtail/src/main.rs
@@ -15,14 +15,14 @@ async fn main() -> Result<()> {
     let listener = TcpListener::bind(&cli.ip_addr).await?;
     println!("listening on {:?}", listener.local_addr()?);
 
-    loop {
-        match listener.accept().await {
-            Err(err) => println!("couldn't get client: {:?}", err),
-            Ok((mut stream, addr)) => {
-                println!("accept a client: {:?}", addr);
-                stream.readable().await?;
-                io::copy(&mut stream, &mut io::stdout()).await?;
+    while let Ok((mut stream, addr)) = listener.accept().await {
+        tokio::spawn(async move {
+            log::info!(target: "logtail", "accept a client: {:?}", addr);
+            let _ = stream.readable().await;
+            if let Err(err) = io::copy(&mut stream, &mut io::stdout()).await {
+                log::error!(target: "logtail", "error in transfer: {:?}", err);
             }
-        }
+        });
     }
+    Ok(())
 }