/* * Copyright 2004-2011 H2 Group. Multiple-Licensed under the H2 License, * Version 1.0, and under the Eclipse Public License, Version 1.0 * (http://h2database.com/html/license.html). * Initial Developer: H2 Group */ package org.h2.jdbcx; import java.io.IOException; import java.io.ObjectInputStream; import java.io.PrintWriter; import java.io.Serializable; import java.sql.Connection; import java.sql.SQLException; import java.util.Properties; import javax.naming.Reference; import javax.naming.Referenceable; import javax.naming.StringRefAddr; import javax.sql.ConnectionPoolDataSource; import javax.sql.DataSource; import javax.sql.PooledConnection; import javax.sql.XAConnection; import javax.sql.XADataSource; import org.h2.Driver; import org.h2.jdbc.JdbcConnection; import org.h2.message.TraceObject; import org.h2.util.StringUtils; /*## Java 1.7 ## import java.util.logging.Logger; //*/ /** * A data source for H2 database connections. It is a factory for XAConnection * and Connection objects. This class is usually registered in a JNDI naming * service. To create a data source object and register it with a JNDI service, * use the following code: * * <pre> * import org.h2.jdbcx.JdbcDataSource; * import javax.naming.Context; * import javax.naming.InitialContext; * JdbcDataSource ds = new JdbcDataSource(); * ds.setURL("jdbc:h2:˜/test"); * ds.setUser("sa"); * ds.setPassword("sa"); * Context ctx = new InitialContext(); * ctx.bind("jdbc/dsName", ds); * </pre> * * To use a data source that is already registered, use the following code: * * <pre> * import java.sql.Connection; * import javax.sql.DataSource; * import javax.naming.Context; * import javax.naming.InitialContext; * Context ctx = new InitialContext(); * DataSource ds = (DataSource) ctx.lookup("jdbc/dsName"); * Connection conn = ds.getConnection(); * </pre> * * In this example the user name and password are serialized as * well; this may be a security problem in some cases. */ public class JdbcDataSource extends TraceObject implements XADataSource, DataSource, ConnectionPoolDataSource, Serializable, Referenceable { private static final long serialVersionUID = 1288136338451857771L; private transient JdbcDataSourceFactory factory; private transient PrintWriter logWriter; private int loginTimeout; private String userName = ""; private char[] passwordChars = { }; private String url = ""; private String description; static { org.h2.Driver.load(); } /** * The public constructor. */ public JdbcDataSource() { initFactory(); int id = getNextId(TraceObject.DATA_SOURCE); setTrace(factory.getTrace(), TraceObject.DATA_SOURCE, id); } /** * Called when de-serializing the object. * * @param in the input stream */ private void readObject(ObjectInputStream in) throws IOException, ClassNotFoundException { initFactory(); in.defaultReadObject(); } private void initFactory() { factory = new JdbcDataSourceFactory(); } /** * Get the login timeout in seconds, 0 meaning no timeout. * * @return the timeout in seconds */ public int getLoginTimeout() { debugCodeCall("getLoginTimeout"); return loginTimeout; } /** * Set the login timeout in seconds, 0 meaning no timeout. * The default value is 0. * This value is ignored by this database. * * @param timeout the timeout in seconds */ public void setLoginTimeout(int timeout) { debugCodeCall("setLoginTimeout", timeout); this.loginTimeout = timeout; } /** * Get the current log writer for this object. * * @return the log writer */ public PrintWriter getLogWriter() { debugCodeCall("getLogWriter"); return logWriter; } /** * Set the current log writer for this object. * This value is ignored by this database. * * @param out the log writer */ public void setLogWriter(PrintWriter out) { debugCodeCall("setLogWriter(out)"); logWriter = out; } /** * Open a new connection using the current URL, user name and password. * * @return the connection */ public Connection getConnection() throws SQLException { debugCodeCall("getConnection"); return getJdbcConnection(userName, StringUtils.cloneCharArray(passwordChars)); } /** * Open a new connection using the current URL and the specified user name * and password. * * @param user the user name * @param password the password * @return the connection */ public Connection getConnection(String user, String password) throws SQLException { if (isDebugEnabled()) { debugCode("getConnection("+quote(user)+", \"\");"); } return getJdbcConnection(user, convertToCharArray(password)); } private JdbcConnection getJdbcConnection(String user, char[] password) throws SQLException { if (isDebugEnabled()) { debugCode("getJdbcConnection("+quote(user)+", new char[0]);"); } Properties info = new Properties(); info.setProperty("user", user); info.put("password", password); Connection conn = Driver.load().connect(url, info); if (conn == null) { throw new SQLException("No suitable driver found for " + url, "08001", 8001); } else if (!(conn instanceof JdbcConnection)) { throw new SQLException("Connecting with old version is not supported: " + url, "08001", 8001); } return (JdbcConnection) conn; } /** * Get the current URL. * * @return the URL */ public String getURL() { debugCodeCall("getURL"); return url; } /** * Set the current URL. * * @param url the new URL */ public void setURL(String url) { debugCodeCall("setURL", url); this.url = url; } /** * Set the current password. * * @param password the new password. */ public void setPassword(String password) { debugCodeCall("setPassword", ""); this.passwordChars = convertToCharArray(password); } /** * Set the current password in the form of a char array. * * @param password the new password in the form of a char array. */ public void setPasswordChars(char[] password) { if (isDebugEnabled()) { debugCode("setPasswordChars(new char[0]);"); } this.passwordChars = password; } private static char[] convertToCharArray(String s) { return s == null ? null : s.toCharArray(); } private static String convertToString(char[] a) { return a == null ? null : new String(a); } /** * Get the current password. * * @return the password */ public String getPassword() { debugCodeCall("getPassword"); return convertToString(passwordChars); } /** * Get the current user name. * * @return the user name */ public String getUser() { debugCodeCall("getUser"); return userName; } /** * Set the current user name. * * @param user the new user name */ public void setUser(String user) { debugCodeCall("setUser", user); this.userName = user; } /** * Get the current description. * * @return the description */ public String getDescription() { debugCodeCall("getDescription"); return description; } /** * Set the description. * * @param description the new description */ public void setDescription(String description) { debugCodeCall("getDescription", description); this.description = description; } /** * Get a new reference for this object, using the current settings. * * @return the new reference */ public Reference getReference() { debugCodeCall("getReference"); String factoryClassName = JdbcDataSourceFactory.class.getName(); Reference ref = new Reference(getClass().getName(), factoryClassName, null); ref.add(new StringRefAddr("url", url)); ref.add(new StringRefAddr("user", userName)); ref.add(new StringRefAddr("password", convertToString(passwordChars))); ref.add(new StringRefAddr("loginTimeout", String.valueOf(loginTimeout))); ref.add(new StringRefAddr("description", description)); return ref; } /** * Open a new XA connection using the current URL, user name and password. * * @return the connection */ public XAConnection getXAConnection() throws SQLException { debugCodeCall("getXAConnection"); int id = getNextId(XA_DATA_SOURCE); return new JdbcXAConnection(factory, id, getJdbcConnection(userName, StringUtils.cloneCharArray(passwordChars))); } /** * Open a new XA connection using the current URL and the specified user * name and password. * * @param user the user name * @param password the password * @return the connection */ public XAConnection getXAConnection(String user, String password) throws SQLException { if (isDebugEnabled()) { debugCode("getXAConnection("+quote(user)+", \"\");"); } int id = getNextId(XA_DATA_SOURCE); return new JdbcXAConnection(factory, id, getJdbcConnection(user, convertToCharArray(password))); } /** * Open a new pooled connection using the current URL, user name and password. * * @return the connection */ public PooledConnection getPooledConnection() throws SQLException { debugCodeCall("getPooledConnection"); return getXAConnection(); } /** * Open a new pooled connection using the current URL and the specified user * name and password. * * @param user the user name * @param password the password * @return the connection */ public PooledConnection getPooledConnection(String user, String password) throws SQLException { if (isDebugEnabled()) { debugCode("getPooledConnection("+quote(user)+", \"\");"); } return getXAConnection(user, password); } /** * [Not supported] Return an object of this class if possible. * * @param iface the class */ //## Java 1.6 ## public <T> T unwrap(Class<T> iface) throws SQLException { throw unsupported("unwrap"); } //*/ /** * [Not supported] Checks if unwrap can return an object of this class. * * @param iface the class */ //## Java 1.6 ## public boolean isWrapperFor(Class<?> iface) throws SQLException { throw unsupported("isWrapperFor"); } //*/ /** * [Not supported] */ /*## Java 1.7 ## public Logger getParentLogger() { return null; } //*/ /** * INTERNAL */ public String toString() { return getTraceObjectName() + ": url=" + url + " user=" + userName; } }