org.jvnet.hyperjaxb3.ejb.strategy.mapping.DefaultAssociationMapping Maven / Gradle / Ivy
package org.jvnet.hyperjaxb3.ejb.strategy.mapping;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jvnet.hyperjaxb3.ejb.schemas.customizations.Customizations;
import org.jvnet.hyperjaxb3.xjc.model.CTypeInfoUtils;
import org.jvnet.jaxb.util.CustomizationUtils;
import jakarta.xml.ns.persistence.orm.AssociationOverride;
import jakarta.xml.ns.persistence.orm.ElementCollection;
import jakarta.xml.ns.persistence.orm.EmbeddableAttributes;
import jakarta.xml.ns.persistence.orm.Embedded;
import jakarta.xml.ns.persistence.orm.JoinColumn;
import jakarta.xml.ns.persistence.orm.JoinTable;
import jakarta.xml.ns.persistence.orm.ManyToMany;
import jakarta.xml.ns.persistence.orm.ManyToOne;
import jakarta.xml.ns.persistence.orm.OneToMany;
import jakarta.xml.ns.persistence.orm.OneToOne;
import jakarta.xml.ns.persistence.orm.OrderColumn;
import jakarta.xml.ns.persistence.orm.PrimaryKeyJoinColumn;
import com.sun.tools.xjc.Options;
import com.sun.tools.xjc.model.CClass;
import com.sun.tools.xjc.model.CClassInfo;
import com.sun.tools.xjc.model.CPropertyInfo;
import com.sun.tools.xjc.model.CTypeInfo;
import com.sun.tools.xjc.outline.ClassOutline;
import com.sun.tools.xjc.outline.FieldOutline;
import com.sun.tools.xjc.outline.Outline;
public class DefaultAssociationMapping implements AssociationMapping {
protected Log logger = LogFactory.getLog(getClass());
public Collection getSourceIdFieldsOutline(Mapping context,
FieldOutline fieldOutline) {
final ClassOutline classOutline = fieldOutline.parent();
return getIdFieldsOutline(classOutline);
}
public Collection getTargetIdFieldsOutline(Mapping context,
FieldOutline fieldOutline) {
final CPropertyInfo propertyInfo = fieldOutline.getPropertyInfo();
final Collection types = context.getGetTypes()
.process(context, propertyInfo);
final CTypeInfo type = CTypeInfoUtils.getCommonBaseTypeInfo(types);
assert type != null;
assert type instanceof CClass;
if (type instanceof CClassInfo) {
final ClassOutline targetClassOutline = fieldOutline.parent()
.parent().getClazz((CClassInfo) type);
return getIdFieldsOutline(targetClassOutline);
} else {
logger.error(MessageFormat
.format("Field outline [{0}] references the type [{1}] which is not present in the XJC model "
+ "(it is probably a class reference due to episodic compilation). "
+ "Due to this reason Hyperjaxb3 cannot generate correct identifier column mapping. "
+ "Please customize your association manually. See also issue HJIII-51.",
propertyInfo.getName(true), type.getType()
.fullName()));
return Collections.emptyList();
}
}
private Collection getIdFieldsOutline(
final ClassOutline classOutline) {
final Collection idFieldOutlines = new ArrayList();
ClassOutline current = classOutline;
while (current != null) {
for (FieldOutline idFieldOutline : current.getDeclaredFields()) {
final CPropertyInfo propertyInfo = idFieldOutline
.getPropertyInfo();
if ((CustomizationUtils.containsCustomization(propertyInfo,
Customizations.ID_ELEMENT_NAME) || CustomizationUtils
.containsCustomization(propertyInfo,
Customizations.EMBEDDED_ID_ELEMENT_NAME))
&& !CustomizationUtils.containsCustomization(
propertyInfo,
Customizations.IGNORED_ELEMENT_NAME)) {
idFieldOutlines.add(idFieldOutline);
}
}
current = current.getSuperClass();
}
return idFieldOutlines;
}
// * 1:1
public void createPrimaryKeyJoinColumns(Mapping context,
FieldOutline fieldOutline,
Collection idFieldOutlines,
List primaryKeyJoinColumns) {
final Iterator joinColumnIterator = new ArrayList(
primaryKeyJoinColumns).iterator();
for (FieldOutline idFieldOutline : idFieldOutlines) {
final PrimaryKeyJoinColumn joinColumn;
if (joinColumnIterator.hasNext()) {
joinColumn = joinColumnIterator.next();
} else {
joinColumn = new PrimaryKeyJoinColumn();
primaryKeyJoinColumns.add(joinColumn);
}
createPrimaryKeyJoinColumn(context, fieldOutline, idFieldOutline,
joinColumn);
}
}
protected void createPrimaryKeyJoinColumn(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
PrimaryKeyJoinColumn primaryKeyJoinColumn) {
createPrimaryKeyJoinColumn$Name(context, fieldOutline, idFieldOutline,
primaryKeyJoinColumn);
}
protected void createPrimaryKeyJoinColumn$Name(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
PrimaryKeyJoinColumn primaryKeyJoinColumn) {
if (primaryKeyJoinColumn.getName() == null
|| "##default".equals(primaryKeyJoinColumn.getName())) {
primaryKeyJoinColumn.setName(context.getNaming()
.getJoinColumn$Name(context, fieldOutline, idFieldOutline));
}
}
// * M:1
// * 1:M
// * 1:1
public void createJoinColumns(Mapping context, FieldOutline fieldOutline,
Collection idFieldOutlines,
List joinColumns) {
final Iterator joinColumnIterator = new ArrayList(
joinColumns).iterator();
for (FieldOutline idFieldOutline : idFieldOutlines) {
final JoinColumn joinColumn;
if (joinColumnIterator.hasNext()) {
joinColumn = joinColumnIterator.next();
} else {
joinColumn = new JoinColumn();
joinColumns.add(joinColumn);
}
createJoinColumn(context, fieldOutline, idFieldOutline, joinColumn);
}
}
protected void createJoinColumn(Mapping context, FieldOutline fieldOutline,
FieldOutline idFieldOutline, JoinColumn joinColumn) {
createJoinColumn$Name(context, fieldOutline, idFieldOutline, joinColumn);
}
protected void createJoinColumn$Name(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
JoinColumn joinColumn) {
if (joinColumn.getName() == null
|| "##default".equals(joinColumn.getName())) {
joinColumn.setName(context.getNaming().getJoinColumn$Name(context,
fieldOutline, idFieldOutline));
}
}
// 1:1
// M:1
// 1:M
// M:M
public void createJoinTable(Mapping context, FieldOutline fieldOutline,
Collection sourceIdFieldOutlines,
Collection targetIdFieldOutlines, JoinTable joinTable) {
createJoinTable$Name(context, fieldOutline, joinTable);
createJoinTable$JoinColumn(context, fieldOutline,
sourceIdFieldOutlines, joinTable);
createJoinTable$InverseJoinColumn(context, fieldOutline,
targetIdFieldOutlines, joinTable);
}
protected void createJoinTable$Name(Mapping context,
FieldOutline fieldOutline, JoinTable joinTable) {
if (joinTable.getName() == null
|| "##default".equals(joinTable.getName())) {
joinTable.setName(context.getNaming().getJoinTable$Name(context,
fieldOutline));
}
}
protected void createJoinTable$JoinColumn(Mapping context,
FieldOutline fieldOutline,
Collection idFieldOutlines, JoinTable joinTable) {
final Iterator joinColumnIterator = new ArrayList(
joinTable.getJoinColumn()).iterator();
for (FieldOutline idFieldOutline : idFieldOutlines) {
final JoinColumn joinColumn;
if (joinColumnIterator.hasNext()) {
joinColumn = joinColumnIterator.next();
} else {
joinColumn = new JoinColumn();
joinTable.getJoinColumn().add(joinColumn);
}
createJoinTable$JoinColumn(context, fieldOutline, idFieldOutline,
joinColumn);
}
}
protected void createJoinTable$InverseJoinColumn(Mapping context,
FieldOutline fieldOutline,
Collection idFieldOutlines, JoinTable joinTable) {
final Iterator joinColumnIterator = new ArrayList(
joinTable.getInverseJoinColumn()).iterator();
final Map map = new IdentityHashMap();
for (FieldOutline idFieldOutline : idFieldOutlines) {
final JoinColumn joinColumn;
if (joinColumnIterator.hasNext()) {
joinColumn = joinColumnIterator.next();
} else {
joinColumn = new JoinColumn();
joinTable.getInverseJoinColumn().add(joinColumn);
}
map.put(idFieldOutline, joinColumn);
createJoinTable$InverseJoinColumn(context, fieldOutline,
idFieldOutline, joinColumn);
}
}
protected void createJoinTable$JoinColumn(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
JoinColumn joinColumn) {
createJoinTable$JoinColumn$Name(context, fieldOutline, idFieldOutline,
joinColumn);
// createJoinTable$JoinColumn$ReferencedColumnName(context,
// fieldOutline,
// idFieldOutline, joinColumn);
}
protected void createJoinTable$InverseJoinColumn(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
JoinColumn joinColumn) {
createManyToOne$JoinTable$InverseJoinColumn$Name(context, fieldOutline,
idFieldOutline, joinColumn);
}
protected void createJoinTable$JoinColumn$Name(Mapping context,
FieldOutline fieldOutline, FieldOutline idFieldOutline,
JoinColumn joinColumn) {
if (joinColumn.getName() == null
|| "##default".equals(joinColumn.getName())) {
joinColumn.setName(context.getNaming()
.getJoinTable$JoinColumn$Name(context, fieldOutline,
idFieldOutline));
}
}
protected void createJoinTable$JoinColumn$ReferencedColumnName(
Mapping context, FieldOutline fieldOutline,
FieldOutline idFieldOutline, JoinColumn joinColumn) {
if (joinColumn.getName() == null
|| "##default".equals(joinColumn.getName())) {
joinColumn.setReferencedColumnName(context.getNaming()
.getColumn$Name(context, idFieldOutline));
}
}
public void createOrderColumn(Mapping context, FieldOutline fieldOutline,
final OrderColumn orderColumn) {
createOrderColumn$Name(context, fieldOutline, orderColumn);
}
protected void createOrderColumn$Name(Mapping context,
FieldOutline fieldOutline, final OrderColumn orderColumn) {
if (orderColumn.getName() == null
|| "##default".equals(orderColumn.getName())) {
orderColumn.setName(context.getNaming().getOrderColumn$Name(
context, fieldOutline));
}
}
public void createElementCollection$OrderColumn(Mapping context,
FieldOutline fieldOutline, final OrderColumn orderColumn) {
createElementCollection$OrderColumn$Name(context, fieldOutline,
orderColumn);
}
protected void createElementCollection$OrderColumn$Name(Mapping context,
FieldOutline fieldOutline, final OrderColumn orderColumn) {
if (orderColumn.getName() == null
|| "##default".equals(orderColumn.getName())) {
orderColumn.setName(context.getNaming()
.getElementCollection$OrderColumn$Name(context,
fieldOutline));
}
}
public void createElementCollection$CollectionTable$JoinColumns(
Mapping context, FieldOutline fieldOutline,
Collection idFieldOutlines,
List joinColumns) {
final Iterator joinColumnIterator = new ArrayList(
joinColumns).iterator();
for (FieldOutline idFieldOutline : idFieldOutlines) {
final JoinColumn joinColumn;
if (joinColumnIterator.hasNext()) {
joinColumn = joinColumnIterator.next();
} else {
joinColumn = new JoinColumn();
joinColumns.add(joinColumn);
}
createElementCollection$CollectionTable$JoinColumn(context,
fieldOutline, idFieldOutline, joinColumn);
}
}
protected void createElementCollection$CollectionTable$JoinColumn(
Mapping context, FieldOutline fieldOutline,
FieldOutline idFieldOutline, JoinColumn joinColumn) {
createElementCollection$CollectionTable$JoinColumn$Name(context,
fieldOutline, idFieldOutline, joinColumn);
}
protected void createElementCollection$CollectionTable$JoinColumn$Name(
Mapping context, FieldOutline fieldOutline,
FieldOutline idFieldOutline, JoinColumn joinColumn) {
if (joinColumn.getName() == null
|| "##default".equals(joinColumn.getName())) {
joinColumn.setName(context.getNaming()
.getElementCollection$CollectionTable$JoinColumn$Name(
context, fieldOutline, idFieldOutline));
}
}
protected void createManyToOne$JoinTable$InverseJoinColumn$Name(
Mapping context, FieldOutline fieldOutline,
FieldOutline idFieldOutline, JoinColumn joinColumn) {
if (joinColumn.getName() == null
|| "##default".equals(joinColumn.getName())) {
joinColumn.setName(context.getNaming()
.getJoinTable$InverseJoinColumn$Name(context, fieldOutline,
idFieldOutline));
}
}
public AssociationMapping createEmbeddedAssociationMapping(
FieldOutline fieldOutline) {
return new EmbeddedAssociationMappingWrapper(this, fieldOutline);
}
public void createAssociationOverride(Mapping context,
FieldOutline fieldOutline,
final List associationOverrides) {
final CPropertyInfo propertyInfo = fieldOutline.getPropertyInfo();
final Collection types = context.getGetTypes()
.process(context, propertyInfo);
assert types.size() == 1;
final CTypeInfo typeInfo = types.iterator().next();
assert typeInfo instanceof CClassInfo;
final CClassInfo classInfo = (CClassInfo) typeInfo;
final Outline outline = fieldOutline.parent().parent();
final ClassOutline classOutline = outline.getClazz(classInfo);
final Options options = outline.getModel().options;
final Map associationOverridesMap = new HashMap();
for (final AssociationOverride associationOverride : associationOverrides) {
associationOverridesMap.put(associationOverride.getName(),
associationOverride);
}
Mapping embeddedMapping = context.createEmbeddedMapping(context, fieldOutline);
final EmbeddableAttributes embeddableAttributes = embeddedMapping
.getEmbeddableAttributesMapping().process(embeddedMapping,
classOutline, options);
for (final ManyToOne source : embeddableAttributes.getManyToOne()) {
final String name = source.getName();
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
associationOverride.setJoinTable(source.getJoinTable());
associationOverride.getJoinColumn().addAll(
source.getJoinColumn());
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
for (final OneToMany source : embeddableAttributes.getOneToMany()) {
final String name = source.getName();
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
associationOverride.setJoinTable(source.getJoinTable());
// Join columns must not be overridden for 1:X
// associationOverride.getJoinColumn().addAll(
// source.getJoinColumn());
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
for (final ManyToMany source : embeddableAttributes.getManyToMany()) {
final String name = source.getName();
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
associationOverride.setJoinTable(source.getJoinTable());
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
for (final OneToOne source : embeddableAttributes.getOneToOne()) {
final String name = source.getName();
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
associationOverride.setJoinTable(source.getJoinTable());
if (source.getMappedBy() == null) {
// Join columns must not be overridden for 1:X
// associationOverride.getJoinColumn().addAll(
// source.getJoinColumn());
} else {
associationOverride.getJoinColumn().addAll(
source.getJoinColumn());
}
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
// TODO Element collection
// http://jira.highsource.org/browse/HJIII-64
for (final ElementCollection source : embeddableAttributes
.getElementCollection()) {
final String name = source.getName();
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
final JoinTable joinTable = new JoinTable();
if (source.getCollectionTable() != null) {
joinTable.setCatalog(source.getCollectionTable()
.getCatalog());
joinTable
.setSchema(source.getCollectionTable().getSchema());
joinTable.setName(source.getCollectionTable().getName());
joinTable.getJoinColumn().addAll(
source.getCollectionTable().getJoinColumn());
}
associationOverride.setJoinTable(joinTable);
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
for (final Embedded embedded : embeddableAttributes.getEmbedded()) {
final String parentName = embedded.getName();
for (AssociationOverride embeddedAssociationOverride : embedded
.getAssociationOverride()) {
final String childName = embeddedAssociationOverride.getName();
final String name = parentName + "." + childName;
final AssociationOverride associationOverride;
if (!associationOverridesMap.containsKey(name)) {
associationOverride = new AssociationOverride();
associationOverride.setName(name);
associationOverride
.setJoinTable(embeddedAssociationOverride
.getJoinTable());
associationOverride.getJoinColumn().addAll(
embeddedAssociationOverride.getJoinColumn());
associationOverridesMap.put(name, associationOverride);
associationOverrides.add(associationOverride);
} else {
associationOverride = associationOverridesMap.get(name);
}
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy