Improve MCM so that it automatically copes with the consumer cancel n…

…otification.
rabbitmq · Oct 27, 2011 · 48bd24a · 48bd24a
1 parent 80aa453
commit 48bd24a
Showing 1 changed file with 22 additions and 13 deletions.
diff --git a/test/src/com/rabbitmq/examples/MulticastMain.java b/test/src/com/rabbitmq/examples/MulticastMain.java
@@ -43,6 +43,7 @@
 import com.rabbitmq.client.ConfirmListener;
 import com.rabbitmq.client.Connection;
 import com.rabbitmq.client.ConnectionFactory;
+import com.rabbitmq.client.ConsumerCancelledException;
 import com.rabbitmq.client.Envelope;
 import com.rabbitmq.client.MessageProperties;
 import com.rabbitmq.client.QueueingConsumer;
@@ -108,12 +109,10 @@ public static void main(String[] args) {
                                              flags.contains("persistent"),
                                              exclusive, autoDelete,
                                              null).getQueue();
-                QueueingConsumer consumer = new QueueingConsumer(channel);
                 if (prefetchCount > 0) channel.basicQos(prefetchCount);
-                channel.basicConsume(qName, autoAck, consumer);
                 channel.queueBind(qName, exchangeName, id);
                 Thread t =
-                    new Thread(new Consumer(consumer, id,
+                    new Thread(new Consumer(channel, id, qName,
                                             consumerTxSize, autoAck,
                                             stats, timeLimit));
                 consumerThreads[i] = t;
@@ -419,18 +418,21 @@ private byte[] createMessage(int sequenceNumber)
     public static class Consumer implements Runnable {
 
         private QueueingConsumer q;
+        private Channel          channel;
         private String           id;
+        private String           queueName;
         private int              txSize;
         private boolean          autoAck;
         private Stats            stats;
         private long             timeLimit;
 
-        public Consumer(QueueingConsumer q, String id,
-                        int txSize, boolean autoAck,
+        public Consumer(Channel channel, String id,
+                        String queueName, int txSize, boolean autoAck,
                         Stats stats, int timeLimit) {
 
-            this.q         = q;
+            this.channel   = channel;
             this.id        = id;
+            this.queueName = queueName;
             this.txSize    = txSize;
             this.autoAck   = autoAck;
             this.stats     = stats;
@@ -444,17 +446,24 @@ public void run() {
             startTime = now = System.currentTimeMillis();
             int totalMsgCount = 0;
 
-            Channel channel = q.getChannel();
-
             try {
+                q = new QueueingConsumer(channel);
+                channel.basicConsume(queueName, autoAck, q);
 
                 while (timeLimit == 0 || now < startTime + timeLimit) {
                     Delivery delivery;
-                    if (timeLimit == 0) {
-                        delivery = q.nextDelivery();
-                    } else {
-                        delivery = q.nextDelivery(startTime + timeLimit - now);
-                        if (delivery == null) break;
+                    try {
+                        if (timeLimit == 0) {
+                            delivery = q.nextDelivery();
+                        } else {
+                            delivery = q.nextDelivery(startTime + timeLimit - now);
+                            if (delivery == null) break;
+                        }
+                    } catch (ConsumerCancelledException e) {
+                        System.out.println("Consumer cancelled by broker. Re-consuming.");
+                        q = new QueueingConsumer(channel);
+                        channel.basicConsume(queueName, autoAck, q);
+                        continue;
                     }
 		    totalMsgCount++;