Application: gvSIG desktop

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

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.SQLException;

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

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

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

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

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

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

public class AppendOperation {

    protected Connection connection = null;

    protected final JDBCHelper helper;

    protected final String database;

    protected final String schema;

    protected final String table;

    protected final FeatureType type;

    protected JDBCSQLBuilderBase sqlbuilder = null;

    protected PreparedStatement preparedStatement;

    protected String sql;

    public AppendOperation(

            JDBCHelper helper, 

            String database,

            String schema, 

            String table, 

            FeatureType type

        ) {

        this.helper = helper;

        this.database = database;

        this.schema = schema;

        this.table = table;

        this.type = type;

    }

    public void begin() throws DataException {

        if (this.sqlbuilder != null) {

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

        }

        try {

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

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

            this.sqlbuilder.insert().table().database(this.database).schema(this.schema).name(this.table);

            for (FeatureAttributeDescriptor attr : type) {

                if( attr.isAutomatic() ) {

                    continue;

                }

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

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

                        sqlbuilder.parameter(attr.getName()).as_geometry().srs( 

                                sqlbuilder.parameter(attr.getSRS()) 

                        ) 

                    );

                } else {

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

                        sqlbuilder.parameter(attr.getName()).as_variable()

                    );

                }

            }

            PreparedStatement st;

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

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

        } catch (SQLException ex) {

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

        }

    }

    public void end() {

        JDBCUtils.closeQuietly(this.connection);

        JDBCUtils.closeQuietly(this.preparedStatement);

        this.connection = null;

        this.preparedStatement = null;

        this.sqlbuilder = null;

        this.sql = null;

    }

    public void abort() {

        this.end();

    }

    public void append(FeatureProvider feature) throws DataException {

        try {

            this.connection.setAutoCommit(false);

            try {

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

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

                if ( n > 0) {

                    this.connection.commit();

                    return;

                }              

                this.connection.rollback();

            } catch(Exception ex) {

                try {

                    this.connection.rollback();

                } catch (Exception e1) {

                    throw new JDBCTransactionRollbackException(e1, ex);

                }                

                throw new JDBCExecuteSQLException(this.sql,ex);

            } 

            throw new JDBCExecuteSQLException(this.sql,null);

        } catch (JDBCExecuteSQLException ex) {

            throw ex;

        } catch (Exception ex) {

            throw new JDBCExecuteSQLException(this.sql,ex);

        }

    }

}