RabbitMQ阻塞读取时数据时，关闭channel引起的问题和解决方案

项目场景：

　　最近在项目中使用了RabbitMq，其中有一个功能必须能随时切断RabbitMq的coumser。第一时间写出来的代码如下：

　　伪代码：

 1 while(flag){
 2     
 3      QueueingConsumer.Delivery delivery=consumer.nextDelivery();
 4      String message = new String(delivery.getBody());
 5      //doing someting strange
 6      //...... 
 7 
 8 }
 9 
10 
11 //另外一个项目开始关闭
12 
13 public void closeConsumer{
14       channel().close();
15       connection().close();
16 }
17 closeConsumer();

 通过关闭channel，消费者自然会关闭。然而，项目开始报错：

 

channel关闭抛出ShutdownSignalException，抛出异常就表示这种关闭方式是不合理的。有必要去探索一下是否有更优雅的链接关闭方式。

 

先看consumer的源码：

public Delivery nextDelivery()throws InterruptedException, ShutdownSignalException, ConsumerCancelledException{
     return handle(_queue.take());
 }

而这里_queue其实是一个 LinkedBlockingQueue，LinkedBlockingQueue是一个单向链表实现的阻塞队列。nextDelivery()方法使用LinkedBlockingQueue的take方法实现了阻塞。这个地方感觉不好操作。但是QueueingConsumer还有另外一个读取数据的方法，源码如下：

public Delivery nextDelivery(long timeout)throws InterruptedException, ShutdownSignalException, ConsumerCancelledException{
       return handle(_queue.poll(timeout, TimeUnit.MILLISECONDS));
}

这边设定了超时时间。虽然没想到优雅关闭消费者的方法，但是利用超时时间来修改一下读取数据的方法还是可以的。代码如下：

try{
    while(flag){
         QueueingConsumer.Delivery delivery = consumer.getDeliveryMessage(10000);
         if(consumer.getState()==Consumer.Status.Stopped.getValue()){
              break;
         }
         if(delivery==null){
                  continue;
         }
　　　　　 String message = new String(delivery.getBody());
   　　　　//....dosomething
    }
}finally {
    if(consumer!=null){
        consumer.closeConnection();
    }
}


public void closeConnection{  
　　channel().close();  
　　connection().close(); 
}

//另外一个线程关闭consumer
consumer.setStatus(Consumer.Status.Stopped);

 

　总结：这种关闭方式可以让Mq关闭连接时不抛出异常，比之前的方式好一点。但可能并不是最好的方法。如果有更好的方案，请留言告诉我，谢谢

 

posted on 2017-12-29 16:20  阿姆斯特朗回旋炮  阅读(9820)  评论(0)  编辑  收藏  举报

导航