io.camunda.zeebe.broker.health.BrokerReadyHealthIndicator Maven / Gradle / Ivy
/*
* Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
* one or more contributor license agreements. See the NOTICE file distributed
* with this work for additional information regarding copyright ownership.
* Licensed under the Zeebe Community License 1.1. You may not use this file
* except in compliance with the Zeebe Community License 1.1.
*/
package io.camunda.zeebe.broker.health;
import io.camunda.zeebe.broker.SpringBrokerBridge;
import io.camunda.zeebe.broker.system.monitoring.BrokerHealthCheckService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.actuate.availability.ReadinessStateHealthIndicator;
import org.springframework.boot.availability.ApplicationAvailability;
import org.springframework.boot.availability.AvailabilityState;
import org.springframework.boot.availability.ReadinessState;
import org.springframework.stereotype.Component;
@Component
public final class BrokerReadyHealthIndicator extends ReadinessStateHealthIndicator {
private final SpringBrokerBridge brokerBridge;
@Autowired
public BrokerReadyHealthIndicator(
final ApplicationAvailability availability, final SpringBrokerBridge brokerBridge) {
super(availability);
this.brokerBridge = brokerBridge;
}
@Override
protected AvailabilityState getState(final ApplicationAvailability applicationAvailability) {
final var isBrokerReady =
brokerBridge
.getBrokerHealthCheckService()
.map(BrokerHealthCheckService::isBrokerReady)
.orElse(false);
return isBrokerReady ? ReadinessState.ACCEPTING_TRAFFIC : ReadinessState.REFUSING_TRAFFIC;
}
}