Add env-logger

Add env-logger and configure with log level filter debug if the logger is not configured via the RUST_LOG env variable.
bschwind · Apr 27, 2022 · 95bcab9 · 95bcab9
1 parent 7e1cdad
commit 95bcab9
Show file tree

Hide file tree

Showing 3 changed files with 84 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/mqtt-v5-broker/Cargo.toml b/mqtt-v5-broker/Cargo.toml
@@ -16,3 +16,4 @@ mqtt-v5 = { path = "../mqtt-v5", version = "0.2" }
 nanoid = "0.3"
 tokio = { version = "1", features = ["net", "rt-multi-thread", "sync", "time", "macros"] }
 tokio-util = { version = "0.6", features = ["codec"] }
+env_logger = "0.8.4"
diff --git a/mqtt-v5-broker/src/main.rs b/mqtt-v5-broker/src/main.rs
@@ -1,3 +1,5 @@
+use std::env;
+
 use crate::{
     broker::{Broker, BrokerMessage},
     client::UnconnectedClient,
@@ -197,6 +199,12 @@ async fn websocket_server_loop(broker_tx: Sender<BrokerMessage>) {
 }
 
 fn main() -> Result<(), Box<dyn std::error::Error>> {
+    if env::var("RUST_LOG").is_err() {
+        env_logger::builder().filter(None, log::LevelFilter::Debug).init();
+    } else {
+        env_logger::init();
+    }
+
     // Creating a Runtime does the following:
     // * Spawn a background thread running a Reactor instance.
     // * Start a ThreadPool for executing futures.