1 /*** 2 * 3 * Copyright 2004 Hiram Chirino 4 * Copyright 2004 Protique Ltd 5 * 6 * Licensed under the Apache License, Version 2.0 (the "License"); 7 * you may not use this file except in compliance with the License. 8 * You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 * 18 **/ 19 package org.codehaus.activemq.store.jdbc; 20 21 import org.apache.commons.logging.Log; 22 import org.apache.commons.logging.LogFactory; 23 24 import java.sql.Connection; 25 import java.util.LinkedList; 26 27 /*** 28 * Helps keep track of the current transaction/JDBC connection. 29 * 30 * @version $Revision: 1.1 $ 31 */ 32 public class TransactionContext { 33 private static final Log log = LogFactory.getLog(TransactionContext.class); 34 private static ThreadLocal threadLocalTxn = new ThreadLocal(); 35 36 /*** 37 * @return the current thread local connection that is associated 38 * with the JMS transaction or null if there is no 39 * transaction in progress. 40 */ 41 public static Connection getConnection() { 42 LinkedList list = (LinkedList) threadLocalTxn.get(); 43 if (list != null && !list.isEmpty()) { 44 return (Connection) list.getFirst(); 45 } 46 throw new IllegalStateException("Attempt to get a connection when no transaction in progress"); 47 } 48 49 /*** 50 * Pops off the current Connection from the stack 51 */ 52 public static Connection popConnection() { 53 LinkedList list = (LinkedList) threadLocalTxn.get(); 54 if (list == null || list.isEmpty()) { 55 log.warn("Attempt to pop connection when no transaction in progress"); 56 return null; 57 } 58 else { 59 return (Connection) list.removeFirst(); 60 } 61 } 62 63 /*** 64 * Sets the current transaction, possibly including nesting 65 */ 66 public static void pushConnection(Connection connection) { 67 LinkedList list = (LinkedList) threadLocalTxn.get(); 68 if (list == null) { 69 list = new LinkedList(); 70 threadLocalTxn.set(list); 71 } 72 list.addLast(connection); 73 } 74 75 public static int getTransactionCount() { 76 LinkedList list = (LinkedList) threadLocalTxn.get(); 77 if (list != null) { 78 return list.size(); 79 } 80 return 0; 81 } 82 83 }