1 files changed, 120 insertions, 0 deletions
diff --git a/java/java/client/test/src/org/apache/qpid/pubsub1/TestSubscriber.java b/java/java/client/test/src/org/apache/qpid/pubsub1/TestSubscriber.java
new file mode 100644
index 0000000000..14cf206f50
--- /dev/null
+++ b/java/java/client/test/src/org/apache/qpid/pubsub1/TestSubscriber.java
@@ -0,0 +1,120 @@
+/*
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ *
+ */
+package org.apache.qpid.pubsub1;
+
+import org.apache.log4j.Logger;
+import org.apache.qpid.client.AMQConnection;
+import org.apache.qpid.client.AMQTopic;
+import org.apache.qpid.jms.Session;
+
+import javax.jms.MessageConsumer;
+import javax.jms.MessageListener;
+import javax.jms.Topic;
+import java.net.InetAddress;
+
+public class TestSubscriber
+{
+    private static final Logger _logger = Logger.getLogger(TestSubscriber.class);
+
+    private static class TestMessageListener implements MessageListener
+    {
+        private String _name;
+
+        private int _expectedMessageCount;
+
+        private int _messageCount;
+
+        private long _startTime = 0;
+
+        public TestMessageListener(String name, int expectedMessageCount)
+        {
+            _name = name;
+            _expectedMessageCount = expectedMessageCount;
+        }
+
+        public void onMessage(javax.jms.Message message)
+        {
+            if (_messageCount++ == 0)
+            {
+                _startTime = System.currentTimeMillis();
+            }
+            if (_logger.isInfoEnabled())
+            {
+                _logger.info(_name + " got message '" + message + "'");
+            }
+            if (_messageCount == _expectedMessageCount)
+            {
+                long totalTime = System.currentTimeMillis() - _startTime;
+                _logger.error(_name + ": Total time to receive " + _messageCount + " messages was " +
+                              totalTime + "ms. Rate is " + (_messageCount/(totalTime/1000.0)));
+            }
+            if (_messageCount > _expectedMessageCount)
+            {
+                _logger.error("Oops! More messages received than expected (" + _messageCount + ")");
+            }
+        }
+    }
+
+    public static void main(String[] args)
+    {
+        _logger.info("Starting...");
+
+        if (args.length != 7)
+        {
+            System.out.println("Usage: host port username password virtual-path expectedMessageCount selector");
+            System.exit(1);
+        }
+        try
+        {
+            InetAddress address = InetAddress.getLocalHost();
+            AMQConnection con1 = new AMQConnection(args[0], Integer.parseInt(args[1]), args[2], args[3],
+                                                  address.getHostName(), args[4]);
+            final org.apache.qpid.jms.Session session1 = (org.apache.qpid.jms.Session) con1.createSession(false, Session.AUTO_ACKNOWLEDGE);
+
+            AMQConnection con2 = new AMQConnection(args[0], Integer.parseInt(args[1]), args[2], args[3],
+                                                  address.getHostName(), args[4]);
+            final org.apache.qpid.jms.Session session2 = (org.apache.qpid.jms.Session) con2.createSession(false, Session.AUTO_ACKNOWLEDGE);
+            String selector = args[6];
+
+            final int expectedMessageCount = Integer.parseInt(args[5]);
+            _logger.info("Message selector is <" + selector + ">...");
+
+            Topic t = new AMQTopic("cbr");
+            MessageConsumer consumer1 = session1.createConsumer(t,
+                                                                100, false, false, selector);
+            MessageConsumer consumer2 = session2.createConsumer(t,
+                                                                100, false, false, selector);
+
+            consumer1.setMessageListener(new TestMessageListener("ML 1", expectedMessageCount));
+            consumer2.setMessageListener(new TestMessageListener("ML 2", expectedMessageCount));
+            con1.start();
+            con2.start();
+        }
+        catch (Throwable t)
+        {
+            System.err.println("Fatal error: " + t);
+            t.printStackTrace();
+        }
+
+        System.out.println("Waiting...");
+    }
+}
+