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 <[email protected]>

Author: gavinking

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

@@ -1541,23 +1541,26 @@ private void missingTypeArgError(String entity, VariableElement parameter, boole
 				Diagnostic.Kind.ERROR );
 	}
 
-	private List<OrderBy> orderByList(ExecutableElement method, TypeElement entityType) {
-		final AnnotationMirror orderByList =
-				getAnnotationMirror( method, "jakarta.data.repository.OrderBy.List" );
-		if ( orderByList != null ) {
-			final List<OrderBy> result = new ArrayList<>();
-			@SuppressWarnings("unchecked")
-			final List<AnnotationValue> list = (List<AnnotationValue>)
-					castNonNull( getAnnotationValue( orderByList, "value" ) ).getValue();
-			for ( AnnotationValue element : list ) {
-				result.add( orderByExpression( castNonNull( (AnnotationMirror) element.getValue() ), entityType, method ) );
-			}
-			return result;
-		}
-		final AnnotationMirror orderBy =
-				getAnnotationMirror( method, "jakarta.data.repository.OrderBy" );
-		if ( orderBy != null ) {
-			return List.of( orderByExpression(orderBy, entityType, method) );
+	private List<OrderBy> orderByList(ExecutableElement method, TypeElement returnType) {
+		final TypeElement entityType = implicitEntityType( returnType );
+		if ( entityType != null ) {
+			final AnnotationMirror orderByList =
+					getAnnotationMirror( method, "jakarta.data.repository.OrderBy.List" );
+			if ( orderByList != null ) {
+				final List<OrderBy> result = new ArrayList<>();
+				@SuppressWarnings("unchecked")
+				final List<AnnotationValue> list = (List<AnnotationValue>)
+						castNonNull( getAnnotationValue( orderByList, "value" ) ).getValue();
+				for ( AnnotationValue element : list ) {
+					result.add( orderByExpression( castNonNull( (AnnotationMirror) element.getValue() ), entityType, method ) );
+				}
+				return result;
+			}
+			final AnnotationMirror orderBy =
+					getAnnotationMirror( method, "jakarta.data.repository.OrderBy" );
+			if ( orderBy != null ) {
+				return List.of( orderByExpression(orderBy, entityType, method) );
+			}
 		}
 		return emptyList();
 	}
@@ -2214,6 +2217,18 @@ else if ( primaryEntity != null ) {
 		}
 	}
 
+	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) {
 		final AnnotationValue name = getAnnotationValue(annotation, "name");
 		if (name != null) {