/*

  * Copyright (c) 2004 International Decision Systems, Inc.  All Rights Reserved.

  *

  * By using this Software, You acknowledge that the Software is a valuable asset

  * and trade secret of either International Decision Systems, Inc. ("IDSI") or a

  * third party supplier of IDSI and constitutes confidential and proprietary

  * information.

  *

  * NEITHER IDSI NOR ANY AGENT OR PERSON ACTING FOR OR WITH IDSI HAS MADE OR DOES

  * MAKE ANY STATEMENTS, AFFIRMATIONS, REPRESENTATIONS OR WARRANTIES WHATSOEVER

  * TO YOU, WHETHER EXPRESS OR IMPLIED, AS TO THE SOFTWARE, THE QUALITY OR

  * CONDITION OF THE SOFTWARE, OR THE OPERATING CHARACTERISTICS OR RELIABILITY OF

  * THE SOFTWARE, OR ITS SUITABILITY FOR ANY GENERAL OR PARTICULAR PURPOSE, OR AS

  * TO ANY OTHER MATTER WHATSOEVER; ANY AND ALL OTHER WARRANTIES INCLUDING

  * WITHOUT LIMITATION ANY WARRANTIES IMPLIED BY LAW, SUCH AS THE IMPLIED

  * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND TITLE,

  * USE AND NON-INFRINGEMENT; ARE HEREBY EXPRESSLY DISCLAIMED AND EXCLUDED.

 */

 package net.sourceforge.addam.selectexec;

 import org.apache.tools.ant.BuildException;

 import org.apache.tools.ant.taskdefs.JDBCTask;

 import java.sql.*;

 import java.util.ArrayList;

 import java.util.Iterator;

 import java.util.List;

 /**

  * This is a strange task, but it can be quite useful... it executes the given

  * SQL and treats the results (the first column, specifically) as a new set of

  * SQL which it then executes.

  *

  * @author TIM3

  * @since Nov 30, 2004 9:07:45 PM

  */

 public class SelectExecTask extends JDBCTask {

     private String text = "";

     private int batchSize = 1;

     /**

      * sets the number of statements to execute simultaneously

      *

      * @param size

      */

     public void setBatchSize(int size) {

         this.batchSize = size;

     }

     /**

      * adds to the SQL statement

      *

      * @param text

      */

     public void addText(String text) {

         this.text += text;

     }

     public void execute() throws BuildException {

         try {

             Connection conn = this.getConnection();

             List resultSQL = new ArrayList();

             log(text);

             PreparedStatement select = conn.prepareStatement(text);

             ResultSet results = select.executeQuery();

             while (results.next()) {

                 resultSQL.add(results.getString(1));

             }

             Statement exec = conn.createStatement();

             int i = 0;

             for (Iterator it = resultSQL.iterator(); it.hasNext(); i++) {

                 String sql = it.next().toString();

                 if (i % batchSize == 0) {

                     exec.executeBatch();

                     exec.clearBatch();

                 }

                 log(sql);

                 exec.addBatch(sql);

             }

             exec.executeBatch();

             conn.commit();

         } catch (SQLException e) {

             e.printStackTrace();

             throw new BuildException("upgrade failed: " + e, e);

         }

     }

 }