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
65
66
67
68
69
70
71
72
73
74

mod query;
mod results;
mod settings;
use std::convert::Infallible;
use std::sync::Arc;
use std::time;
use warp::Filter;

#[macro_use]
extern crate log;

async fn get_root(s: Arc<settings::Settings>) -> Result<impl warp::Reply, Infallible> {
    Ok(warp::reply::html(format!(
        "
<!DOCTYPE html>
<html>
    <body>
        <a href=\"config\">Configuration</a><br>
        <a href=\"{}\">Metrics</a>
    </body>
</html>
",
        s.metrics_path.clone().unwrap()
    )))
}

async fn get_metrics(s: Arc<settings::Settings>) -> Result<impl warp::Reply, Infallible> {
    let start = time::Instant::now();
    let result = query::process_targets(&s).await;
    let duration = start.elapsed();
    Ok(vec![
        result.to_string(),
        format!(
            "web_exporter_scrape_duration_milliseconds {}",
            duration.as_millis()
        ),
    ]
    .join("\n"))
}

async fn get_settings(s: Arc<settings::Settings>) -> Result<impl warp::Reply, Infallible> {
    match serde_yaml::to_string(&*s) {
        Ok(settings_str) => Ok(settings_str),
        Err(err) => Ok(format!("Invalid Settings: {}", err)),
    }
}

pub async fn run() {
    pretty_env_logger::init_custom_env("WEB_EXPORTER_LOG_LEVEL");
    match settings::Settings::new() {
        Ok(setting) => {
            let addr = setting.ip_address.unwrap();
            let port = setting.port.unwrap();
            let path = setting.metrics_path.clone().unwrap();
            let s = Arc::new(setting);

            info!("settings: {:?}", s.clone());
            let state = warp::any().map(move || s.clone());
            let root_route = warp::path::end().and(state.clone()).and_then(get_root);
            let settings_route = warp::path("config")
                .and(state.clone())
                .and_then(get_settings);
            let metrics_route = warp::path(path).and(state).and_then(get_metrics);
            let routes = metrics_route
                .or(settings_route)
                .or(root_route)
                .with(warp::compression::gzip());
            let server = warp::serve(routes).run((addr, port));
            server.await;
            info!("Initialization Complete!");
        }
        Err(err) => error!("Cannot parse configuration file: {:?}", err),
    }
}