PreparedStatementFactory.java

  1. /*
  2.  *
  3.  * The DbUnit Database Testing Framework
  4.  * Copyright (C)2002-2004, DbUnit.org
  5.  *
  6.  * This library is free software; you can redistribute it and/or
  7.  * modify it under the terms of the GNU Lesser General Public
  8.  * License as published by the Free Software Foundation; either
  9.  * version 2.1 of the License, or (at your option) any later version.
  10.  *
  11.  * This library is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14.  * Lesser General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU Lesser General Public
  17.  * License along with this library; if not, write to the Free Software
  18.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19.  *
  20.  */

  21. package org.dbunit.database.statement;

  22. import org.slf4j.Logger;
  23. import org.slf4j.LoggerFactory;

  24. import org.dbunit.database.DatabaseConfig;
  25. import org.dbunit.database.IDatabaseConnection;

  26. import java.sql.SQLException;

  27. /**
  28.  * @author Manuel Laflamme
  29.  * @version $Revision$
  30.  * @since Mar 20, 2002
  31.  */
  32. public class PreparedStatementFactory extends AbstractStatementFactory
  33. {

  34.     /**
  35.      * Logger for this class
  36.      */
  37.     private static final Logger logger = LoggerFactory.getLogger(PreparedStatementFactory.class);

  38.     public IBatchStatement createBatchStatement(IDatabaseConnection connection)
  39.             throws SQLException
  40.     {

  41.         logger.debug("createBatchStatement(connection={}) - start", connection);

  42.         if (supportBatchStatement(connection))
  43.         {
  44.             return new BatchStatement(connection.getConnection());
  45.         }
  46.         else
  47.         {
  48.             return new SimpleStatement(connection.getConnection());
  49.         }
  50.     }

  51.     public IPreparedBatchStatement createPreparedBatchStatement(String sql,
  52.             IDatabaseConnection connection) throws SQLException
  53.     {
  54.         if (logger.isDebugEnabled())
  55.         {
  56.             logger.debug("createPreparedBatchStatement(sql={}, connection={}) - start", sql, connection);
  57.         }
  58.        
  59.         Integer batchSize = (Integer)connection.getConfig().getProperty(DatabaseConfig.PROPERTY_BATCH_SIZE);

  60.         IPreparedBatchStatement statement = null;
  61.         if (supportBatchStatement(connection))
  62.         {
  63.             statement = new PreparedBatchStatement(sql, connection.getConnection());
  64.         }
  65.         else
  66.         {
  67.             statement = new SimplePreparedStatement(sql, connection.getConnection());
  68.         }
  69.         return new AutomaticPreparedBatchStatement(statement, batchSize.intValue());
  70.     }
  71. }