Examples with XASession javax.jms.XASession used on opensource projects

Example 26 with XASession

use of javax.jms.XASession in project activemq-artemis by apache.

the class XATest method testOneSessionTwoTransactionsRollbackAcknowledge.

@Test
public void testOneSessionTwoTransactionsRollbackAcknowledge() throws Exception {
    XAConnection conn = null;
    Connection conn2 = null;
    try {
        // First send 2 messages
        conn2 = cf.createConnection();
        Session sessProducer = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        MessageProducer prod = sessProducer.createProducer(queue1);
        Message m = sessProducer.createTextMessage("jellyfish1");
        prod.send(m);
        m = sessProducer.createTextMessage("jellyfish2");
        prod.send(m);
        conn = xacf.createXAConnection();
        // Create a session
        XASession sess1 = conn.createXASession();
        XAResource res1 = sess1.getXAResource();
        conn.start();
        MessageConsumer cons1 = sess1.createConsumer(queue1);
        tm.begin();
        Transaction tx1 = tm.getTransaction();
        tx1.enlistResource(res1);
        // Receive one message in one tx
        TextMessage r1 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r1);
        Assert.assertEquals("jellyfish1", r1.getText());
        tx1.delistResource(res1, XAResource.TMSUCCESS);
        // suspend the tx
        Transaction suspended = tm.suspend();
        tm.begin();
        Transaction tx2 = tm.getTransaction();
        tx2.enlistResource(res1);
        // Receive 2nd message in a different tx
        TextMessage r2 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r2);
        Assert.assertEquals("jellyfish2", r2.getText());
        cons1.close();
        tx2.delistResource(res1, XAResource.TMSUCCESS);
        // rollback this transaction
        tm.rollback();
        // verify that second message is available
        conn2.close();
        conn2 = cf.createConnection();
        Session sess = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        conn2.start();
        MessageConsumer cons = sess.createConsumer(queue1);
        TextMessage r3 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r3);
        Assert.assertEquals("jellyfish2", r3.getText());
        r3 = (TextMessage) cons.receive(100);
        Assert.assertNull(r3);
        // rollback the other tx
        tm.resume(suspended);
        tm.rollback();
        // Verify the first message is now available
        r3 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r3);
        Assert.assertEquals("jellyfish1", r3.getText());
        r3 = (TextMessage) cons.receive(100);
        Assert.assertNull(r3);
    } finally {
        if (conn != null) {
            conn.close();
        }
        if (conn2 != null) {
            conn2.close();
        }
    }
}

Example 27 with XASession

use of javax.jms.XASession in project activemq-artemis by apache.

the class XATest method testMultipleSessionsOneTxRollbackAcknowledgeForceFailureInCommit.

@Test
public void testMultipleSessionsOneTxRollbackAcknowledgeForceFailureInCommit() throws Exception {
    XAConnection conn = null;
    Connection conn2 = null;
    try {
        // First send 4 messages
        conn2 = cf.createConnection();
        Session sessProducer = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        MessageProducer prod = sessProducer.createProducer(queue1);
        Message m = sessProducer.createTextMessage("jellyfish1");
        prod.send(m);
        m = sessProducer.createTextMessage("jellyfish2");
        prod.send(m);
        m = sessProducer.createTextMessage("jellyfish3");
        prod.send(m);
        m = sessProducer.createTextMessage("jellyfish4");
        prod.send(m);
        conn = xacf.createXAConnection();
        conn.start();
        tm.begin();
        XASession sess1 = conn.createXASession();
        XAResource res1 = sess1.getXAResource();
        DummyXAResource res2 = new DummyXAResource(true);
        Transaction tx = tm.getTransaction();
        tx.enlistResource(res1);
        tx.enlistResource(res2);
        MessageConsumer cons1 = sess1.createConsumer(queue1);
        TextMessage r1 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r1);
        Assert.assertEquals("jellyfish1", r1.getText());
        r1 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r1);
        Assert.assertEquals("jellyfish2", r1.getText());
        r1 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r1);
        Assert.assertEquals("jellyfish3", r1.getText());
        r1 = (TextMessage) cons1.receive(5000);
        Assert.assertNotNull(r1);
        Assert.assertEquals("jellyfish4", r1.getText());
        r1 = (TextMessage) cons1.receive(100);
        Assert.assertNull(r1);
        cons1.close();
        // try and commit - and we're going to make the dummyxaresource throw an exception on commit,
        // which should cause rollback to be called on the other resource
        tx.delistResource(res1, XAResource.TMSUCCESS);
        tx.delistResource(res2, XAResource.TMSUCCESS);
        // rollback will cause an attempt to deliver messages locally to the original consumers.
        // the original consumer has closed, so it will cancelled to the server
        // the server cancel is asynch, so we need to sleep for a bit to make sure it completes
        ExtrasTestLogger.LOGGER.trace("Forcing failure");
        try {
            tm.commit();
            Assert.fail("should not get here");
        } catch (Exception e) {
        // We should expect this
        }
        Thread.sleep(1000);
        Session sess = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        MessageConsumer cons = sess.createConsumer(queue1);
        conn2.start();
        TextMessage r = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r);
        Assert.assertEquals("jellyfish1", r.getText());
        r = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r);
        Assert.assertEquals("jellyfish2", r.getText());
        r = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r);
        Assert.assertEquals("jellyfish3", r.getText());
        r = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(r);
        Assert.assertEquals("jellyfish4", r.getText());
        r = (TextMessage) cons.receive(100);
        Assert.assertNull(r);
    } finally {
        if (conn != null) {
            conn.close();
        }
        if (conn2 != null) {
            conn2.close();
        }
    }
}

Example 28 with XASession

use of javax.jms.XASession in project activemq-artemis by apache.

the class XATest method test2PCReceiveCommit1PCOptimization.

@Test
public void test2PCReceiveCommit1PCOptimization() throws Exception {
    // Since both resources have some RM, TM will probably use 1PC optimization
    XAConnection conn = null;
    Connection conn2 = null;
    try {
        conn2 = cf.createConnection();
        conn2.start();
        Session sessProducer = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        MessageProducer prod = sessProducer.createProducer(queue1);
        Message m = sessProducer.createTextMessage("XATest1");
        prod.send(m);
        m = sessProducer.createTextMessage("XATest2");
        prod.send(m);
        conn = xacf.createXAConnection();
        conn.start();
        tm.begin();
        XASession sess = conn.createXASession();
        XAResource res = sess.getXAResource();
        XAResource res2 = new DummyXAResource();
        Transaction tx = tm.getTransaction();
        tx.enlistResource(res);
        tx.enlistResource(res2);
        MessageConsumer cons = sess.createConsumer(queue1);
        TextMessage m2 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m2);
        Assert.assertEquals("XATest1", m2.getText());
        m2 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m2);
        Assert.assertEquals("XATest2", m2.getText());
        tx.delistResource(res, XAResource.TMSUCCESS);
        tx.delistResource(res2, XAResource.TMSUCCESS);
        tm.commit();
        // New tx
        tm.begin();
        tx = tm.getTransaction();
        tx.enlistResource(res);
        tx.enlistResource(res2);
        Message m3 = cons.receive(100);
        Assert.assertNull(m3);
        tx.delistResource(res, XAResource.TMSUCCESS);
        tx.delistResource(res2, XAResource.TMSUCCESS);
        tm.commit();
    } finally {
        if (conn != null) {
            conn.close();
        }
        if (conn2 != null) {
            conn2.close();
        }
    }
}

Example 29 with XASession

use of javax.jms.XASession in project activemq-artemis by apache.

the class XATest method test2PCReceiveRollback1PCOptimization.

@Test
public void test2PCReceiveRollback1PCOptimization() throws Exception {
    // Since both resources have some RM, TM will probably use 1PC optimization
    XAConnection conn = null;
    Connection conn2 = null;
    try {
        conn2 = cf.createConnection();
        Session sessProducer = conn2.createSession(false, Session.AUTO_ACKNOWLEDGE);
        MessageProducer prod = sessProducer.createProducer(queue1);
        Message m = sessProducer.createTextMessage("XATest1");
        prod.send(m);
        m = sessProducer.createTextMessage("XATest2");
        prod.send(m);
        conn = xacf.createXAConnection();
        conn.start();
        tm.begin();
        XASession sess = conn.createXASession();
        XAResource res = sess.getXAResource();
        XAResource res2 = new DummyXAResource();
        Transaction tx = tm.getTransaction();
        tx.enlistResource(res);
        tx.enlistResource(res2);
        MessageConsumer cons = sess.createConsumer(queue1);
        TextMessage m2 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m2);
        Assert.assertEquals("XATest1", m2.getText());
        m2 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m2);
        Assert.assertEquals("XATest2", m2.getText());
        tx.delistResource(res, XAResource.TMSUCCESS);
        tx.delistResource(res2, XAResource.TMSUCCESS);
        tm.rollback();
        // Message should be redelivered
        // New tx
        tm.begin();
        tx = tm.getTransaction();
        tx.enlistResource(res);
        tx.enlistResource(res2);
        TextMessage m3 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m3);
        Assert.assertEquals("XATest1", m3.getText());
        m3 = (TextMessage) cons.receive(5000);
        Assert.assertNotNull(m3);
        Assert.assertEquals("XATest2", m3.getText());
        Assert.assertTrue(m3.getJMSRedelivered());
        tx.delistResource(res, XAResource.TMSUCCESS);
        tx.delistResource(res2, XAResource.TMSUCCESS);
        tm.commit();
    } finally {
        if (conn != null) {
            conn.close();
        }
        if (conn2 != null) {
            conn2.close();
        }
    }
}

Example 30 with XASession

use of javax.jms.XASession in project activemq-artemis by apache.

the class SessionTest method testGetXAResource2.

@Test
public void testGetXAResource2() throws Exception {
    XAConnection conn = getXAConnectionFactory().createXAConnection();
    XASession sess = conn.createXASession();
    sess.getXAResource();
    conn.close();
}