me.legrange.services.rabbitmq.RabbitMqComponent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of service-rabbit-mq Show documentation
There is a newer version: 4.5.4
package me.legrange.services.rabbitmq;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.ShutdownSignalException;

import java.io.IOException;
import java.net.URISyntaxException;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import me.legrange.service.Component;
import me.legrange.service.ComponentException;
import me.legrange.service.Service;
import me.legrange.services.logging.WithLogging;

/**
 * @author gideon
 */
public final class RabbitMqComponent extends Component implements WithLogging {

    private Connection rabbitMq;
    private Channel channel;
    private RabbitMqConfig conf;

    public RabbitMqComponent(Service service) {
        super(service);
    }

    @Override
    public void start(RabbitMqConfig conf) throws ComponentException {
        this.conf = conf;
        if (!conf.isStartOnRequest()) {
            startRabbitMq();
        }
    }

    private void startRabbitMq() throws ComponentException {
        boolean connected = false;
        int retries = 0;
        while (!connected) {
            try {
                info("Connecting to RabbitMQ server %s", conf.getHostname());
                ConnectionFactory factory = new ConnectionFactory();
                factory.setUsername(conf.getUsername());
                factory.setPassword(conf.getPassword());
                factory.setVirtualHost(conf.getVirtualHost());
                if (conf.isSecure()) {
                    try {
                        factory.setUri("amqps://" + conf.getHostname());
                    } catch (URISyntaxException | NoSuchAlgorithmException | KeyManagementException e) {
                        error("Error configuring amqps URI: %s", e.getMessage());
                    }
                } else {
                    factory.setHost(conf.getHostname());
                }
                factory.setPort(conf.getPort());
                factory.setAutomaticRecoveryEnabled(true);
                rabbitMq = factory.newConnection();
                rabbitMq.addShutdownListener((ShutdownSignalException signal) -> {
                    warning("RabbitMQ connection to %s shut down. Reason: %s", conf.getHostname(), signal.getMessage());
                });
                channel = rabbitMq.createChannel();
                connected = true;
                info("Connected to RabbitMQ server %s", conf.getHostname());
            } catch (IOException ex) {
                error(ex, "Error connecting to RabbitMQ server %s: %s", conf.getHostname(), ex.getMessage());
            } catch (TimeoutException ex) {
                error(ex, "Timeout connecting to RabbitMQ server %s: %s", conf.getHostname(), ex.getMessage());
            }
            if (!connected) {
                warning("Could not connect to RabbitMQ server %s. Retrying in %d seconds", conf.getHostname(), conf.getRetryTime());
                try {
                    retries++;
                    TimeUnit.SECONDS.sleep(conf.getRetryTime());
                } catch (InterruptedException ex) {
                }
            }
        }
    }

    public Connection getConnection() throws ComponentException {
        if (rabbitMq == null) {
            startRabbitMq();
        }
        return rabbitMq;
    }

    public Channel getChannel() throws ComponentException {
        if (rabbitMq == null) {
            startRabbitMq();
        }
        return channel;
    }

    @Override
    public String getName() {
        return "rabbitMq";
    }


}