Created
February 2, 2019 13:09
-
-
Save breezewish/bb10bccd13a7fe332ef534ff0306ceb5 to your computer and use it in GitHub Desktop.
A getting started example for rust-prometheus + hyper
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #[macro_use] | |
| extern crate lazy_static; | |
| use hyper::rt::Future; | |
| use hyper::service::service_fn_ok; | |
| use hyper::{Body, Request, Response, Server}; | |
| use prometheus::*; | |
| use rand::prelude::*; | |
| lazy_static! { | |
| static ref REQUEST_DURATION: HistogramVec = register_histogram_vec!( | |
| "http_requests_duration", | |
| "Histogram of HTTP request duration in seconds", | |
| &["method"], | |
| exponential_buckets(0.005, 2.0, 20).unwrap() | |
| ) | |
| .unwrap(); | |
| } | |
| fn thread_simulate_requests() { | |
| let mut rng = rand::thread_rng(); | |
| loop { | |
| // Simulate duration 0s ~ 2s | |
| let duration = rng.gen_range(0f64, 2f64); | |
| // Simulate HTTP method | |
| let method = ["GET", "POST", "PUT", "DELETE"].choose(&mut rng).unwrap(); | |
| // Record metrics | |
| println!("{}\t{:.3}s", method, duration); | |
| REQUEST_DURATION | |
| .with_label_values(&[method]) | |
| .observe(duration); | |
| // One request per second | |
| std::thread::sleep(std::time::Duration::from_secs(1)); | |
| } | |
| } | |
| fn metric_service(_req: Request<Body>) -> Response<Body> { | |
| let encoder = TextEncoder::new(); | |
| let mut buffer = vec![]; | |
| let mf = prometheus::gather(); | |
| encoder.encode(&mf, &mut buffer).unwrap(); | |
| Response::builder() | |
| .header(hyper::header::CONTENT_TYPE, encoder.format_type()) | |
| .body(Body::from(buffer)) | |
| .unwrap() | |
| } | |
| fn main() { | |
| std::thread::spawn(thread_simulate_requests); | |
| let addr = ([127, 0, 0, 1], 12345).into(); | |
| let service = || service_fn_ok(metric_service); | |
| let server = Server::bind(&addr) | |
| .serve(service) | |
| .map_err(|e| panic!("{}", e)); | |
| hyper::rt::run(server); | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment