01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.test.jca.ejb;
23:
24: import java.sql.Connection;
25:
26: import javax.naming.InitialContext;
27: import javax.sql.DataSource;
28:
29: /**
30: * Thread Local Database
31: *
32: * @author <a href="mailto:adrian@jboss.org">Adrian Brock</a>
33: * @version <tt>$Revision: 57211 $</tt>
34: */
35: public class ThreadLocalDB {
36: private static ThreadLocal tl = new ThreadLocal();
37:
38: public static Connection open() throws Exception {
39: Connection c = getConnection();
40: if (c == null) {
41: InitialContext ctx = new InitialContext();
42: DataSource ds = (DataSource) ctx.lookup("java:/DefaultDS");
43: ctx.close();
44: c = ds.getConnection();
45: tl.set(c);
46: }
47: return c;
48: }
49:
50: public static void close() {
51: Connection c = getConnection();
52: tl.set(null);
53: if (c != null) {
54: try {
55: c.close();
56: } catch (Exception ignored) {
57: }
58: }
59: }
60:
61: private static Connection getConnection() {
62: return (Connection) tl.get();
63: }
64: }
|