io.quarkus.smallrye.health.runtime.QuarkusAsyncHealthCheckFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-smallrye-health Show documentation

Monitor service health

The newest version!

package io.quarkus.smallrye.health.runtime;

import java.util.concurrent.Callable;
import java.util.concurrent.Executor;

import jakarta.inject.Singleton;

import org.eclipse.microprofile.health.HealthCheck;
import org.eclipse.microprofile.health.HealthCheckResponse;

import io.smallrye.common.vertx.VertxContext;
import io.smallrye.health.AsyncHealthCheckFactory;
import io.smallrye.health.api.AsyncHealthCheck;
import io.smallrye.mutiny.Uni;
import io.smallrye.mutiny.vertx.MutinyHelper;
import io.vertx.core.Context;
import io.vertx.core.Vertx;

/**
 * Quarkus specific health check factory that runs blocking and reactive
 * health checks with different executors provided by {@link MutinyHelper}.
 */
@Singleton
public class QuarkusAsyncHealthCheckFactory extends AsyncHealthCheckFactory {

    private final Vertx vertx;

    public QuarkusAsyncHealthCheckFactory(Vertx vertx) {
        this.vertx = vertx;
    }

    @Override
    public Uni callSync(HealthCheck healthCheck) {
        Uni healthCheckResponseUni = super.callSync(healthCheck);
        return healthCheckResponseUni.runSubscriptionOn(new Executor() {
            @Override
            public void execute(Runnable command) {
                Context duplicatedContext = VertxContext.createNewDuplicatedContext(vertx.getOrCreateContext());
                duplicatedContext.executeBlocking(new Callable() {
                    @Override
                    public Void call() throws Exception {
                        command.run();
                        return null;
                    }
                }, false);
            }
        });
    }

    @Override
    public Uni callAsync(AsyncHealthCheck asyncHealthCheck) {
        Uni healthCheckResponseUni = super.callAsync(asyncHealthCheck);
        return healthCheckResponseUni.runSubscriptionOn(MutinyHelper.executor(vertx));
    }
}