Application: gvSIG desktop

package org.gvsig.fmap.dal.store.jdbc2.spi.operations;

import java.sql.Connection;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.gvsig.fmap.dal.store.jdbc2.JDBCUtils;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

public class DropTableOperation extends AbstractConnectionWritableOperation {

    private final TableReference table;

    public DropTableOperation(

            JDBCHelper helper

    ) {

        this(helper, null);

    }

    public DropTableOperation(

            JDBCHelper helper,

            TableReference table

    ) {

        super(helper);

        this.table = table;

    }

    @Override

    public final Object perform(Connection conn) throws DataException {

        this.dropTable(conn);

        return true;

    }

    public List<String> getSQLs() {

        JDBCSQLBuilderBase sqlbuilder = this.createSQLBuilder();

        sqlbuilder.drop_table().table()

                .database(this.table.getDatabase())

                .schema(this.table.getSchema())

                .name(this.table.getTable());

      List<String> sqls = sqlbuilder.drop_table().toStrings();

      return sqls;

    }

    public void dropTable(Connection conn) throws DataException {

        Statement st = null;

        try {

            st = conn.createStatement();

            for (String sql : this.getSQLs()) {

                JDBCUtils.execute(st, sql);

            }

        } catch (SQLException ex) {

            throw new JDBCSQLException(ex);

        } finally {

            JDBCUtils.closeQuietly(st);

        }

    }

}