Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.hibernate.cfg.reveng.OverrideBinder Maven / Gradle / Ivy
package org.hibernate.cfg.reveng;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import org.apache.commons.collections4.MultiValuedMap;
import org.apache.commons.collections4.multimap.HashSetValuedHashMap;
import org.hibernate.MappingException;
import org.hibernate.cfg.reveng.utils.JdbcToHibernateTypeHelper;
import org.hibernate.cfg.reveng.utils.MetaAttributeHelper;
import org.hibernate.cfg.reveng.utils.MetaAttributeHelper.SimpleMetaAttribute;
import org.hibernate.cfg.reveng.utils.RevengUtils;
import org.hibernate.internal.util.StringHelper;
import org.hibernate.mapping.Column;
import org.hibernate.mapping.ForeignKey;
import org.hibernate.mapping.Table;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
public class OverrideBinder {
public static void bindRoot(OverrideRepository repository, Document doc) {
Element rootElement = doc.getDocumentElement();
bindSchemaSelections(getChildElements(rootElement, "schema-selection"), repository);
bindTypeMappings(getChildElements(rootElement, "type-mapping"), repository);
bindTableFilters(getChildElements(rootElement, "table-filter"), repository);
bindTables(getChildElements(rootElement, "table"), repository);
}
private static void bindSchemaSelections(
ArrayList schemaSelections,
OverrideRepository repository) {
for (Element schemaSelection : schemaSelections) {
bindSchemaSelection(schemaSelection, repository);
}
}
private static void bindTypeMappings(
ArrayList typeMappings,
OverrideRepository repository) {
if (typeMappings.size() > 0) {
bindTypeMapping(typeMappings.get(0), repository);
}
}
private static void bindTableFilters(
ArrayList tableFilters,
OverrideRepository repository) {
for (Element element : tableFilters) {
TableFilter tableFilter = new TableFilter();
tableFilter.setMatchCatalog(getAttribute(element, "match-catalog"));
tableFilter.setMatchSchema(getAttribute(element, "match-schema"));
tableFilter.setMatchName(getAttribute(element, "match-name"));
tableFilter.setExclude(Boolean.valueOf(getAttribute(element, "exclude")));
tableFilter.setPackage(getAttribute(element, "package"));
MultiValuedMap map =
MetaAttributeHelper.loadAndMergeMetaMap(
element,
new HashSetValuedHashMap());
if (map != null && !map.isEmpty()) {
tableFilter.setMetaAttributes(map);
} else {
tableFilter.setMetaAttributes(null);
}
repository.addTableFilter(tableFilter);
}
}
private static void bindTables(
ArrayList tables,
OverrideRepository repository) {
for (Element element : tables) {
Table table = new Table();
table.setCatalog(getAttribute(element, "catalog"));
table.setSchema(getAttribute(element, "schema"));
table.setName(getAttribute(element, "name"));
ArrayList primaryKeys = getChildElements(element, "primary-key");
if (primaryKeys.size() > 0) {
bindPrimaryKey(primaryKeys.get(0), table, repository);
}
bindColumns(getChildElements(element, "column"), table, repository);
bindForeignKeys(getChildElements(element, "foreign-key"), table, repository);
bindMetaAttributes(element, table, repository);
repository.addTable(table, getAttribute(element, "class"));
}
}
private static void bindPrimaryKey(
Element element,
Table table,
OverrideRepository repository) {
String propertyName = getAttribute(element, "property");
String compositeIdName = getAttribute(element, "id-class");
ArrayList generators = getChildElements(element, "generator");
if (generators.size() > 0) {
Element generator = generators.get(0);
String identifierClass = getAttribute(generator, "class");
Properties params = new Properties();
ArrayList parameterList = getChildElements(generator, "param");
for (int i = 0; i < parameterList.size(); i++) {
Element parameter = parameterList.get(i);
params.setProperty(getAttribute(parameter, "name"), parameter.getTextContent());
}
repository.addTableIdentifierStrategy(table, identifierClass, params);
}
List boundColumnNames = bindColumns(getChildElements(element, "key-column"), table, repository);
repository.addPrimaryKeyNamesForTable(table, boundColumnNames, propertyName, compositeIdName);
}
private static List bindColumns(
ArrayList columns,
Table table,
OverrideRepository repository) {
List columnNames = new ArrayList();
for (Element element : columns) {
Column column = new Column();
column.setName(getAttribute(element, "name"));
String attributeValue = getAttribute(element, "jdbc-type");
if (StringHelper.isNotEmpty(attributeValue)) {
column.setSqlTypeCode(Integer.valueOf(JdbcToHibernateTypeHelper.getJDBCType(attributeValue)));
}
TableIdentifier tableIdentifier = TableIdentifier.create(table);
if (table.getColumn(column) != null) {
throw new MappingException("Column " + column.getName() + " already exists in table " + tableIdentifier );
}
MultiValuedMap map =
MetaAttributeHelper.loadAndMergeMetaMap(
element,
new HashSetValuedHashMap());
if(map!=null && !map.isEmpty()) {
repository.addMetaAttributeInfo( tableIdentifier, column.getName(), map);
}
table.addColumn(column);
columnNames.add(column.getName());
repository.setTypeNameForColumn(
tableIdentifier,
column.getName(),
getAttribute(element, "type"));
repository.setPropertyNameForColumn(
tableIdentifier,
column.getName(),
getAttribute(element, "property"));
boolean excluded = Boolean.valueOf(element.getAttribute("exclude") );
if(excluded) {
repository.setExcludedColumn(tableIdentifier, column.getName());
}
if (element.hasAttribute("foreign-table")) {
String foreignTableName = element.getAttribute("foreign-table");
List localColumns = new ArrayList();
localColumns.add(column);
List foreignColumns = new ArrayList();
Table foreignTable = new Table();
foreignTable.setName(foreignTableName);
foreignTable.setCatalog(
element.hasAttribute("foreign-catalog") ?
element.getAttribute("foreign-catalog") :
table.getCatalog());
foreignTable.setSchema(
element.hasAttribute("foreign-schema") ?
element.getAttribute("foreign-schema") :
table.getSchema());
if (element.hasAttribute("foreign-column")) {
String foreignColumnName = element.getAttribute("foreign-column");
Column foreignColumn = new Column();
foreignColumn.setName(foreignColumnName);
foreignColumns.add(foreignColumn);
} else {
throw new MappingException("foreign-column is required when foreign-table is specified on " + column);
}
ForeignKey key = table.createForeignKey(
null,
localColumns,
foreignTableName,
null,
foreignColumns);
key.setReferencedTable(foreignTable); // only possible if foreignColumns is explicitly specified (workaround on aligncolumns)
}
}
return columnNames;
}
private static void bindForeignKeys(
ArrayList foreignKeys,
Table table,
OverrideRepository repository) {
for (Element element : foreignKeys) {
String constraintName = getAttribute(element, "constraint-name");
String foreignTableName = getAttribute(element, "foreign-table");
if (foreignTableName != null) {
Table foreignTable = new Table();
foreignTable.setName(foreignTableName);
foreignTable.setCatalog(
element.hasAttribute("foreign-catalog") ?
element.getAttribute("foreign-catalog") :
table.getCatalog());
foreignTable.setSchema(
element.hasAttribute("foreign-schema") ?
element.getAttribute("foreign-schema") :
table.getSchema());
List localColumns = new ArrayList();
List foreignColumns = new ArrayList();
ArrayList columnRefs = getChildElements(element, "column-ref");
for (Element columnRef : columnRefs) {
localColumns.add(new Column(columnRef.getAttribute("local-column")));
foreignColumns.add(new Column(columnRef.getAttribute("foreign-column")));
}
ForeignKey key = table.createForeignKey(
constraintName,
localColumns,
foreignTableName,
null,
foreignColumns);
key.setReferencedTable(foreignTable); // only possible if foreignColumns is explicitly specified (workaround on aligncolumns)
}
if (StringHelper.isNotEmpty(constraintName)) {
if (!validateFkAssociations(element)) {
throw new IllegalArgumentException("you can't mix oneToOnes = getChildElements(element, "one-to-one");
Element oneToOne = null;
AssociationInfo associationInfo = null;
if(oneToOnes.size() > 0) {
oneToOne = oneToOnes.get(0);
oneToOneProperty = getAttribute(oneToOne, "property");
excludeOneToOne = Boolean.valueOf(oneToOne.getAttribute("exclude"));
associationInfo = extractAssociationInfo(oneToOne);
}
String inverseOneToOneProperty = null;
Boolean excludeInverseOneToOne = null;
ArrayList inverseOneToOnes = getChildElements(element, "inverse-one-to-one");
Element inverseOneToOne = null;
AssociationInfo inverseAssociationInfo = null;
if(inverseOneToOnes.size() > 0) {
inverseOneToOne = inverseOneToOnes.get(0);
inverseOneToOneProperty = getAttribute(inverseOneToOne, "property");
excludeInverseOneToOne = Boolean.valueOf(inverseOneToOne.getAttribute("exclude"));
inverseAssociationInfo = extractAssociationInfo(inverseOneToOne);
}
// having oneToOne = null and inverseOneToOne != null doesn't make sense
// we cannot have the inverse side without the owning side in this case
if ( (oneToOne!=null) ) {
repository.addForeignKeyInfo(
constraintName,
oneToOneProperty,
excludeOneToOne,
inverseOneToOneProperty,
excludeInverseOneToOne,
associationInfo,
inverseAssociationInfo);
}
}
private static boolean bindManyToOneAndCollection(
Element element,
String constraintName,
OverrideRepository repository) {
String manyToOneProperty = null;
Boolean excludeManyToOne = null;
AssociationInfo associationInfo = null;
AssociationInfo inverseAssociationInfo = null;
ArrayList manyToOnes = getChildElements(element, "many-to-one");
Element manyToOne = null;
if (manyToOnes.size() > 0) {
manyToOne = manyToOnes.get(0);
manyToOneProperty = getAttribute(manyToOne, "property");
excludeManyToOne = Boolean.valueOf(manyToOne.getAttribute("exclude"));
associationInfo = extractAssociationInfo(manyToOne);
}
String collectionProperty = null;
Boolean excludeCollection = null;
ArrayList sets = getChildElements(element, "set");
Element set = null;
if (sets.size() > 0) {
set = sets.get(0);
collectionProperty = getAttribute(set, "property");
excludeCollection = Boolean.valueOf(set.getAttribute("exclude"));
inverseAssociationInfo = extractAssociationInfo(set);
}
if ( (manyToOne!=null) || (set!=null) ) {
repository.addForeignKeyInfo(
constraintName,
manyToOneProperty,
excludeManyToOne,
collectionProperty,
excludeCollection,
associationInfo,
inverseAssociationInfo);
return true;
} else {
return false;
}
}
private static AssociationInfo extractAssociationInfo(Element manyToOne) {
return RevengUtils.createAssociationInfo(
manyToOne.hasAttribute("cascade") ? manyToOne.getAttribute("cascade") : null,
manyToOne.hasAttribute("fetch") ? manyToOne.getAttribute("fetch") : null,
manyToOne.hasAttribute("insert") ?
Boolean.parseBoolean(manyToOne.getAttribute("insert")) : null,
manyToOne.hasAttribute("update") ?
Boolean.parseBoolean(manyToOne.getAttribute("update")) : null);
}
private static boolean validateFkAssociations(Element element){
ArrayList manyToOnes = getChildElements(element, "many-to-one");
ArrayList oneToOnes = getChildElements(element, "one-to-one");
ArrayList sets = getChildElements(element, "set");
ArrayList inverseOneToOnes = getChildElements(element, "inverse-one-to-one");
if (manyToOnes.size() != 0 &&
(oneToOnes.size() != 0 || inverseOneToOnes.size() != 0)) {
return false;
}
if (oneToOnes.size() != 0 && sets.size() != 0) {
return false;
}
if (inverseOneToOnes.size() != 0 && sets.size() != 0) {
return false;
}
return true;
}
private static void bindMetaAttributes(
Element element,
Table table,
OverrideRepository repository) {
MultiValuedMap map =
MetaAttributeHelper.loadAndMergeMetaMap(
element,
new HashSetValuedHashMap());
if(map!=null && !map.isEmpty()) {
repository.addMetaAttributeInfo( table, map);
}
}
private static void bindSchemaSelection(
Element schemaSelectionElement,
OverrideRepository repository) {
repository.addSchemaSelection(
new SchemaSelection() {
@Override
public String getMatchCatalog() {
return getAttribute(schemaSelectionElement, "match-catalog");
}
@Override
public String getMatchSchema() {
return getAttribute(schemaSelectionElement, "match-schema");
}
@Override
public String getMatchTable() {
return getAttribute(schemaSelectionElement, "match-table");
}
});
}
private static void bindTypeMapping(
Element typeMapping,
OverrideRepository repository) {
ArrayList sqlTypes = getChildElements(typeMapping, "sql-type");
for (int i = 0; i < sqlTypes.size(); i++) {
bindSqlType(sqlTypes.get(i), repository);
}
}
private static void bindSqlType(Element sqlType, OverrideRepository repository) {
int jdbcType = JdbcToHibernateTypeHelper.getJDBCType(
getAttribute(sqlType, "jdbc-type"));
SQLTypeMapping sqlTypeMapping = new SQLTypeMapping(jdbcType);
sqlTypeMapping.setHibernateType(getHibernateType(sqlType));
sqlTypeMapping.setLength(getInteger(
getAttribute(sqlType, "length"),
SQLTypeMapping.UNKNOWN_LENGTH));
sqlTypeMapping.setPrecision(getInteger(
getAttribute(sqlType, "precision"),
SQLTypeMapping.UNKNOWN_PRECISION));
sqlTypeMapping.setScale(getInteger(
getAttribute(sqlType, "scale"),
SQLTypeMapping.UNKNOWN_SCALE));
String notNull = getAttribute(sqlType, "not-null");
if (StringHelper.isEmpty(notNull)) {
sqlTypeMapping.setNullable(null);
} else {
sqlTypeMapping.setNullable(notNull.equals("false"));
}
if (StringHelper.isEmpty(sqlTypeMapping.getHibernateType())) {
throw new MappingException(
"No hibernate-type specified for " +
sqlType.getAttribute("jdbc-type") +
" at " +
sqlType.getTagName());
}
repository.addTypeMapping(sqlTypeMapping);
}
private static String getHibernateType(Element element) {
String attributeValue = getAttribute(element, "hibernate-type");
if(StringHelper.isEmpty(attributeValue)) {
ArrayList hibernateTypes = getChildElements(element, "hibernate-type");
if (hibernateTypes.size() > 0) {
Element hibernateType = hibernateTypes.get(0);
if (hibernateType.hasAttribute("name")) {
return hibernateType.getAttribute("name");
}
} else {
return null;
}
}
return attributeValue;
}
private static int getInteger(String string, int defaultValue) {
if(string==null) {
return defaultValue;
}
else {
return Integer.parseInt(string);
}
}
private static ArrayList getChildElements(Element parent, String tagName) {
ArrayList result = new ArrayList();
NodeList nodeList = parent.getChildNodes();
for (int i = 0; i < nodeList.getLength(); i++) {
Node node = nodeList.item(i);
if (node instanceof Element) {
if (tagName.equals(((Element)node).getTagName())) {
result.add((Element)node);
}
}
}
return result;
}
private static String getAttribute(Element element, String attributeName) {
String result = null;
if (element.hasAttribute(attributeName)) {
result = element.getAttribute(attributeName);
}
return result;
}
}
