fix to @OrderBy for @Query methods with primary entity type

don't think this is even strictly-speaking required by Jakarta Data
but we will allow it

Signed-off-by: Gavin King <gavin@hibernate.org>
This commit is contained in:
Gavin King 2024-03-31 20:20:08 +02:00
parent c7852feb55
commit e21d139a84
1 changed files with 31 additions and 16 deletions

View File

@ -1541,7 +1541,9 @@ public class AnnotationMetaEntity extends AnnotationMeta {
Diagnostic.Kind.ERROR ); Diagnostic.Kind.ERROR );
} }
private List<OrderBy> orderByList(ExecutableElement method, TypeElement entityType) { private List<OrderBy> orderByList(ExecutableElement method, TypeElement returnType) {
final TypeElement entityType = implicitEntityType( returnType );
if ( entityType != null ) {
final AnnotationMirror orderByList = final AnnotationMirror orderByList =
getAnnotationMirror( method, "jakarta.data.repository.OrderBy.List" ); getAnnotationMirror( method, "jakarta.data.repository.OrderBy.List" );
if ( orderByList != null ) { if ( orderByList != null ) {
@ -1559,6 +1561,7 @@ public class AnnotationMetaEntity extends AnnotationMeta {
if ( orderBy != null ) { if ( orderBy != null ) {
return List.of( orderByExpression(orderBy, entityType, method) ); return List.of( orderByExpression(orderBy, entityType, method) );
} }
}
return emptyList(); return emptyList();
} }
@ -2214,6 +2217,18 @@ public class AnnotationMetaEntity extends AnnotationMeta {
} }
} }
private @Nullable TypeElement implicitEntityType(@Nullable TypeElement resultType) {
if ( resultType != null && hasAnnotation(resultType, ENTITY) ) {
return resultType;
}
else if ( primaryEntity != null ) {
return primaryEntity;
}
else {
return null;
}
}
private static String entityName(DeclaredType resultType, AnnotationMirror annotation) { private static String entityName(DeclaredType resultType, AnnotationMirror annotation) {
final AnnotationValue name = getAnnotationValue(annotation, "name"); final AnnotationValue name = getAnnotationValue(annotation, "name");
if (name != null) { if (name != null) {