MessageHandlerHealthIndicator

MessageHandlerHealthIndicator.java

public class MessageHandlerHealthIndicator extends AbstractHealthIndicator {

    private final AbstractBrokerMessageHandler abstractBrokerMessageHandler;

    public MessageHandlerHealthIndicator(AbstractBrokerMessageHandler messageHandler) {
        Assert.notNull(messageHandler, "MessageHandler can not be null");
        this.abstractBrokerMessageHandler = messageHandler;
    }

    @Override
    protected void doHealthCheck(Health.Builder builder) throws Exception {

        if (abstractBrokerMessageHandler.isRunning() && abstractBrokerMessageHandler.isBrokerAvailable()) {
            builder.up();
        } else {
            builder.down();
        }
    }
}

WebsocketBrokerHealthIndicatorConfiguration.java

@Configuration
protected static class WebsocketBrokerHealthIndicatorConfiguration {

    @Autowired(required = false)
    private AbstractBrokerMessageHandler stompBrokerRelayMessageHandler;

    @Bean
    @ConditionalOnMissingBean(name = "stompBrokerRelayHealthIndicator")
    public HealthIndicator stompBrokerRelayHealthIndicator() {

        return new MessageHandlerHealthIndicator(stompBrokerRelayMessageHandler);
    }
}