feat: export runtime metric to promethues

src/common/runtime/Cargo.toml

@@ -17,5 +17,8 @@ snafu.workspace = true
 tokio-util.workspace = true
 tokio.workspace = true
 
+tokio-metrics = "0.3"
+tokio-metrics-collector = { version = "0.2" }
+
 [dev-dependencies]
 tokio-test = "0.4"

src/common/runtime/src/runtime.rs

@@ -95,7 +95,7 @@ pub struct Builder {
 impl Default for Builder {
     fn default() -> Self {
         Self {
-            runtime_name: format!("runtime-{}", RUNTIME_ID.fetch_add(1, Ordering::Relaxed)),
+            runtime_name: format!("runtime_{}", RUNTIME_ID.fetch_add(1, Ordering::Relaxed)),
             thread_name: "default-worker".to_string(),
             builder: RuntimeBuilder::new_multi_thread(),
         }
@@ -152,15 +152,18 @@ impl Builder {
             .build()
             .context(BuildRuntimeSnafu)?;
 
+        let name = self.runtime_name.clone();
         let handle = runtime.handle().clone();
         let (send_stop, recv_stop) = oneshot::channel();
         // Block the runtime to shutdown.
         let _ = thread::Builder::new()
             .name(format!("{}-blocker", self.thread_name))
             .spawn(move || runtime.block_on(recv_stop));
 
+        register_collector(name.clone(), &handle);
+
         Ok(Runtime {
-            name: self.runtime_name.clone(),
+            name,
             handle,
             _dropper: Arc::new(Dropper {
                 close: Some(send_stop),
@@ -169,6 +172,12 @@ impl Builder {
     }
 }
 
+pub fn register_collector(name: String, handle: &Handle) {
+    let monitor = tokio_metrics::RuntimeMonitor::new(handle);
+    let collector = tokio_metrics_collector::RuntimeCollector::new(monitor, name.clone());
+    let _ = prometheus::register(Box::new(collector));
+}
+
 fn on_thread_start(thread_name: String) -> impl Fn() + 'static {
     move || {
         METRIC_RUNTIME_THREADS_ALIVE
@@ -241,6 +250,9 @@ mod tests {
 
         assert!(metric_text.contains("runtime_threads_idle{thread_name=\"test_runtime_metric\"}"));
         assert!(metric_text.contains("runtime_threads_alive{thread_name=\"test_runtime_metric\"}"));
+        assert!(metric_text.contains("runtime_0_tokio_budget_forced_yield_count"));
+        assert!(metric_text.contains("runtime_0_tokio_injection_queue_depth"));
+        assert!(metric_text.contains("runtime_0_tokio_workers_count"));
     }
 
     #[test]