clean up some minor formatting / code style issues

Author: gavinking

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

@@ -30,6 +30,7 @@
 import javax.lang.model.type.DeclaredType;
 import javax.lang.model.type.TypeKind;
 import javax.lang.model.type.TypeMirror;
+import javax.lang.model.util.Elements;
 import javax.tools.Diagnostic;
 import javax.tools.StandardLocation;
 import java.io.IOException;
@@ -41,6 +42,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
+import java.util.function.Function;
 
 import static java.lang.Boolean.parseBoolean;
 import static javax.lang.model.util.ElementFilter.fieldsIn;
@@ -374,7 +376,8 @@ private void processElement(Element element, @Nullable Element parent) {
 				// skip it completely
 				return;
 			}
-			else if ( isEntityOrEmbeddable( element ) && !element.getModifiers().contains( Modifier.PRIVATE )) {
+			else if ( isEntityOrEmbeddable( element )
+					  && !element.getModifiers().contains( Modifier.PRIVATE ) ) {
 				context.logMessage( Diagnostic.Kind.OTHER, "Processing annotated entity class '" + element + "'" );
 				handleRootElementAnnotationMirrors( element, parent );
 			}
@@ -409,30 +412,14 @@ else if ( element instanceof TypeElement typeElement ) {
 						}
 					}
 					if ( enclosesEntityOrEmbeddable( element ) ) {
-						AnnotationMetaEntity parentMeta = null;
-						if ( parent instanceof TypeElement parentElement ) {
-							final String key = parentElement.getQualifiedName().toString();
-							if ( context.getMetamodel( key ) instanceof AnnotationMetaEntity parentMetaEntity ) {
-								parentMeta = parentMetaEntity;
-							}
-						}
-						final NonManagedMetamodel  metaEntity =
-								NonManagedMetamodel .create(
-										typeElement, context,
-										false, parentMeta );
+						final AnnotationMetaEntity parentMeta = parentMetadata( parent, context::getMetamodel );
+						final NonManagedMetamodel metaEntity =
+								NonManagedMetamodel.create( typeElement, context, false, parentMeta );
 						context.addMetaEntity( metaEntity.getQualifiedName(), metaEntity );
-						if ( context.generateJakartaDataStaticMetamodel()) {
-							AnnotationMetaEntity parentDataMeta = null;
-							if ( parent instanceof TypeElement parentElement ) {
-								final String key = parentElement.getQualifiedName().toString();
-								if ( context.getDataMetaEntity( key ) instanceof AnnotationMetaEntity parentMetaEntity ) {
-									parentDataMeta = parentMetaEntity;
-								}
-							}
-							final NonManagedMetamodel  dataMetaEntity =
-									NonManagedMetamodel .create(
-											typeElement, context,
-											true, parentDataMeta );
+						if ( context.generateJakartaDataStaticMetamodel() ) {
+							final AnnotationMetaEntity parentDataMeta = parentMetadata( parent, context::getDataMetaEntity );
+							final NonManagedMetamodel dataMetaEntity =
+									NonManagedMetamodel.create( typeElement, context, true, parentDataMeta );
 							context.addDataMetaEntity( dataMetaEntity.getQualifiedName(), dataMetaEntity );
 						}
 
@@ -458,6 +445,17 @@ else if ( element instanceof TypeElement typeElement ) {
 		}
 	}
 
+	private AnnotationMetaEntity parentMetadata(Element parent, Function<String,Object> metamodel) {
+		if ( parent instanceof TypeElement parentElement
+				&& metamodel.apply( parentElement.getQualifiedName().toString() )
+						instanceof AnnotationMetaEntity parentMetaEntity ) {
+			return parentMetaEntity;
+		}
+		else {
+			return null;
+		}
+	}
+
 	private boolean hasPackageAnnotation(Element element, String annotation) {
 		final PackageElement pack = context.getElementUtils().getPackageOf( element ); // null for module descriptor
 		return pack != null && hasAnnotation( pack, annotation );
@@ -562,15 +560,18 @@ private boolean modelGenerationNeedsToBeDeferred(Collection<Metamodel> entities,
 	}
 
 	private static boolean enclosesEntityOrEmbeddable(Element element) {
-		if ( !(element instanceof TypeElement typeElement) ) {
+		if ( element instanceof TypeElement typeElement ) {
+			for ( final Element enclosedElement : typeElement.getEnclosedElements() ) {
+				if ( isEntityOrEmbeddable( enclosedElement )
+					 || enclosesEntityOrEmbeddable( enclosedElement ) ) {
+					return true;
+				}
+			}
 			return false;
 		}
-		for ( final Element enclosedElement : typeElement.getEnclosedElements() ) {
-			if ( isEntityOrEmbeddable( enclosedElement ) || enclosesEntityOrEmbeddable( enclosedElement ) ) {
-				return true;
-			}
+		else {
+			return false;
 		}
-		return false;
 	}
 
 	private static boolean isEntityOrEmbeddable(Element element) {
@@ -783,13 +784,14 @@ else if ( hasAnnotation( typeElement, EMBEDDABLE ) ) {
 
 	private void writeIndex() {
 		final ProcessingEnvironment processingEnvironment = context.getProcessingEnvironment();
+		final Elements elementUtils = processingEnvironment.getElementUtils();
 		context.getEntityNameMappings().forEach((entityName, className) -> {
 			try (Writer writer = processingEnvironment.getFiler()
 					.createResource(
 							StandardLocation.SOURCE_OUTPUT,
 							ENTITY_INDEX,
 							entityName,
-							processingEnvironment.getElementUtils().getTypeElement( className )
+							elementUtils.getTypeElement( className )
 					)
 					.openWriter()) {
 				writer.append(className);
@@ -801,11 +803,12 @@ private void writeIndex() {
 			}
 		});
 		context.getEnumTypesByValue().forEach((valueName, enumTypeNames) -> {
-			try (Writer writer = processingEnvironment.getFiler().createResource(
+			try (Writer writer = processingEnvironment.getFiler()
+					.createResource(
 							StandardLocation.SOURCE_OUTPUT,
 							ENTITY_INDEX,
 							'.' + valueName,
-							processingEnvironment.getElementUtils().getTypeElement( enumTypeNames.iterator().next() )
+							elementUtils.getTypeElement( enumTypeNames.iterator().next() )
 					)
 					.openWriter()) {
 				for (String enumTypeName : enumTypeNames) {