org.specrunner.dbms.AbstractBaseTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of specrunner-dbms Show documentation
The newest version!
/*
    SpecRunner - Acceptance Test Driven Development Tool
    Copyright (C) 2011-2014  Thiago Santos

    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, see 
 */
package org.specrunner.dbms;

import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

import org.specrunner.dbms.core.PairingDefault;
import org.specrunner.dbms.listeners.IColumnListener;
import org.specrunner.dbms.listeners.ITableListener;
import org.specrunner.dbms.util.UtilIO;

import schemacrawler.schema.Column;
import schemacrawler.schema.Database;
import schemacrawler.schema.Schema;
import schemacrawler.schema.Table;

/**
 * Perform base operations.
 * 
 * @author Thiago Santos
 */
public abstract class AbstractBaseTool {

    protected String process(String msg, ConnectionInfo old, ConnectionInfo current, String fileTableListeners, String fileColumnListeners) throws Exception {
        ConnectionDatabase pairOld = null;
        ConnectionDatabase pairCurrent = null;
        try {
            pairOld = new ConnectionDatabase(old);
            Database databaseOld = pairOld.database;

            pairCurrent = new ConnectionDatabase(current);
            Database databaseCurrent = pairCurrent.database;

            StringBuilder sb = process(databaseOld, databaseOld.getSchema(old.getSchema()), databaseCurrent, databaseCurrent.getSchema(current.getSchema()), fileTableListeners, fileColumnListeners);
            if (sb.length() > 0) {
                return sb.toString();
            } else {
                return msg;
            }
        } finally {
            if (pairOld != null) {
                pairOld.finalize();
            }
            if (pairCurrent != null) {
                pairCurrent.finalize();
            }
        }
    }

    protected StringBuilder process(Database database1, Schema schema1, Database database2, Schema schema2, String fileTableListeners, String fileColumnListeners) {
        Iterable> tables = new PairingDefault().pair(children(database1, schema1), children(database2, schema2), comparatorTable());
        Iterator> iterTables = tables.iterator();
        StringBuilder report = new StringBuilder();
        while (iterTables.hasNext()) {
            StringBuilder sbTable = new StringBuilder();
            boolean showTable = false;
            Pair table = iterTables.next();
            for (ITableListener lis : getTableListeners(fileTableListeners)) {
                IPart p = lis.process(table);
                if (p.hasData()) {
                    showTable = showTable || p.isOptional();
                    sbTable.append(p.getData());
                }
            }
            Iterable> columns = new PairingDefault().pair(children(schema1, table.getOld()), children(schema2, table.getCurrent()), comparatorColumn());
            Iterator> iterColumns = columns.iterator();
            StringBuilder sbColumns = new StringBuilder();
            boolean showColumns = false;
            while (iterColumns.hasNext()) {
                StringBuilder sbColumn = new StringBuilder();
                boolean showColumn = false;
                Pair column = iterColumns.next();
                for (IColumnListener lis : getColumnListeners(fileColumnListeners)) {
                    IPart p = lis.process(column);
                    if (p.hasData()) {
                        showColumn = showColumn || p.isOptional();
                        sbColumn.append(p.getData());
                    }
                }
                if (showColumn) {
                    sbColumns.append(sbColumn);
                    showColumns = true;
                }
            }
            if (showTable || showColumns) {
                report.append('\n');
                report.append(sbTable);
            }
            if (showColumns) {
                report.append(sbColumns);
            }
        }
        return report;
    }

    protected List getTableListeners(String file) {
        return UtilIO.load(ITableListener.class, file);
    }

    protected List getColumnListeners(String file) {
        return UtilIO.load(IColumnListener.class, file);
    }

    protected List children(Database database, Schema schema) {
        if (database == null || schema == null) {
            return new LinkedList();
        }
        List
 tables = new LinkedList(database.getTables(schema));
        Collections.sort(tables, comparatorTable());
        return tables;
    }

    protected Comparator comparatorTable() {
        return new Comparator() {
            @Override
            public int compare(Table o1, Table o2) {
                return o1 == null ? (o2 == null ? 0 : 1) : (o2 == null ? -1 : o1.getName().compareToIgnoreCase(o2.getName()));
            }
        };
    }

    protected List children(Schema schema, Table table) {
        if (schema == null || table == null) {
            return new LinkedList();
        }
        List columns = table.getColumns();
        Collections.sort(columns, comparatorColumn());
        return columns;
    }

    protected Comparator comparatorColumn() {
        return new Comparator() {
            @Override
            public int compare(Column o1, Column o2) {
                return o1 == null ? (o2 == null ? 0 : 1) : (o2 == null ? -1 : o1.getName().compareToIgnoreCase(o2.getName()));
            }
        };
    }
}    

    







    © 2015 - 2025 Weber Informatics LLC | Privacy Policy