io.quarkiverse.openfga.runtime.health.OpenFGAHealthCheck Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-openfga-client Show documentation

There is a newer version: 2.1.0

Show newest version

package io.quarkiverse.openfga.runtime.health;

import javax.enterprise.context.ApplicationScoped;
import javax.inject.Inject;

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

import io.quarkiverse.openfga.client.api.API;
import io.smallrye.health.api.AsyncHealthCheck;
import io.smallrye.mutiny.Uni;

@ApplicationScoped
public class OpenFGAHealthCheck implements AsyncHealthCheck {

    private final API api;

    @Inject
    public OpenFGAHealthCheck(API api) {
        this.api = api;
    }

    @Override
    public Uni call() {

        final HealthCheckResponseBuilder builder = HealthCheckResponse.named("OpenFGA client connection health check");

        return api.health()
                .map(response -> {

                    if (response.getStatus().equalsIgnoreCase("SERVING")) {
                        return builder.up().build();
                    }

                    return builder.down().withData("reported-status", response.getStatus()).build();
                })
                .onFailure().recoverWithItem(x -> builder.withData("failure", x.getMessage()).build());
    }
}