feat: [#1403] expose udp-tracker-core metrics expose in REST API

Author: josecelano

packages/axum-rest-tracker-api-server/src/v1/context/stats/handlers.rs

@@ -70,11 +70,19 @@ pub async fn get_metrics_handler(
         Arc<InMemoryTorrentRepository>,
         Arc<RwLock<BanService>>,
         Arc<bittorrent_http_tracker_core::statistics::repository::Repository>,
+        Arc<bittorrent_udp_tracker_core::statistics::repository::Repository>,
         Arc<torrust_udp_tracker_server::statistics::repository::Repository>,
     )>,
     params: Query<QueryParams>,
 ) -> Response {
-    let metrics = get_labeled_metrics(state.0.clone(), state.1.clone(), state.2.clone(), state.3.clone()).await;
+    let metrics = get_labeled_metrics(
+        state.0.clone(),
+        state.1.clone(),
+        state.2.clone(),
+        state.3.clone(),
+        state.4.clone(),
+    )
+    .await;
 
     match params.0.format {
         Some(format) => match format {

packages/axum-rest-tracker-api-server/src/v1/context/stats/routes.rs

@@ -29,6 +29,7 @@ pub fn add(prefix: &str, router: Router, http_api_container: &Arc<TrackerHttpApi
                 http_api_container.tracker_core_container.in_memory_torrent_repository.clone(),
                 http_api_container.ban_service.clone(),
                 http_api_container.http_stats_repository.clone(),
+                http_api_container.udp_core_stats_repository.clone(),
                 http_api_container.udp_server_stats_repository.clone(),
             )),
         )

packages/rest-tracker-api-core/src/statistics/services.rs

@@ -84,22 +84,36 @@ pub struct TrackerLabeledMetrics {
 }
 
 /// It returns all the [`TrackerLabeledMetrics`]
+///
+/// # Panics
+///
+/// Will panic if the metrics cannot be merged. This could happen if the
+/// packages are producing duplicate metric names, for example.
 #[allow(deprecated)]
 pub async fn get_labeled_metrics(
     in_memory_torrent_repository: Arc<InMemoryTorrentRepository>,
     ban_service: Arc<RwLock<BanService>>,
     http_stats_repository: Arc<bittorrent_http_tracker_core::statistics::repository::Repository>,
+    udp_stats_repository: Arc<bittorrent_udp_tracker_core::statistics::repository::Repository>,
     udp_server_stats_repository: Arc<udp_server_statistics::repository::Repository>,
 ) -> TrackerLabeledMetrics {
     let _torrents_metrics = in_memory_torrent_repository.get_torrents_metrics();
     let _udp_banned_ips_total = ban_service.read().await.get_banned_ips_total();
     let _udp_server_stats = udp_server_stats_repository.get_stats().await;
 
     let http_stats = http_stats_repository.get_stats().await;
-
-    TrackerLabeledMetrics {
-        metrics: http_stats.metric_collection.clone(),
-    }
+    let udp_stats_repository = udp_stats_repository.get_stats().await;
+
+    // Merge the metrics from the HTTP and UDP metrics
+    let mut metrics = MetricCollection::default();
+    metrics
+        .merge(&http_stats.metric_collection)
+        .expect("msg: failed to merge HTTP core metrics");
+    metrics
+        .merge(&udp_stats_repository.metric_collection)
+        .expect("failed to merge UDP core metrics");
+
+    TrackerLabeledMetrics { metrics }
 }
 
 #[cfg(test)]