mirror of https://github.com/apache/activemq.git
commit test case with policy maxPageSize workaround in place that demonstrates it works as expected - https://issues.apache.org/activemq/browse/AMQ-2217
git-svn-id: https://svn.apache.org/repos/asf/activemq/trunk@995381 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Gary Tully
parent
adb76e5eb1
commit
9f0c86cc6d
|
|
@ -0,0 +1,336 @@
|
|||
/**
|
||||
* 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.activemq.usecases;
|
||||
|
||||
import javax.jms.Connection;
|
||||
import javax.jms.DeliveryMode;
|
||||
import javax.jms.Destination;
|
||||
import javax.jms.JMSException;
|
||||
import javax.jms.Message;
|
||||
import javax.jms.MessageConsumer;
|
||||
import javax.jms.MessageProducer;
|
||||
import javax.jms.Queue;
|
||||
import javax.jms.Session;
|
||||
import javax.jms.TextMessage;
|
||||
|
||||
import org.apache.activemq.ActiveMQConnectionFactory;
|
||||
import org.apache.activemq.JmsConnectionStartStopTest;
|
||||
import org.apache.activemq.broker.BrokerService;
|
||||
import org.apache.activemq.broker.region.policy.PolicyEntry;
|
||||
import org.apache.activemq.broker.region.policy.PolicyMap;
|
||||
|
||||
/**
|
||||
* Test case intended to demonstrate delivery interruption to queue consumers when
|
||||
* a JMS selector leaves some messages on the queue (due to use of a JMS Selector)
|
||||
*
|
||||
* testNonDiscriminatingConsumer() demonstrates proper functionality for consumers that don't use
|
||||
* a selector to qualify their input.
|
||||
*
|
||||
* testDiscriminatingConsumer() demonstrates the failure condition in which delivery to the consumer
|
||||
* eventually halts.
|
||||
*
|
||||
* The expected behavior is for the delivery to the client to be maintained regardless of the depth
|
||||
* of the queue, particularly when the messages in the queue do not meet the selector criteria of the
|
||||
* client.
|
||||
*
|
||||
* https://issues.apache.org/activemq/browse/AMQ-2217
|
||||
*
|
||||
*/
|
||||
public class DiscriminatingConsumerLoadTest extends TestSupport {
|
||||
|
||||
private static final org.apache.commons.logging.Log LOG = org.apache.commons.logging.LogFactory
|
||||
.getLog(DiscriminatingConsumerLoadTest.class);
|
||||
|
||||
private Connection producerConnection;
|
||||
private Connection consumerConnection;
|
||||
private int counterSent = 0;
|
||||
private int counterReceived = 0;
|
||||
|
||||
public static final String JMSTYPE_EATME = "DiscriminatingLoadClient.EatMe";
|
||||
public static final String JMSTYPE_IGNOREME = "DiscriminatingLoadClient.IgnoreMe";
|
||||
|
||||
private int testSize = 5000; // setting this to a small number will pass all tests
|
||||
|
||||
BrokerService broker;
|
||||
|
||||
protected void setUp() throws Exception {
|
||||
broker = new BrokerService();
|
||||
broker.setPersistent(false);
|
||||
|
||||
// workaround is to ensure sufficient dispatch buffer for the destination
|
||||
PolicyMap policyMap = new PolicyMap();
|
||||
PolicyEntry defaultPolicy = new PolicyEntry();
|
||||
defaultPolicy.setMaxPageSize(testSize);
|
||||
policyMap.setDefaultEntry(defaultPolicy);
|
||||
broker.setDestinationPolicy(policyMap);
|
||||
broker.start();
|
||||
|
||||
super.setUp();
|
||||
this.producerConnection = this.createConnection();
|
||||
this.consumerConnection = this.createConnection();
|
||||
}
|
||||
|
||||
/**
|
||||
* @see junit.framework.TestCase#tearDown()
|
||||
*/
|
||||
protected void tearDown() throws Exception {
|
||||
if (producerConnection != null) {
|
||||
producerConnection.close();
|
||||
producerConnection = null;
|
||||
}
|
||||
if (consumerConnection != null) {
|
||||
consumerConnection.close();
|
||||
consumerConnection = null;
|
||||
}
|
||||
super.tearDown();
|
||||
broker.stop();
|
||||
}
|
||||
|
||||
/**
|
||||
* Test to check if a single consumer with no JMS selector will receive all intended messages
|
||||
*
|
||||
* @throws java.lang.Exception
|
||||
*/
|
||||
public void testNonDiscriminatingConsumer() throws Exception {
|
||||
|
||||
consumerConnection = createConnection();
|
||||
consumerConnection.start();
|
||||
LOG.info("consumerConnection = " +consumerConnection);
|
||||
|
||||
try {Thread.sleep(1000); } catch (Exception e) {}
|
||||
|
||||
// here we pass in null for the JMS selector
|
||||
Consumer consumer = new Consumer(consumerConnection, null);
|
||||
Thread consumerThread = new Thread(consumer);
|
||||
|
||||
consumerThread.start();
|
||||
|
||||
producerConnection = createConnection();
|
||||
producerConnection.start();
|
||||
LOG.info("producerConnection = " +producerConnection);
|
||||
|
||||
try {Thread.sleep(3000); } catch (Exception e) {}
|
||||
|
||||
Producer producer = new Producer(producerConnection);
|
||||
Thread producerThread = new Thread(producer);
|
||||
producerThread.start();
|
||||
|
||||
// now that everything is running, let's wait for the consumer thread to finish ...
|
||||
consumerThread.join();
|
||||
producer.stop = true;
|
||||
|
||||
if (consumer.getCount() == testSize )
|
||||
LOG.info("test complete .... all messsages consumed!!");
|
||||
else
|
||||
LOG.info("test failed .... Sent " + (testSize / 1) +
|
||||
" messages intended to be consumed ( " + testSize + " total), but only consumed " + consumer.getCount());
|
||||
|
||||
|
||||
assertTrue("Sent " + testSize + " messages intended to be consumed, but only consumed " + consumer.getCount(),
|
||||
(consumer.getCount() == testSize ));
|
||||
assertFalse("Delivery of messages to consumer was halted during this test", consumer.deliveryHalted());
|
||||
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Test to check if a single consumer with a JMS selector will receive all intended messages
|
||||
*
|
||||
* @throws java.lang.Exception
|
||||
*/
|
||||
public void testDiscriminatingConsumer() throws Exception {
|
||||
|
||||
consumerConnection = createConnection();
|
||||
consumerConnection.start();
|
||||
LOG.info("consumerConnection = " +consumerConnection);
|
||||
|
||||
try {Thread.sleep(1000); } catch (Exception e) {}
|
||||
|
||||
// here we pass the JMS selector we intend to consume
|
||||
Consumer consumer = new Consumer(consumerConnection, JMSTYPE_EATME);
|
||||
Thread consumerThread = new Thread(consumer);
|
||||
|
||||
consumerThread.start();
|
||||
|
||||
producerConnection = createConnection();
|
||||
producerConnection.start();
|
||||
LOG.info("producerConnection = " +producerConnection);
|
||||
|
||||
try {Thread.sleep(3000); } catch (Exception e) {}
|
||||
|
||||
Producer producer = new Producer(producerConnection);
|
||||
Thread producerThread = new Thread(producer);
|
||||
producerThread.start();
|
||||
|
||||
// now that everything is running, let's wait for the consumer thread to finish ...
|
||||
consumerThread.join();
|
||||
producer.stop = true;
|
||||
|
||||
if (consumer.getCount() == (testSize / 2))
|
||||
{
|
||||
LOG.info("test complete .... all messsages consumed!!");
|
||||
}
|
||||
else
|
||||
{
|
||||
LOG.info("test failed .... Sent " + testSize + " original messages, only half of which (" + (testSize / 2) +
|
||||
") were intended to be consumed: consumer paused at: " + consumer.getCount());
|
||||
//System.out.println("test failed .... Sent " + testSize + " original messages, only half of which (" + (testSize / 2) +
|
||||
// ") were intended to be consumed: consumer paused at: " + consumer.getCount());
|
||||
|
||||
}
|
||||
|
||||
assertTrue("Sent " + testSize + " original messages, only half of which (" + (testSize / 2) +
|
||||
") were intended to be consumed: consumer paused at: " + consumer.getCount(),
|
||||
(consumer.getCount() == (testSize / 2)));
|
||||
assertTrue("Delivery of messages to consumer was halted during this test as it only wants half", consumer.deliveryHalted());
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper class that will publish 2 * testSize messages. The messages will be distributed evenly
|
||||
* between the following two JMS types:
|
||||
*
|
||||
* @see JMSTYPE_INTENDED_FOR_CONSUMPTION
|
||||
* @see JMSTYPE_NOT_INTENDED_FOR_CONSUMPTION
|
||||
*
|
||||
* @author jlyons
|
||||
*
|
||||
*/
|
||||
private class Producer extends Thread
|
||||
{
|
||||
private int counterSent = 0;
|
||||
private Connection connection = null;
|
||||
public boolean stop = false;
|
||||
|
||||
public Producer(Connection connection)
|
||||
{
|
||||
this.connection = connection;
|
||||
}
|
||||
|
||||
public void run() {
|
||||
try {
|
||||
final Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
final Queue queue = session.createQueue("test");
|
||||
|
||||
// wait for 10 seconds to allow consumer.receive to be run
|
||||
// first
|
||||
Thread.sleep(10000);
|
||||
MessageProducer producer = session.createProducer(queue);
|
||||
|
||||
while (!stop && (counterSent < testSize))
|
||||
{
|
||||
// first send a message intended to be consumed ....
|
||||
TextMessage message = session.createTextMessage("*** Ill ....... Ini ***"); // alma mater ...
|
||||
message.setJMSType(JMSTYPE_EATME);
|
||||
//LOG.info("sending .... JMSType = " + message.getJMSType());
|
||||
producer.send(message,DeliveryMode.NON_PERSISTENT,0,1800000);
|
||||
|
||||
counterSent++;
|
||||
|
||||
// now send a message intended to be consumed by some other consumer in the the future
|
||||
// ... we expect these messages to accrue in the queue
|
||||
message = session.createTextMessage("*** Ill ....... Ini ***"); // alma mater ...
|
||||
message.setJMSType(JMSTYPE_IGNOREME);
|
||||
//LOG.info("sending .... JMSType = " + message.getJMSType());
|
||||
producer.send(message,DeliveryMode.NON_PERSISTENT,0,1800000);
|
||||
|
||||
counterSent++;
|
||||
}
|
||||
|
||||
session.close();
|
||||
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
LOG.info("producer thread complete ... " + counterSent + " messages sent to the queue");
|
||||
}
|
||||
|
||||
public int getCount()
|
||||
{
|
||||
return this.counterSent;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper class that will consume messages from the queue based on the supplied JMS selector.
|
||||
* Thread will stop after the first receive(..) timeout, or once all expected messages have
|
||||
* been received (see testSize). If the thread stops due to a timeout, it is experiencing the
|
||||
* delivery pause that is symptomatic of a bug in the broker.
|
||||
*
|
||||
* @author jlyons
|
||||
*
|
||||
*/
|
||||
private class Consumer extends Thread
|
||||
{
|
||||
protected int counterReceived = 0;
|
||||
private Connection connection = null;
|
||||
private String jmsSelector = null;
|
||||
private boolean deliveryHalted = false;
|
||||
|
||||
public Consumer(Connection connection, String jmsSelector)
|
||||
{
|
||||
this.connection = connection;
|
||||
this.jmsSelector = jmsSelector;
|
||||
}
|
||||
|
||||
public void run() {
|
||||
boolean testComplete = false;
|
||||
try {
|
||||
Session session = consumerConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
final Queue queue = session.createQueue("test");
|
||||
MessageConsumer consumer = null;
|
||||
if (null != this.jmsSelector)
|
||||
{
|
||||
consumer = session.createConsumer(queue, "JMSType='" + this.jmsSelector + "'");
|
||||
}
|
||||
else
|
||||
{
|
||||
consumer = session.createConsumer(queue);
|
||||
}
|
||||
|
||||
while (!deliveryHalted && (counterReceived < testSize))
|
||||
{
|
||||
TextMessage result = (TextMessage) consumer.receive(30000);
|
||||
if (result != null) {
|
||||
counterReceived++;
|
||||
//System.out.println("consuming .... JMSType = " + result.getJMSType() + " received = " + counterReceived);
|
||||
LOG.info("consuming .... JMSType = " + result.getJMSType() + " received = " + counterReceived);
|
||||
} else
|
||||
{
|
||||
LOG.info("consuming .... timeout while waiting for a message ... broker must have stopped delivery ... received = " + counterReceived);
|
||||
deliveryHalted = true;
|
||||
}
|
||||
}
|
||||
session.close();
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public int getCount()
|
||||
{
|
||||
return this.counterReceived;
|
||||
}
|
||||
|
||||
public boolean deliveryHalted()
|
||||
{
|
||||
return this.deliveryHalted;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue