Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal.operations;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.Iterator;

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

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

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

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

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.PerformChangesOperation;

import org.gvsig.postgresql.dal.PostgreSQLHelper;

public class PostgreSQLPerformChangesOperation extends PerformChangesOperation {

    public PostgreSQLPerformChangesOperation(JDBCHelper helper) {

        this(helper, null, null, null, null, null, null, null, null);

    }

    public PostgreSQLPerformChangesOperation(JDBCHelper helper,

            String dbName,

            String schemaName,

            String tableName,

            FeatureType featureType,

            Iterator<FeatureReferenceProviderServices> deleteds,

            Iterator<FeatureProvider> inserteds,

            Iterator<FeatureProvider> updateds,

            Iterator<FeatureStoreProvider.FeatureTypeChanged> featureTypesChanged) {

        super(helper);

        this.dbName = dbName;

        this.deleteds = deleteds;

        this.inserteds = inserteds;

        this.updateds = updateds;

        this.schemaName = schemaName;

        this.tableName = tableName;

        this.featureType = featureType;

        this.featureTypesChanged = featureTypesChanged;

    }

    private PostgreSQLHelper getHelper() {

        return (PostgreSQLHelper) this.helper;

    }

    @Override

    public void performInserts(Connection conn,

            String database,

            String schema,

            String table,

            FeatureType type,

            Iterator<FeatureProvider> inserteds) throws DataException {

        JDBCSQLBuilderBase sqlbuilder = buildInsertSQL(database, schema, table, type);

        PreparedStatement st;

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

        try {

            st = conn.prepareStatement(sql);

            while (inserteds.hasNext()) {

                FeatureProvider feature = inserteds.next();

                getHelper().setPreparedStatementParameters(st, sqlbuilder, type, feature);

                if (JDBCUtils.executeUpdate(st,sql) == 0) {

                    throw new JDBCExecuteSQLException(

                            sqlbuilder.insert().toString(),

                            null

                    );

                }

            }

        } catch (JDBCExecuteSQLException ex) {

            throw ex;

        } catch (Exception ex) {

            throw new JDBCExecuteSQLException(sql,ex);

        }

    }

    @Override

    public void performUpdates(Connection conn,

            String database,

            String schema,

            String table,

            FeatureType type,

            Iterator<FeatureProvider> updateds) throws DataException {

        JDBCSQLBuilderBase sqlbuilder = buildUpdateSQL(database, schema, table, type);

        PreparedStatement st = null;

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

        try {

            st = conn.prepareStatement(sql);

            while (updateds.hasNext()) {

                FeatureProvider feature = (FeatureProvider) updateds.next();

                getHelper().setPreparedStatementParameters(st, sqlbuilder, type, feature);

                if (JDBCUtils.executeUpdate(st,sql) == 0) {

                    throw new JDBCUpdateWithoutChangesException(sql,null);

                }

            }

        } catch (SQLException e) {

            throw new JDBCSQLException(e);

        } finally {

            JDBCUtils.closeQuietly(st);

        }

    }

    @Override

    public void performUpdateTable(Connection conn,

            String database,

            String schema,

            String table,

            FeatureType original,

            FeatureType target) throws DataException {

        SQLBuilderBase sqlbuilder = buildUpdateTableSQL(database, schema, table, original, target);

        Statement st = null;

        try {

            st = conn.createStatement();

            for (String sql : sqlbuilder.alter_table().toStrings()) {

                JDBCUtils.execute(st, sql);

            }

        } catch (SQLException e) {

            throw new JDBCSQLException(e);

        } finally {

            JDBCUtils.closeQuietly(st);

        }

    }

}