Skip to content

Commit 654ff21

Browse files
committed
feat: tracing
1 parent e26739f commit 654ff21

File tree

3 files changed

+27
-12
lines changed

3 files changed

+27
-12
lines changed

Cargo.lock

Lines changed: 17 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

crates/sock/Cargo.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ rmp-serde = "1.3.0"
1515
serde = { version = "1.0.188", features = ["serde_derive"] }
1616
serde_json = "1.0.107"
1717
tokio = { version = "1.32.0", features = ["full"] }
18+
tracing = "0.1.41"
1819

1920
[dev-dependencies]
2021
pretty_assertions = "1.4.0"

crates/sock/src/main.rs

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ use tokio::{
77
io::{AsyncReadExt, AsyncWriteExt},
88
net::UnixListener,
99
};
10+
use tracing::{debug, error, info};
1011

1112
#[derive(Deserialize)]
1213
#[serde(tag = "tag", content = "value")]
@@ -46,18 +47,18 @@ async fn run() -> Result<()> {
4647

4748
let listener = UnixListener::bind(socket_path)?;
4849

49-
eprintln!("listening for streams...");
50+
info!("listening for streams...");
5051
while let Ok((mut stream, _addr)) = listener.accept().await {
51-
eprintln!("accepted stream");
52+
debug!("accepted stream");
5253
let mf = markdown_files.clone();
5354
tokio::spawn(async move {
5455
let mut buf = vec![0; 1024];
5556

5657
loop {
57-
eprintln!("reading stream");
58+
debug!("reading stream");
5859
match stream.read(&mut buf).await {
5960
Ok(0) => {
60-
eprintln!("stream terminated");
61+
debug!("stream terminated");
6162
break;
6263
}
6364
Ok(n) => match rmp_serde::from_slice(&buf[..n]) {
@@ -66,7 +67,7 @@ async fn run() -> Result<()> {
6667
sort_key,
6768
order_desc,
6869
}) => {
69-
eprintln!("Received request: {name:?} {sort_key:?} {order_desc:?}");
70+
debug!("Received request: {name:?} {sort_key:?} {order_desc:?}");
7071
let resp_buf = {
7172
let keeper = mf.lock().unwrap();
7273
let response = custard_lib::single::get(
@@ -75,19 +76,19 @@ async fn run() -> Result<()> {
7576
sort_key.as_deref(),
7677
order_desc,
7778
);
78-
eprintln!("Sending response: {response:?}");
79+
debug!("Sending response: {response:?}");
7980
rmp_serde::to_vec(&response).unwrap()
8081
};
8182
stream.write_all(&resp_buf).await.unwrap();
8283
}
8384
Err(err) => {
84-
eprintln!("stream decode failed: {err}");
85+
error!("stream decode failed: {err}");
8586
let nun = rmp_serde::to_vec(&Option::<Response>::None).unwrap();
8687
stream.write_all(&nun).await.unwrap();
8788
}
8889
},
8990
Err(err) => {
90-
eprintln!("stream read failed: {err}");
91+
error!("stream read failed: {err}");
9192
}
9293
}
9394
}

0 commit comments

Comments
 (0)