FAELIX/rexaping
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: add healthcheck endpoint

Browse Source
This commit is contained in:
Jan Christian Grünhage
2021-04-19 10:01:35 +02:00
parent 44251fa2f7
commit d6900c7aa9
3 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/metrics.rs
View File

@ -49,7 +49,15 @@ lazy_static! {
.unwrap();
}
async fn serve_req(_req: Request<Body>) -> std::result::Result<Response<Body>, hyper::Error> {
async fn serve_req(req: Request<Body>) -> std::result::Result<Response<Body>, hyper::Error> {
match req.uri().path() {
"/metrics" => serve_metrics().await,
"/health" => serve_health_check().await,
_ => serve_not_found().await,
}
}
async fn serve_metrics() -> std::result::Result<Response<Body>, hyper::Error> {
let encoder = TextEncoder::new();
HTTP_COUNTER.inc();
@ -71,6 +79,20 @@ async fn serve_req(_req: Request<Body>) -> std::result::Result<Response<Body>, h
Ok(response)
}
async fn serve_health_check() -> std::result::Result<Response<Body>, hyper::Error> {
Ok(Response::builder()
.status(200)
.body(Body::from(vec![]))
.unwrap())
}
async fn serve_not_found() -> std::result::Result<Response<Body>, hyper::Error> {
Ok(Response::builder()
.status(404)
.body(Body::from(vec![]))
.unwrap())
}
pub(crate) async fn start_serving_metrics(config: Config) -> std::result::Result<(), hyper::Error> {
let serve_future = Server::bind(&config.listener).serve(make_service_fn(|_| async {
Ok::<_, hyper::Error>(service_fn(serve_req))