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.Statement;

import org.gvsig.expressionevaluator.Expression;

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;

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

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

public class UpdatePassThroughOperation extends AbstractConnectionOperation {

    private final Object[] parameters;

    private final Expression filter;

    private final TableReference table;

    public UpdatePassThroughOperation(

            JDBCHelper helper,

            TableReference table,

            Object[] parameters,

            Expression filter

        ) {

        super(helper);

        this.table = table;

        this.parameters = parameters;

        this.filter = filter;

    }

    @Override

    public final Object perform(JDBCConnection conn) throws DataException {

        return execute(this.table, this.parameters, this.filter);

    }

    public Object execute(

        TableReference table,

        Object[] parameters,

        Expression filter

    ) {

        Statement st = null;

        String sql = getSQL(table, parameters, filter);

        try {

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

            JDBCUtils.executeUpdate(st, sql);

            JDBCUtils.closeQuietly(st);

            return null;

        } catch (Exception ex) {

            JDBCUtils.closeQuietly(st);

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

        }

    }

    public String getSQL(

            TableReference table,

            Object[] parameters,

            Expression filter) {

        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();

        sqlbuilder.update().table()

            .database(this.table.getDatabase())

            .schema(this.table.getSchema())

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

        for (int i = 0; i < parameters.length-1; i+=2){

            String field = (String) parameters[i];

            sqlbuilder.update().column().name(field).with_value(sqlbuilder.expression().constant(parameters[i+1]));

        }

        if(filter != null ){

            sqlbuilder.update().where().set(filter.getCode().toValue());

        }

        String sql = sqlbuilder.update().toString();

        return sql;

    }

}