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.SQLException;
32
33 /**
34 * Decorates an operation and close the database connection after executing it.
35 *
36 * @author Manuel Laflamme
37 * @version $Revision$
38 * @since Mar 6, 2002
39 */
40 public class CloseConnectionOperation extends DatabaseOperation
41 {
42
43 /**
44 * Logger for this class
45 */
46 private static final Logger logger = LoggerFactory.getLogger(CloseConnectionOperation.class);
47
48 private final DatabaseOperation _operation;
49
50 /**
51 * Creates a CloseConnectionOperation object that decorates the specified
52 * operation.
53 */
54 public CloseConnectionOperation(DatabaseOperation operation)
55 {
56 _operation = operation;
57 }
58
59 ////////////////////////////////////////////////////////////////////////////
60 // DatabaseOperation class
61
62 public void execute(IDatabaseConnection connection,
63 IDataSet dataSet) throws DatabaseUnitException, SQLException
64 {
65 logger.debug("execute(connection={}, dataSet={}) - start", connection, dataSet);
66
67 try
68 {
69 _operation.execute(connection, dataSet);
70 }
71 finally
72 {
73 connection.close();
74 }
75 }
76 }