You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.5 KiB
Rust
77 lines
2.5 KiB
Rust
6 years ago
|
#[macro_use]
|
||
|
|
extern crate serde_derive;
|
||
|
|
extern crate toml;
|
||
|
|
extern crate hyper;
|
||
|
|
#[macro_use]
|
||
|
|
extern crate lazy_static;
|
||
|
|
#[macro_use]
|
||
|
|
extern crate prometheus;
|
||
|
|
extern crate futures;
|
||
|
|
|
||
|
|
use hyper::header::CONTENT_TYPE;
|
||
|
|
use hyper::{Body, Request, Response, Server};
|
||
|
|
use hyper::rt::Future;
|
||
|
|
use hyper::service::service_fn_ok;
|
||
|
|
|
||
|
|
use prometheus::{Counter, Encoder, Gauge, HistogramVec, TextEncoder};
|
||
|
|
|
||
|
|
lazy_static! {
|
||
|
|
static ref HTTP_COUNTER: Counter = register_counter!(opts!(
|
||
|
|
"example_http_requests_total",
|
||
|
|
"Total number of HTTP requests made.",
|
||
|
|
labels! {"handler" => "all",}
|
||
|
|
))
|
||
|
|
.unwrap();
|
||
|
|
static ref HTTP_BODY_GAUGE: Gauge = register_gauge!(opts!(
|
||
|
|
"example_http_response_size_bytes",
|
||
|
|
"The HTTP response sizes in bytes.",
|
||
|
|
labels! {"handler" => "all",}
|
||
|
|
))
|
||
|
|
.unwrap();
|
||
|
|
static ref HTTP_REQ_HISTOGRAM: HistogramVec = register_histogram_vec!(
|
||
|
|
"example_http_request_duration_seconds",
|
||
|
|
"The HTTP request latencies in seconds.",
|
||
|
|
&["handler"]
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
}
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
//TODO: Add service that does the pinging, based on oping and the example over at https://tokio.rs/docs/futures/spawning/
|
||
|
|
//TODO: Clean this shameful mess up!
|
||
|
|
//TODO: Ivestigate why all samples end up in all histogram buckets.
|
||
|
|
//TODO: Do config reading etc
|
||
|
|
let serve_metrics = || {
|
||
|
|
service_fn_ok(|_req| {
|
||
|
|
HTTP_COUNTER.inc();
|
||
|
|
let timer = HTTP_REQ_HISTOGRAM.with_label_values(&["all"]).start_timer();
|
||
|
|
|
||
|
|
let metric_families = prometheus::gather();
|
||
|
|
let mut buffer = vec![];
|
||
|
|
let encoder = TextEncoder::new();
|
||
|
|
encoder.encode(&metric_families, &mut buffer).unwrap();
|
||
|
|
HTTP_BODY_GAUGE.set(buffer.len() as f64);
|
||
|
|
let mut res = Response::new(Body::from(buffer));
|
||
|
|
res.headers_mut().insert(CONTENT_TYPE, encoder.format_type().parse().unwrap());
|
||
|
|
timer.observe_duration();
|
||
|
|
res
|
||
|
|
})
|
||
|
|
};
|
||
|
|
println!("listening addr 127.0.0.1:9898");
|
||
|
|
let server = Server::bind(&([127, 0, 0, 1], 9898).into()).serve(serve_metrics);
|
||
|
|
|
||
|
|
// Prepare some signal for when the server should start
|
||
|
|
// shutting down...
|
||
|
|
let (tx, rx) = futures::sync::oneshot::channel::<()>();
|
||
|
|
|
||
|
|
let graceful = server.with_graceful_shutdown(rx).map_err(|err| {
|
||
|
|
eprintln!("server error: {}", err)
|
||
|
|
});
|
||
|
|
|
||
|
|
// Spawn `server` onto an Executor...
|
||
|
|
hyper::rt::run(graceful);
|
||
|
|
|
||
|
|
// And later, trigger the signal by calling `tx.send(())`.
|
||
|
|
let _ = tx.send(());
|
||
|
|
}
|