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

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

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

import org.gvsig.expressionevaluator.ExpressionUtils;

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

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

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

import org.gvsig.fmap.geom.DataTypes;

public class CreateTableOperation extends AbstractConnectionWritableOperation {

    protected final TableReference table;

    protected FeatureType type;

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

    protected final List<String> additionalSQLs;

    public CreateTableOperation(

            JDBCHelper helper

        ) {

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

    }

    public CreateTableOperation(

            JDBCHelper helper,

            TableReference table,

            FeatureType type,

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

            List<String> additionalSQLs

        ) {

        super(helper);

        this.table = table;

        this.type = type;

        this.userAndPrivileges = userAndPrivileges;

        this.additionalSQLs = additionalSQLs;

    }

    public FeatureType getType() {

        return type;

    }

    @Override

    public final Object perform(JDBCConnection conn) throws DataException {

        return this.performCreateTable(conn);

    }

    protected List<String> buildCreateIndexesSQL(

            TableReference table,

            FeatureType type

        ) {

        ArrayList<String> sqls = new ArrayList<>();

        for (FeatureAttributeDescriptor attr : type) {

            if( attr.isIndexed() && !attr.isPrimaryKey() ) {

                JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();

                SQLBuilder.CreateIndexBuilder createIndex = sqlbuilder.create_index();

                createIndex.setFeatureType(type);

                if( attr.getType()==org.gvsig.fmap.dal.DataTypes.GEOMETRY ) {

                    createIndex.spatial();

                }

                createIndex.if_not_exist();

                createIndex.name("idx_" + table.toString().replace(".", "_") + "_" + attr.getName());

                createIndex.column(attr.getName());

                createIndex.table()

                        .database(this.table.getDatabase())

                        .schema(this.table.getSchema())

                        .name(this.table.getTable()

                );

                if(!attr.allowIndexDuplicateds()){

                    createIndex.unique();

                }

                sqls.addAll(createIndex.toStrings());

            }

        }

        return sqls;

    }

    public List<String> getSQLs() throws DataException {

        JDBCSQLBuilderBase sqlbuilder = this.createSQLBuilder();

        sqlbuilder.create_table().table()

                .database(table.getDatabase())

                .schema(table.getSchema())

                .name(table.getTable());

        for (FeatureAttributeDescriptor attr : type) {

            if( attr.isComputed() ) {

                continue;

            }

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

                sqlbuilder.create_table().add_column(attr);

//                sqlbuilder.create_table().add_geometry_column(attr);

//                        attr.getName(),

//                        attr.getGeomType().getType(),

//                        attr.getGeomType().getSubType(),

//                        attr.getSRS(),

//                        attr.isIndexed(),

//                        attr.allowNull()

//                );

            } else {

                Object defaultValue = attr.getDefaultValue();

                if( defaultValue instanceof CharSequence ) {

                    if( ExpressionUtils.isDynamicText(defaultValue.toString())) {

                        defaultValue = null;

                    }

                }

                sqlbuilder.create_table().add_column(

                        attr.getName(),

                        attr.getType(),

                        attr.getSize(),

                        attr.getPrecision(),

                        attr.getScale(),

                        attr.isPrimaryKey(),

                        attr.isIndexed(),

                        attr.allowNull(),

                        attr.isAutomatic(),

                        defaultValue

                );

            }

        }

        if( userAndPrivileges!=null ) {

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

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

          }

        }

        List<String> sqls;

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

        sqls.addAll(buildCreateIndexesSQL(this.table,type));

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

        if( additionalSQLs!=null ) {

          sqls.addAll(additionalSQLs);

        }

        return sqls;

    }

    public boolean performCreateTable(JDBCConnection conn) throws DataException {

        List<String> sqls = this.getSQLs();

        Statement st = null; String currsql = null;

        try {

            st = conn.createStatement(StringUtils.join(sqls, "; "));

            for (String sql : sqls) {

                currsql = sql; JDBCUtils.execute(st, sql);

            }

        } catch (SQLException ex) {

            throw new JDBCSQLException(ex, currsql);

        } finally {

            JDBCUtils.closeQuietly(st);

        }

        return true;

    }

}