KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > test > jca > ejb > ThreadLocalDB


1 /*
2   * JBoss, Home of Professional Open Source
3   * Copyright 2005, JBoss Inc., and individual contributors as indicated
4   * by the @authors tag. See the copyright.txt in the distribution for a
5   * full listing of individual contributors.
6   *
7   * This is free software; you can redistribute it and/or modify it
8   * under the terms of the GNU Lesser General Public License as
9   * 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 JavaDoc;
25
26 import javax.naming.InitialContext JavaDoc;
27 import javax.sql.DataSource JavaDoc;
28
29 /**
30  * Thread Local Database
31  *
32  * @author <a HREF="mailto:adrian@jboss.org">Adrian Brock</a>
33  * @version <tt>$Revision: 37406 $</tt>
34  */

35 public class ThreadLocalDB
36 {
37    private static ThreadLocal JavaDoc tl = new ThreadLocal JavaDoc();
38
39    public static Connection JavaDoc open()
40       throws Exception JavaDoc
41    {
42       Connection JavaDoc c = getConnection();
43       if (c == null)
44       {
45          InitialContext JavaDoc ctx = new InitialContext JavaDoc();
46          DataSource JavaDoc ds = (DataSource JavaDoc) ctx.lookup("java:/DefaultDS");
47          ctx.close();
48          c = ds.getConnection();
49          tl.set(c);
50       }
51       return c;
52    }
53
54    public static void close()
55    {
56       Connection JavaDoc c = getConnection();
57       tl.set(null);
58       if (c != null)
59       {
60          try
61          {
62             c.close();
63          }
64          catch (Exception JavaDoc ignored)
65          {
66          }
67       }
68    }
69
70    private static Connection JavaDoc getConnection()
71    {
72       return (Connection JavaDoc) tl.get();
73    }
74 }
75
Popular Tags