| /** |
| * 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.bugs; |
| |
| import java.util.concurrent.atomic.AtomicInteger; |
| |
| import javax.jms.ConnectionFactory; |
| import javax.jms.DeliveryMode; |
| import javax.jms.JMSException; |
| import javax.jms.MessageProducer; |
| import javax.jms.ResourceAllocationException; |
| import javax.jms.Session; |
| import javax.jms.TextMessage; |
| |
| import org.apache.activemq.ActiveMQConnection; |
| import org.apache.activemq.ActiveMQConnectionFactory; |
| import org.apache.activemq.EmbeddedBrokerTestSupport; |
| import org.apache.activemq.command.ActiveMQDestination; |
| import org.apache.activemq.transport.RequestTimedOutIOException; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| public class JmsTimeoutTest extends EmbeddedBrokerTestSupport { |
| |
| static final Logger LOG = LoggerFactory.getLogger(JmsTimeoutTest.class); |
| |
| protected final int messageSize=1024*64; |
| protected final int messageCount=10000; |
| protected final AtomicInteger exceptionCount = new AtomicInteger(0); |
| |
| /** |
| * Test the case where the broker is blocked due to a memory limit |
| * and a producer timeout is set on the connection. |
| * @throws Exception |
| */ |
| public void testBlockedProducerConnectionTimeout() throws Exception { |
| final ActiveMQConnection cx = (ActiveMQConnection)createConnection(); |
| final ActiveMQDestination queue = createDestination("testqueue"); |
| |
| // we should not take longer than 10 seconds to return from send |
| cx.setSendTimeout(10000); |
| |
| Runnable r = new Runnable() { |
| public void run() { |
| try { |
| LOG.info("Sender thread starting"); |
| Session session = cx.createSession(false, 1); |
| MessageProducer producer = session.createProducer(queue); |
| producer.setDeliveryMode(DeliveryMode.PERSISTENT); |
| |
| TextMessage message = session.createTextMessage(createMessageText()); |
| for(int count=0; count<messageCount; count++){ |
| producer.send(message); |
| } |
| LOG.info("Done sending.."); |
| } catch (JMSException e) { |
| if (e.getCause() instanceof RequestTimedOutIOException) { |
| exceptionCount.incrementAndGet(); |
| } else { |
| e.printStackTrace(); |
| } |
| return; |
| } |
| |
| } |
| }; |
| cx.start(); |
| Thread producerThread = new Thread(r); |
| producerThread.start(); |
| producerThread.join(30000); |
| cx.close(); |
| // We should have a few timeout exceptions as memory store will fill up |
| assertTrue("No exception from the broker", exceptionCount.get() > 0); |
| } |
| |
| /** |
| * Test the case where the broker is blocked due to a memory limit |
| * with a fail timeout |
| * @throws Exception |
| */ |
| public void testBlockedProducerUsageSendFailTimeout() throws Exception { |
| final ActiveMQConnection cx = (ActiveMQConnection)createConnection(); |
| final ActiveMQDestination queue = createDestination("testqueue"); |
| |
| broker.getSystemUsage().setSendFailIfNoSpaceAfterTimeout(5000); |
| Runnable r = new Runnable() { |
| public void run() { |
| try { |
| LOG.info("Sender thread starting"); |
| Session session = cx.createSession(false, 1); |
| MessageProducer producer = session.createProducer(queue); |
| producer.setDeliveryMode(DeliveryMode.PERSISTENT); |
| |
| TextMessage message = session.createTextMessage(createMessageText()); |
| for(int count=0; count<messageCount; count++){ |
| producer.send(message); |
| } |
| LOG.info("Done sending.."); |
| } catch (JMSException e) { |
| if (e instanceof ResourceAllocationException || e.getCause() instanceof RequestTimedOutIOException) { |
| exceptionCount.incrementAndGet(); |
| } else { |
| e.printStackTrace(); |
| } |
| return; |
| } |
| } |
| }; |
| cx.start(); |
| Thread producerThread = new Thread(r); |
| producerThread.start(); |
| producerThread.join(30000); |
| cx.close(); |
| // We should have a few timeout exceptions as memory store will fill up |
| assertTrue("No exception from the broker", exceptionCount.get() > 0); |
| } |
| |
| protected void setUp() throws Exception { |
| exceptionCount.set(0); |
| bindAddress = "tcp://localhost:0"; |
| broker = createBroker(); |
| broker.setDeleteAllMessagesOnStartup(true); |
| broker.getSystemUsage().getMemoryUsage().setLimit(5*1024*1024); |
| |
| super.setUp(); |
| } |
| |
| @Override |
| protected ConnectionFactory createConnectionFactory() throws Exception { |
| return new ActiveMQConnectionFactory( |
| broker.getTransportConnectors().get(0).getPublishableConnectString()); |
| } |
| |
| protected String createMessageText() { |
| StringBuffer buffer = new StringBuffer(); |
| buffer.append("<filler>"); |
| for (int i = buffer.length(); i < messageSize; i++) { |
| buffer.append('X'); |
| } |
| buffer.append("</filler>"); |
| return buffer.toString(); |
| } |
| |
| } |