/*
* Copyright 2009 Red Hat, Inc.
* Red Hat 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.hornetq.tests.integration.xa;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import javax.transaction.xa.XAException;
import javax.transaction.xa.XAResource;
import javax.transaction.xa.Xid;
import junit.framework.Assert;
import org.hornetq.api.core.HornetQException;
import org.hornetq.api.core.SimpleString;
import org.hornetq.api.core.TransportConfiguration;
import org.hornetq.api.core.client.*;
import org.hornetq.core.config.impl.ConfigurationImpl;
import org.hornetq.core.remoting.impl.invm.InVMConnectorFactory;
import org.hornetq.core.server.HornetQServer;
import org.hornetq.core.server.HornetQServers;
import org.hornetq.core.server.MessageReference;
import org.hornetq.core.server.Queue;
import org.hornetq.core.settings.impl.AddressSettings;
import org.hornetq.core.transaction.Transaction;
import org.hornetq.core.transaction.TransactionOperation;
import org.hornetq.core.transaction.impl.XidImpl;
import org.hornetq.tests.util.UnitTestCase;
import org.hornetq.utils.UUIDGenerator;
/**
* @author <a href="mailto:andy.taylor@jboss.org">Andy Taylor</a>
*/
public class XaTimeoutTest extends UnitTestCase
{
private final Map<String, AddressSettings> addressSettings = new HashMap<String, AddressSettings>();
private HornetQServer messagingService;
private ClientSession clientSession;
private ClientProducer clientProducer;
private ClientConsumer clientConsumer;
private ClientSessionFactory sessionFactory;
private ConfigurationImpl configuration;
private final SimpleString atestq = new SimpleString("atestq");
@Override
protected void setUp() throws Exception
{
super.setUp();
addressSettings.clear();
configuration = createBasicConfig();
configuration.setSecurityEnabled(false);
configuration.setTransactionTimeoutScanPeriod(500);
TransportConfiguration transportConfig = new TransportConfiguration(UnitTestCase.INVM_ACCEPTOR_FACTORY);
configuration.getAcceptorConfigurations().add(transportConfig);
messagingService = HornetQServers.newHornetQServer(configuration, false);
// start the server
messagingService.start();
// then we create a client as normal
ServerLocator locator = HornetQClient.createServerLocatorWithoutHA(new TransportConfiguration(InVMConnectorFactory.class.getName()));
sessionFactory = locator.createSessionFactory();
clientSession = sessionFactory.createSession(true, false, false);
clientSession.createQueue(atestq, atestq, null, true);
clientProducer = clientSession.createProducer(atestq);
clientConsumer = clientSession.createConsumer(atestq);
}
@Override
protected void tearDown() throws Exception
{
if (clientSession != null)
{
try
{
clientSession.close();
}
catch (HornetQException e1)
{
//
}
}
if (messagingService != null && messagingService.isStarted())
{
try
{
messagingService.stop();
}
catch (Exception e1)
{
//
}
}
messagingService = null;
clientSession = null;
clientProducer = null;;
clientConsumer = null;
sessionFactory = null;
configuration = null;;
super.tearDown();
}
public void testSimpleTimeoutOnSendOnCommit() throws Exception
{
Xid xid = new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
ClientMessage m1 = createTextMessage(clientSession, "m1");
ClientMessage m2 = createTextMessage(clientSession, "m2");
ClientMessage m3 = createTextMessage(clientSession, "m3");
ClientMessage m4 = createTextMessage(clientSession, "m4");
clientSession.setTransactionTimeout(1);
clientSession.start(xid, XAResource.TMNOFLAGS);
clientProducer.send(m1);
clientProducer.send(m2);
clientProducer.send(m3);
clientProducer.send(m4);
clientSession.end(xid, XAResource.TMSUCCESS);
CountDownLatch latch = new CountDownLatch(1);
messagingService.getResourceManager().getTransaction(xid).addOperation(new RollbackCompleteOperation(latch));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
try
{
clientSession.commit(xid, true);
}
catch (XAException e)
{
Assert.assertTrue(e.errorCode == XAException.XAER_NOTA);
}
clientSession.start();
ClientMessage m = clientConsumer.receiveImmediate();
Assert.assertNull(m);
}
public void testSimpleTimeoutOnReceive() throws Exception
{
Xid xid = new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
ClientMessage m1 = createTextMessage(clientSession, "m1");
ClientMessage m2 = createTextMessage(clientSession, "m2");
ClientMessage m3 = createTextMessage(clientSession, "m3");
ClientMessage m4 = createTextMessage(clientSession, "m4");
ClientSession clientSession2 = sessionFactory.createSession(false, true, true);
ClientProducer clientProducer2 = clientSession2.createProducer(atestq);
clientProducer2.send(m1);
clientProducer2.send(m2);
clientProducer2.send(m3);
clientProducer2.send(m4);
clientSession2.close();
clientSession.setTransactionTimeout(2);
clientSession.start(xid, XAResource.TMNOFLAGS);
clientSession.start();
ClientMessage m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m1");
m = clientConsumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m3");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m4");
clientSession.end(xid, XAResource.TMSUCCESS);
CountDownLatch latch = new CountDownLatch(1);
messagingService.getResourceManager().getTransaction(xid).addOperation(new RollbackCompleteOperation(latch));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
try
{
clientSession.commit(xid, true);
}
catch (XAException e)
{
Assert.assertTrue(e.errorCode == XAException.XAER_NOTA);
}
clientSession.setTransactionTimeout(0);
clientConsumer.close();
clientSession2 = sessionFactory.createSession(false, true, true);
ClientConsumer consumer = clientSession2.createConsumer(atestq);
clientSession2.start();
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m1");
m = consumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m3");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m4");
clientSession2.close();
}
public void testSimpleTimeoutOnSendAndReceive() throws Exception
{
Xid xid = new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
ClientMessage m1 = createTextMessage(clientSession, "m1");
ClientMessage m2 = createTextMessage(clientSession, "m2");
ClientMessage m3 = createTextMessage(clientSession, "m3");
ClientMessage m4 = createTextMessage(clientSession, "m4");
ClientMessage m5 = createTextMessage(clientSession, "m5");
ClientMessage m6 = createTextMessage(clientSession, "m6");
ClientMessage m7 = createTextMessage(clientSession, "m7");
ClientMessage m8 = createTextMessage(clientSession, "m8");
ClientSession clientSession2 = sessionFactory.createSession(false, true, true);
ClientProducer clientProducer2 = clientSession2.createProducer(atestq);
clientProducer2.send(m1);
clientProducer2.send(m2);
clientProducer2.send(m3);
clientProducer2.send(m4);
clientSession2.close();
clientSession.setTransactionTimeout(2);
clientSession.start(xid, XAResource.TMNOFLAGS);
clientSession.start();
clientProducer.send(m5);
clientProducer.send(m6);
clientProducer.send(m7);
clientProducer.send(m8);
ClientMessage m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m1");
m = clientConsumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m3");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m4");
clientSession.end(xid, XAResource.TMSUCCESS);
CountDownLatch latch = new CountDownLatch(1);
messagingService.getResourceManager().getTransaction(xid).addOperation(new RollbackCompleteOperation(latch));
Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
try
{
clientSession.commit(xid, true);
}
catch (XAException e)
{
Assert.assertTrue(e.errorCode == XAException.XAER_NOTA);
}
clientSession.setTransactionTimeout(0);
clientConsumer.close();
clientSession2 = sessionFactory.createSession(false, true, true);
ClientConsumer consumer = clientSession2.createConsumer(atestq);
clientSession2.start();
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m1");
m = consumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m3");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m4");
m = consumer.receiveImmediate();
Assert.assertNull(m);
clientSession2.close();
}
public void testPreparedTransactionNotTimedOut() throws Exception
{
Xid xid = new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
ClientMessage m1 = createTextMessage(clientSession, "m1");
ClientMessage m2 = createTextMessage(clientSession, "m2");
ClientMessage m3 = createTextMessage(clientSession, "m3");
ClientMessage m4 = createTextMessage(clientSession, "m4");
ClientMessage m5 = createTextMessage(clientSession, "m5");
ClientMessage m6 = createTextMessage(clientSession, "m6");
ClientMessage m7 = createTextMessage(clientSession, "m7");
ClientMessage m8 = createTextMessage(clientSession, "m8");
ClientSession clientSession2 = sessionFactory.createSession(false, true, true);
ClientProducer clientProducer2 = clientSession2.createProducer(atestq);
clientProducer2.send(m1);
clientProducer2.send(m2);
clientProducer2.send(m3);
clientProducer2.send(m4);
clientSession2.close();
clientSession.setTransactionTimeout(2);
clientSession.start(xid, XAResource.TMNOFLAGS);
clientSession.start();
clientProducer.send(m5);
clientProducer.send(m6);
clientProducer.send(m7);
clientProducer.send(m8);
ClientMessage m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m1");
m = clientConsumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m3");
m = clientConsumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m4");
clientSession.end(xid, XAResource.TMSUCCESS);
clientSession.prepare(xid);
CountDownLatch latch = new CountDownLatch(1);
messagingService.getResourceManager().getTransaction(xid).addOperation(new RollbackCompleteOperation(latch));
Assert.assertFalse(latch.await(2600, TimeUnit.MILLISECONDS));
clientSession.commit(xid, false);
clientSession.setTransactionTimeout(0);
clientConsumer.close();
clientSession2 = sessionFactory.createSession(false, true, true);
ClientConsumer consumer = clientSession2.createConsumer(atestq);
clientSession2.start();
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m5");
m = consumer.receive(500);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m6");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m7");
m = consumer.receive(500);
m.acknowledge();
Assert.assertNotNull(m);
Assert.assertEquals(m.getBodyBuffer().readString(), "m8");
m = consumer.receiveImmediate();
Assert.assertNull(m);
clientSession2.close();
}
public void testChangingTimeoutGetsPickedUp() throws Exception
{
Xid xid = new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
ClientMessage m1 = createTextMessage(clientSession, "m1");
ClientMessage m2 = createTextMessage(clientSession, "m2");
ClientMessage m3 = createTextMessage(clientSession, "m3");
ClientMessage m4 = createTextMessage(clientSession, "m4");
clientSession.start(xid, XAResource.TMNOFLAGS);
clientProducer.send(m1);
clientProducer.send(m2);
clientProducer.send(m3);
clientProducer.send(m4);
clientSession.end(xid, XAResource.TMSUCCESS);
clientSession.commit(xid, true);
clientSession.setTransactionTimeout(1);
clientSession.start(xid, XAResource.TMNOFLAGS);
CountDownLatch latch = new CountDownLatch(1);
messagingService.getResourceManager().getTransaction(xid).addOperation(new RollbackCompleteOperation(latch));
clientProducer.send(m1);
clientProducer.send(m2);
clientProducer.send(m3);
clientProducer.send(m4);
clientSession.end(xid, XAResource.TMSUCCESS);
Assert.assertTrue(latch.await(2600, TimeUnit.MILLISECONDS));
try
{
clientSession.commit(xid, true);
}
catch (XAException e)
{
Assert.assertTrue(e.errorCode == XAException.XAER_NOTA);
}
clientSession.start();
ClientMessage m = clientConsumer.receiveImmediate();
Assert.assertNotNull(m);
m = clientConsumer.receiveImmediate();
Assert.assertNotNull(m);
m = clientConsumer.receiveImmediate();
Assert.assertNotNull(m);
m = clientConsumer.receiveImmediate();
Assert.assertNotNull(m);
m = clientConsumer.receiveImmediate();
Assert.assertNull(m);
}
public void testMultipleTransactionsTimedOut() throws Exception
{
Xid[] xids = new XidImpl[100];
for (int i = 0; i < xids.length; i++)
{
xids[i] = new XidImpl(("xa" + i).getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes());
}
ClientSession[] clientSessions = new ClientSession[xids.length];
for (int i = 0; i < clientSessions.length; i++)
{
clientSessions[i] = sessionFactory.createSession(true, false, false);
clientSessions[i].setTransactionTimeout(i < 50?2:5000);
}
ClientProducer[] clientProducers = new ClientProducer[xids.length];
for (int i = 0; i < clientProducers.length; i++)
{
clientProducers[i] = clientSessions[i].createProducer(atestq);
}
ClientMessage[] messages = new ClientMessage[xids.length];
for (int i = 0; i < messages.length; i++)
{
messages[i] = createTextMessage(clientSession, "m" + i);
}
for (int i = 0; i < clientSessions.length; i++)
{
clientSessions[i].start(xids[i], XAResource.TMNOFLAGS);
}
for (int i = 0; i < clientProducers.length; i++)
{
clientProducers[i].send(messages[i]);
}
for (int i = 0; i < clientSessions.length; i++)
{
clientSessions[i].end(xids[i], XAResource.TMSUCCESS);
}
CountDownLatch[] latches = new CountDownLatch[xids.length];
for (int i1 = 0; i1 < latches.length; i1++)
{
latches[i1] = new CountDownLatch(1);
messagingService.getResourceManager()
.getTransaction(xids[i1])
.addOperation(new RollbackCompleteOperation(latches[i1]));
}
for (int i1 = 0;i1 < latches.length/2; i1++)
{
Assert.assertTrue(latches[i1].await(5, TimeUnit.SECONDS));
}
for (int i = 0; i < clientSessions.length/2; i++)
{
try
{
clientSessions[i].commit(xids[i], true);
}
catch (XAException e)
{
Assert.assertTrue(e.errorCode == XAException.XAER_NOTA);
}
}
for (int i = 50; i < clientSessions.length; i++)
{
clientSessions[i].commit(xids[i], true);
}
for (ClientSession session : clientSessions)
{
session.close();
}
clientSession.start();
for(int i = 0; i < clientSessions.length/2; i++)
{
ClientMessage m = clientConsumer.receiveImmediate();
Assert.assertNotNull(m);
}
ClientMessage m = clientConsumer.receiveImmediate();
Assert.assertNull(m);
}
class RollbackCompleteOperation implements TransactionOperation
{
final CountDownLatch latch;
public RollbackCompleteOperation(final CountDownLatch latch)
{
this.latch = latch;
}
public void beforePrepare(final Transaction tx) throws Exception
{
}
public void beforeCommit(final Transaction tx) throws Exception
{
}
public void beforeRollback(final Transaction tx) throws Exception
{
}
public void afterPrepare(final Transaction tx)
{
}
public void afterCommit(final Transaction tx)
{
}
public void afterRollback(final Transaction tx)
{
latch.countDown();
}
public Collection<Queue> getDistinctQueues()
{
return Collections.emptySet();
}
public List<MessageReference> getRelatedMessageReferences()
{
return null;
}
}
}