Skip to content

Commit 002ecf9

Browse files
cigalycigaly
committed
HHH-18693 Changed Hibernate Processor implementation to generate inner classes under "parent" meta class
1 parent c883ed0 commit 002ecf9

File tree

7 files changed

+274
-44
lines changed

7 files changed

+274
-44
lines changed

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

Lines changed: 59 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,14 @@
55
package org.hibernate.processor;
66

77
import jakarta.annotation.Nullable;
8+
import org.hibernate.processor.annotation.AnnotationMetaEntity;
9+
import org.hibernate.processor.annotation.InnerClassMetaAttribute;
810
import org.hibernate.processor.model.MetaAttribute;
911
import org.hibernate.processor.model.Metamodel;
1012

1113
import javax.annotation.processing.FilerException;
1214
import javax.lang.model.element.ElementKind;
15+
import javax.lang.model.element.Modifier;
1316
import javax.lang.model.element.TypeElement;
1417
import javax.tools.Diagnostic;
1518
import javax.tools.FileObject;
@@ -19,7 +22,11 @@
1922
import java.io.StringWriter;
2023
import java.time.OffsetDateTime;
2124
import java.time.format.DateTimeFormatter;
25+
import java.util.Arrays;
2226
import java.util.List;
27+
import java.util.Set;
28+
import java.util.stream.Collectors;
29+
2330

2431
/**
2532
* Helper class to write the actual metamodel class using the {@link javax.annotation.processing.Filer} API.
@@ -33,14 +40,15 @@ private ClassWriter() {
3340
}
3441

3542
public static void writeFile(Metamodel entity, Context context) {
43+
if (entity.hasParent()) throw new IllegalStateException();
3644
try {
3745
String metaModelPackage = entity.getPackageName();
3846
// need to generate the body first, since this will also update
3947
// the required imports which need to be written out first
4048
String body = generateBody( entity, context ).toString();
4149

4250
FileObject fo = context.getProcessingEnvironment().getFiler().createSourceFile(
43-
getFullyQualifiedClassName( entity, metaModelPackage ),
51+
getFullyQualifiedClassName( entity ),
4452
entity.getElement()
4553
);
4654
OutputStream os = fo.openOutputStream();
@@ -102,6 +110,14 @@ private static StringBuffer generateBody(Metamodel entity, Context context) {
102110
pw.println();
103111

104112
final List<MetaAttribute> members = entity.getMembers();
113+
for ( MetaAttribute metaMember : members ) {
114+
if ( metaMember instanceof InnerClassMetaAttribute innerClass ) {
115+
generateBody( innerClass.getMetaEntity(), context )
116+
.toString().lines()
117+
.forEach(line -> pw.println('\t' + line));
118+
}
119+
}
120+
105121
for ( MetaAttribute metaMember : members ) {
106122
if ( metaMember.hasStringAttribute() ) {
107123
metaMember.getAttributeNameDeclarationString().lines()
@@ -134,7 +150,18 @@ private static StringBuffer generateBody(Metamodel entity, Context context) {
134150
}
135151

136152
private static void printClassDeclaration(Metamodel entity, PrintWriter pw, Context context) {
137-
pw.print( "public " );
153+
if ( entity.hasParent() ) {
154+
final Set<Modifier> modifiers = entity.getElement().getModifiers();
155+
if (modifiers.contains( Modifier.PUBLIC )){
156+
pw.print( "public " );
157+
}else if (modifiers.contains( Modifier.PROTECTED )) {
158+
pw.print( "protected " );
159+
}
160+
pw.print( "static " );
161+
}
162+
else {
163+
pw.print( "public " );
164+
}
138165
if ( !entity.isImplementation() && !entity.isJakartaDataStyle() ) {
139166
pw.print( "abstract " );
140167
}
@@ -147,32 +174,51 @@ private static void printClassDeclaration(Metamodel entity, PrintWriter pw, Cont
147174
if (superClassEntity == null) {
148175
superClassEntity = context.getMetaEmbeddable( superClassName );
149176
}
150-
pw.print( " extends " + getGeneratedSuperclassName(entity, superClassName, superClassEntity ) );
177+
final String superclassName = getGeneratedSuperclassName(
178+
superClassName, superClassEntity, entity.isJakartaDataStyle() );
179+
if (entity.getSimpleName().equals( "OffsetDateTimeTest" )) {
180+
System.err.printf( "Extends (?) %s - superClassName is not null%n", superClassName );
181+
}
182+
pw.print( " extends " + entity.importType(superclassName) );
151183
}
152184
if ( entity.isImplementation() ) {
153185
pw.print( entity.getElement().getKind() == ElementKind.CLASS ? " extends " : " implements " );
154186
pw.print( entity.getSimpleName() );
187+
if (entity.getSimpleName().equals( "OffsetDateTimeTest" )) {
188+
System.err.printf( "Extends (?) %s -- implementation (?!?)%n", entity.getSimpleName() );
189+
}
155190
}
156191

157192
pw.println( " {" );
158193
}
159194

160-
private static String getFullyQualifiedClassName(Metamodel entity, String metaModelPackage) {
195+
private static String getFullyQualifiedClassName(Metamodel entity) {
196+
final String metaModelPackage = entity.getPackageName();
161197
String fullyQualifiedClassName = "";
162198
if ( !metaModelPackage.isEmpty() ) {
163199
fullyQualifiedClassName = fullyQualifiedClassName + metaModelPackage + ".";
164200
}
165-
fullyQualifiedClassName = fullyQualifiedClassName + getGeneratedClassName( entity );
166-
return fullyQualifiedClassName;
201+
final String className;
202+
if ( entity.getElement().getKind() == ElementKind.PACKAGE ) {
203+
className = getGeneratedClassName( entity );
204+
}
205+
else {
206+
className = Arrays.stream(
207+
entity.getQualifiedName().substring( fullyQualifiedClassName.length() ).split( "\\." ) )
208+
.map( AnnotationMetaEntity::removeDollar )
209+
.map( part -> entity.isJakartaDataStyle() ? '_' + part : part + '_' )
210+
.collect( Collectors.joining( "." ) );
211+
}
212+
return fullyQualifiedClassName + className;
167213
}
168214

169215
private static String getGeneratedClassName(Metamodel entity) {
170-
final String className = entity.getSimpleName().replace( '.', '_' );
171-
return entity.isJakartaDataStyle() ? '_' + className : className + '_';
216+
final String className = entity.getSimpleName();
217+
return entity.isJakartaDataStyle() ? '_' + className : className + '_'; // TODO : check inner class!!!
172218
}
173219

174-
private static String getGeneratedSuperclassName(Metamodel entity, String superClassName, @Nullable Metamodel superClassEntity) {
175-
if ( entity.isJakartaDataStyle() ) {
220+
private static String getGeneratedSuperclassName(String superClassName, @Nullable Metamodel superClassEntity, boolean jakartaDataStyle) {
221+
if ( jakartaDataStyle ) {
176222
int lastDot = superClassName.lastIndexOf('.');
177223
if ( lastDot<0 ) {
178224
return '_' + superClassName;
@@ -184,7 +230,7 @@ private static String getGeneratedSuperclassName(Metamodel entity, String superC
184230
}
185231
else {
186232
return superClassEntity == null ? superClassName + '_'
187-
: getFullyQualifiedClassName( superClassEntity, superClassEntity.getPackageName() );
233+
: getFullyQualifiedClassName( superClassEntity );
188234
}
189235
}
190236

@@ -233,6 +279,7 @@ private static String writeStaticMetaModelAnnotation(Metamodel entity) {
233279
final String annotation = entity.isJakartaDataStyle()
234280
? "jakarta.data.metamodel.StaticMetamodel"
235281
: "jakarta.persistence.metamodel.StaticMetamodel";
236-
return "@" + entity.importType( annotation ) + "(" + entity.getSimpleName() + ".class)";
282+
final String simpleName = entity.importType( entity.getQualifiedName() );
283+
return "@" + entity.importType( annotation ) + "(" + simpleName + ".class)";
237284
}
238285
}

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,15 @@ public Collection<Metamodel> getDataMetaEmbeddables() {
304304
return dataMetaEmbeddables.values();
305305
}
306306

307+
public @Nullable Metamodel getMetamodel(String qualifiedName) {
308+
if ( metaEntities.containsKey( qualifiedName ) ) {
309+
return metaEntities.get( qualifiedName );
310+
}
311+
else {
312+
return metaEmbeddables.get( qualifiedName );
313+
}
314+
}
315+
307316
public @Nullable Metamodel getMetaAuxiliary(String qualifiedName) {
308317
return metaAuxiliaries.get( qualifiedName );
309318
}

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

Lines changed: 62 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import org.checkerframework.checker.nullness.qual.Nullable;
88
import org.hibernate.processor.annotation.AnnotationMetaEntity;
99
import org.hibernate.processor.annotation.AnnotationMetaPackage;
10+
import org.hibernate.processor.annotation.NonManagedMetamodel;
1011
import org.hibernate.processor.model.Metamodel;
1112
import org.hibernate.processor.util.Constants;
1213
import org.hibernate.processor.xml.JpaDescriptorParser;
@@ -361,11 +362,11 @@ private void processClasses(RoundEnvironment roundEnvironment) {
361362
}
362363

363364
for ( Element element : roundEnvironment.getRootElements() ) {
364-
processElement( element );
365+
processElement( element, null );
365366
}
366367
}
367368

368-
private void processElement(Element element) {
369+
private void processElement(Element element, @Nullable Element parent) {
369370
try {
370371
if ( !included( element )
371372
|| hasAnnotation( element, Constants.EXCLUDE )
@@ -375,7 +376,7 @@ private void processElement(Element element) {
375376
}
376377
else if ( isEntityOrEmbeddable( element ) && !element.getModifiers().contains( Modifier.PRIVATE )) {
377378
context.logMessage( Diagnostic.Kind.OTHER, "Processing annotated entity class '" + element + "'" );
378-
handleRootElementAnnotationMirrors( element );
379+
handleRootElementAnnotationMirrors( element, parent );
379380
}
380381
else if ( hasAuxiliaryAnnotations( element ) ) {
381382
context.logMessage( Diagnostic.Kind.OTHER, "Processing annotated class '" + element + "'" );
@@ -407,12 +408,29 @@ else if ( element instanceof TypeElement typeElement ) {
407408
break;
408409
}
409410
}
411+
if ( enclosesEntityOrEmbeddable( element ) ) {
412+
AnnotationMetaEntity parentMeta = null;
413+
if ( parent instanceof TypeElement parentElement ) {
414+
final String key = parentElement.getQualifiedName().toString();
415+
if ( context.getMetamodel( key ) instanceof AnnotationMetaEntity parentMetaEntity ) {
416+
parentMeta = parentMetaEntity;
417+
}
418+
}
419+
if (element.getSimpleName().toString().equals( "OffsetDateTimeTest" )) {
420+
System.err.println("Create OffsetDateTimeTest as non-entity class#1");
421+
}
422+
final NonManagedMetamodel metaEntity =
423+
NonManagedMetamodel .create(
424+
typeElement, context,
425+
parentMeta );
426+
context.addMetaEntity( metaEntity.getQualifiedName(), metaEntity );
427+
}
410428
}
411429
}
412430
if ( isClassOrRecordType( element ) ) {
413431
for ( final Element child : element.getEnclosedElements() ) {
414432
if ( isClassOrRecordType( child ) ) {
415-
processElement( child );
433+
processElement( child, element );
416434
}
417435
}
418436
}
@@ -445,7 +463,16 @@ private void createMetaModelClasses() {
445463
}
446464

447465
for ( Metamodel entity : context.getMetaEntities() ) {
448-
if ( !context.isAlreadyGenerated(entity) ) {
466+
if ( !context.isAlreadyGenerated(entity) && !entity.hasParent()) {
467+
context.logMessage( Diagnostic.Kind.OTHER,
468+
"Writing Jakarta Persistence metamodel for entity '" + entity + "'" );
469+
ClassWriter.writeFile( entity, context );
470+
context.markGenerated(entity);
471+
}
472+
}
473+
474+
for ( Metamodel entity : context.getMetaEntities() ) {
475+
if ( !context.isAlreadyGenerated(entity) && !entity.hasParent()) {
449476
context.logMessage( Diagnostic.Kind.OTHER,
450477
"Writing Jakarta Persistence metamodel for entity '" + entity + "'" );
451478
ClassWriter.writeFile( entity, context );
@@ -477,7 +504,7 @@ private void processEmbeddables(Collection<Metamodel> models) {
477504
final int toProcessCountBeforeLoop = models.size();
478505
for ( Metamodel metamodel : models ) {
479506
// see METAGEN-36
480-
if ( context.isAlreadyGenerated(metamodel) ) {
507+
if ( context.isAlreadyGenerated(metamodel) || metamodel.hasParent()) {
481508
processed.add( metamodel );
482509
}
483510
else if ( !modelGenerationNeedsToBeDeferred(models, metamodel ) ) {
@@ -531,6 +558,18 @@ private boolean modelGenerationNeedsToBeDeferred(Collection<Metamodel> entities,
531558
return false;
532559
}
533560

561+
private static boolean enclosesEntityOrEmbeddable(Element element) {
562+
if ( !(element instanceof TypeElement typeElement) ) {
563+
return false;
564+
}
565+
for ( final Element enclosedElement : typeElement.getEnclosedElements() ) {
566+
if ( isEntityOrEmbeddable( enclosedElement ) || enclosesEntityOrEmbeddable( enclosedElement ) ) {
567+
return true;
568+
}
569+
}
570+
return false;
571+
}
572+
534573
private static boolean isEntityOrEmbeddable(Element element) {
535574
return hasAnnotation(
536575
element,
@@ -562,7 +601,7 @@ private boolean hasAuxiliaryAnnotations(Element element) {
562601
);
563602
}
564603

565-
private void handleRootElementAnnotationMirrors(final Element element) {
604+
private void handleRootElementAnnotationMirrors(final Element element, @Nullable Element parent) {
566605
if ( isClassOrRecordType( element ) ) {
567606
if ( isEntityOrEmbeddable( element ) ) {
568607
final TypeElement typeElement = (TypeElement) element;
@@ -579,12 +618,23 @@ private void handleRootElementAnnotationMirrors(final Element element) {
579618
+ "' since XML configuration is metadata complete.");
580619
}
581620
else {
621+
AnnotationMetaEntity parentMetaEntity = null;
622+
if ( parent instanceof TypeElement parentTypeElement ) {
623+
if ( context.getMetamodel(
624+
parentTypeElement.getQualifiedName().toString() )
625+
instanceof AnnotationMetaEntity pme ) {
626+
parentMetaEntity = pme;
627+
}
628+
}
582629
final boolean requiresLazyMemberInitialization
583630
= hasAnnotation( element, EMBEDDABLE, MAPPED_SUPERCLASS );
631+
if (element.getSimpleName().toString().equals( "OffsetDateTimeTest" )) {
632+
System.err.println("Create OffsetDateTimeTest in handleRootElementAnnotationMirrors#2");
633+
}
584634
final AnnotationMetaEntity metaEntity =
585635
AnnotationMetaEntity.create( typeElement, context,
586636
requiresLazyMemberInitialization,
587-
true, false );
637+
true, false, parentMetaEntity );
588638
if ( alreadyExistingMetaEntity != null ) {
589639
metaEntity.mergeInMembers( alreadyExistingMetaEntity );
590640
}
@@ -598,10 +648,13 @@ && hasAnnotation( element, ENTITY, MAPPED_SUPERCLASS )
598648
// let a handwritten metamodel "override" the generated one
599649
// (this is used in the Jakarta Data TCK)
600650
&& !hasHandwrittenMetamodel(element) ) {
651+
if (element.getSimpleName().toString().equals( "OffsetDateTimeTest" )) {
652+
System.err.println("Create OffsetDateTimeTest in handleRootElementAnnotationMirrors#3");
653+
}
601654
final AnnotationMetaEntity dataMetaEntity =
602655
AnnotationMetaEntity.create( typeElement, context,
603656
requiresLazyMemberInitialization,
604-
true, true );
657+
true, true, parentMetaEntity );
605658
// final Metamodel alreadyExistingDataMetaEntity =
606659
// tryGettingExistingDataEntityFromContext( mirror, '_' + qualifiedName );
607660
// if ( alreadyExistingDataMetaEntity != null ) {

0 commit comments

Comments
 (0)