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);
         }
     }
 }