pg_exporter 0.4.1

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
55
56
57
58
59
60

use crate::collectors::Collector;
use anyhow::Result;
use futures::future::BoxFuture;
use prometheus::Registry;
use sqlx::PgPool;
use std::sync::Arc;

mod connections;
use connections::ConnectionsCollector;

mod locks;
use locks::LocksCollector;

mod wait;
use wait::WaitEventsCollector;

/// Main Activity Collector (aggregates sub-collectors)
#[derive(Clone, Default)]
pub struct ActivityCollector {
    subs: Vec<Arc<dyn Collector + Send + Sync>>,
}

impl ActivityCollector {
    pub fn new() -> Self {
        Self {
            subs: vec![
                Arc::new(ConnectionsCollector::new()),
                Arc::new(LocksCollector::new()),
                Arc::new(WaitEventsCollector::new()),
            ],
        }
    }
}

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

    fn register_metrics(&self, registry: &Registry) -> Result<()> {
        for sub in &self.subs {
            sub.register_metrics(registry)?;
        }
        Ok(())
    }

    fn collect<'a>(&'a self, pool: &'a PgPool) -> BoxFuture<'a, Result<()>> {
        let subs = &self.subs;
        Box::pin(async move {
            for sub in subs {
                sub.collect(pool).await?;
            }
            Ok(())
        })
    }

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