org.specrunner.sql.meta.Table 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 , IMergeable, IResetableExtended {
/**
* The parent schema.
*/
private Schema parent;
/**
* The table alias.
*/
private String alias;
/**
* The table name.
*/
private String name;
/**
* List of column names.
*/
private List columns = new LinkedList();
/**
* Map from alias to columns.
*/
private Map aliasToColumns = new HashMap();
/**
* Map form names to columns.
*/
private Map namesToColumns = new HashMap();
/**
* Map form alias to tables.
*/
private Map aliasToTables = new HashMap();
/**
* Get the schema.
*
* @return The schema.
*/
public Schema getParent() {
return parent;
}
/**
* Set the schema.
*
* @param parent
* The schema.
* @return The table itself.
*/
public Table setParent(Schema parent) {
this.parent = parent;
return this;
}
/**
* Get the alias.
*
* @return The alias.
*/
public String getAlias() {
return alias;
}
/**
* Set table alias.
*
* @param alias
* The alias.
* @return The table itself.
*/
public Table setAlias(String alias) {
this.alias = alias == null ? null : UtilNames.normalize(alias.toLowerCase());
return this;
}
/**
* Get the name.
*
* @return The table name.
*/
public String getName() {
return name;
}
/**
* Set the name.
*
* @param name
* The name.
* @return The table itself.
*/
public Table setName(String name) {
this.name = name == null ? null : name.toUpperCase();
return this;
}
/**
* Get the columns.
*
* @return The columns.
*/
public List getColumns() {
return columns;
}
/**
* Set the columns.
*
* @param columns
* The columns.
* @return The table itself.
*/
public Table setColumns(List columns) {
this.columns = columns;
return this;
}
/**
* Add a column.
*
* @param column
* A column.
* @return The table itself.
* @throws SchemaException
* On repetition errors.
*/
public Table add(Column column) throws SchemaException {
if (namesToColumns.containsKey(column.getName())) {
throw new SchemaException("Column with name '" + column.getName() + "' already exist in table (" + name + "," + alias + "), read before error: " + namesToColumns.keySet());
}
if (aliasToColumns.containsKey(column.getAlias())) {
throw new SchemaException("Column with alias '" + column.getAlias() + "' already exist in table (" + name + "," + alias + "), read before error: " + aliasToColumns.keySet());
}
columns.add(column);
if (column.getTable() != null) {
aliasToTables.put(column.getAlias(), column.getTable());
aliasToColumns.put(createAlias(column.getAlias(), column.getTable()), column);
} else {
aliasToColumns.put(column.getAlias(), column);
}
namesToColumns.put(column.getName(), column);
column.setParent(this);
return this;
}
/**
* Get an alias with prefix and suffix.
*
* @param prefix
* A prefix.
* @param suffix
* A suffix.
* @return A composite alias.
*/
protected String createAlias(String prefix, String suffix) {
return prefix + "(table:" + suffix + ")";
}
/**
* Get a column by its alias.
*
* @param alias
* The alias.
* @return The column.
*/
public Column getAlias(String alias) {
String key = alias == null ? null : UtilNames.normalize(alias);
String table = aliasToTables.get(key);
if (table != null) {
key = createAlias(key, table);
}
Column column = aliasToColumns.get(key);
if (column == null) {
column = getName(key);
}
return column;
}
/**
* Get the column by its name.
*
* @param name
* The name.
* @return The column name.
*/
public Column getName(String name) {
return namesToColumns.get(name == null ? null : name.toUpperCase());
}
/**
* Get the alias mapping.
*
* @return The mapping.
*/
public Map getAliasToColumns() {
return aliasToColumns;
}
/**
* Set the alias mapping.
*
* @param aliasToColumns
* A new mapping.
*/
public void setAliasToColumns(Map aliasToColumns) {
this.aliasToColumns = aliasToColumns;
}
/**
* Get the name mapping.
*
* @return The mapping
*/
public Map getNamesToColumns() {
return namesToColumns;
}
/**
* Set the name mapping.
*
* @param namesToColumns
* A new mapping.
*/
public void setNamesToColumns(Map namesToColumns) {
this.namesToColumns = namesToColumns;
}
/**
* Get the columns which are keys.
*
* @return The table keys.
*/
public List getKeys() {
List result = new LinkedList();
for (Column c : columns) {
if (c.isKey()) {
result.add(c);
}
}
return result;
}
/**
* Get the columns which are references.
*
* @return The table references.
*/
public List getReferences() {
List result = new LinkedList();
for (Column c : columns) {
if (c.isReference()) {
result.add(c);
}
}
return result;
}
/**
* Get the columns which are references.
*
* @return The table references.
*/
public List getVirtual() {
List result = new LinkedList();
for (Column c : columns) {
if (c.isVirtual()) {
result.add(c);
}
}
return result;
}
@Override
public Table copy() throws ReplicableException {
Table copy = new Table().setParent(parent).setName(name).setAlias(alias);
for (Column c : columns) {
try {
copy.add(c.copy());
} catch (SchemaException e) {
throw new ReplicableException(e);
}
}
return copy;
}
@Override
public void merge(Table other) throws MergeableException {
if (other == null) {
return;
}
setName(other.name);
setAlias(other.alias);
for (Column c : other.getColumns()) {
Column old = getName(c.getName());
if (old == null) {
try {
add(c.copy());
} catch (SchemaException e) {
throw new MergeableException(e);
}
} else {
old.merge(c);
}
}
}
@Override
public void initialize(IContext context) {
for (Column c : columns) {
c.initialize(context);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy