001 /**
002 * Licensed to the Apache Software Foundation (ASF) under one or more
003 * contributor license agreements. See the NOTICE file distributed with
004 * this work for additional information regarding copyright ownership.
005 * The ASF licenses this file to You under the Apache License, Version 2.0
006 * (the "License"); you may not use this file except in compliance with
007 * the License. You may obtain a copy of the License at
008 *
009 * http://www.apache.org/licenses/LICENSE-2.0
010 *
011 * Unless required by applicable law or agreed to in writing, software
012 * distributed under the License is distributed on an "AS IS" BASIS,
013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014 * See the License for the specific language governing permissions and
015 * limitations under the License.
016 */
017 package org.apache.activemq.pool;
018
019 import javax.jms.JMSException;
020 import javax.jms.Session;
021 import javax.transaction.RollbackException;
022 import javax.transaction.Status;
023 import javax.transaction.SystemException;
024 import javax.transaction.TransactionManager;
025 import javax.transaction.xa.XAResource;
026
027 import org.apache.activemq.ActiveMQConnection;
028 import org.apache.commons.pool.ObjectPoolFactory;
029
030 /**
031 * An XA-aware connection pool. When a session is created and an xa transaction is active,
032 * the session will automatically be enlisted in the current transaction.
033 *
034 * @author gnodet
035 */
036 public class XaConnectionPool extends ConnectionPool {
037
038 private TransactionManager transactionManager;
039
040 public XaConnectionPool(ActiveMQConnection connection, ObjectPoolFactory poolFactory, TransactionManager transactionManager) {
041 super(connection, poolFactory);
042 this.transactionManager = transactionManager;
043 }
044
045 public Session createSession(boolean transacted, int ackMode) throws JMSException {
046 try {
047 boolean isXa = (transactionManager != null && transactionManager.getStatus() != Status.STATUS_NO_TRANSACTION);
048 if (isXa) {
049 transacted = true;
050 ackMode = Session.SESSION_TRANSACTED;
051 }
052 PooledSession session = (PooledSession) super.createSession(transacted, ackMode);
053 if (isXa) {
054 session.setIgnoreClose(true);
055 transactionManager.getTransaction().registerSynchronization(new Synchronization(session));
056 incrementReferenceCount();
057 transactionManager.getTransaction().enlistResource(createXaResource(session));
058 }
059 return session;
060 } catch (RollbackException e) {
061 final JMSException jmsException = new JMSException("Rollback Exception");
062 jmsException.initCause(e);
063 throw jmsException;
064 } catch (SystemException e) {
065 final JMSException jmsException = new JMSException("System Exception");
066 jmsException.initCause(e);
067 throw jmsException;
068 }
069 }
070
071 protected XAResource createXaResource(PooledSession session) throws JMSException {
072 return session.getXAResource();
073 }
074
075
076 protected class Synchronization implements javax.transaction.Synchronization {
077 private final PooledSession session;
078
079 private Synchronization(PooledSession session) {
080 this.session = session;
081 }
082
083 public void beforeCompletion() {
084 }
085
086 public void afterCompletion(int status) {
087 try {
088 // This will return session to the pool.
089 session.setIgnoreClose(false);
090 session.close();
091 decrementReferenceCount();
092 } catch (JMSException e) {
093 throw new RuntimeException(e);
094 }
095 }
096 }
097
098 }