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

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.util.Collections;

import java.util.List;

import org.gvsig.expressionevaluator.GeometryExpressionBuilder;

import org.gvsig.fmap.dal.DataTypes;

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

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.feature.exception.AlreadyEditingException;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

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

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;

import org.gvsig.tools.dispose.Disposable;

import org.gvsig.tools.dispose.DisposeUtils;

import org.slf4j.LoggerFactory;

public class AppendOperation {

    protected static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(AppendOperation.class);

    protected Connection connection = null;

    protected final JDBCHelper helper;

    protected final TableReference table;

    protected final FeatureType type;

    protected JDBCSQLBuilderBase sqlbuilder = null;

    protected GeometryExpressionBuilder expbuilder;

    protected PreparedStatement preparedStatement;

    protected String insertSQL;

    protected int batchCount;

    private final int batchSize;

    private final Disposable[] disposableParameters;

    public AppendOperation(

            JDBCHelper helper, 

            TableReference table, 

            FeatureType type

        ) {

        this.helper = helper;

        this.table = table;

        this.type = type;

        this.batchSize = this.helper.getConnectionParameters().getBatchSize();

        if( this.batchSize>0 ) {

            this.disposableParameters = new Disposable[this.batchSize];

        } else {

            this.disposableParameters = null;

        }

    }

    public void begin() throws DataException {

        if (this.sqlbuilder != null) {

            throw new AlreadyEditingException(this.helper.getSourceId());

        }

        try {

            this.connection = this.helper.getConnectionWritable();

            this.sqlbuilder = this.helper.createSQLBuilder();

            this.expbuilder = this.sqlbuilder.expression();

            this.sqlbuilder.insert().table()

                    .database(this.table.getDatabase())

                    .schema(this.table.getSchema())

                    .name(this.table.getTable()

            );

            for (FeatureAttributeDescriptor attr : type) {

                if( attr.isAutomatic() || attr.isComputed() ) {

                    continue;

                }

                if (attr.getType() == DataTypes.GEOMETRY) {

                        this.sqlbuilder.insert().column().name(attr.getName()).with_value( 

                            expbuilder.parameter(attr.getName()).as_geometry_variable().srs( 

                                    expbuilder.parameter().value(attr.getSRS()) 

                            ) 

                        );

                    } else {

                        this.sqlbuilder.insert().column().name(attr.getName()).with_value(

                        expbuilder.parameter(attr.getName()).as_variable()

                    );

                }

            }

            this.insertSQL = this.sqlbuilder.insert().toString();

            if( this.connection != null ) { // Not in test mode ???

              this.preparedStatement = this.connection.prepareStatement(insertSQL);

              this.connection.setAutoCommit(false);

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

                JDBCUtils.execute(this.connection, sql);

              }

            }

        } catch (SQLException ex) {

            throw new JDBCPreparingSQLException(this.sqlbuilder.toString(),ex);

        }

    }

    protected void clean() {

        if( this.batchCount > 0 ) {

            this.clearBatch();

        }

        JDBCUtils.closeQuietly(this.preparedStatement);

        this.helper.closeConnection(this.connection);

        this.connection = null;

        this.preparedStatement = null;

        this.sqlbuilder = null;

        this.insertSQL = null;        

    }

    public void end() {

        try {

            if( this.connection == null ) {

              return; // In test mode ???

            }

            if( this.batchCount > 0 ) {

                this.executeBatch();

            }

            this.connection.commit();

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

              JDBCUtils.execute(this.connection, sql);

            }

        } catch (SQLException ex) {

            try {

                this.connection.rollback();

            } catch (SQLException ex1) {

            }

            throw new RuntimeException("Can't commit transaction", ex);

        } finally {

            clean();

        }

    }

    public void abort() {        

        try {

            if( this.connection == null ) {

              return; // In test mode ???

            }

            if( this.batchCount > 0 ) {

                this.clearBatch();

            }

            this.connection.rollback();

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

              JDBCUtils.execute(this.connection, sql);

            }

        } catch (SQLException ex) {

        } finally {

          clean();

        }

    }

    public String getSQL() { // For test

      return this.insertSQL;

    }

    public List<String> getPreviousSQLs() {

      return Collections.EMPTY_LIST;

    }

    public List<String> getPostSQLs() {

      return Collections.EMPTY_LIST;

    }    

    public List<Object> getSQLParameters(FeatureProvider feature) {

      return this.sqlbuilder.getParameters(feature);

    }

    @SuppressWarnings("UseSpecificCatch")

    public void append(FeatureProvider feature) throws DataException {

        try {

            if( this.batchSize>0 ) {

                this.addBatch(feature);

                if( this.batchCount >= this.batchSize ) {

                    this.executeBatch();

                }

            } else {

                Disposable theParametersDisposable = null;

                try {

                    theParametersDisposable = this.sqlbuilder.setParameters(this.preparedStatement, feature);

                    int n = JDBCUtils.executeUpdate(this.preparedStatement,this.insertSQL);

                    if( n<1 ) {

                        throw new RuntimeException("Can't insert feature (n="+n+").");

                    }

                } finally {

                    DisposeUtils.disposeQuietly(theParametersDisposable);

                }

            }

        } catch(Exception ex) {

            throw new RuntimeException("Can't insert feature.", ex);

        }

    }

    private void addBatch(FeatureProvider feature) throws SQLException {

        Disposable theParametersDisposable = this.sqlbuilder.setParameters(this.preparedStatement, feature);

        JDBCUtils.addBatch(this.preparedStatement,this.insertSQL);

        this.disposableParameters[this.batchCount++] = theParametersDisposable;

    }

    private void executeBatch() throws SQLException {

        int[] status = JDBCUtils.executeBatch(this.preparedStatement,this.insertSQL);

        this.clearBatch();

        for (int n : status) {

            if( n<1 ) {

                throw new RuntimeException("Can't insert feature (n="+n+").");

            }

        }

    }

    private void clearBatch() {

        try {

            this.preparedStatement.clearParameters();

            this.preparedStatement.clearBatch();

            for (int i = 0; i < this.batchCount && i < this.disposableParameters.length; i++) {

                DisposeUtils.dispose(this.disposableParameters[i]);

                this.disposableParameters[i] = null;

            }

            this.batchCount = 0;

        } catch (SQLException ex) {

            LOGGER.warn("Can't clear batch statement", ex);

        }

    }    

}