Observe tokio runtime metrics

Author: sandhose

crates/cli/src/main.rs

@@ -50,9 +50,17 @@ impl sentry::TransportFactory for SentryTransportFactory {
 }
 
 fn main() -> anyhow::Result<ExitCode> {
-    let runtime = tokio::runtime::Builder::new_multi_thread()
-        .enable_all()
-        .build()?;
+    let mut builder = tokio::runtime::Builder::new_multi_thread();
+    builder.enable_all();
+
+    #[cfg(tokio_unstable)]
+    builder
+        .enable_metrics_poll_time_histogram()
+        .metrics_poll_time_histogram_configuration(tokio::runtime::HistogramConfiguration::log(
+            tokio::runtime::LogHistogram::default(),
+        ));
+
+    let runtime = builder.build()?;
 
     runtime.block_on(async_main())
 }

crates/cli/src/telemetry.rs

@@ -4,6 +4,8 @@
 // SPDX-License-Identifier: AGPL-3.0-only
 // Please see LICENSE in the repository root for full details.
 
+mod tokio;
+
 use std::sync::{LazyLock, OnceLock};
 
 use anyhow::Context as _;
@@ -60,6 +62,9 @@ pub fn setup(config: &TelemetryConfig) -> anyhow::Result<()> {
     init_tracer(&config.tracing).context("Failed to configure traces exporter")?;
     init_meter(&config.metrics).context("Failed to configure metrics exporter")?;
 
+    let handle = ::tokio::runtime::Handle::current();
+    self::tokio::observe(handle.metrics());
+
     Ok(())
 }