relayrl_framework 0.5.0-beta.2

A system-oriented multi-agent reinforcement learning framework.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! RelayRL Prometheus Metrics Exporter
//!
//! This module provides Prometheus integration for the RelayRL metrics system,
//! enabling export of metrics to Prometheus monitoring infrastructure.

use prometheus::{Encoder, Registry, TextEncoder};

/// Initialize the Prometheus exporter with default settings
///
/// This function starts a web server on the configured host and port
/// to expose metrics in Prometheus format.
#[cfg(feature = "prometheus")]
pub fn create_prometheus_registry() -> Registry {
    Registry::new()
}

/// Return the current metrics in Prometheus text format
///
/// # Returns
///
/// * `String` - The metrics in Prometheus text format
#[cfg(feature = "prometheus")]
#[allow(unused)]
pub fn get_metrics_as_string(registry: &Registry) -> String {
    let mut buffer = Vec::new();
    let encoder = TextEncoder::new();
    let metric_families = registry.gather();
    encoder.encode(&metric_families, &mut buffer).unwrap();
    String::from_utf8(buffer).unwrap()
}