jjanella/ningy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: improved logging

Browse Source
This commit is contained in:
Jeremy Janella
2025-12-06 02:12:53 -05:00
parent 606880f928
commit 4cdf2db0c9
2 changed files with 26 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/netutils.rs
View File

@@ -23,14 +23,15 @@ impl fmt::Display for Backend {
pub async fn tunnel(client_stream: TcpStream, backend: Backend) -> Result<(), Box<dyn Error>> {
let backend_address: String = backend.address.clone();
tokio::spawn(async move {
let backend_stream: TcpStream = match TcpStream::connect(&backend_address).await {
Ok(s) => {
println!("Connected to backend {backend_address}");
info!("connected to backend {backend_address}");
s
}
Err(e) => {
eprintln!("Failed connecting to backend {backend_address}: {e}");
error!("failed connecting to backend {backend_address}: {e}");
return;
}
};
@@ -38,23 +39,11 @@ pub async fn tunnel(client_stream: TcpStream, backend: Backend) -> Result<(), Bo
let (mut read_client, mut write_client) = client_stream.into_split();
let (mut read_backend, mut write_backend) = backend_stream.into_split();
let client_to_backend = tokio::spawn(async move {
match io::copy(&mut read_client, &mut write_backend)
.await
.unwrap()
{
n => println!("{n}B ==> backend"),
}
});
let client_to_backend =
tokio::spawn(async move { io::copy(&mut read_client, &mut write_backend).await });
let backend_to_client = tokio::spawn(async move {
match io::copy(&mut read_backend, &mut write_client)
.await
.unwrap()
{
n => println!("{n}B ==> client"),
}
});
let backend_to_client =
tokio::spawn(async move { io::copy(&mut read_backend, &mut write_client).await });
let _ = tokio::join!(client_to_backend, backend_to_client);
});