Application: gvSIG desktop

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

import java.sql.Connection;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.List;

import org.apache.commons.lang3.tuple.Pair;

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

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

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

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

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

import org.gvsig.fmap.geom.DataTypes;

public class CreateTableOperation extends AbstractConnectionWritableOperation {

    private final String schemaName;

    private final String tableName;

    private final FeatureType type;

    private final List<Pair<String, SQLBuilder.Privilege>> userAndPrivileges;

    private final List<String> additionalSQLs;

    private final String dbName;

    public CreateTableOperation(

            JDBCHelper helper

        ) {

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

    }

    public CreateTableOperation(

            JDBCHelper helper,

            String dbName,

            String schemaName,

            String tableName,

            FeatureType type,

            List<Pair<String, SQLBuilder.Privilege>> userAndPrivileges,

            List<String> additionalSQLs

        ) {

        super(helper);

        this.dbName = dbName;

        this.schemaName = schemaName;

        this.tableName = tableName;

        this.type = type;

        this.userAndPrivileges = userAndPrivileges;

        this.additionalSQLs = additionalSQLs;

    }

    @Override

    public final Object perform(Connection conn) throws DataException {

        return this.performCreateTable(conn, 

                dbName, schemaName, tableName,

                type, userAndPrivileges, additionalSQLs);

    }

    public boolean performCreateTable(Connection conn,

            String dbName, 

            String schemaName,

            String tableName,

            FeatureType type,

            List<Pair<String, SQLBuilder.Privilege>> rolesAndPrivileges,

            List<String> additionalSQLs) throws DataException {

        JDBCSQLBuilderBase sqlbuilder = this.createSQLBuilder();

        sqlbuilder.create_table().table().database(dbName).schema(schemaName).name(tableName);

        for (FeatureAttributeDescriptor attr : type) {

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

                sqlbuilder.create_table().add_geometry_column(

                        attr.getName(),

                        attr.getGeomType().getType(),

                        attr.getGeomType().getSubType(),

                        attr.getSRS(),

                        attr.isIndexed(),

                        attr.allowNull()

                );

            } else {

                sqlbuilder.create_table().add_column(

                        attr.getName(),

                        attr.getType(),

                        attr.getSize(),

                        attr.getPrecision(),

                        attr.isPrimaryKey(),

                        attr.isIndexed(),

                        attr.allowNull(),

                        attr.isAutomatic(),

                        attr.getDefaultValue()

                );

            }

        }

        for (Pair<String, SQLBuilder.Privilege> roleAndPrivilege : rolesAndPrivileges) {

            sqlbuilder.grant().role(roleAndPrivilege.getLeft()).privilege(roleAndPrivilege.getRight());

        }

        List<String> sqls;

        sqls = sqlbuilder.create_table().toStrings();

        sqls.addAll(sqlbuilder.grant().toStrings());

        sqls.addAll(additionalSQLs);

        Statement st = null;

        try {

            st = conn.createStatement();

            for (String sql : sqls) {

                JDBCUtils.execute(st, sql);

            }

        } catch (SQLException ex) {

            throw new JDBCSQLException(ex);

        } finally {

            JDBCUtils.closeQuietly(st);

        }

        return true;

    }

}