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.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  }