feat: support tokio-console (#40)

* feat: support `tokio-console`

* chore: rename feature to `tokio-console`

.cargo/config.toml

@@ -5,10 +5,17 @@ lld = true
 # [target.x86_64-unknown-linux-gnu]
 # rustflags = ["-C", "linker=clang", "-C", "link-arg=-fuse-ld=lld"]
 
-# Enable UUID Unstable Features
-# https://docs.rs/uuid/1.4.0/uuid/index.html#unstable-features
 [target.'cfg(all())']
-rustflags = ["--cfg", "uuid_unstable"]
+rustflags = [
+    # UUID Unstable for Uuid::now_v7()
+    # https://docs.rs/uuid/1.4.0/uuid/index.html#unstable-features
+    "--cfg",
+    "uuid_unstable",
+    # Tokio Unstable for `console` feature
+    # https://github.com/tokio-rs/console#instrumenting-your-program
+    "--cfg",
+    "tokio_unstable",
+]
 
 # x86_64-v3 (Intel Haswell / AMD Excavator and above)
 # https://en.wikipedia.org/wiki/X86-64#Microarchitecture_levels

Cargo.toml

@@ -39,6 +39,10 @@ opt-level = "z"
 panic = "abort"
 strip = "symbols"
 
+[features]
+default = ["tokio-console"]
+tokio-console = ["dep:console-subscriber", "tokio/tracing"]
+
 [workspace]
 members = [
     ".",
@@ -134,3 +138,6 @@ tokio-graceful-shutdown = { workspace = true }
 tracing = { workspace = true }
 tracing-error = { workspace = true }
 tracing-subscriber = { version = "0.3", features = ["env-filter", "json"] }
+
+# optional-dependencies
+console-subscriber = { version = "0.2", optional = true }

src/main.rs

@@ -19,10 +19,14 @@ use tracing_subscriber::prelude::*;
 async fn main() -> Result<(), AppError> {
     setup_panic!(metadata!().homepage("https://github.com/importantimport/hatsu/issues"));
 
-    tracing_subscriber::registry()
+    let registry = tracing_subscriber::registry()
         .with(tracing_subscriber::fmt::layer())
-        .with(tracing_error::ErrorLayer::default())
-        .init();
+        .with(tracing_error::ErrorLayer::default());
+
+    #[cfg(feature = "tokio-console")]
+    let registry = registry.with(console_subscriber::spawn());
+
+    registry.init();
 
     tracing::info!("loading environment variables");
     if dotenvy::dotenv().is_err() {