bees-prometheus-exporter 0.1.2

Prometheus exporter for the bees deduplication daemon
Documentation 
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use anyhow::Context;
use axum::{Router, response::Html, routing::get};
use log::{error, info};
use prometheus_client::registry::Registry;
use std::sync::{Arc, Mutex};
use tokio::net::TcpListener;

async fn handler(registry: Arc<Mutex<Registry>>) -> String {
    let mut buffer = String::new();
    {
        let registry = registry.lock().unwrap();
        match prometheus_client::encoding::text::encode(&mut buffer, &registry) {
            Ok(_) => {}
            Err(e) => {
                error!("Failed to encode metrics: {}", e);
                buffer = String::new();
            }
        }
    }
    buffer
}

async fn root_handler() -> Html<&'static str> {
    Html(
        r#"
<!DOCTYPE html>
<html>
<head>
    <title>Bees Prometheus Exporter</title>
</head>
<body>
    <h1>Bees Prometheus Exporter</h1>
    <p><a href="/metrics">Metrics</a></p>
</body>
</html>
    "#,
    )
}

fn init_app(registry: Arc<Mutex<Registry>>) -> Router {
    Router::new().route("/", get(root_handler)).route(
        "/metrics",
        get({
            let registry = registry.clone();
            move || async move { handler(registry.clone()).await }
        }),
    )
}

pub async fn start_server(
    registry: Arc<Mutex<Registry>>,
    address: &str,
    port: u16,
) -> anyhow::Result<()> {
    let app = init_app(registry);
    let listener = TcpListener::bind((address, port))
        .await
        .with_context(|| format!("Failed to bind to {}:{}", address, port))?;

    info!("Listening on http://{}:{}", address, port);
    axum::serve(listener, app).await.context("Server error")?;

    Ok(())
}