View Javadoc
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  
22  package org.dbunit.database.statement;
23  
24  import org.slf4j.Logger;
25  import org.slf4j.LoggerFactory;
26  
27  import org.dbunit.dataset.ITable;
28  import org.dbunit.dataset.datatype.DataType;
29  import org.dbunit.dataset.datatype.TypeCastException;
30  
31  import java.sql.Connection;
32  import java.sql.SQLException;
33  
34  /**
35   * @author Manuel Laflamme
36   * @version $Revision$
37   * @since Mar 16, 2002
38   */
39  public class SimplePreparedStatement extends AbstractPreparedBatchStatement
40  {
41  
42      /**
43       * Logger for this class
44       */
45      private static final Logger logger = LoggerFactory.getLogger(SimplePreparedStatement.class);
46  
47      private int _index;
48      private int _result;
49  
50      public SimplePreparedStatement(String sql, Connection connection)
51              throws SQLException
52      {
53          super(sql, connection);
54          _index = 0;
55          _result = 0;
56      }
57  
58      ////////////////////////////////////////////////////////////////////////////
59      // IPreparedBatchStatement interface
60  
61      public void addValue(Object value, DataType dataType)
62              throws TypeCastException, SQLException
63      {
64      	logger.debug("addValue(value={}, dataType={}) - start", value, dataType);
65  
66          // Special NULL handling
67          if (value == null || value == ITable.NO_VALUE)
68          {
69              String sqlTypeName = dataType .getSqlTypeName();
70              if (sqlTypeName == null) {
71                  _statement.setNull(++_index, dataType.getSqlType());
72              } else {
73                  _statement.setNull(++_index, dataType.getSqlType(), sqlTypeName);
74              }
75              return;
76          }
77  
78          dataType.setSqlValue(value, ++_index, _statement);
79      }
80  
81      public void addBatch() throws SQLException
82      {
83          logger.debug("addBatch() - start");
84  
85          boolean result = _statement.execute();
86          if (!result)
87          {
88              _result += _statement.getUpdateCount();
89          }
90          _index = 0;
91      }
92  
93      public int executeBatch() throws SQLException
94      {
95          logger.debug("executeBatch() - start");
96  
97          int result = _result;
98          clearBatch();
99          return result;
100     }
101 
102     public void clearBatch() throws SQLException
103     {
104         logger.debug("clearBatch() - start");
105 
106         _index = 0;
107         _result = 0;
108     }
109 
110 }
111 
112 
113 
114 
115 
116