io.jooby.metrics.HealthCheckHandler Maven / Gradle / Ivy
/*
* Jooby https://jooby.io
* Apache License Version 2.0 https://jooby.io/LICENSE.txt
* Copyright 2014 Edgar Espina
*/
package io.jooby.metrics;
import java.util.SortedMap;
import java.util.TreeMap;
import com.codahale.metrics.health.HealthCheck.Result;
import com.codahale.metrics.health.HealthCheckRegistry;
import edu.umd.cs.findbugs.annotations.NonNull;
import io.jooby.Context;
import io.jooby.Route;
import io.jooby.StatusCode;
public class HealthCheckHandler implements Route.Handler {
@NonNull @Override
public Object apply(@NonNull Context ctx) {
HealthCheckRegistry registry = ctx.require(HealthCheckRegistry.class);
SortedMap checks =
ctx.query("name")
.toOptional()
.map(
name -> {
SortedMap map = new TreeMap<>();
map.put(name, registry.runHealthCheck(name));
return map;
})
.orElseGet(registry::runHealthChecks);
ctx.setResponseCode(
checks.isEmpty()
? StatusCode.NOT_IMPLEMENTED
: checks.values().stream()
.filter(it -> !it.isHealthy())
.findFirst()
.map(it -> StatusCode.SERVER_ERROR)
.orElse(StatusCode.OK));
ctx.setResponseHeader(MetricsModule.CACHE_HEADER_NAME, MetricsModule.CACHE_HEADER_VALUE);
return checks;
}
}