Merged 19219 into default

Author: 0x6e6562

test/src/com/rabbitmq/client/test/Bug19219Test.java

@@ -0,0 +1,179 @@
+//   The contents of this file are subject to the Mozilla Public License
+//   Version 1.1 (the "License"); you may not use this file except in
+//   compliance with the License. You may obtain a copy of the License at
+//   http://www.mozilla.org/MPL/
+//
+//   Software distributed under the License is distributed on an "AS IS"
+//   basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
+//   License for the specific language governing rights and limitations
+//   under the License.
+//
+//   The Original Code is RabbitMQ.
+//
+//   The Initial Developers of the Original Code are LShift Ltd.,
+//   Cohesive Financial Technologies LLC., and Rabbit Technologies Ltd.
+//
+//   Portions created by LShift Ltd., Cohesive Financial Technologies
+//   LLC., and Rabbit Technologies Ltd. are Copyright (C) 2007-2008
+//   LShift Ltd., Cohesive Financial Technologies LLC., and Rabbit
+//   Technologies Ltd.;
+//
+//   All Rights Reserved.
+//
+//   Contributor(s): ______________________________________.
+//
+package com.rabbitmq.client.test;
+
+import java.io.IOException;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.Semaphore;
+
+import junit.framework.TestCase;
+import junit.framework.TestSuite;
+
+import com.rabbitmq.client.AMQP;
+import com.rabbitmq.client.Channel;
+import com.rabbitmq.client.Connection;
+import com.rabbitmq.client.Consumer;
+import com.rabbitmq.client.DefaultConsumer;
+import com.rabbitmq.client.MessageProperties;
+import com.rabbitmq.client.ShutdownSignalException;
+
+import com.rabbitmq.client.test.functional.BrokerTestCase;
+
+/**
+ * Test for bug 19219 - timeouts due to task parallelism in channel
+ * closure code.
+ */
+public class Bug19219Test extends BrokerTestCase {
+
+    /*
+      These settings require careful tuning. Depending on them we get
+      one of three outcomes:
+      1) this program terminates normally
+      2) the broker runs out of memory and eventually dies, and the
+      this program barfs and/or dies
+      3) there are lots of timeout errors in the broker log
+
+      It's the last case we are interested in.
+
+      The settings below work on tanto against default.
+    */
+    private static final int Q_COUNT = 1500;
+    private static final int PUB_THREAD_COUNT = 100;
+    private static final int CLOSE_DELAY = 2000;
+
+    private static final Semaphore init = new Semaphore(0);
+    private static final CountDownLatch resume = new CountDownLatch(1);
+
+    @Override protected void setUp() throws Exception {
+        super.setUp();
+        openConnection();
+        openChannel();
+    }
+
+    @Override protected void tearDown() throws Exception {
+        closeChannel();
+        closeConnection();
+        super.tearDown();
+    }
+
+    public static TestSuite suite() {
+        TestSuite suite = new TestSuite("Bug19219");
+        suite.addTestSuite(Bug19219Test.class);
+        return suite;
+    }
+
+    private static void publish(final Channel ch,
+                                final int ticket)
+        throws IOException {
+        ch.basicPublish(ticket, "amq.fanout", "",
+                        MessageProperties.PERSISTENT_TEXT_PLAIN,
+                        new byte[0]);
+    }
+
+    public void testIt() throws IOException, InterruptedException {
+
+        final Consumer c = new DefaultConsumer(channel);
+
+        //1. create lots of auto-delete queues, bind them to the
+        //amq.fanout exchange, and set up a non-auto-ack consumer for
+        //each.
+        for (int i = 0; i < Q_COUNT; i++) {
+            String qName = channel.queueDeclare(ticket).getQueue();
+            channel.queueBind(ticket, qName, "amq.fanout", "");
+            channel.basicConsume(ticket, qName, false, c);
+        }
+
+        //2. send lots of messages in background, to keep the server,
+        //and especially the queues, busy
+        final Runnable r = new Runnable() {
+                public void run() {
+                    try {
+                        startPublisher();
+                    } catch (IOException e) {
+                    } catch (InterruptedException e) {
+                    }
+                }
+            };
+
+        for (int i = 0; i < PUB_THREAD_COUNT; i++) {
+            final Thread t = new Thread(r);
+            t.start();
+            //wait for thread to finish initialisation
+            init.acquire();
+        }
+
+        //tell all threads to resume
+        resume.countDown();
+
+        //wait for threads to get into full swing
+        Thread.sleep(CLOSE_DELAY);
+
+        //3. close channel. This will result in the server notifying
+        //all the queues in parallel, which in turn will requeue all
+        //the messages. The combined workload may result in some
+        //notifications timing out.
+        boolean success = false;
+        try {
+            channel.close(AMQP.REPLY_SUCCESS, "bye");
+            success = true;
+        } catch (ShutdownSignalException e) {
+        } finally {
+            //We deliberately do not perform a clean shutdown of all
+            //the connections. This test is pushing the server really
+            //hard, so we chose the quickest way to end things.
+            channel = null;
+            connection = null;
+
+            assertTrue(success);
+        }
+    }
+
+    private void startPublisher() throws IOException, InterruptedException {
+
+        final Connection conn = connectionFactory.newConnection("localhost");
+        final Channel pubCh = conn.createChannel();
+        final int pubTicket = pubCh.accessRequest("/data");
+
+        //This forces the initialisation of the guid generation, which
+        //is an interaction with the persister and not something we
+        //want to see delay things.
+        publish(pubCh, pubTicket);
+
+        //a synchronous request, to make sure the publish is done
+        pubCh.accessRequest("/data");
+
+        //signal the main thread
+        init.release();
+        //wait for main thread to let us resume
+        resume.await();
+
+        //publish lots of messages
+        while(true) {
+            publish(pubCh, pubTicket);
+        }
+
+    }
+
+}