org.specrunner.tools.dbms.AbstractBaseTool Maven / Gradle / Ivy
/*
SpecRunner - Acceptance Test Driven Development Tool
Copyright (C) 2011-2016 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 tableListeners = getTableListeners(fileTableListeners);
List columnListeners = getColumnListeners(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 : tableListeners) {
IPart p = lis.process(table);
if (p.hasData()) {
showTable = showTable || p.isMandatory();
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 : columnListeners) {
IPart p = lis.process(column);
if (p.hasData()) {
showColumn = showColumn || p.isMandatory();
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(ConfigurationFiles fileTableListeners) {
return load(ITableListener.class, fileTableListeners);
}
protected List getColumnListeners(ConfigurationFiles fileColumnListeners) {
return load(IColumnListener.class, fileColumnListeners);
}
/**
* Load object type from a file. Try first simple File reference, of not
* exists, lookup in classpath.
*
* @param type
* Class type of listed elements in file.
* @param file
* File name to find.
* @return A list of instances.
*/
@SuppressWarnings("unchecked")
protected List load(Class type, ConfigurationFiles files) {
List result = new LinkedList();
for (String file : files) {
InputStream in = null;
BufferedReader br = null;
InputStreamReader fr = null;
try {
File f = new File(file);
if (!f.exists()) {
in = BaseComparator.class.getResourceAsStream(file);
if (in == null) {
throw new RuntimeException("Invalid configuration file (" + type + "): " + file);
}
} else {
in = new FileInputStream(f);
}
fr = new InputStreamReader(in);
br = new BufferedReader(fr);
String input;
while ((input = br.readLine()) != null) {
result.add((T) Class.forName(input.trim()).newInstance());
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
if (fr != null) {
try {
fr.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if (br != null) {
try {
br.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
return result;
}
protected List children(Catalog 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