com.gs.obevo.reladomo.ReladomoSchemaConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of obevo-reladomo-util Show documentation
Collection of utility modules that allow Obevo integration with other third-party modules.
There is a newer version: 8.2.1
/**
 * Copyright 2017 Goldman Sachs.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package com.gs.obevo.reladomo;

import java.io.File;

import com.gs.obevo.api.platform.ChangeType;
import com.gs.obevo.api.platform.Platform;
import com.gs.obevo.db.api.factory.DbPlatformConfiguration;
import com.gs.obevo.db.apps.reveng.AquaRevengMain;
import com.gs.obevo.db.apps.reveng.ChangeEntry;
import com.gs.obevo.db.apps.reveng.RevEngDestination;
import com.gs.obevo.db.apps.reveng.RevengWriter;
import com.gs.obevo.util.DAStringUtil;
import com.gs.obevo.util.FileUtilsCobra;
import org.apache.commons.io.FilenameUtils;
import org.eclipse.collections.api.RichIterable;
import org.eclipse.collections.api.block.function.Function;
import org.eclipse.collections.api.list.ListIterable;
import org.eclipse.collections.api.list.MutableList;
import org.eclipse.collections.api.tuple.Pair;
import org.eclipse.collections.impl.factory.Lists;
import org.eclipse.collections.impl.list.fixed.ArrayAdapter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ReladomoSchemaConverter {
    private static final Logger LOG = LoggerFactory.getLogger(ReladomoSchemaConverter.class);

    public void convertDdlsToDaFormat(ReladomoSchemaConverterArgs args) {
        convertDdlsToDaFormat(DbPlatformConfiguration.getInstance().valueOf(args.getPlatform()), args.getInputDir(), args.getOutputDir(), args.getDbSchema(), !args.isDontGenerateBaseline(), args.getExcludeObjects());
    }

    public void convertDdlsToDaFormat(final Platform platform, File inputDir, File outputDir, final String schemaName, boolean generateBaseline, String excludeObjects) {
        if (!inputDir.canRead()) {
            throw new IllegalArgumentException("Cannot read input directory: " + inputDir);
        }
        final MutableList changeEntries = ArrayAdapter.adapt(inputDir.listFiles(AquaRevengMain.VCS_FILE_FILTER)).flatCollect(new Function>() {
            @Override
            public Iterable valueOf(File file) {
                final String objectName = FilenameUtils.getBaseName(file.getName());
                final String fileExtension = FilenameUtils.getExtension(file.getName());

                final Function, ChangeEntry> func;
                if (fileExtension.equalsIgnoreCase("ddl")) {
                    func = new Function, ChangeEntry>() {
                        @Override
                        public ChangeEntry valueOf(Pair sqlIndexPair) {
                            return getTableChange(platform, schemaName, objectName, sqlIndexPair.getOne(), sqlIndexPair.getTwo());
                        }
                    };
                } else if (fileExtension.equalsIgnoreCase("fk")) {
                    func = new Function, ChangeEntry>() {
                        @Override
                        public ChangeEntry valueOf(Pair sqlIndexPair) {
                            return getFkChange(platform, schemaName, objectName, sqlIndexPair.getOne(), sqlIndexPair.getTwo());
                        }
                    };
                } else if (fileExtension.equalsIgnoreCase("idx")) {
                    func = new Function, ChangeEntry>() {
                        @Override
                        public ChangeEntry valueOf(Pair sqlIndexPair) {
                            return getIndexChange(platform, schemaName, objectName, sqlIndexPair.getOne(), sqlIndexPair.getTwo());
                        }
                    };
                } else {
                    LOG.warn("Unexpected file extension {} for file {}, so we are not processing this file. Expecting extensions: .ddl, .fx, .idx", fileExtension, file);
                    // unexpected
                    return Lists.mutable.empty();
                }

                final RichIterable sqls = splitSqlBySemicolon(FileUtilsCobra.readFileToString(file));
                return sqls.asLazy().reject(DAStringUtil.STRING_IS_BLANK).zipWithIndex().collect(func);
            }
        });

        new RevengWriter().write(platform, changeEntries, outputDir, generateBaseline, RevengWriter.defaultShouldOverwritePredicate(), null, null, null, null, excludeObjects);
    }

    /*
     * public for testing
     */
    static ListIterable splitSqlBySemicolon(String sql) {
        return ArrayAdapter.adapt(sql.split(";(\n|$|\r\n)"));
    }

    private ChangeEntry getTableChange(Platform platform, String schemaName, String objectName, String sql, int index) {
        return new ChangeEntry(new RevEngDestination(schemaName, platform.getChangeType(ChangeType.TABLE_STR), objectName, false), sql, "\"init-table\"", null, 0);
    }
    private ChangeEntry getIndexChange(Platform platform, String schemaName, String objectName, String sql, int index) {
        return new ChangeEntry(new RevEngDestination(schemaName, platform.getChangeType(ChangeType.TABLE_STR), objectName, false), sql, "\"index" + index + "\"", "INDEX", index + 1);
    }
    private ChangeEntry getFkChange(Platform platform, String schemaName, String objectName, String sql, int index) {
        return new ChangeEntry(new RevEngDestination(schemaName, platform.getChangeType(ChangeType.TABLE_STR), objectName, false), sql, "\"fk" + index + "\"", "FK", index + 1);
    }
}