| /** |
| * 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.io.File; |
| import java.util.ArrayList; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| import javax.jms.BytesMessage; |
| import javax.jms.Connection; |
| import javax.jms.MessageConsumer; |
| import javax.jms.MessageProducer; |
| import javax.jms.Queue; |
| import javax.jms.Session; |
| import junit.framework.TestCase; |
| import org.apache.activemq.ActiveMQConnectionFactory; |
| import org.apache.activemq.broker.BrokerService; |
| import org.apache.activemq.broker.region.policy.FilePendingQueueMessageStoragePolicy; |
| import org.apache.activemq.broker.region.policy.PolicyEntry; |
| import org.apache.activemq.broker.region.policy.PolicyMap; |
| import org.apache.activemq.command.ActiveMQQueue; |
| import org.apache.activemq.store.kahadb.KahaDBPersistenceAdapter; |
| import org.apache.activemq.util.IOHelper; |
| |
| public class AMQ2616Test extends TestCase { |
| private static final int NUMBER = 2000; |
| private BrokerService brokerService; |
| private final ArrayList<Thread> threads = new ArrayList<Thread>(); |
| String ACTIVEMQ_BROKER_BIND = "tcp://0.0.0.0:61616"; |
| AtomicBoolean shutdown = new AtomicBoolean(); |
| |
| public void testQueueResourcesReleased() throws Exception{ |
| ActiveMQConnectionFactory fac = new ActiveMQConnectionFactory(ACTIVEMQ_BROKER_BIND); |
| Connection tempConnection = fac.createConnection(); |
| tempConnection.start(); |
| Session tempSession = tempConnection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| Queue tempQueue = tempSession.createTemporaryQueue(); |
| final MessageConsumer tempConsumer = tempSession.createConsumer(tempQueue); |
| |
| Connection testConnection = fac.createConnection(); |
| long startUsage = brokerService.getSystemUsage().getMemoryUsage().getUsage(); |
| Session testSession = testConnection.createSession(false, Session.AUTO_ACKNOWLEDGE); |
| MessageProducer testProducer = testSession.createProducer(tempQueue); |
| byte[] payload = new byte[1024*4]; |
| for (int i = 0; i < NUMBER; i++ ) { |
| BytesMessage msg = testSession.createBytesMessage(); |
| msg.writeBytes(payload); |
| testProducer.send(msg); |
| } |
| long endUsage = brokerService.getSystemUsage().getMemoryUsage().getUsage(); |
| assertFalse(startUsage==endUsage); |
| tempConnection.close(); |
| Thread.sleep(1000); |
| endUsage = brokerService.getSystemUsage().getMemoryUsage().getUsage(); |
| assertEquals(startUsage,endUsage); |
| } |
| |
| |
| @Override |
| protected void setUp() throws Exception { |
| // Start an embedded broker up. |
| brokerService = new BrokerService(); |
| |
| KahaDBPersistenceAdapter adaptor = new KahaDBPersistenceAdapter(); |
| adaptor.setEnableJournalDiskSyncs(false); |
| File file = new File("target/AMQ2616Test"); |
| IOHelper.mkdirs(file); |
| IOHelper.deleteChildren(file); |
| adaptor.setDirectory(file); |
| brokerService.setPersistenceAdapter(adaptor); |
| |
| PolicyMap policyMap = new PolicyMap(); |
| PolicyEntry pe = new PolicyEntry(); |
| pe.setMemoryLimit(10 * 1024 * 1024); |
| pe.setOptimizedDispatch(true); |
| pe.setProducerFlowControl(false); |
| pe.setExpireMessagesPeriod(1000); |
| pe.setPendingQueuePolicy(new FilePendingQueueMessageStoragePolicy()); |
| policyMap.put(new ActiveMQQueue(">"), pe); |
| brokerService.setDestinationPolicy(policyMap); |
| brokerService.getSystemUsage().getMemoryUsage().setLimit(20 * 1024 * 1024); |
| brokerService.getSystemUsage().getTempUsage().setLimit(200 * 1024 * 1024); |
| brokerService.addConnector(ACTIVEMQ_BROKER_BIND); |
| brokerService.start(); |
| new ActiveMQQueue(getName()); |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| // Stop any running threads. |
| shutdown.set(true); |
| for (Thread t : threads) { |
| t.interrupt(); |
| t.join(); |
| } |
| brokerService.stop(); |
| } |
| |
| } |