/*
* Hibernate Search, full-text search for your domain model
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.search.test.integration.jbossjta.infra;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
import java.util.logging.Logger;
import javax.sql.DataSource;
import javax.sql.XAConnection;
import javax.sql.XADataSource;
import com.arjuna.ats.jdbc.TransactionalDriver;
//import com.arjuna.common.util.propertyservice.PropertyManager;
/**
* This class provides a DataSource based approach to
* management of transaction aware database connections.
* <p>
* It's a XADataSource from which they can obtain a XAResource.
* Hence it implements both DataSource and XADataSource.
*
* @author Jonathan Halliday jonathan.halliday@redhat.com
* @author Emmanuel Bernard
* @since 2008-05
*/
class XADataSourceWrapper implements XADataSource, DataSource {
private XADataSource _theXADataSource;
private final TransactionalDriver _theTransactionalDriver = new TransactionalDriver();
private String _name;
private Properties defaultProperties = new Properties();
/**
* Create a wrapper around the provided XADataSource implementation,
* which should be registered in tomcat's global JNDI with the specified name.
* Note: the registration is not done here, it's someone elses problem.
* See TransactionalResourceFactory for example usage.
*
* @param name should be the fully qualifed JNDI name of the XADataSource, in
* tomcat's global JNDI, not a webapp specific JNDI context.
* @param theDataSource
*/
public XADataSourceWrapper(String name, XADataSource theDataSource) {
_theXADataSource = theDataSource;
_name = name;
}
public void setProperty(String key, String value) {
defaultProperties.setProperty( key, value );
}
/**
* Obtain a direct reference to the wrapped object. This is not
* recommended but may be necessary to e.g. call vendor specific methods.
*
* @return an XA datasource
*/
public XADataSource getUnwrappedXADataSource() {
return _theXADataSource;
}
///////////////////////
// Implementation of the DataSource API is done by reusing the arjuna
// TransactionalDriver. Its already got all the smarts for checking tx
// context, enlisting resources etc so we just delegate to it.
// All we need is some fudging to make the JNDI name stuff behave.
/**
* Obtain a connection to the database.
* Note: Pooling behaviour depends on the vendor's underlying XADataSource implementation.
*
* @return
*
* @throws SQLException
*/
@Override
public Connection getConnection() throws SQLException {
String url = TransactionalDriver.arjunaDriver + _name;
// although we are not setting any properties, the driver will barf if we pass 'null'.
Properties properties = new Properties( defaultProperties );
return getTransactionalConnection( url, properties );
}
/**
* Obtain a connection to the database using the supplied authentication credentials.
*
* @param username
* @param password
*
* @return
*
* @throws SQLException
*/
@Override
public Connection getConnection(String username, String password) throws SQLException {
String url = TransactionalDriver.arjunaDriver + _name;
Properties properties = new Properties( defaultProperties );
properties.setProperty( TransactionalDriver.userName, username );
properties.setProperty( TransactionalDriver.password, password );
return getTransactionalConnection( url, properties );
}
/*
* This is where most of the tomcat specific weirdness resides. You probably
* want to subclass and override this method for reuse in env other than tomcat.
*/
protected Connection getTransactionalConnection(String url, Properties properties) throws SQLException {
/*
// For ref, the url the TransactionalDriver expects is the arjuna driver's
// special prefix followed by a JNDI name.
// via ConnectionImple the IndirectRecoverableConnection.createDataSource method
// attempts to look it up in JNDI. There are two problems with this:
// First problem,
// it always calls InitialContext(env), never InitalContext().
// This we work around by copying into the arjuna config, the system
// properties it needs to populate the env:
// caution: ensure the tx lifecycle listener is configured in tomcat or there will be a
// possible race here, as recovery needs these properties too and may start first
jdbcPropertyManager.propertyManager.setProperty("Context.INITIAL_CONTEXT_FACTORY", System.getProperty(Context.INITIAL_CONTEXT_FACTORY));
jdbcPropertyManager.propertyManager.setProperty("Context.URL_PKG_PREFIXES", System.getProperty(Context.URL_PKG_PREFIXES));
*/
Connection connection;
connection = _theTransactionalDriver.connect( url, properties );
return connection;
}
///////////////////////
// Implementation of XADataSource API is just a straightforward wrap/delegate.
// Note that some of these methods also appear in the DataSource API.
// We don't really care, it's the underlying implementations problem
// to disambiguate them if required.
@Override
public boolean isWrapperFor(Class<?> iface) throws SQLException {
return iface.isAssignableFrom( XADataSource.class );
}
@Override
public <T> T unwrap(Class<T> iface) throws SQLException {
if ( isWrapperFor( iface ) ) {
return (T) getUnwrappedXADataSource();
}
else {
throw new SQLException( "Not a wrapper for " + iface.getCanonicalName() );
}
}
@Override
public XAConnection getXAConnection() throws SQLException {
return _theXADataSource.getXAConnection();
}
@Override
public XAConnection getXAConnection(String user, String password) throws SQLException {
return _theXADataSource.getXAConnection( user, password );
}
@Override
public PrintWriter getLogWriter() throws SQLException {
return _theXADataSource.getLogWriter();
}
@Override
public void setLogWriter(PrintWriter out) throws SQLException {
_theXADataSource.setLogWriter( out );
}
@Override
public void setLoginTimeout(int seconds) throws SQLException {
_theXADataSource.setLoginTimeout( seconds );
}
@Override
public int getLoginTimeout() throws SQLException {
return _theXADataSource.getLoginTimeout();
}
@Override
public Logger getParentLogger() {
// getParentLogger() was introduced in Java 7:
// don't use @Override nor invoke super.getParenLogger() or it breaks on Java 6
// Method must be defined to compile on Java 7.
return null;
}
}