HHH-19450 generated typesafe references to EnabledFetchProfile objects

Author: gavinking

tooling/metamodel-generator/src/main/java/org/hibernate/processor/annotation/AnnotationMeta.java

@@ -23,6 +23,7 @@
 
 import static java.lang.Character.isJavaIdentifierStart;
 import static org.hibernate.processor.util.Constants.ENTITY_GRAPH;
+import static org.hibernate.processor.util.Constants.HIB_ENABLED_FETCH_PROFILE;
 import static org.hibernate.processor.util.Constants.JAVA_OBJECT;
 import static org.hibernate.processor.util.Constants.NAMED_QUERY;
 import static org.hibernate.processor.util.Constants.TYPED_QUERY_REFERENCE;
@@ -182,22 +183,26 @@ private void addAuxiliaryMembersForMirror(AnnotationMirror mirror, String prefix
 	}
 
 	private NameMetaAttribute auxiliaryMember(AnnotationMirror mirror, String prefix, String name) {
-		if ( "QUERY_".equals(prefix) ) {
-			final AnnotationValue resultClass = getAnnotationValue( mirror, "resultClass" );
-			// if there is no explicit result class, we will infer it later by
-			// type checking the query (this is allowed but not required by JPA)
-			// and then we will replace this TypedMetaAttribute
-			return new TypedMetaAttribute( this, name, prefix,
-					resultClass == null ? JAVA_OBJECT : resultClass.getValue().toString(),
-					TYPED_QUERY_REFERENCE, null );
-		}
-		else if ( "GRAPH_".equals(prefix) ) {
-			return new TypedMetaAttribute( this, name, prefix, getQualifiedName(),
-					ENTITY_GRAPH, null );
-		}
-		else {
-			return new NameMetaAttribute( this, name, prefix);
-		}
+		return switch (prefix) {
+			case "QUERY_" -> {
+				final AnnotationValue resultClass = getAnnotationValue( mirror, "resultClass" );
+				// if there is no explicit result class, we will infer it later by
+				// type checking the query (this is allowed but not required by JPA)
+				// and then we will replace this TypedMetaAttribute
+				final String resultTypeName =
+						resultClass == null ? JAVA_OBJECT : resultClass.getValue().toString();
+				yield new TypedMetaAttribute( this, name, prefix, resultTypeName,
+						TYPED_QUERY_REFERENCE, null );
+			}
+			case "GRAPH_" ->
+					new TypedMetaAttribute( this, name, prefix, getQualifiedName(),
+							ENTITY_GRAPH, null );
+			case "PROFILE_" ->
+					new EnabledFetchProfileMetaAttribute( this, name, prefix,
+							HIB_ENABLED_FETCH_PROFILE );
+			default ->
+					new NameMetaAttribute( this, name, prefix );
+		};
 	}
 
 	protected String getSessionVariableName() {

tooling/metamodel-generator/src/main/java/org/hibernate/processor/annotation/EnabledFetchProfileMetaAttribute.java

@@ -0,0 +1,80 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.processor.annotation;
+
+import org.hibernate.processor.model.Metamodel;
+
+import static org.hibernate.processor.util.StringUtil.nameToMethodName;
+
+/**
+ * Represents a named fetch profile.
+ *
+ * @author Gavin King
+ */
+class EnabledFetchProfileMetaAttribute extends NameMetaAttribute {
+	private final String prefix;
+	private final String referenceType;
+
+	public EnabledFetchProfileMetaAttribute(
+			Metamodel annotationMetaEntity,
+			String name,
+			String prefix,
+			String referenceType) {
+		super( annotationMetaEntity, name, prefix );
+		this.prefix = prefix;
+		this.referenceType = referenceType;
+	}
+
+	@Override
+	public boolean hasTypedAttribute() {
+		return true;
+	}
+
+	@Override
+	public String getAttributeNameDeclarationString() {
+		StringBuilder declaration = new StringBuilder();
+		declaration
+				.append("\n/**\n * @see ")
+				.append("#");
+		appendFieldName( declaration );
+		return declaration
+				.append( "\n **/\n" )
+				.append(super.getAttributeNameDeclarationString())
+				.toString();
+	}
+
+	@Override
+	public String getAttributeDeclarationString() {
+		final Metamodel entity = getHostingEntity();
+		final StringBuilder declaration = new StringBuilder();
+		declaration
+				.append("\n/**")
+				.append("\n * The fetch profile named {@value ")
+				.append(prefix)
+				.append(fieldName())
+				.append("}\n")
+				.append(" *\n * @see ")
+				.append(entity.getQualifiedName())
+				.append("\n **/\n")
+				.append("public static final ")
+				.append(entity.importType(referenceType))
+				.append(' ');
+		appendFieldName( declaration );
+		declaration
+				.append(" = new ")
+				.append(entity.importType(referenceType))
+				.append("(")
+				.append(prefix)
+				.append(fieldName())
+				.append(");");
+		return declaration.toString();
+	}
+
+	private void appendFieldName(StringBuilder declaration) {
+		declaration
+				.append('_')
+				.append(nameToMethodName(getPropertyName()));
+	}
+}

tooling/metamodel-generator/src/main/java/org/hibernate/processor/util/Constants.java

@@ -109,6 +109,7 @@ public final class Constants {
 	public static final String MUTINY_SESSION = "org.hibernate.reactive.mutiny.Mutiny.Session";
 	public static final String MUTINY_STATELESS_SESSION = "org.hibernate.reactive.mutiny.Mutiny.StatelessSession";
 	public static final String QUARKUS_SESSION_OPERATIONS = "io.quarkus.hibernate.reactive.panache.common.runtime.SessionOperations";
+	public static final String HIB_ENABLED_FETCH_PROFILE = "org.hibernate.EnabledFetchProfile";
 
 	public static final String TUPLE = "jakarta.persistence.Tuple";