https://issues.apache.org/jira/browse/AMQ-3272 - Handle RejectedExecutionException. In this case avoid the root cause, by allowing the semaphore to limit the enqueue tasks. Issue was in the cancel case locks were being release and the semaphore dropped while the executore queue still had the task pending, it needs to execute before the the samaphore can be released as it gates the task queue

git-svn-id: https://svn.apache.org/repos/asf/activemq/trunk@1102014 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Gary Tully 2011-05-11 18:43:43 +00:00
parent 0aa5c2a889
commit 717bca1a40
2 changed files with 109 additions and 1 deletions

View File

@ -1067,7 +1067,6 @@ public class KahaDBStore extends MessageDatabase implements PersistenceAdapter {
}
public boolean cancel() {
releaseLocks();
if (this.done.compareAndSet(false, true)) {
return this.future.cancel(false);
}

View File

@ -0,0 +1,109 @@
/**
* 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.store.kahadb;
import java.util.Vector;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import org.apache.activemq.command.ActiveMQDestination;
import org.apache.activemq.command.ActiveMQMessage;
import org.apache.activemq.command.ActiveMQQueue;
import org.apache.activemq.command.Message;
import org.apache.activemq.command.MessageAck;
import org.apache.activemq.command.MessageId;
import org.apache.activemq.command.ProducerId;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.assertTrue;
public class KahaDBStoreTest {
KahaDBStore.KahaDBMessageStore underTest;
KahaDBStore store;
ActiveMQMessage message;
ProducerId producerId = new ProducerId("1.1.1");
private static final int MESSAGE_COUNT = 2000;
private Vector<Throwable> exceptions = new Vector<Throwable>();
@Before
public void initStore() throws Exception {
ActiveMQDestination destination = new ActiveMQQueue("Test");
store = new KahaDBStore();
store.setMaxAsyncJobs(100);
store.setDeleteAllMessages(true);
store.start();
underTest = store.new KahaDBMessageStore(destination);
underTest.start();
message = new ActiveMQMessage();
message.setDestination(destination);
}
@After
public void destroyStore() throws Exception {
if (store != null) {
store.stop();
}
}
@Test
public void testConcurrentStoreAndDispatchQueue() throws Exception {
ExecutorService executor = Executors.newCachedThreadPool();
for (int i=0; i<MESSAGE_COUNT; i++) {
final int id = ++i;
executor.execute(new Runnable() {
public void run() {
try {
Message msg = message.copy();
msg.setMessageId(new MessageId(producerId, id));
underTest.asyncAddQueueMessage(null, msg);
} catch (Exception e) {
exceptions.add(e);
}
}
});
}
ExecutorService executor2 = Executors.newCachedThreadPool();
for (int i=0; i<MESSAGE_COUNT; i++) {
final int id = ++i;
executor2.execute(new Runnable() {
public void run() {
try {
MessageAck ack = new MessageAck();
ack.setLastMessageId(new MessageId(producerId, id));
underTest.removeAsyncMessage(null, ack);
} catch (Exception e) {
exceptions.add(e);
}
}
});
}
executor.shutdown();
executor.awaitTermination(60, TimeUnit.SECONDS);
executor2.shutdown();
executor2.awaitTermination(60, TimeUnit.SECONDS);
assertTrue("no exceptions " + exceptions, exceptions.isEmpty());
}
}