Improve telemetry

Author: LGY07

Cargo.lock

@@ -28,11 +28,13 @@ zip = "8.0"
 
 # Telemetry
 console-subscriber = { version = "0.5", optional = true }
+dhat = { version = "0.3", optional = true }
+tracing-chrome = { version = "0.7", optional = true }
 tracing-flame = { version = "0.2", optional = true }
 
 [features]
 default = []
-telemetry = ["console-subscriber", "tracing-flame"]
+telemetry = ["console-subscriber", "tracing-chrome", "tracing-flame", "dhat"]
 
 [profile.release]
 strip = true

Cargo.toml

@@ -1,6 +1,10 @@
 #![feature(async_fn_traits)]
 #![feature(unboxed_closures)]
 
+#[cfg(feature = "telemetry")]
+#[global_allocator]
+static ALLOC: dhat::Alloc = dhat::Alloc;
+
 mod command;
 mod core;
 mod plugin;
@@ -53,38 +57,45 @@ enum Commands {
     Info,
 }
 
-fn init_tracing() {
+#[tokio::main(flavor = "current_thread")]
+async fn main() -> anyhow::Result<()> {
+    // 初始化日志
     let fmt_layer = tracing_subscriber::fmt::Layer::default()
         .with_filter(tracing_subscriber::filter::LevelFilter::INFO);
 
     #[cfg(not(feature = "telemetry"))]
     let subscriber_builder = Registry::default().with(fmt_layer);
 
     #[cfg(feature = "telemetry")]
-    let subscriber_builder = {
+    let (subscriber_builder, _g1, _g2) = {
         // console layer
         let console_layer = console_subscriber::ConsoleLayer::builder()
             .with_default_env()
             .spawn();
 
+        // chrome layer
+        let (chrome_layer, chrome_guard) = tracing_chrome::ChromeLayerBuilder::new().build();
+
         // flame layer
-        let flame_layer = tracing_flame::FlameLayer::with_file("flamegraph.folded")
-            .unwrap()
-            .0;
-
-        Registry::default()
-            .with(fmt_layer)
-            .with(console_layer)
-            .with(flame_layer)
+        let (flame_layer, flame_guard) =
+            tracing_flame::FlameLayer::with_file("flamegraph.folded").unwrap();
+
+        (
+            Registry::default()
+                .with(fmt_layer)
+                .with(console_layer)
+                .with(chrome_layer)
+                .with(flame_layer),
+            chrome_guard,
+            flame_guard,
+        )
     };
 
     subscriber_builder.init();
-}
 
-#[tokio::main(flavor = "current_thread")]
-async fn main() -> anyhow::Result<()> {
-    // 初始化日志
-    init_tracing();
+    #[cfg(feature = "telemetry")]
+    dhat::Profiler::new_heap();
+
     // 参数解析
     let cli = Cli::parse();
     // 尝试从当前目录获取配置文件