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

/*

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

 *

 * Copyright 2011 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.correct;

import org.openconcerto.sql.Configuration;

import org.openconcerto.sql.changer.Changer;

import org.openconcerto.sql.element.SQLElement;

import org.openconcerto.sql.element.SQLElementDirectory;

import org.openconcerto.sql.element.SQLElementLink;

import org.openconcerto.sql.element.SQLElementLink.LinkType;

import org.openconcerto.sql.model.DBSystemRoot;

import org.openconcerto.sql.model.SQLField;

import org.openconcerto.sql.model.SQLRowValues;

import org.openconcerto.sql.model.SQLRowValuesListFetcher;

import org.openconcerto.sql.model.SQLSelect;

import org.openconcerto.sql.model.SQLSelect.ArchiveMode;

import org.openconcerto.sql.model.SQLSystem;

import org.openconcerto.sql.model.SQLTable;

import org.openconcerto.sql.model.Where;

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

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

import org.openconcerto.sql.utils.SQLUtils;

import org.openconcerto.sql.utils.SQLUtils.SQLFactory;

import org.openconcerto.utils.cc.ITransformer;

import java.sql.SQLException;

import java.util.EnumSet;

import java.util.List;

/**

 * Find the private foreign fields for the passed tables and copy the shared private rows. Eg if

 * CPI1 -> OBS1{DES='pb'} and CPI2 -> OBS1{DES='pb'} this clones OBS1 : CPI1 -> OBS1{DES='pb'} and

 * CPI2 -> OBS2{DES='pb'}.

 *

 * @author Sylvain

 */

public class FixSharedPrivate extends Changer<SQLTable> {

    private final SQLElementDirectory dir;

    public FixSharedPrivate(DBSystemRoot b) {

        this(b, null);

    }

    public FixSharedPrivate(DBSystemRoot b, final SQLElementDirectory dir) {

        super(b);

        if (dir == null) {

            if (Configuration.getInstance() == null)

                throw new IllegalStateException("no conf");

            this.dir = Configuration.getInstance().getDirectory();

            if (this.dir == null)

                throw new IllegalStateException("no directory in conf");

        } else {

            this.dir = dir;

        }

        assert this.dir != null;

    }

    public final SQLElementDirectory getDir() {

        return this.dir;

    }

    @Override

    protected EnumSet<SQLSystem> getCompatibleSystems() {

        // When executing deleteReq :

        // - if PRIVATE_REFS is temporary : "Can't reopen table"

        // - if not : "You can't specify target table 'PRIVATE_REFS' for update in FROM clause"

        return EnumSet.complementOf(EnumSet.of(SQLSystem.MYSQL));

    }

    @Override

    protected void changeImpl(final SQLTable t) throws SQLException {

        getStream().print(t);

        final SQLElement elem = this.getDir().getElement(t);

        if (elem == null) {

            getStream().println(" : no element");

            return;

        } else {

            getStream().println("... ");

        }

        for (final SQLElementLink elemLink : elem.getOwnedLinks().getByType(LinkType.COMPOSITION)) {

            // e.g. to either MISSION (i.e. empty path) or MISSION_Q18

            final Path pathToForeign = elemLink.getPath().minusLast();

            // e.g. ID_Q18

            final Step toPrivateStep = elemLink.getPath().getStep(-1);

            // eg Q18

            final SQLElement privateElement = elemLink.getOwned();

            final SQLTable privateTable = privateElement.getTable();

            // SELECT q.ID FROM Ideation_2007.Q18 q

            // JOIN Ideation_2007.MISSION m on m.ID_Q18 = q.ID

            // where q.ID != 1

            // GROUP BY q.ID

            // HAVING count(q.ID) > 1;

            final SQLSelect sel = new SQLSelect();

            sel.setArchivedPolicy(ArchiveMode.BOTH);

            sel.addSelect(privateTable.getKey());

            // The last step of a ElementLink is always foreign, be it a simple foreign key or with

            // a join table. Thus we don't need to go back to the main row to find duplicates.

            assert toPrivateStep.isForeign();

            sel.addJoin("INNER", null, toPrivateStep.reverse(), "m");

            final String req = sel.asString() + " GROUP BY " + privateTable.getKey().getFieldRef() + " HAVING count(" + privateTable.getKey().getFieldRef() + ")>1";

            @SuppressWarnings("unchecked")

            final List<Number> privateIDs = t.getDBSystemRoot().getDataSource().executeCol(req);

            if (privateIDs.size() > 0) {

                getStream().println("\t" + elemLink + " fixing " + privateIDs.size() + " ... ");

                final SQLField archF = t.getArchiveField();

                final SQLField privateArchF = privateTable.getArchiveField();

                if ((archF == null) != (privateArchF == null))

                    throw new IllegalStateException("Incoherent archive field : " + archF + " / " + privateArchF);

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

                    @Override

                    public Object create() throws SQLException {

                        // for each private pointed by more than one parent

                        for (final Number privateID : privateIDs) {

                            final SQLRowValues vals = new SQLRowValues(t);

                            if (archF != null)

                                vals.putNulls(archF.getName());

                            // e.g. ID_OBSERVATION

                            final SQLField ff = toPrivateStep.getSingleField();

                            vals.assurePath(pathToForeign).putNulls(ff.getName());

                            final SQLRowValuesListFetcher fetcher = SQLRowValuesListFetcher.create(vals);

                            fetcher.setSelTransf(new ITransformer<SQLSelect, SQLSelect>() {

                                @Override

                                public SQLSelect transformChecked(SQLSelect fixSel) {

                                    fixSel.setArchivedPolicy(ArchiveMode.BOTH);

                                    fixSel.setWhere(new Where(fixSel.getAlias(ff), "=", privateID));

                                    return fixSel;

                                }

                            });

                            final List<SQLRowValues> tIDs = fetcher.fetch();

                            for (final SQLRowValues tID : tIDs) {

                                // the first one can keep its private

                                final SQLRowValues reallyPrivate;

                                if (tID == tIDs.get(0))

                                    reallyPrivate = new SQLRowValues(privateElement.getTable()).setID(privateID);

                                else

                                    reallyPrivate = privateElement.createCopy(privateID.intValue());

                                // keep archive coherence

                                if (archF != null)

                                    reallyPrivate.put(privateArchF.getName(), tID.getObject(archF.getName()));

                                new SQLRowValues(pathToForeign.getLast()).setID(tID.followPath(pathToForeign).getIDNumber()).put(ff.getName(), reallyPrivate).update();

                            }

                        }

                        return null;

                    }

                });

            }

        }

        getStream().println(t + " done");

    }

}