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 }