Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2020 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

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

import java.sql.ResultSet;

import java.sql.Statement;

import org.apache.commons.lang3.StringUtils;

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

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

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

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

public class ExecuteOperation extends AbstractConnectionOperation {

    private final String sql;

    public ExecuteOperation(

            JDBCHelper helper,

            String sql

        ) {

        super(helper);

        this.sql = sql;

    }

    @Override

    public final Object perform(JDBCConnection conn) throws DataException {

        return execute(sql);

    }

    public Object execute(String sql) {

        Statement st = null;

        ResultSet rs = null;

        try {

            st = this.getConnection().createStatement(sql);

            String mode;

            try {

                String sqltrim = sql.trim();

                mode = StringUtils.left(sqltrim, sqltrim.indexOf(' ')).toLowerCase();

            } catch (Exception e) {

                mode = "unknown";

            }

            switch(mode){

                case "select":

                    rs = JDBCUtils.executeQuery(st, sql);

                    break;

                case "update":

                    JDBCUtils.executeUpdate(st, sql);

                    break;

                default:

                    JDBCUtils.execute(st, sql);

                    break;

            }

            if(rs == null) {

                JDBCUtils.closeQuietly(st);

                return null;

            }

            if(rs.getMetaData().getColumnCount() > 1) {

                return rs;

            }

            if(!rs.next()){

                rs.close();

                JDBCUtils.closeQuietly(st);

                return null;

            }

            if(rs.isFirst() && rs.isLast()) {

                Object res = rs.getObject(1);

                rs.close();

                JDBCUtils.closeQuietly(st);

                return res;

            }

            rs.first();

            return rs;

        } catch (Exception ex) {

            JDBCUtils.closeQuietly(rs);

            JDBCUtils.closeQuietly(st);

            throw new RuntimeException("Can't execute query ["+sql+"].",ex);

        }

    }

    protected void closeConnection(Object result) throws Exception {

        if (!(result instanceof ResultSet)) {

            this.getConnection().close();

            conn = null;

        }

    }

}