com.shesse.jdbcproxy.HaConnection Maven / Gradle / Ivy
/**
* (c) DICOS GmbH, 2011
*
* $Id$
*/
package com.shesse.jdbcproxy;
import java.lang.reflect.Field;
import java.sql.Array;
import java.sql.Blob;
import java.sql.CallableStatement;
import java.sql.Clob;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.NClob;
import java.sql.PreparedStatement;
import java.sql.SQLClientInfoException;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.SQLWarning;
import java.sql.SQLXML;
import java.sql.Savepoint;
import java.sql.Statement;
import java.sql.Struct;
import java.util.ArrayList;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.Executor;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.h2.engine.Session;
import org.h2.jdbc.JdbcConnection;
import org.h2.jdbcx.JdbcXAConnection;
import org.h2.value.Transfer;
import com.shesse.jdbcproxy.AlternatingConnectionFactory.RegisteredHaConnection;
/**
*
* @author sth
*/
public class HaConnection
implements Connection, RegisteredHaConnection
{
// /////////////////////////////////////////////////////////
// Class Members
// /////////////////////////////////////////////////////////
/** */
private static Logger log = Logger.getLogger(HaConnection.class.getName());
/** */
private AlternatingConnectionFactory connectionFactory;
/** */
private ServerMonitor monitoredBy;
/** */
@SuppressWarnings("unused")
private JdbcXAConnection h2XaConnection;
/** */
private JdbcConnection h2Connection;
/** */
private Session session;
// /////////////////////////////////////////////////////////
// Constructors
// /////////////////////////////////////////////////////////
/**
* @param connectionFactory
* @throws SQLException
*/
public HaConnection(AlternatingConnectionFactory connectionFactory, ServerMonitor monitoredBy,
JdbcXAConnection h2XaConnection)
throws SQLException
{
this.connectionFactory = connectionFactory;
this.monitoredBy = monitoredBy;
this.h2XaConnection = h2XaConnection;
this.h2Connection = (JdbcConnection)h2XaConnection.getConnection();
this.session = h2Connection.getSession();
connectionFactory.register(this);
}
// /////////////////////////////////////////////////////////
// Methods
// /////////////////////////////////////////////////////////
/**
* cleanup if the close was not called explicitly
*
* {@inheritDoc}
*
* @see java.lang.Object#finalize()
*/
protected void finalize()
{
try {
close();
} catch (SQLException x) {
}
}
/**
* @return the h2Connection
*/
public JdbcConnection getH2Connection()
{
return h2Connection;
}
/**
* {@inheritDoc}
*
* @see com.shesse.jdbcproxy.AlternatingConnectionFactory.RegisteredHaConnection#getMonitoredBy()
*/
@Override
public ServerMonitor getMonitoredBy()
{
return monitoredBy;
}
/**
* {@inheritDoc}
*
* @see com.shesse.jdbcproxy.AlternatingConnectionFactory.RegisteredHaConnection#cleanup()
*/
@Override
public void cleanup()
{
forceCloseCommunicationSocket(session);
}
/**
* This does a bit of forbidden magic: we want to close the socket
* support the argument session. However, all the relevant Methods
* are synchronized snd are probably already locked when we run
* in the situation where closeCommunicationSocket is needed.
*
* To overcome this, directly access the private member transferList,
* which is an ArrayList. From there, we can access the
* Transfer objects and their sockets.
*
* @param session
*/
public static void forceCloseCommunicationSocket(Session session)
{
if (session == null) {
log.fine("not closing session because it is null");
return;
}
try {
log.fine("forcing close on session "+session.getClass().getName());
Field transferListField = session.getClass().getDeclaredField("transferList");
transferListField.setAccessible(true);
@SuppressWarnings("unchecked")
ArrayList transferList = (ArrayList)transferListField.get(session);
Transfer[] transfers = transferList.toArray(new Transfer[transferList.size()]);
for (Transfer transfer: transfers) {
try {
log.fine("closing socket for transfer");
transfer.getSocket().close();
} catch (Exception x) {
log.log(Level.FINE, "got exception when forcing close on socket", x);
}
}
} catch (Exception x) {
log.log(Level.FINE, "got exception when forcing close on connection", x);
}
}
/**
* @throws SQLException
* @see java.sql.Connection#clearWarnings()
*/
public void clearWarnings()
throws SQLException
{
h2Connection.clearWarnings();
}
/**
* @throws SQLException
* @see java.sql.Connection#close()
*/
public void close()
throws SQLException
{
connectionFactory.deregister(this);
h2Connection.close();
}
/**
* @throws SQLException
* @see java.sql.Connection#commit()
*/
public void commit()
throws SQLException
{
h2Connection.commit();
}
/**
* @param paramString
* @param paramArrayOfObject
* @return
* @throws SQLException
* @see java.sql.Connection#createArrayOf(java.lang.String,
* java.lang.Object[])
*/
public Array createArrayOf(String paramString, Object[] paramArrayOfObject)
throws SQLException
{
return new HaArray(this, h2Connection.createArrayOf(paramString, paramArrayOfObject));
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#createBlob()
*/
public Blob createBlob()
throws SQLException
{
return new HaBlob(this, h2Connection.createBlob());
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#createClob()
*/
public Clob createClob()
throws SQLException
{
return new HaClob(this, h2Connection.createClob());
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#createNClob()
*/
public NClob createNClob()
throws SQLException
{
return new HaNClob(this, h2Connection.createNClob());
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#createSQLXML()
*/
public SQLXML createSQLXML()
throws SQLException
{
return new HaSQLXML(this, h2Connection.createSQLXML());
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#createStatement()
*/
public Statement createStatement()
throws SQLException
{
return new HaStatement(this, h2Connection.createStatement());
}
/**
* @param resultSetType
* @param resultSetConcurrency
* @param resultSetHoldability
* @return
* @throws SQLException
* @see java.sql.Connection#createStatement(int, int, int)
*/
public Statement createStatement(int resultSetType, int resultSetConcurrency,
int resultSetHoldability)
throws SQLException
{
return new HaStatement(this, h2Connection.createStatement(resultSetType,
resultSetConcurrency, resultSetHoldability));
}
/**
* @param resultSetType
* @param resultSetConcurrency
* @return
* @throws SQLException
* @see java.sql.Connection#createStatement(int, int)
*/
public Statement createStatement(int resultSetType, int resultSetConcurrency)
throws SQLException
{
return new HaStatement(this, h2Connection.createStatement(resultSetType,
resultSetConcurrency));
}
/**
* @param paramString
* @param paramArrayOfObject
* @return
* @throws SQLException
* @see java.sql.Connection#createStruct(java.lang.String,
* java.lang.Object[])
*/
public Struct createStruct(String paramString, Object[] paramArrayOfObject)
throws SQLException
{
return new HaStruct(this, h2Connection.createStruct(paramString, paramArrayOfObject));
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getAutoCommit()
*/
public boolean getAutoCommit()
throws SQLException
{
return h2Connection.getAutoCommit();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getCatalog()
*/
public String getCatalog()
throws SQLException
{
return h2Connection.getCatalog();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getClientInfo()
*/
public Properties getClientInfo()
throws SQLException
{
return h2Connection.getClientInfo();
}
/**
* @param paramString
* @return
* @throws SQLException
* @see java.sql.Connection#getClientInfo(java.lang.String)
*/
public String getClientInfo(String paramString)
throws SQLException
{
return h2Connection.getClientInfo(paramString);
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getHoldability()
*/
public int getHoldability()
throws SQLException
{
return h2Connection.getHoldability();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getMetaData()
*/
public DatabaseMetaData getMetaData()
throws SQLException
{
return new HaDatabaseMetaData(this, h2Connection.getMetaData());
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getTransactionIsolation()
*/
public int getTransactionIsolation()
throws SQLException
{
return h2Connection.getTransactionIsolation();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getTypeMap()
*/
public Map> getTypeMap()
throws SQLException
{
return h2Connection.getTypeMap();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#getWarnings()
*/
public SQLWarning getWarnings()
throws SQLException
{
return h2Connection.getWarnings();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#isClosed()
*/
public boolean isClosed()
throws SQLException
{
return h2Connection.isClosed();
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#isReadOnly()
*/
public boolean isReadOnly()
throws SQLException
{
return h2Connection.isReadOnly();
}
/**
* @param paramInt
* @return
* @throws SQLException
* @see java.sql.Connection#isValid(int)
*/
public boolean isValid(int paramInt)
throws SQLException
{
return h2Connection.isValid(paramInt);
}
/**
* @param paramClass
* @return
* @throws SQLException
* @see java.sql.Wrapper#isWrapperFor(java.lang.Class)
*/
public boolean isWrapperFor(Class paramClass)
throws SQLException
{
return h2Connection.isWrapperFor(paramClass);
}
/**
* @param paramString
* @return
* @throws SQLException
* @see java.sql.Connection#nativeSQL(java.lang.String)
*/
public String nativeSQL(String paramString)
throws SQLException
{
return h2Connection.nativeSQL(paramString);
}
/**
* @param paramString
* @param paramInt1
* @param paramInt2
* @param paramInt3
* @return
* @throws SQLException
* @see java.sql.Connection#prepareCall(java.lang.String, int, int, int)
*/
public CallableStatement prepareCall(String paramString, int paramInt1, int paramInt2,
int paramInt3)
throws SQLException
{
return new HaCallableStatement(this, h2Connection.prepareCall(paramString, paramInt1,
paramInt2, paramInt3));
}
/**
* @param paramString
* @param paramInt1
* @param paramInt2
* @return
* @throws SQLException
* @see java.sql.Connection#prepareCall(java.lang.String, int, int)
*/
public CallableStatement prepareCall(String paramString, int paramInt1, int paramInt2)
throws SQLException
{
return new HaCallableStatement(this, h2Connection.prepareCall(paramString, paramInt1,
paramInt2));
}
/**
* @param paramString
* @return
* @throws SQLException
* @see java.sql.Connection#prepareCall(java.lang.String)
*/
public CallableStatement prepareCall(String paramString)
throws SQLException
{
return new HaCallableStatement(this, h2Connection.prepareCall(paramString));
}
/**
* @param paramString
* @param paramInt1
* @param paramInt2
* @param paramInt3
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String, int, int,
* int)
*/
public PreparedStatement prepareStatement(String paramString, int paramInt1, int paramInt2,
int paramInt3)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString, paramInt1,
paramInt2, paramInt3));
}
/**
* @param paramString
* @param paramInt1
* @param paramInt2
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String, int, int)
*/
public PreparedStatement prepareStatement(String paramString, int paramInt1, int paramInt2)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString, paramInt1,
paramInt2));
}
/**
* @param paramString
* @param paramInt
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String, int)
*/
public PreparedStatement prepareStatement(String paramString, int paramInt)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString, paramInt));
}
/**
* @param paramString
* @param paramArrayOfInt
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String, int[])
*/
public PreparedStatement prepareStatement(String paramString, int[] paramArrayOfInt)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString,
paramArrayOfInt));
}
/**
* @param paramString
* @param paramArrayOfString
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String,
* java.lang.String[])
*/
public PreparedStatement prepareStatement(String paramString, String[] paramArrayOfString)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString,
paramArrayOfString));
}
/**
* @param paramString
* @return
* @throws SQLException
* @see java.sql.Connection#prepareStatement(java.lang.String)
*/
public PreparedStatement prepareStatement(String paramString)
throws SQLException
{
return new HaPreparedStatement(this, h2Connection.prepareStatement(paramString));
}
/**
* @param paramSavepoint
* @throws SQLException
* @see java.sql.Connection#releaseSavepoint(java.sql.Savepoint)
*/
public void releaseSavepoint(Savepoint paramSavepoint)
throws SQLException
{
h2Connection.releaseSavepoint(paramSavepoint);
}
/**
* @throws SQLException
* @see java.sql.Connection#rollback()
*/
public void rollback()
throws SQLException
{
h2Connection.rollback();
}
/**
* @param paramSavepoint
* @throws SQLException
* @see java.sql.Connection#rollback(java.sql.Savepoint)
*/
public void rollback(Savepoint paramSavepoint)
throws SQLException
{
h2Connection.rollback(paramSavepoint);
}
/**
* @param paramBoolean
* @throws SQLException
* @see java.sql.Connection#setAutoCommit(boolean)
*/
public void setAutoCommit(boolean paramBoolean)
throws SQLException
{
h2Connection.setAutoCommit(paramBoolean);
}
/**
* @param paramString
* @throws SQLException
* @see java.sql.Connection#setCatalog(java.lang.String)
*/
public void setCatalog(String paramString)
throws SQLException
{
h2Connection.setCatalog(paramString);
}
/**
* @param paramProperties
* @throws SQLClientInfoException
* @see java.sql.Connection#setClientInfo(java.util.Properties)
*/
public void setClientInfo(Properties paramProperties)
throws SQLClientInfoException
{
h2Connection.setClientInfo(paramProperties);
}
/**
* @param paramString1
* @param paramString2
* @throws SQLClientInfoException
* @see java.sql.Connection#setClientInfo(java.lang.String,
* java.lang.String)
*/
public void setClientInfo(String paramString1, String paramString2)
throws SQLClientInfoException
{
h2Connection.setClientInfo(paramString1, paramString2);
}
/**
* @param paramInt
* @throws SQLException
* @see java.sql.Connection#setHoldability(int)
*/
public void setHoldability(int paramInt)
throws SQLException
{
h2Connection.setHoldability(paramInt);
}
/**
* @param paramBoolean
* @throws SQLException
* @see java.sql.Connection#setReadOnly(boolean)
*/
public void setReadOnly(boolean paramBoolean)
throws SQLException
{
h2Connection.setReadOnly(paramBoolean);
}
/**
* @return
* @throws SQLException
* @see java.sql.Connection#setSavepoint()
*/
public Savepoint setSavepoint()
throws SQLException
{
return h2Connection.setSavepoint();
}
/**
* @param paramString
* @return
* @throws SQLException
* @see java.sql.Connection#setSavepoint(java.lang.String)
*/
public Savepoint setSavepoint(String paramString)
throws SQLException
{
return h2Connection.setSavepoint(paramString);
}
/**
* @param paramInt
* @throws SQLException
* @see java.sql.Connection#setTransactionIsolation(int)
*/
public void setTransactionIsolation(int paramInt)
throws SQLException
{
h2Connection.setTransactionIsolation(paramInt);
}
/**
* @param paramMap
* @throws SQLException
* @see java.sql.Connection#setTypeMap(java.util.Map)
*/
public void setTypeMap(Map> paramMap)
throws SQLException
{
h2Connection.setTypeMap(paramMap);
}
/**
* @param
* @param paramClass
* @return
* @throws SQLException
* @see java.sql.Wrapper#unwrap(java.lang.Class)
*/
public T unwrap(Class paramClass)
throws SQLException
{
return h2Connection.unwrap(paramClass);
}
/**
* {@inheritDoc}
*
* @see java.sql.Connection#setSchema(java.lang.String)
*/
public void setSchema(String schema)
throws SQLException
{
throw new SQLFeatureNotSupportedException("");
}
/**
* {@inheritDoc}
*
* @see java.sql.Connection#getSchema()
*/
public String getSchema()
throws SQLException
{
throw new SQLFeatureNotSupportedException("");
}
/**
* {@inheritDoc}
*
* @see java.sql.Connection#abort(java.util.concurrent.Executor)
*/
public void abort(Executor executor)
throws SQLException
{
throw new SQLFeatureNotSupportedException("");
}
/**
* {@inheritDoc}
*
* @see java.sql.Connection#setNetworkTimeout(java.util.concurrent.Executor,
* int)
*/
public void setNetworkTimeout(Executor executor, int milliseconds)
throws SQLException
{
throw new SQLFeatureNotSupportedException("");
}
/**
* {@inheritDoc}
*
* @see java.sql.Connection#getNetworkTimeout()
*/
public int getNetworkTimeout()
throws SQLException
{
throw new SQLFeatureNotSupportedException("");
}
// /////////////////////////////////////////////////////////
// Inner Classes
// /////////////////////////////////////////////////////////
}