/*
* Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
*
* This file is part of Resin(R) Open Source
*
* Each copy or derived work must preserve the copyright notice and this
* notice unmodified.
*
* Resin Open Source is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Resin Open Source is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
* of NON-INFRINGEMENT. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with Resin Open Source; if not, write to the
*
* Free SoftwareFoundation, Inc.
* 59 Temple Place, Suite 330
* Boston, MA 02111-1307 USA
*
* @author Scott Ferguson
*/
package com.caucho.jdbc;
import com.caucho.util.L10N;
import com.caucho.util.Log;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* Abstract way of grabbing data from the JDBC connection.
*/
public class ConnectionContext {
private static final L10N L = new L10N(ConnectionContext.class);
private static final Logger log = Log.open(ConnectionContext.class);
private static InitialContext _initialContext;
private static ThreadLocal<HashMap<String,ConnectionContext>> _localConn
= new ThreadLocal<HashMap<String,ConnectionContext>>();
private int _depth;
private Connection _conn;
public static void begin(String jndiName)
{
HashMap<String,ConnectionContext> map = _localConn.get();
if (map == null) {
map = new HashMap<String,ConnectionContext>(8);
_localConn.set(map);
}
ConnectionContext cxt = map.get(jndiName);
if (cxt == null) {
cxt = new ConnectionContext();
map.put(jndiName, cxt);
}
cxt._depth++;
}
public static Connection getConnection(String jndiName)
throws SQLException
{
HashMap<String,ConnectionContext> map = _localConn.get();
if (map == null)
throw new IllegalStateException(L.l("'{0}' is not an available connection.",
jndiName));
ConnectionContext cxt = map.get(jndiName);
if (cxt == null || cxt._depth == 0)
throw new IllegalStateException(L.l("'{0}' is not an available connection.",
jndiName));
if (cxt._conn == null) {
try {
DataSource ds = (DataSource) _initialContext.lookup(jndiName);
cxt._conn = ds.getConnection();
} catch (NamingException e) {
throw new IllegalStateException(e);
}
}
return null;
}
public static void end(String jdbcName)
{
HashMap<String,ConnectionContext> map = _localConn.get();
if (map == null)
return;
ConnectionContext cxt = map.get(jdbcName);
if (cxt == null)
return;
if (--cxt._depth == 0) {
Connection conn = cxt._conn;
try {
cxt._conn = null;
conn.close();
} catch (SQLException e) {
log.log(Level.FINE, e.toString(), e);
}
}
}
private static Context getInitialContext()
{
if (_initialContext == null) {
try {
_initialContext = new InitialContext();
} catch (NamingException e) {
}
}
return _initialContext;
}
}