2020-04-06 10:56:39 +00:00
|
|
|
use crate::config::Config;
|
|
|
|
use hyper::{
|
|
|
|
header::CONTENT_TYPE,
|
|
|
|
service::{make_service_fn, service_fn},
|
|
|
|
Body, Request, Response, Server,
|
|
|
|
};
|
2019-02-06 18:29:27 +00:00
|
|
|
use lazy_static::lazy_static;
|
|
|
|
use prometheus::*;
|
2020-04-06 10:56:39 +00:00
|
|
|
use prometheus::{Counter, Gauge, HistogramVec, TextEncoder};
|
2019-02-06 18:29:27 +00:00
|
|
|
|
|
|
|
lazy_static! {
|
|
|
|
static ref HTTP_COUNTER: Counter = register_counter!(opts!(
|
|
|
|
"http_requests_total",
|
|
|
|
"Total number of HTTP requests made.",
|
|
|
|
labels! {"handler" => "all",}
|
|
|
|
))
|
|
|
|
.unwrap();
|
|
|
|
static ref HTTP_BODY_GAUGE: Gauge = register_gauge!(opts!(
|
|
|
|
"http_response_size_bytes",
|
|
|
|
"The HTTP response sizes in bytes.",
|
|
|
|
labels! {"handler" => "all",}
|
|
|
|
))
|
|
|
|
.unwrap();
|
|
|
|
static ref HTTP_REQ_HISTOGRAM: HistogramVec = register_histogram_vec!(
|
|
|
|
"http_request_duration_seconds",
|
|
|
|
"The HTTP request latencies in seconds.",
|
|
|
|
&["handler"]
|
|
|
|
)
|
|
|
|
.unwrap();
|
|
|
|
}
|
|
|
|
|
2020-04-06 10:56:39 +00:00
|
|
|
async fn serve_req(_req: Request<Body>) -> std::result::Result<Response<Body>, hyper::Error> {
|
|
|
|
let encoder = TextEncoder::new();
|
|
|
|
|
|
|
|
HTTP_COUNTER.inc();
|
|
|
|
let timer = HTTP_REQ_HISTOGRAM.with_label_values(&["all"]).start_timer();
|
|
|
|
|
|
|
|
let metric_families = prometheus::gather();
|
|
|
|
let mut buffer = vec![];
|
|
|
|
encoder.encode(&metric_families, &mut buffer).unwrap();
|
|
|
|
HTTP_BODY_GAUGE.set(buffer.len() as f64);
|
|
|
|
|
|
|
|
let response = Response::builder()
|
|
|
|
.status(200)
|
|
|
|
.header(CONTENT_TYPE, encoder.format_type())
|
|
|
|
.body(Body::from(buffer))
|
|
|
|
.unwrap();
|
|
|
|
|
|
|
|
timer.observe_duration();
|
|
|
|
|
|
|
|
Ok(response)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub(crate) async fn start_serving_metrics(config: Config) {
|
|
|
|
let serve_future = Server::bind(&config.listener).serve(make_service_fn(|_| async {
|
|
|
|
Ok::<_, hyper::Error>(service_fn(serve_req))
|
|
|
|
}));
|
|
|
|
|
|
|
|
if let Err(err) = serve_future.await {
|
|
|
|
eprintln!("server error: {}", err);
|
|
|
|
}
|
|
|
|
}
|