refactor: replace RdKafkaStats with Guarded metrics (#18462)

Signed-off-by: tabVersion <[email protected]>

src/common/metrics/src/guarded_metrics.rs

@@ -83,6 +83,22 @@ macro_rules! register_guarded_int_gauge_vec_with_registry {
     }};
 }
 
+#[macro_export]
+macro_rules! register_guarded_uint_gauge_vec_with_registry {
+    ($NAME:expr, $HELP:expr, $LABELS_NAMES:expr, $REGISTRY:expr $(,)?) => {{
+        let inner = prometheus::core::GenericGaugeVec::<AtomicU64>::new(
+            prometheus::opts!($NAME, $HELP),
+            $LABELS_NAMES,
+        );
+        inner.and_then(|inner| {
+            let inner = $crate::__extract_gauge_builder(inner);
+            let label_guarded = $crate::LabelGuardedUintGaugeVec::new(inner, { $LABELS_NAMES });
+            let result = ($REGISTRY).register(Box::new(label_guarded.clone()));
+            result.map(move |()| label_guarded)
+        })
+    }};
+}
+
 #[macro_export]
 macro_rules! register_guarded_int_counter_vec_with_registry {
     ($NAME:expr, $HELP:expr, $LABELS_NAMES:expr, $REGISTRY:expr $(,)?) => {{
@@ -131,6 +147,8 @@ pub type LabelGuardedIntCounterVec<const N: usize> =
     LabelGuardedMetricVec<VecBuilderOfCounter<AtomicU64>, N>;
 pub type LabelGuardedIntGaugeVec<const N: usize> =
     LabelGuardedMetricVec<VecBuilderOfGauge<AtomicI64>, N>;
+pub type LabelGuardedUintGaugeVec<const N: usize> =
+    LabelGuardedMetricVec<VecBuilderOfGauge<AtomicU64>, N>;
 pub type LabelGuardedGaugeVec<const N: usize> =
     LabelGuardedMetricVec<VecBuilderOfGauge<AtomicF64>, N>;
 

src/common/src/lib.rs

@@ -76,7 +76,7 @@ pub mod memory;
 pub use risingwave_common_metrics::{
     monitor, register_guarded_gauge_vec_with_registry,
     register_guarded_histogram_vec_with_registry, register_guarded_int_counter_vec_with_registry,
-    register_guarded_int_gauge_vec_with_registry,
+    register_guarded_int_gauge_vec_with_registry, register_guarded_uint_gauge_vec_with_registry,
 };
 pub use {
     risingwave_common_metrics as metrics, risingwave_common_secret as secret,