io.smallrye.health.SmallRyeHealthReporter Java Examples

public void afterDeploymentValidation(@Observes final AfterDeploymentValidation abd, BeanManager beanManager) {
    try {
        if (delegate != null) {
            Unmanaged<SmallRyeHealthReporter> unmanagedHealthCheck =
                new Unmanaged<>(beanManager, SmallRyeHealthReporter.class);
            reporterInstance = unmanagedHealthCheck.newInstance();
            reporter =  reporterInstance.produce().inject().postConstruct().get();
            monitor.registerHealthReporter(reporter);

            // THORN-2195: Use the correct TCCL when health checks are obtained
            // In WildFly, the TCCL should always be set to the top-level deployment CL during extension notification
            monitor.registerContextClassLoader(Thread.currentThread().getContextClassLoader());

            log.info(">> Added health reporter bean " + reporter);
            delegate = null;
        }

    } catch (Exception e) {
        throw new RuntimeException("Failed to register health reporter bean", e);
    }
}
 

@Override
public void handleRequest(HttpServerExchange exchange) throws Exception {
    if (NODE.equals(exchange.getRequestPath())) {
        nodeInfo(exchange);
        return;
    } else if (HEAP.equals(exchange.getRequestPath())) {
        heap(exchange);
        return;
    } else if (THREADS.equals(exchange.getRequestPath())) {
        threads(exchange);
        return;
    } else if (HEALTH.equals(exchange.getRequestPath())) {
        health(exchange, SmallRyeHealthReporter::getHealth, this::defaultUpHealthInfo);
        return;
    } else if (HEALTH_LIVE.equals(exchange.getRequestPath())) {
        health(exchange, SmallRyeHealthReporter::getLiveness, this::defaultUpHealthInfo);
        return;
    } else if (HEALTH_READY.equals(exchange.getRequestPath())) {
        health(exchange, SmallRyeHealthReporter::getReadiness, this::defaultDownHealthInfo);
        return;
    }

    next.handleRequest(exchange);
}
 

private void doHandle(RoutingContext ctx) {
    SmallRyeHealthReporter reporter = Arc.container().instance(SmallRyeHealthReporter.class).get();
    SmallRyeHealth health = getHealth(reporter, ctx);
    HttpServerResponse resp = ctx.response();
    if (health.isDown()) {
        resp.setStatusCode(503);
    }
    resp.headers().set(HttpHeaders.CONTENT_TYPE, "application/json; charset=UTF-8");
    try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
        reporter.reportHealth(outputStream, health);
        resp.end(Buffer.buffer(outputStream.toByteArray()));
    } catch (IOException e) {
        throw new UncheckedIOException(e);
    }
}
 

@BuildStep
@Record(ExecutionTime.STATIC_INIT)
@SuppressWarnings("unchecked")
void build(SmallRyeHealthRecorder recorder, RecorderContext recorderContext,
        BuildProducer<FeatureBuildItem> feature,
        BuildProducer<AdditionalBeanBuildItem> additionalBean,
        BuildProducer<BeanDefiningAnnotationBuildItem> beanDefiningAnnotation,
        BuildProducer<NotFoundPageDisplayableEndpointBuildItem> displayableEndpoints,
        LaunchModeBuildItem launchModeBuildItem) throws IOException {

    feature.produce(new FeatureBuildItem(Feature.SMALLRYE_HEALTH));

    // add health endpoints to not found page
    if (launchModeBuildItem.getLaunchMode().isDevOrTest()) {
        displayableEndpoints.produce(new NotFoundPageDisplayableEndpointBuildItem(health.rootPath));
        displayableEndpoints.produce(new NotFoundPageDisplayableEndpointBuildItem(health.rootPath + health.livenessPath));
        displayableEndpoints
                .produce(new NotFoundPageDisplayableEndpointBuildItem(health.rootPath + health.readinessPath));
        displayableEndpoints.produce(new NotFoundPageDisplayableEndpointBuildItem(health.rootPath + health.groupPath));
    }

    // Discover the beans annotated with @Health, @Liveness, @Readiness, @HealthGroup and @HealthGroups even if no scope is defined
    beanDefiningAnnotation.produce(new BeanDefiningAnnotationBuildItem(HEALTH));
    beanDefiningAnnotation.produce(new BeanDefiningAnnotationBuildItem(LIVENESS));
    beanDefiningAnnotation.produce(new BeanDefiningAnnotationBuildItem(READINESS));
    beanDefiningAnnotation.produce(new BeanDefiningAnnotationBuildItem(HEALTH_GROUP));
    beanDefiningAnnotation.produce(new BeanDefiningAnnotationBuildItem(HEALTH_GROUPS));

    // Add additional beans
    additionalBean.produce(new AdditionalBeanBuildItem(SmallRyeHealthReporter.class));

    // Make ArC discover @HealthGroup as a qualifier
    additionalBean.produce(new AdditionalBeanBuildItem(HealthGroup.class));

    // Discover and register the HealthCheckResponseProvider
    Set<String> providers = ServiceUtil.classNamesNamedIn(getClass().getClassLoader(),
            "META-INF/services/" + HealthCheckResponseProvider.class.getName());

    if (providers.isEmpty()) {
        throw new IllegalStateException("No HealthCheckResponseProvider implementation found.");
    } else if (providers.size() > 1) {
        throw new IllegalStateException(
                String.format("Multiple HealthCheckResponseProvider implementations found: %s", providers));
    }

    recorder.registerHealthCheckResponseProvider(
            (Class<? extends HealthCheckResponseProvider>) recorderContext.classProxy(providers.iterator().next()));
}
 

@Override
protected SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext ctx) {
    return reporter.getHealthGroups();
}
 

@Override
protected SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext ctx) {
    return reporter.getHealth();
}
 

@Override
protected SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext ctx) {
    return reporter.getLiveness();
}
 

@Override
protected SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext ctx) {
    String group = ctx.normalisedPath().substring(ctx.normalisedPath().lastIndexOf("/") + 1);
    return reporter.getHealthGroup(group);
}
 

@Override
protected SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext routingContext) {
    return reporter.getReadiness();
}
 

public void observeResources(@Observes ProcessAnnotatedType<SmallRyeHealthReporter> event) {
    delegate = event.getAnnotatedType();
}
 

protected abstract SmallRyeHealth getHealth(SmallRyeHealthReporter reporter, RoutingContext routingContext);