HHH-6321 : Add org.hibernate.metamodel.binding.CascadeType.toCascadeStyle()

Author: gbadner

hibernate-core/src/main/java/org/hibernate/metamodel/binding/CascadeType.java

@@ -26,6 +26,9 @@
 import java.util.HashMap;
 import java.util.Map;
 
+import org.hibernate.MappingException;
+import org.hibernate.engine.spi.CascadeStyle;
+
 
 /**
  * @author Hardy Ferentschik
@@ -120,6 +123,22 @@ public enum CascadeType {
 		jpaCascadeTypeToHibernateCascadeType.put( javax.persistence.CascadeType.DETACH, EVICT );
 	}
 
+	private static final Map<CascadeType, CascadeStyle> cascadeTypeToCascadeStyle = new HashMap<CascadeType, CascadeStyle>();
+	static {
+		cascadeTypeToCascadeStyle.put( ALL, CascadeStyle.ALL );
+		cascadeTypeToCascadeStyle.put( ALL_DELETE_ORPHAN, CascadeStyle.ALL_DELETE_ORPHAN );
+		cascadeTypeToCascadeStyle.put( UPDATE, CascadeStyle.UPDATE );
+		cascadeTypeToCascadeStyle.put( PERSIST, CascadeStyle.PERSIST );
+		cascadeTypeToCascadeStyle.put( MERGE, CascadeStyle.MERGE );
+		cascadeTypeToCascadeStyle.put( LOCK, CascadeStyle.LOCK );
+		cascadeTypeToCascadeStyle.put( REFRESH, CascadeStyle.REFRESH );
+		cascadeTypeToCascadeStyle.put( REPLICATE, CascadeStyle.REPLICATE );
+		cascadeTypeToCascadeStyle.put( EVICT, CascadeStyle.EVICT );
+		cascadeTypeToCascadeStyle.put( DELETE, CascadeStyle.DELETE );
+		cascadeTypeToCascadeStyle.put( DELETE_ORPHAN, CascadeStyle.DELETE_ORPHAN );
+		cascadeTypeToCascadeStyle.put( NONE, CascadeStyle.NONE );
+	}
+
 	/**
 	 * @param hbmOptionName the cascading option as specified in the hbm mapping file
 	 *
@@ -137,4 +156,17 @@ public static CascadeType getCascadeType(String hbmOptionName) {
 	public static CascadeType getCascadeType(javax.persistence.CascadeType jpaCascade) {
 		return jpaCascadeTypeToHibernateCascadeType.get( jpaCascade );
 	}
+
+	/**
+	 * @return Returns the {@code CascadeStyle} that corresponds to this {@code CascadeType}
+	 *
+	 * @throws MappingException if there is not corresponding {@code CascadeStyle}
+	 */
+	public CascadeStyle toCascadeStyle() {
+		CascadeStyle cascadeStyle = cascadeTypeToCascadeStyle.get( this );
+		if ( cascadeStyle == null ) {
+			throw new MappingException( "No CascadeStyle that corresponds with CascadeType=" + this.name() );
+		}
+		return cascadeStyle;
+	}
 }