org.eclipse.jetty.jndi.DataSourceCloser Maven / Gradle / Ivy
package org.eclipse.jetty.jndi;
import java.lang.reflect.Method;
import java.sql.Statement;
import javax.sql.DataSource;
import org.eclipse.jetty.util.component.AggregateLifeCycle;
import org.eclipse.jetty.util.component.Destroyable;
import org.eclipse.jetty.util.log.Log;
import org.eclipse.jetty.util.log.Logger;
/**
* Close a DataSource.
* Some {@link DataSource}'s need to be close (eg. Atomikos). This bean is a {@link Destroyable} and
* may be added to any {@link AggregateLifeCycle} so that {@link #destroy()}
* will be called. The {@link #destroy()} method calls any no-arg method called "close" on the passed DataSource.
*
*/
public class DataSourceCloser implements Destroyable
{
private static final Logger LOG = Log.getLogger(DataSourceCloser.class);
final DataSource _datasource;
final String _shutdown;
public DataSourceCloser(DataSource datasource)
{
if (datasource==null)
throw new IllegalArgumentException();
_datasource=datasource;
_shutdown=null;
}
public DataSourceCloser(DataSource datasource,String shutdownSQL)
{
if (datasource==null)
throw new IllegalArgumentException();
_datasource=datasource;
_shutdown=shutdownSQL;
}
public void destroy()
{
try
{
if (_shutdown!=null)
{
LOG.info("Shutdown datasource {}",_datasource);
Statement stmt = _datasource.getConnection().createStatement();
stmt.executeUpdate(_shutdown);
stmt.close();
}
}
catch (Exception e)
{
LOG.warn(e);
}
try
{
Method close = _datasource.getClass().getMethod("close", new Class[]{});
LOG.info("Close datasource {}",_datasource);
close.invoke(_datasource, new Object[]{});
}
catch (Exception e)
{
LOG.warn(e);
}
}
}