net.leanix.dropkit.amqp.AMQPHealthCheck Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of leanix-dropkit Show documentation

Base functionality for leanIX dropwizard-based services

There is a newer version: 2.0.10

package net.leanix.dropkit.amqp;

import com.codahale.metrics.health.HealthCheck;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;

public class AMQPHealthCheck extends HealthCheck {

    private static final Logger LOG = LoggerFactory.getLogger(AMQPHealthCheck.class);
    private final ConnectionFactory connectionFactory;

    public AMQPHealthCheck(ConnectionFactory connectionFactory) {
        super();
        this.connectionFactory = connectionFactory;
    }

    @Override
    protected Result check() throws Exception {
        Connection conn = null;
        try {
            conn = connectionFactory.newConnection();
            if (conn.isOpen()) {
                return Result.healthy();
            } else {
                return Result.unhealthy("AMQP connection is closed.");
            }
        } catch (IOException e) {
            return Result.unhealthy("Cannot open AMQP connection. " + e.getMessage());
        } finally {
            if (conn != null) {
                try {
                    LOG.info("Closing AMQP connection.");
                    conn.close();
                } catch (IOException e) {
                    conn = null;
                    LOG.info("Error closing AMQP connection.");
                }
            }
        }
    }
}