RabbitMQ连接池、生产者、消费者实例

1、本文分享RabbitMQ的工具类，经过实际项目长期测试，在此分享给发家，各位大神有什么建议请指正 ！！！

2、下面是链接池主要代码：

import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

/**
 * 获取RabbitMq连接
 * @author skyfeng
 */
public class RabbitMqConnectFactory {
    static Logger log = LoggerFactory.getLogger(RabbitMqConnectFactory.class);
    /**
     * 缓存连接工厂,将建立的链接放入map缓存
     */
    private static Map<String, ConnectionFactory> connectionFactoryMap = new HashMap<String, ConnectionFactory>();
    /**
     * 根据rabbitMqName获取一个连接，使用完记得要自己关闭连接 conn.close()
     */
    public static Connection getConnection(String rabbitMqName) {
        if(StringUtils.isEmpty(rabbitMqName)){
            log.error("rabbitMqName不能为空!");
            throw new java.lang.NullPointerException("rabbitMqName为空");
        }
        if(connectionFactoryMap.get(rabbitMqName)==null){
            initConnectionFactory(rabbitMqName);
        }
        ConnectionFactory connectionFactory = connectionFactoryMap.get(rabbitMqName);
        if(connectionFactory==null){
            log.info("没有找到对应的rabbitmq,name={}",rabbitMqName);
        }
        try {
            return connectionFactory.newConnection();
        }catch (Exception e) {
            log.error("创建rabbitmq连接异常！",e);
            return null;
        }
    }
    /**
     * 初始化一个连接工厂
     * @param rabbitMqName
     */
    private static void initConnectionFactory(String rabbitMqName){
        
        try {
            ConnectionFactory factory = new ConnectionFactory();
            //新增代码，如果连接断开会自动重连
            //factory.setAutomaticRecoveryEnabled(true);
            factory.setHost("127.0.0.1");
            factory.setPort(5672);
            //factory.setVirtualHost(vhost);
            factory.setUsername("test");
            factory.setPassword("test");
            connectionFactoryMap.put(rabbitMqName, factory);
        } catch (Exception e) {
            e.printStackTrace();
        }finally{
        }
    }
    
}

 3、消费端的代码：

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.Consumer;

/**
 * RabbitMQq客户端代码
 * @author skyfeng
 *
 */
public class CustomerMqClient {

    final static Logger log = LoggerFactory.getLogger(CustomerMqClient.class);
    private final static String RABBITMQ_NAME = "mq_name";
    private final static String EXCHANGE_NAME = "Exchange_name";
    private final static String QUEUE_NAME = "queue_name";
    private static Channel channel = null;
    private static Connection connection = null;
    
    /**
     * 初始化客户端代码
     */
    public static void initClient() {
        //重新链接时判断之前的channel是否关闭，没有关闭先关闭
        if(null != channel  && channel.isOpen()){
            try {
                channel.close();
            } catch (Exception e) {
                log.error("mq name =[" +RABBITMQ_NAME+"] close old channel exception.e={}",e);
            }finally {
                channel = null;
            }
        }
        //重新链接时判断之前的connection是否关闭，没有关闭先关闭
        if (null != connection && connection.isOpen()) {
            try {
                connection.close();
            } catch (Exception e) {
                log.error("mq name =[" +RABBITMQ_NAME+"] close old connection exception.e={}",e);
            }finally{
                connection = null;
            }
        }
        //从链接池中获取链接
        connection = RabbitMqConnectFactory.getConnection(RABBITMQ_NAME);
        try {
            channel = connection.createChannel();
            channel.exchangeDeclare(EXCHANGE_NAME, "topic", true);
            channel.queueDeclare(QUEUE_NAME, false, false, false, null);
            channel.queueBind(QUEUE_NAME, EXCHANGE_NAME, "#");//#号接收所有的数据
            Consumer consumer = new CustomerMqConsumer(channel);//具体的业务逻辑在CustomerMqConsumer中
            channel.basicConsume(QUEUE_NAME, false, consumer);
        } catch (Exception e) {
            log.error("CustomerMqClient mq client connection fail .....{}", e);
            //发生异常时，重连
            reConnect();
        }
    }

    // 异常时，重连的方法
    public static void reConnect() {
        log.error("等待5s后重连");
        try {
            Thread.sleep(5000);
        } catch (InterruptedException e) {
        }
        initClient();
    }

}

4、生产端代码：

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.rabbitmq.client.AlreadyClosedException;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;

/**
 * 把数据发送到rabbitmq的exchange，
 */
public class SendToExchange {
    static Logger log = LoggerFactory.getLogger(SendToExchange.class);
    
    final static String TYPE = "topic";
    final static String CHARSET_UTF8 = "UTF-8";
    //MQ生产者exchange，把数据发给这个exchange
    final static String rabbitExchangeName = "ExchangeName";
    static boolean mqConnected = false;//mq当前处于连接状态
    
    static Channel channel=null;
    static{
        init();
    }
    public static void init(){
        log.info(" rabbit mq init begin...");
        try {
            //在mq连接中断后，发送程序判断已经断开，启动重连的时候会执行
            if(channel!=null){
                try {
                    channel.close();
                } catch (Exception e) {
                    log.error("关闭老channel 异常",e);
                }finally{
                    channel = null;
                }
            }
            Connection connection = RabbitMqConnectFactory.getConnection("connection");
            channel = connection.createChannel();
            /*
             *这里只定义exchange，因为每个业务模块都会从这里接入数据，所以不在这里定义队列
             *队列的定义在各个业务模块自己的消费端定义
             */
            channel.exchangeDeclare(rabbitExchangeName, TYPE, true, false, null);
            log.info(" rabbit mq init OK");
            mqConnected = true;
        } catch (Exception e) {
            log.error("rabbitmq初始化错误",e);
            mqConnected = false;
        }
    }
    /**
     * 往rabbitmq发数据
     * @param message
     */
    public static void sendToRabbitMq(String message,String routingKey){
        try {
            if(StringUtils.isEmpty(message)){
                log.debug("message is empty");
                return;
            }
            channel.basicPublish(rabbitExchangeName, routingKey, null, message.getBytes(CHARSET_UTF8));
        }catch(AlreadyClosedException ex){
            log.error("往rabbitmq发数据报错,可能连接已关闭,尝试重连,data:",message,ex);
            init();
        }catch (Exception e) {
            log.error("往rabbitmq发数据报错,data:",message,e);
        }
    }
}