HHH-7436 Initial many-to-many annotation support

brmeyer · brmeyer · commit 727e51c0e527 · 2012-11-29T16:28:19.000-05:00
diff --git a/hibernate-core/src/main/java/org/hibernate/metamodel/internal/source/annotations/ManyToManyPluralAttributeElementSourceImpl.java b/hibernate-core/src/main/java/org/hibernate/metamodel/internal/source/annotations/ManyToManyPluralAttributeElementSourceImpl.java
@@ -31,19 +31,47 @@
 import org.hibernate.engine.spi.CascadeStyle;
 import org.hibernate.metamodel.internal.source.annotations.attribute.Column;
 import org.hibernate.metamodel.internal.source.annotations.attribute.PluralAssociationAttribute;
+import org.hibernate.metamodel.internal.source.annotations.util.EnumConversionHelper;
+import org.hibernate.metamodel.internal.source.annotations.util.JPADotNames;
+import org.hibernate.metamodel.internal.source.annotations.util.JandexHelper;
 import org.hibernate.metamodel.spi.binding.CascadeType;
+import org.hibernate.metamodel.spi.relational.Value;
 import org.hibernate.metamodel.spi.source.ForeignKeyContributingSource;
 import org.hibernate.metamodel.spi.source.ManyToManyPluralAttributeElementSource;
 import org.hibernate.metamodel.spi.source.RelationalValueSource;
+import org.jboss.jandex.AnnotationInstance;
 
 /**
  * @author Hardy Ferentschik
+ * @author Brett Meyer
  */
 public class ManyToManyPluralAttributeElementSourceImpl implements ManyToManyPluralAttributeElementSource {
+	
 	private final PluralAssociationAttribute associationAttribute;
-
-	public ManyToManyPluralAttributeElementSourceImpl(PluralAssociationAttribute associationAttribute) {
+	private final List<RelationalValueSource> relationalValueSources
+			= new ArrayList<RelationalValueSource>();
+	private final Collection<String> referencedColumnNames
+			= new HashSet<String>();
+	private final Iterable<CascadeStyle> cascadeStyles;
+
+	public ManyToManyPluralAttributeElementSourceImpl(
+			PluralAssociationAttribute associationAttribute) {
 		this.associationAttribute = associationAttribute;
+		
+		for ( Column column : associationAttribute.getJoinColumnValues() ) {
+			relationalValueSources.add( new ColumnSourceImpl( 
+					associationAttribute, null, column ) );
+		}
+		
+		for ( Column column : associationAttribute.getJoinColumnValues() ) {
+			if ( column.getReferencedColumnName() != null ) {
+				referencedColumnNames.add( column.getReferencedColumnName() );
+			}
+		}
+		
+		cascadeStyles = EnumConversionHelper.cascadeTypeToCascadeStyleSet(
+				associationAttribute.getCascadeTypes(),
+				associationAttribute.getContext() );
 	}
 
 	@Override
@@ -59,28 +87,16 @@ public String getReferencedEntityAttributeName() {
 
 	@Override
 	public Collection<String> getReferencedColumnNames() {
-		HashSet<String> referencedColumnNames = new HashSet<String>();
-		for ( Column column : associationAttribute.getJoinColumnValues() ) {
-			if ( column.getReferencedColumnName() != null ) {
-				referencedColumnNames.add( column.getReferencedColumnName() );
-			}
-		}
 		return referencedColumnNames;
 	}
 
 	@Override
 	public List<RelationalValueSource> relationalValueSources() {
-		List<RelationalValueSource> valueSources = new ArrayList<RelationalValueSource>();
-		// todo
-		return valueSources;
+		return relationalValueSources;
 	}
 
 	@Override
 	public Iterable<CascadeStyle> getCascadeStyles() {
-		List<CascadeStyle> cascadeStyles = new ArrayList<CascadeStyle>();
-		for ( javax.persistence.CascadeType cascadeType : associationAttribute.getCascadeTypes() ) {
-			cascadeStyles.add( CascadeType.getCascadeType( cascadeType ).toCascadeStyle() );
-		}
 		return cascadeStyles;
 	}
 
@@ -96,7 +112,8 @@ public String getExplicitForeignKeyName() {
 
 	@Override
 	public JoinColumnResolutionDelegate getForeignKeyTargetColumnResolutionDelegate() {
-		return null;  //To change body of implemented methods use File | Settings | File Templates.
+		return associationAttribute.getJoinColumnValues()
+				.isEmpty() ? null : new AnnotationJoinColumnResolutionDelegate();
 	}
 
 	@Override
@@ -138,6 +155,55 @@ public boolean areValuesIncludedInUpdateByDefault() {
 	public boolean areValuesNullableByDefault() {
 		return false;
 	}
+
+	// TODO: Taken from and in duplicate of ToOneAttributeSourceImpl.  Look into
+	// abstracting some of this out.
+	public class AnnotationJoinColumnResolutionDelegate
+			implements ForeignKeyContributingSource.JoinColumnResolutionDelegate {
+		private final String logicalJoinTableName;
+
+		public AnnotationJoinColumnResolutionDelegate() {
+			logicalJoinTableName = resolveLogicalJoinTableName();
+		}
+
+		@Override
+		public List<Value> getJoinColumns(JoinColumnResolutionContext context) {
+			final List<Value> values = new ArrayList<Value>();
+			for ( Column column : associationAttribute.getJoinColumnValues() ) {
+				if ( column.getReferencedColumnName() == null ) {
+					return context.resolveRelationalValuesForAttribute( null );
+				}
+				org.hibernate.metamodel.spi.relational.Column resolvedColumn = context.resolveColumn(
+						column.getReferencedColumnName(),
+						logicalJoinTableName,
+						null,
+						null
+				);
+				values.add( resolvedColumn );
+			}
+			return values;
+		}
+
+		@Override
+		public String getReferencedAttributeName() {
+			// in annotations we are not referencing attribute but column names via @JoinColumn(s)
+			return null;
+		}
+
+		private String resolveLogicalJoinTableName() {
+			final AnnotationInstance joinTableAnnotation = JandexHelper.getSingleAnnotation(
+					associationAttribute.annotations(),
+					JPADotNames.JOIN_TABLE
+			);
+
+			if ( joinTableAnnotation != null ) {
+				return JandexHelper.getValue( joinTableAnnotation, "name", String.class );
+			}
+
+			// todo : this ties into the discussion about naming strategies.  This would be part of a logical naming strategy...
+			return null;
+		}
+	}
 }
 
 
diff --git a/hibernate-core/src/test/java/org/hibernate/test/annotations/manytomany/CatPk.java b/hibernate-core/src/test/java/org/hibernate/test/annotations/manytomany/CatPk.java
@@ -1,11 +1,14 @@
 //$Id$
 package org.hibernate.test.annotations.manytomany;
 import java.io.Serializable;
+
 import javax.persistence.Column;
+import javax.persistence.Embeddable;
 
 /**
  * @author Emmanuel Bernard
  */
+@Embeddable
 public class CatPk implements Serializable {
 	private String name;
 	private String thoroughbred;
