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 java.sql.Connection;
28  import java.sql.SQLException;
29  import java.util.ArrayList;
30  import java.util.List;
31  
32  /**
33   * @author Manuel Laflamme
34   * @version $Revision$
35   * @since Feb 20, 2002
36   */
37  public class SimpleStatement extends AbstractBatchStatement
38  {
39  
40      /**
41       * Logger for this class
42       */
43      private static final Logger logger = LoggerFactory.getLogger(SimpleStatement.class);
44  
45      private final List _list = new ArrayList();
46  
47      SimpleStatement(Connection connection) throws SQLException
48      {
49          super(connection);
50      }
51  
52      public void addBatch(String sql) throws SQLException
53      {
54          logger.debug("addBatch(sql={}) - start", sql);
55          _list.add(sql);
56      }
57  
58      public int executeBatch() throws SQLException
59      {
60          logger.debug("executeBatch() - start");
61  
62          int result = 0;
63          for (int i = 0; i < _list.size(); i++)
64          {
65              String sql = (String)_list.get(i);
66              if(logger.isDebugEnabled())
67                  logger.debug("DbUnit SQL: " + sql);
68              
69              boolean r = _statement.execute(sql);
70              if(!r)
71              {
72                  result += _statement.getUpdateCount();
73              }
74          }
75          return result;
76      }
77  
78      public void clearBatch() throws SQLException
79      {
80          logger.debug("clearBatch() - start");
81          _list.clear();
82      }
83  }
84  
85  
86  
87  
88