log span events

tlsnotary · Jan 23, 2024 · fa231af · fa231af
1 parent a164191
commit fa231af
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/tlsn/benches/Cargo.toml b/tlsn/benches/Cargo.toml
@@ -20,7 +20,7 @@ tokio = { workspace = true, features = [
     "fs",
 ] }
 tokio-util.workspace = true
-tracing-subscriber.workspace = true
+tracing-subscriber = { workspace = true, features = ["env-filter"] }
 
 [[bin]]
 name = "setup_network"

diff --git a/tlsn/benches/src/prover.rs b/tlsn/benches/src/prover.rs
@@ -6,10 +6,14 @@ use tlsn_server_fixture::{CA_CERT_DER, SERVER_DOMAIN};
 use tokio_util::compat::TokioAsyncReadCompatExt;
 
 use tlsn_prover::tls::{Prover, ProverConfig};
+use tracing_subscriber::{fmt::format::FmtSpan, EnvFilter};
 
 #[tokio::main]
 async fn main() {
-    tracing_subscriber::fmt::init();
+    tracing_subscriber::fmt()
+        .with_env_filter(EnvFilter::from_default_env())
+        .with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)
+        .init();
 
     let (client_conn, server_conn) = tokio::io::duplex(2 << 16);
     let server_task = tokio::spawn(tlsn_server_fixture::bind(server_conn.compat()));

diff --git a/tlsn/benches/src/verifier.rs b/tlsn/benches/src/verifier.rs
@@ -3,10 +3,14 @@ use tlsn_server_fixture::CA_CERT_DER;
 use tokio_util::compat::TokioAsyncReadCompatExt;
 
 use tlsn_verifier::tls::{Verifier, VerifierConfig};
+use tracing_subscriber::{fmt::format::FmtSpan, EnvFilter};
 
 #[tokio::main]
 async fn main() {
-    tracing_subscriber::fmt::init();
+    tracing_subscriber::fmt()
+        .with_env_filter(EnvFilter::from_default_env())
+        .with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)
+        .init();
 
     let ip = std::env::var("VERIFIER_IP").unwrap_or_else(|_| "10.10.1.1".to_string());
     let port: u16 = std::env::var("VERIFIER_PORT")