Explore returning Search Results.

Author: mp911de

spring-data-jpa/pom.xml

@@ -88,6 +88,12 @@
 			<optional>true</optional>
 		</dependency>
 
+		<dependency>
+			<groupId>org.springframework</groupId>
+			<artifactId>spring-test</artifactId>
+			<scope>test</scope>
+		</dependency>
+
 		<dependency>
 			<groupId>org.junit.platform</groupId>
 			<artifactId>junit-platform-launcher</artifactId>
@@ -183,6 +189,13 @@
 			</exclusions>
 		</dependency>
 
+		<dependency>
+			<groupId>${hibernate.groupId}.orm</groupId>
+			<artifactId>hibernate-vector</artifactId>
+			<version>${hibernate}</version>
+			<optional>true</optional>
+		</dependency>
+
 		<dependency>
 			<groupId>${hibernate.groupId}.orm</groupId>
 			<artifactId>hibernate-jpamodelgen</artifactId>

spring-data-jpa/src/main/java/org/springframework/data/jpa/convert/VectorConverters.java

@@ -0,0 +1,61 @@
+/*
+ * Copyright 2025 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.jpa.convert;
+
+import jakarta.persistence.AttributeConverter;
+import jakarta.persistence.Converter;
+
+import org.jspecify.annotations.Nullable;
+
+import org.springframework.data.domain.Vector;
+
+/**
+ * JPA {@link Converter} for {@link Vector} types.
+ *
+ * @author Mark Paluch
+ * @since 4.0
+ */
+public class VectorConverters {
+
+	@Converter(autoApply = true)
+	public static class VectorAsFloatArrayConverter implements AttributeConverter<@Nullable Vector, @Nullable float[]> {
+
+		@Override
+		public @Nullable float[] convertToDatabaseColumn(@Nullable Vector vector) {
+			return vector == null ? null : vector.toFloatArray();
+		}
+
+		@Override
+		public @Nullable Vector convertToEntityAttribute(@Nullable float[] floats) {
+			return floats == null ? null : Vector.of(floats);
+		}
+	}
+
+	@Converter(autoApply = true)
+	public static class VectorAsDoubleArrayConverter implements AttributeConverter<@Nullable Vector, @Nullable double[]> {
+
+		@Override
+		public @Nullable double[] convertToDatabaseColumn(@Nullable Vector vector) {
+			return vector == null ? null : vector.toDoubleArray();
+		}
+
+		@Override
+		public @Nullable Vector convertToEntityAttribute(@Nullable double[] doubles) {
+			return doubles == null ? null : Vector.of(doubles);
+		}
+	}
+
+}

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/aot/QueriesFactory.java

@@ -224,7 +224,8 @@ private AotQuery createQuery(PartTree partTree, ReturnedType returnedType, JpaPa
 
 		ParameterMetadataProvider metadataProvider = new ParameterMetadataProvider(parameters, EscapeCharacter.DEFAULT,
 				templates);
-		JpaQueryCreator queryCreator = new JpaQueryCreator(partTree, returnedType, metadataProvider, templates, metamodel);
+		JpaQueryCreator queryCreator = new JpaQueryCreator(partTree, false, returnedType, metadataProvider, templates,
+				metamodel);
 
 		return StringAotQuery.jpqlQuery(queryCreator.createQuery(), metadataProvider.getBindings(),
 				partTree.getResultLimit(), partTree.isDelete(), partTree.isExistsProjection());

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/AbstractJpaQuery.java

@@ -101,7 +101,7 @@ public AbstractJpaQuery(JpaQueryMethod method, EntityManager em) {
 				return new StreamExecution();
 			} else if (method.isProcedureQuery()) {
 				return new ProcedureExecution(method.isCollectionQuery());
-			} else if (method.isCollectionQuery()) {
+			} else if (method.isCollectionQuery() || method.isSearchQuery()) {
 				return new CollectionExecution();
 			} else if (method.isSliceQuery()) {
 				return new SlicedExecution();
@@ -140,17 +140,18 @@ protected JpaMetamodel getMetamodel() {
 
 	@Override
 	public @Nullable Object execute(Object[] parameters) {
-		return doExecute(getExecution(), parameters);
+
+		JpaParametersParameterAccessor accessor = obtainParameterAccessor(parameters);
+		return doExecute(getExecution(accessor), accessor);
 	}
 
 	/**
 	 * @param execution
 	 * @param values
 	 * @return
 	 */
-	private @Nullable Object doExecute(JpaQueryExecution execution, Object[] values) {
+	private @Nullable Object doExecute(JpaQueryExecution execution, JpaParametersParameterAccessor accessor) {
 
-		JpaParametersParameterAccessor accessor = obtainParameterAccessor(values);
 		Object result = execution.execute(this, accessor);
 
 		ResultProcessor withDynamicProjection = method.getResultProcessor().withDynamicProjection(accessor);
@@ -167,10 +168,17 @@ private JpaParametersParameterAccessor obtainParameterAccessor(Object[] values)
 		return new JpaParametersParameterAccessor(method.getParameters(), values);
 	}
 
-	protected JpaQueryExecution getExecution() {
+	protected JpaQueryExecution getExecution(JpaParametersParameterAccessor accessor) {
 
 		JpaQueryExecution execution = this.execution.getNullable();
 
+		if (method.isSearchQuery()) {
+
+			ReturnedType returnedType = method.getResultProcessor().withDynamicProjection(accessor).getReturnedType();
+			return new JpaQueryExecution.SearchResultExecution(execution == null ? new SingleEntityExecution() : execution,
+					returnedType, accessor.getScoringFunction());
+		}
+
 		if (execution != null) {
 			return execution;
 		}

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/JpaCountQueryCreator.java

@@ -48,7 +48,7 @@ public class JpaCountQueryCreator extends JpaQueryCreator {
 	public JpaCountQueryCreator(PartTree tree, ReturnedType returnedType, ParameterMetadataProvider provider,
 			JpqlQueryTemplates templates, EntityManager em) {
 
-		super(tree, returnedType, provider, templates, em);
+		super(tree, returnedType, provider, templates, em.getMetamodel());
 
 		this.distinct = tree.isDistinct();
 		this.returnedType = returnedType;

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/JpaKeysetScrollQueryCreator.java

@@ -23,6 +23,8 @@
 import java.util.List;
 import java.util.Map;
 
+import java.util.concurrent.atomic.AtomicInteger;
+
 import org.jspecify.annotations.Nullable;
 
 import org.springframework.data.domain.KeysetScrollPosition;
@@ -49,7 +51,7 @@ public JpaKeysetScrollQueryCreator(PartTree tree, ReturnedType type, ParameterMe
 			JpqlQueryTemplates templates, JpaEntityInformation<?, ?> entityInformation, KeysetScrollPosition scrollPosition,
 			EntityManager em) {
 
-		super(tree, type, provider, templates, em);
+		super(tree, type, provider, templates, em.getMetamodel());
 
 		this.entityInformation = entityInformation;
 		this.scrollPosition = scrollPosition;

spring-data-jpa/src/main/java/org/springframework/data/jpa/repository/query/JpaParametersParameterAccessor.java

@@ -17,6 +17,9 @@
 
 import org.jspecify.annotations.Nullable;
 
+import org.springframework.data.domain.Range;
+import org.springframework.data.domain.Score;
+import org.springframework.data.domain.ScoringFunction;
 import org.springframework.data.jpa.repository.query.JpaParameters.JpaParameter;
 import org.springframework.data.repository.query.Parameter;
 import org.springframework.data.repository.query.Parameters;
@@ -68,4 +71,28 @@ protected Object potentiallyUnwrap(Object parameterValue) {
 		return parameterValue;
 	}
 
+	public ScoringFunction getScoringFunction() {
+
+		Score score = getScore();
+		if (score != null) {
+			return score.getFunction();
+		}
+
+		JpaParameters parameters = getParameters();
+		if (parameters.hasScoreRangeParameter()) {
+
+			Range<Score> range = getScoreRange();
+
+			if (range.getUpperBound().isBounded()) {
+				return range.getUpperBound().getValue().get().getFunction();
+			}
+
+			if (range.getLowerBound().isBounded()) {
+				return range.getLowerBound().getValue().get().getFunction();
+			}
+		}
+
+		return ScoringFunction.UNSPECIFIED;
+	}
+
 }