org.reactivecommons.async.kafka.health.KafkaReactiveHealthIndicator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of async-kafka-starter Show documentation

Abstract your broker with semantic async messages

There is a newer version: 5.1.2

package org.reactivecommons.async.kafka.health;

import lombok.AllArgsConstructor;
import lombok.extern.log4j.Log4j2;
import org.apache.kafka.clients.admin.AdminClient;
import org.springframework.boot.actuate.health.AbstractReactiveHealthIndicator;
import org.springframework.boot.actuate.health.Health;
import reactor.core.publisher.Mono;

import static org.reactivecommons.async.starter.config.health.ReactiveCommonsHealthIndicator.DOMAIN;
import static org.reactivecommons.async.starter.config.health.ReactiveCommonsHealthIndicator.VERSION;

@Log4j2
@AllArgsConstructor
public class KafkaReactiveHealthIndicator extends AbstractReactiveHealthIndicator {
    private final String domain;
    private final AdminClient adminClient;

    @Override
    protected Mono doHealthCheck(Health.Builder builder) {
        builder.withDetail(DOMAIN, domain);
        return checkKafkaHealth()
                .map(clusterId -> builder.up().withDetail(VERSION, clusterId).build())
                .onErrorReturn(builder.down().build());
    }

    private Mono checkKafkaHealth() {
        return Mono.fromFuture(adminClient.describeCluster().clusterId()
                        .toCompletionStage()
                        .toCompletableFuture())
                .doOnError(e -> log.error("Error checking Kafka health in domain {}", domain, e));
    }

}