pg_exporter 0.3.2

PostgreSQL metric exporter for Prometheus
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

use crate::collectors::Collector;
use anyhow::Result;
use prometheus::{IntGauge, Registry};
use sqlx::PgPool;

#[derive(Clone)]
pub struct VacuumCollector {
    // Store metric handles for updating during collection
    connection_count: IntGauge,
}

impl VacuumCollector {
    pub fn new() -> Self {
        Self {
            connection_count: IntGauge::new(
                "pg_connections_total_2",
                "Total number of connections",
            )
            .expect("Failed to create connection_count metric"),
        }
    }
}

impl Default for VacuumCollector {
    fn default() -> Self {
        Self::new()
    }
}

impl Collector for VacuumCollector {
    fn name(&self) -> &'static str {
        "vacuum"
    }

    fn enabled_by_default(&self) -> bool {
        false
    }

    fn register_metrics(&self, registry: &Registry) -> Result<()> {
        registry.register(Box::new(self.connection_count.clone()))?;
        Ok(())
    }

    async fn collect(&self, pool: &PgPool) -> Result<()> {
        // Query the database and update metrics
        let row: (i64,) = sqlx::query_as("SELECT count(*) FROM pg_stat_activity")
            .fetch_one(pool)
            .await?;

        self.connection_count.set(row.0);

        Ok(())
    }
}