/*
*
* The DbUnit Database Testing Framework
* Copyright (C)2002-2004, DbUnit.org
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library 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. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
package org.dbunit.database;
import org.dbunit.database.statement.IStatementFactory;
import org.dbunit.dataset.*;
import org.dbunit.dataset.datatype.IDataTypeFactory;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
/**
* @author Manuel Laflamme
* @version $Revision: 421 $
* @since Mar 6, 2002
*/
public abstract class AbstractDatabaseConnection implements IDatabaseConnection
{
private IDataSet _dataSet = null;
private DatabaseConfig _databaseConfig;
public AbstractDatabaseConnection()
{
_databaseConfig = new DatabaseConfig();
}
////////////////////////////////////////////////////////////////////////////
// IDatabaseConnection interface
public IDataSet createDataSet() throws SQLException
{
if (_dataSet == null)
{
_dataSet = new DatabaseDataSet(this);
}
return _dataSet;
}
public IDataSet createDataSet(String[] tableNames) throws SQLException
{
return new FilteredDataSet(tableNames, createDataSet());
}
public ITable createQueryTable(String resultName, String sql)
throws DataSetException, SQLException
{
Statement statement = getConnection().createStatement();
try
{
ResultSet resultSet = statement.executeQuery(sql);
try
{
IDataTypeFactory typeFactory = (IDataTypeFactory)_databaseConfig.getProperty(
DatabaseConfig.PROPERTY_DATATYPE_FACTORY);
ITableMetaData metaData = DatabaseTableMetaData.createMetaData(
resultName, resultSet, typeFactory);
return new CachedResultSetTable(metaData, resultSet);
}
finally
{
resultSet.close();
}
}
finally
{
statement.close();
}
}
public int getRowCount(String tableName) throws SQLException
{
return getRowCount(tableName, null);
}
public int getRowCount(String tableName, String whereClause) throws SQLException
{
StringBuffer sqlBuffer = new StringBuffer(128);
sqlBuffer.append("select count(*) from ");
sqlBuffer.append(tableName);
if (whereClause != null)
{
sqlBuffer.append(" ");
sqlBuffer.append(whereClause);
}
Statement statement = getConnection().createStatement();
try
{
ResultSet resultSet = statement.executeQuery(sqlBuffer.toString());
try
{
resultSet.next();
return resultSet.getInt(1);
}
finally
{
resultSet.close();
}
}
finally
{
statement.close();
}
}
public DatabaseConfig getConfig()
{
return _databaseConfig;
}
public IStatementFactory getStatementFactory()
{
return (IStatementFactory)_databaseConfig.getProperty(
DatabaseConfig.PROPERTY_STATEMENT_FACTORY);
}
}