Dépôt officiel du code source de l'ERP OpenConcerto

/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 2011-2019 OpenConcerto, by ILM Informatique. All rights reserved.

 *

 * The contents of this file are subject to the terms of the GNU General Public License Version 3

 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a

 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific

 * language governing permissions and limitations under the License.

 *

 * When distributing the software, include this License Header Notice in each file.

 */

 package org.openconcerto.sql.changer.convert;

import static java.util.Collections.singletonList;

import org.openconcerto.sql.changer.Changer;

import org.openconcerto.sql.model.ConnectionHandlerNoSetup;

import org.openconcerto.sql.model.DBSystemRoot;

import org.openconcerto.sql.model.SQLDataSource;

import org.openconcerto.sql.model.SQLSystem;

import org.openconcerto.sql.model.SQLTable;

import org.openconcerto.sql.model.graph.Link;

import org.openconcerto.sql.utils.AlterTable;

import org.openconcerto.sql.utils.SQLUtils;

import java.sql.SQLException;

import java.util.EnumSet;

import java.util.Set;

/**

 * Rename the primary key of a table to "ID".

 *

 * @author Sylvain

 */

public class RenamePK extends Changer<SQLTable> {

    public RenamePK(DBSystemRoot b) {

        super(b);

    }

    @Override

    protected EnumSet<SQLSystem> getCompatibleSystems() {

        return EnumSet.of(SQLSystem.MYSQL, SQLSystem.POSTGRESQL);

    }

    protected void changeImpl(SQLTable t) throws SQLException {

        renamePrimary(t, "ID");

    }

    private void renamePrimary(final SQLTable t, final String newName) throws SQLException {

        final SQLSystem system = t.getBase().getServer().getSQLSystem();

        if (t.getKey() != null) {

            final String keyName = t.getKey().getName();

            if (!keyName.equals(newName) && keyName.startsWith("ID")) {

                getStream().println(t);

                if (system == SQLSystem.MYSQL) {

                    final Set<Link> referentLinks = this.getSystemRoot().getGraph().getReferentLinks(t);

                    SQLUtils.executeAtomic(t.getDBSystemRoot().getDataSource(), new ConnectionHandlerNoSetup<Object, SQLException>() {

                        @Override

                        public Object handle(SQLDataSource ds) throws SQLException {

                            for (final Link refLink : referentLinks) {

                                final String dropIndex = new AlterTable(refLink.getSource()).dropForeignConstraint(refLink.getName()).asString();

                                getStream().println(dropIndex);

                                getDS().execute(dropIndex);

                            }

                            final String alter = t.getBase().quote("ALTER TABLE %n CHANGE COLUMN %n %i " + getSyntax().getPrimaryIDDefinitionShort(), t, t.getKey(), newName);

                            getStream().println(alter);

                            getDS().execute(alter);

                            t.fetchFields();

                            for (final Link l : referentLinks) {

                                final String addFK = new AlterTable(l.getSource()).addForeignConstraint(l.getCols(), l.getContextualName(), false, singletonList(newName)).asString();

                                getStream().println(addFK);

                                getDS().execute(addFK);

                            }

                            return null;

                        }

                    });

                } else if (system == SQLSystem.POSTGRESQL) {

                    final String alter = t.getBase().quote("ALTER TABLE %f  RENAME COLUMN %n TO %i", t, t.getKey(), newName);

                    getStream().println(alter);

                    this.getDS().execute(alter);

                } else

                    throw new UnsupportedOperationException("for " + system);

            }

        }

    }

}