This closes #777

This commit is contained in:
Martyn Taylor 2016-09-15 16:02:49 +01:00
commit 4e444d53f9
9 changed files with 292 additions and 62 deletions

View File

@ -28,14 +28,18 @@ import org.jboss.logging.Logger;
/** This is for components with a scheduled at a fixed rate. */ /** This is for components with a scheduled at a fixed rate. */
public abstract class ActiveMQScheduledComponent implements ActiveMQComponent, Runnable { public abstract class ActiveMQScheduledComponent implements ActiveMQComponent, Runnable {
private static final Logger logger = Logger.getLogger(ActiveMQScheduledComponent.class); private static final Logger logger = Logger.getLogger(ActiveMQScheduledComponent.class);
private final ScheduledExecutorService scheduledExecutorService; private final ScheduledExecutorService scheduledExecutorService;
private long period; private long period;
private long millisecondsPeriod;
private TimeUnit timeUnit; private TimeUnit timeUnit;
private final Executor executor; private final Executor executor;
private ScheduledFuture future; private ScheduledFuture future;
private final boolean onDemand; private final boolean onDemand;
long lastTime = 0;
private final AtomicInteger delayed = new AtomicInteger(0); private final AtomicInteger delayed = new AtomicInteger(0);
public ActiveMQScheduledComponent(ScheduledExecutorService scheduledExecutorService, public ActiveMQScheduledComponent(ScheduledExecutorService scheduledExecutorService,
@ -58,6 +62,8 @@ public abstract class ActiveMQScheduledComponent implements ActiveMQComponent, R
if (future != null) { if (future != null) {
return; return;
} }
this.millisecondsPeriod = timeUnit.convert(period, TimeUnit.MILLISECONDS);
if (onDemand) { if (onDemand) {
return; return;
} }
@ -113,11 +119,6 @@ public abstract class ActiveMQScheduledComponent implements ActiveMQComponent, R
} }
@Override
public void run() {
delayed.decrementAndGet();
}
@Override @Override
public synchronized boolean isStarted() { public synchronized boolean isStarted() {
return future != null; return future != null;
@ -132,10 +133,30 @@ public abstract class ActiveMQScheduledComponent implements ActiveMQComponent, R
} }
} }
final Runnable runForExecutor = new Runnable() {
@Override
public void run() {
if (onDemand && delayed.get() > 0) {
delayed.decrementAndGet();
}
if (!onDemand && lastTime > 0) {
if (System.currentTimeMillis() - lastTime < millisecondsPeriod) {
logger.trace("Execution ignored due to too many simultaneous executions, probably a previous delayed execution");
return;
}
}
lastTime = System.currentTimeMillis();
ActiveMQScheduledComponent.this.run();
}
};
final Runnable runForScheduler = new Runnable() { final Runnable runForScheduler = new Runnable() {
@Override @Override
public void run() { public void run() {
executor.execute(ActiveMQScheduledComponent.this); executor.execute(runForExecutor);
} }
}; };

View File

@ -0,0 +1,80 @@
/**
* 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.artemis.utils;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import org.apache.activemq.artemis.core.server.ActiveMQScheduledComponent;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
public class ActiveMQScheduledComponentTest {
@Rule
public ThreadLeakCheckRule rule = new ThreadLeakCheckRule();
ScheduledExecutorService scheduledExecutorService;
ExecutorService executorService;
@Before
public void before() {
scheduledExecutorService = new ScheduledThreadPoolExecutor(5);
executorService = Executors.newSingleThreadExecutor();
}
@After
public void after() {
executorService.shutdown();
scheduledExecutorService.shutdown();
}
@Test
public void testAccumulation() throws Exception {
final AtomicInteger count = new AtomicInteger(0);
final ActiveMQScheduledComponent local = new ActiveMQScheduledComponent(scheduledExecutorService, executorService, 100, TimeUnit.MILLISECONDS, false) {
public void run() {
if (count.get() == 0) {
try {
Thread.sleep(800);
}
catch (Exception e) {
}
}
count.incrementAndGet();
}
};
local.start();
Thread.sleep(1000);
local.stop();
Assert.assertTrue("just because one took a lot of time, it doesn't mean we can accumulate many, we got " + count + " executions", count.get() < 5);
}
}

View File

@ -38,7 +38,6 @@ public class JDBCJournalSync extends ActiveMQScheduledComponent {
@Override @Override
public void run() { public void run() {
super.run();
if (journal.isStarted()) { if (journal.isStarted()) {
journal.sync(); journal.sync();
} }

View File

@ -78,7 +78,6 @@ final class PageSyncTimer extends ActiveMQScheduledComponent {
@Override @Override
public void run() { public void run() {
super.run();
tick(); tick();
} }

View File

@ -27,13 +27,13 @@ import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap; import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.CountDownLatch; import java.util.concurrent.Executor;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean; import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.activemq.artemis.api.core.ActiveMQAddressFullException; import org.apache.activemq.artemis.api.core.ActiveMQAddressFullException;
import org.apache.activemq.artemis.api.core.ActiveMQDuplicateIdException; import org.apache.activemq.artemis.api.core.ActiveMQDuplicateIdException;
import org.apache.activemq.artemis.api.core.ActiveMQInterruptedException;
import org.apache.activemq.artemis.api.core.ActiveMQNonExistentQueueException; import org.apache.activemq.artemis.api.core.ActiveMQNonExistentQueueException;
import org.apache.activemq.artemis.api.core.Message; import org.apache.activemq.artemis.api.core.Message;
import org.apache.activemq.artemis.api.core.Pair; import org.apache.activemq.artemis.api.core.Pair;
@ -57,6 +57,7 @@ import org.apache.activemq.artemis.core.postoffice.PostOffice;
import org.apache.activemq.artemis.core.postoffice.QueueInfo; import org.apache.activemq.artemis.core.postoffice.QueueInfo;
import org.apache.activemq.artemis.core.postoffice.RoutingStatus; import org.apache.activemq.artemis.core.postoffice.RoutingStatus;
import org.apache.activemq.artemis.core.server.ActiveMQMessageBundle; import org.apache.activemq.artemis.core.server.ActiveMQMessageBundle;
import org.apache.activemq.artemis.core.server.ActiveMQScheduledComponent;
import org.apache.activemq.artemis.core.server.ActiveMQServer; import org.apache.activemq.artemis.core.server.ActiveMQServer;
import org.apache.activemq.artemis.core.server.ActiveMQServerLogger; import org.apache.activemq.artemis.core.server.ActiveMQServerLogger;
import org.apache.activemq.artemis.core.server.LargeServerMessage; import org.apache.activemq.artemis.core.server.LargeServerMessage;
@ -112,8 +113,6 @@ public class PostOfficeImpl implements PostOffice, NotificationListener, Binding
private Reaper reaperRunnable; private Reaper reaperRunnable;
private volatile Thread reaperThread;
private final long reaperPeriod; private final long reaperPeriod;
private final int reaperPriority; private final int reaperPriority;
@ -198,12 +197,6 @@ public class PostOfficeImpl implements PostOffice, NotificationListener, Binding
if (reaperRunnable != null) if (reaperRunnable != null)
reaperRunnable.stop(); reaperRunnable.stop();
if (reaperThread != null) {
reaperThread.join();
reaperThread = null;
}
addressManager.clear(); addressManager.clear();
queueInfos.clear(); queueInfos.clear();
@ -1244,12 +1237,9 @@ public class PostOfficeImpl implements PostOffice, NotificationListener, Binding
if (reaperPeriod > 0) { if (reaperPeriod > 0) {
if (reaperRunnable != null) if (reaperRunnable != null)
reaperRunnable.stop(); reaperRunnable.stop();
reaperRunnable = new Reaper(); reaperRunnable = new Reaper(server.getScheduledPool(), server.getExecutorFactory().getExecutor(), reaperPeriod, TimeUnit.MILLISECONDS, false);
reaperThread = new Thread(reaperRunnable, "activemq-expiry-reaper-thread");
reaperThread.setPriority(reaperPriority); reaperRunnable.start();
reaperThread.start();
} }
} }
@ -1268,29 +1258,20 @@ public class PostOfficeImpl implements PostOffice, NotificationListener, Binding
return message; return message;
} }
private final class Reaper implements Runnable { private final class Reaper extends ActiveMQScheduledComponent {
private final CountDownLatch latch = new CountDownLatch(1); Reaper(ScheduledExecutorService scheduledExecutorService,
Executor executor,
public void stop() { long checkPeriod,
latch.countDown(); TimeUnit timeUnit,
boolean onDemand) {
super(scheduledExecutorService, executor, checkPeriod, timeUnit, onDemand);
} }
@Override @Override
public void run() { public void run() {
// The reaper thread should be finished case the PostOffice is gone // The reaper thread should be finished case the PostOffice is gone
// This is to avoid leaks on PostOffice between stops and starts // This is to avoid leaks on PostOffice between stops and starts
while (isStarted()) {
try {
if (latch.await(reaperPeriod, TimeUnit.MILLISECONDS))
return;
}
catch (InterruptedException e1) {
throw new ActiveMQInterruptedException(e1);
}
if (!isStarted())
return;
Map<SimpleString, Binding> nameMap = addressManager.getBindings(); Map<SimpleString, Binding> nameMap = addressManager.getBindings();
List<Queue> queues = new ArrayList<>(); List<Queue> queues = new ArrayList<>();
@ -1313,7 +1294,6 @@ public class PostOfficeImpl implements PostOffice, NotificationListener, Binding
} }
} }
} }
}
public static final class AddOperation implements TransactionOperation { public static final class AddOperation implements TransactionOperation {

View File

@ -74,7 +74,6 @@ public class FileStoreMonitor extends ActiveMQScheduledComponent {
@Override @Override
public void run() { public void run() {
super.run();
tick(); tick();
} }

View File

@ -1100,13 +1100,17 @@ public class QueueImpl implements Queue {
} }
} }
@Override
public void expire(final MessageReference ref) throws Exception { public void expire(final MessageReference ref) throws Exception {
if (expiryAddress != null) { SimpleString messageExpiryAddress = expiryAddressFromMessageAddress(ref);
if (logger.isTraceEnabled()) { if (messageExpiryAddress == null) {
logger.trace("moving expired reference " + ref + " to address = " + expiryAddress + " from queue=" + this.getName()); messageExpiryAddress = expiryAddressFromAddressSettings(ref);
} }
move(null, expiryAddress, ref, false, AckReason.EXPIRED);
if (messageExpiryAddress != null) {
if (logger.isTraceEnabled()) {
logger.trace("moving expired reference " + ref + " to address = " + messageExpiryAddress + " from queue=" + this.getName());
}
move(null, messageExpiryAddress, ref, false, AckReason.EXPIRED);
} }
else { else {
if (logger.isTraceEnabled()) { if (logger.isTraceEnabled()) {
@ -1116,6 +1120,40 @@ public class QueueImpl implements Queue {
} }
} }
private SimpleString expiryAddressFromMessageAddress(MessageReference ref) {
SimpleString messageAddress = extractAddress(ref.getMessage());
SimpleString expiryAddress = null;
if (messageAddress == null || messageAddress.equals(getAddress())) {
expiryAddress = getExpiryAddress();
}
return expiryAddress;
}
private SimpleString expiryAddressFromAddressSettings(MessageReference ref) {
SimpleString messageAddress = extractAddress(ref.getMessage());
SimpleString expiryAddress = null;
if (messageAddress != null) {
AddressSettings addressSettings = addressSettingsRepository.getMatch(messageAddress.toString());
expiryAddress = addressSettings.getExpiryAddress();
}
return expiryAddress;
}
private SimpleString extractAddress(ServerMessage message) {
if (message.containsProperty(Message.HDR_ORIG_MESSAGE_ID)) {
return message.getSimpleStringProperty(Message.HDR_ORIGINAL_ADDRESS);
}
else {
return message.getAddress();
}
}
@Override @Override
public SimpleString getExpiryAddress() { public SimpleString getExpiryAddress() {
return this.expiryAddress; return this.expiryAddress;

View File

@ -44,7 +44,6 @@ public class ReloadManagerImpl extends ActiveMQScheduledComponent implements Rel
@Override @Override
public void run() { public void run() {
super.run();
tick(); tick();
} }

View File

@ -0,0 +1,115 @@
/**
* 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.artemis.tests.integration.cluster.distribution;
import org.apache.activemq.artemis.api.core.SimpleString;
import org.apache.activemq.artemis.api.core.client.ClientConsumer;
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.ClientSession;
import org.apache.activemq.artemis.api.core.client.ClientSessionFactory;
import org.apache.activemq.artemis.core.server.cluster.ClusterConnection;
import org.apache.activemq.artemis.core.server.cluster.MessageFlowRecord;
import org.apache.activemq.artemis.core.server.cluster.impl.ClusterConnectionImpl;
import org.apache.activemq.artemis.core.server.cluster.impl.MessageLoadBalancingType;
import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
public class ExpireWhileLoadBalanceTest extends ClusterTestBase {
@Before
public void setUp() throws Exception {
super.setUp();
setupServer(0, isFileStorage(), true);
setupServer(1, isFileStorage(), true);
setupServer(2, isFileStorage(), true);
for (int i = 0; i < 3; i++) {
servers[i].getConfiguration().setMessageExpiryScanPeriod(100);
}
setupClusterConnection("cluster0", "queues", MessageLoadBalancingType.STRICT, 1, true, 0, 1, 2);
setupClusterConnection("cluster1", "queues", MessageLoadBalancingType.STRICT, 1, true, 1, 0, 2);
setupClusterConnection("cluster2", "queues", MessageLoadBalancingType.STRICT, 1, true, 2, 0, 1);
startServers(0, 1, 2);
setupSessionFactory(0, true);
setupSessionFactory(1, true);
setupSessionFactory(2, true);
}
@Test
public void testSend() throws Exception {
waitForTopology(getServer(0), 3);
waitForTopology(getServer(1), 3);
waitForTopology(getServer(2), 3);
SimpleString expiryQueue = SimpleString.toSimpleString("expiryQueue");
AddressSettings as = new AddressSettings();
as.setDeadLetterAddress(expiryQueue);
as.setExpiryAddress(expiryQueue);
for (int i = 0; i <= 2; i++) {
createQueue(i, "queues.testaddress", "queue0", null, true);
getServer(i).createQueue(expiryQueue, expiryQueue, null, true, false);
getServer(i).getAddressSettingsRepository().addMatch("queues.*", as);
}
// this will pause all the cluster bridges
for (ClusterConnection clusterConnection : getServer(0).getClusterManager().getClusterConnections()) {
for (MessageFlowRecord record : ((ClusterConnectionImpl) clusterConnection).getRecords().values()) {
record.getBridge().pause();
}
}
ClientSessionFactory sf = sfs[0];
ClientSession session = sf.createSession(false, false);
ClientProducer producer = session.createProducer("queues.testaddress");
for (int i = 0; i < 1000; i++) {
ClientMessage message = session.createMessage(true);
message.setExpiration(500);
producer.send(message);
}
session.commit();
session.start();
ClientConsumer consumer = session.createConsumer("expiryQueue");
for (int i = 0; i < 1000; i++) {
ClientMessage message = consumer.receive(2000);
Assert.assertNotNull(message);
message.acknowledge();
}
session.commit();
session.close();
}
}