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.operation;
23  
24  import org.slf4j.Logger;
25  import org.slf4j.LoggerFactory;
26  
27  import org.dbunit.DatabaseUnitException;
28  import org.dbunit.database.IDatabaseConnection;
29  import org.dbunit.dataset.IDataSet;
30  
31  import java.sql.Connection;
32  import java.sql.SQLException;
33  
34  /**
35   * Decorates an operation and executes within the context of a transaction.
36   *
37   * @author Manuel Laflamme
38   * @version $Revision$
39   * @since Feb 21, 2002
40   */
41  public class TransactionOperation extends DatabaseOperation
42  {
43  
44      /**
45       * Logger for this class
46       */
47      private static final Logger logger = LoggerFactory.getLogger(TransactionOperation.class);
48  
49      private final DatabaseOperation _operation;
50  
51      /**
52       * Creates a TransactionOperation that decorates the specified operation.
53       */
54      public TransactionOperation(DatabaseOperation operation)
55      {
56          _operation = operation;
57      }
58  
59      ////////////////////////////////////////////////////////////////////////////
60      // DatabaseOperation class
61  
62      public void execute(IDatabaseConnection connection, IDataSet dataSet)
63              throws DatabaseUnitException, SQLException
64      {
65          logger.debug("execute(connection={}, dataSet={}) - start", connection, dataSet);
66  
67          IDatabaseConnection databaseConnection = connection;
68          Connection jdbcConnection = databaseConnection.getConnection();
69  
70          if (jdbcConnection.getAutoCommit() == false)
71          {
72              throw new ExclusiveTransactionException();
73          }
74  
75          jdbcConnection.setAutoCommit(false);
76          try
77          {
78              _operation.execute(databaseConnection, dataSet);
79              jdbcConnection.commit();
80          }
81          catch (DatabaseUnitException e)
82          {
83              jdbcConnection.rollback();
84              throw e;
85          }
86          catch (SQLException e)
87          {
88              jdbcConnection.rollback();
89              throw e;
90          }
91          catch (RuntimeException e)
92          {
93              jdbcConnection.rollback();
94              throw e;
95          }
96          finally
97          {
98              jdbcConnection.setAutoCommit(true);
99          }
100     }
101 }