This commit is contained in:
Justin Bertram 2022-08-24 17:17:09 -05:00
commit 451514f915
No known key found for this signature in database
GPG Key ID: F41830B875BB8633
2 changed files with 52 additions and 3 deletions

View File

@ -576,7 +576,7 @@ public class ScaleDownHandler {
MessageReference initialRef = null; MessageReference initialRef = null;
for (int i = 0; i < numberOfScans; i++) { for (int i = 0; i < numberOfScans; i++) {
logger.debug("iterating on queue " + queue + " while looking for reference " + reference); logger.debug("Iterating on queue " + queue + " while looking for reference " + reference);
memoryIterator = queue.iterator(); memoryIterator = queue.iterator();
while (memoryIterator.hasNext()) { while (memoryIterator.hasNext()) {
@ -610,8 +610,10 @@ public class ScaleDownHandler {
// if we reached two iterations without finding anything.. we just go away by cleaning everything up // if we reached two iterations without finding anything.. we just go away by cleaning everything up
lastRef = null; lastRef = null;
memoryIterator.close(); if (memoryIterator != null) {
memoryIterator = null; memoryIterator.close();
memoryIterator = null;
}
return false; return false;
} }

View File

@ -16,6 +16,7 @@
*/ */
package org.apache.activemq.artemis.tests.integration.server; package org.apache.activemq.artemis.tests.integration.server;
import javax.management.MBeanServer;
import java.lang.reflect.Field; import java.lang.reflect.Field;
import java.net.URI; import java.net.URI;
import java.util.Arrays; import java.util.Arrays;
@ -28,6 +29,7 @@ import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import org.apache.activemq.artemis.api.core.Message; import org.apache.activemq.artemis.api.core.Message;
import org.apache.activemq.artemis.api.core.QueueConfiguration;
import org.apache.activemq.artemis.api.core.RoutingType; import org.apache.activemq.artemis.api.core.RoutingType;
import org.apache.activemq.artemis.api.core.SimpleString; import org.apache.activemq.artemis.api.core.SimpleString;
import org.apache.activemq.artemis.api.core.client.ActiveMQClient; import org.apache.activemq.artemis.api.core.client.ActiveMQClient;
@ -36,6 +38,7 @@ import org.apache.activemq.artemis.api.core.client.ClientMessage;
import org.apache.activemq.artemis.api.core.client.ClientProducer; import org.apache.activemq.artemis.api.core.client.ClientProducer;
import org.apache.activemq.artemis.api.core.client.ClientSession; import org.apache.activemq.artemis.api.core.client.ClientSession;
import org.apache.activemq.artemis.api.core.client.ClientSessionFactory; import org.apache.activemq.artemis.api.core.client.ClientSessionFactory;
import org.apache.activemq.artemis.api.core.management.QueueControl;
import org.apache.activemq.artemis.core.client.impl.ClientSessionFactoryImpl; import org.apache.activemq.artemis.core.client.impl.ClientSessionFactoryImpl;
import org.apache.activemq.artemis.core.client.impl.ServerLocatorImpl; import org.apache.activemq.artemis.core.client.impl.ServerLocatorImpl;
import org.apache.activemq.artemis.core.client.impl.ServerLocatorInternal; import org.apache.activemq.artemis.core.client.impl.ServerLocatorInternal;
@ -48,6 +51,7 @@ import org.apache.activemq.artemis.core.server.cluster.impl.MessageLoadBalancing
import org.apache.activemq.artemis.core.server.impl.QueueImpl; import org.apache.activemq.artemis.core.server.impl.QueueImpl;
import org.apache.activemq.artemis.core.settings.impl.AddressSettings; import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
import org.apache.activemq.artemis.tests.integration.cluster.distribution.ClusterTestBase; import org.apache.activemq.artemis.tests.integration.cluster.distribution.ClusterTestBase;
import org.apache.activemq.artemis.tests.integration.management.ManagementControlHelper;
import org.apache.activemq.artemis.tests.util.ActiveMQTestBase; import org.apache.activemq.artemis.tests.util.ActiveMQTestBase;
import org.apache.activemq.artemis.tests.util.Wait; import org.apache.activemq.artemis.tests.util.Wait;
import org.apache.activemq.transport.amqp.client.AmqpClient; import org.apache.activemq.transport.amqp.client.AmqpClient;
@ -796,4 +800,47 @@ public class ScaleDownTest extends ClusterTestBase {
consumer.close(); consumer.close();
} }
@Test
public void testScaleDownPagedMessageWithMultipleAutoCreatedDLAResources() throws Exception {
final SimpleString dla = new SimpleString("DLA");
final SimpleString qName = new SimpleString("Q");
final SimpleString adName = new SimpleString("ADDR");
final String sampleText = "Put me on DLQ";
final int messageCount = 10;
final int numQueues = 2;
AddressSettings addressSettings = new AddressSettings().setDeadLetterAddress(dla).setAutoCreateDeadLetterResources(true).setDeadLetterQueuePrefix(dla);
AddressSettings dlaAddressSettings = new AddressSettings().setDeadLetterAddress(dla).setMaxSizeBytes(200L).setAutoCreateQueues(true);
servers[0].getAddressSettingsRepository().addMatch( "#", addressSettings);
servers[0].getAddressSettingsRepository().addMatch(dla.toString(), dlaAddressSettings);
ClientSessionFactory sf = sfs[0];
ClientSession session = addClientSession(sf.createSession(true, true));
MBeanServer mbeanServer = servers[0].getMBeanServer();
for (int i = 0; i < numQueues; i++) {
SimpleString curAddr = adName.concat(String.valueOf(i));
SimpleString curQ = qName.concat(String.valueOf(i));
SimpleString dlq = dla.concat(curAddr);
session.createQueue(new QueueConfiguration(curQ).setAddress(curAddr).setDurable(true));
ClientProducer producer = session.createProducer(curAddr);
for (int p = 0; p < messageCount; p++) {
producer.send(createTextMessage(session, sampleText));
}
QueueControl queueControl = ManagementControlHelper.createQueueControl(curAddr, curQ, mbeanServer);
Assert.assertEquals(messageCount, queueControl.sendMessagesToDeadLetterAddress(null));
Assert.assertEquals(0, queueControl.getMessageCount());
Wait.assertTrue(servers[0].locateQueue(dlq).getPagingStore()::isPaging);
}
servers[0].getActiveMQServerControl().scaleDown(servers[0].getConfiguration().getClusterConfigurations().iterator().next().getStaticConnectors().get(0));
Wait.assertEquals((messageCount * numQueues), () -> servers[1].getTotalMessageCount());
}
} }