Using EnvFilter instead hardcoded with_max_level. (#19)

ihciah · Oct 4, 2022 · 246ec8f · 246ec8f
1 parent e8c3eb2
commit 246ec8f
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -20,7 +20,7 @@ pin-project-lite = "0.2"
 rustls = {version = "0.20", default-features = false}
 sha1 = "0.10"
 tracing = "0.1"
-tracing-subscriber = "0.3"
+tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 webpki-roots = "0.22"
 
 [profile.release]

diff --git a/src/main.rs b/src/main.rs
@@ -11,7 +11,8 @@ use std::{rc::Rc, sync::Arc};
 
 use clap::{Parser, Subcommand};
 use monoio::net::TcpListener;
-use tracing::{error, info, Level};
+use tracing::{error, info};
+use tracing_subscriber::{fmt, prelude::*, EnvFilter};
 
 use crate::{client::ShadowTlsClient, server::ShadowTlsServer, util::set_tcp_keepalive};
 
@@ -68,7 +69,10 @@ enum Commands {
 }
 
 fn main() {
-    tracing_subscriber::fmt().with_max_level(Level::INFO).init();
+    tracing_subscriber::registry()
+        .with(fmt::layer())
+        .with(EnvFilter::from_default_env())
+        .init();
     let args = Arc::new(Args::parse());
     let mut threads = Vec::new();
     let parallelism = get_parallelism(&args);