homestar-runtime 0.3.0

Homestar runtime implementation
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

//! Collect metrics and setup recorder and scrape endpoint.

use crate::settings;
use anyhow::Result;
use metrics_exporter_prometheus::PrometheusHandle;
#[cfg(feature = "monitoring")]
use tokio::runtime::Handle;

mod exporter;
#[cfg(feature = "monitoring")]
#[cfg_attr(docsrs, doc(cfg(feature = "monitoring")))]
mod node;

/// Start metrics collection and setup scrape endpoint.
/// Also, spawn a task to collect process metrics at a regular interval.
#[cfg(feature = "monitoring")]
#[cfg_attr(docsrs, doc(cfg(feature = "monitoring")))]
pub(crate) async fn start(
    monitor_settings: &settings::Monitoring,
    network_settings: &settings::Network,
) -> Result<PrometheusHandle> {
    let metrics_hdl = exporter::setup_metrics_recorder(network_settings)?;

    // Spawn tick-driven process collection task
    let handle = Handle::current();
    handle.spawn(node::collect_metrics(
        monitor_settings.process_collector_interval,
    ));

    Ok(metrics_hdl)
}

/// Start metrics collection and setup scrape endpoint.
#[cfg(not(feature = "monitoring"))]
pub(crate) async fn start(network_settings: &settings::Network) -> Result<PrometheusHandle> {
    let metrics_hdl = exporter::setup_metrics_recorder(network_settings)?;
    Ok(metrics_hdl)
}