Spring/spring-data-keyvalue
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Enable KeyValuePartTreeQuery usage for projections.

Browse Source 
Closes #563
This commit is contained in:
Mark Paluch
2024-02-20 11:16:22 +01:00
parent f8e38a50c3
commit 95ec0c1eab
2 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/org/springframework/data/keyvalue/repository/query/KeyValuePartTreeQuery.java
View File

@@ -134,15 +134,13 @@ public class KeyValuePartTreeQuery implements RepositoryQuery {
} else if (queryMethod.isCollectionQuery()) {
return this.keyValueOperations.find(query, queryMethod.getEntityInformation().getJavaType());
} else if (queryMethod.isQueryForEntity()) {
} else if (partTree.get().isExistsProjection()) {
return keyValueOperations.exists(query, queryMethod.getEntityInformation().getJavaType());
} else {
Iterable<?> result = this.keyValueOperations.find(query, queryMethod.getEntityInformation().getJavaType());
return result.iterator().hasNext() ? result.iterator().next() : null;
} else if (partTree.get().isExistsProjection()) {
return keyValueOperations.exists(query, queryMethod.getEntityInformation().getJavaType());
}
throw new UnsupportedOperationException("Query method not supported");
}
protected KeyValueQuery<?> prepareQuery(Object[] parameters) {

27
src/test/java/org/springframework/data/keyvalue/repository/query/KeyValuePartTreeQueryUnitTests.java
View File

@@ -93,6 +93,27 @@ class KeyValuePartTreeQueryUnitTests {
assertThat(query.getRows()).isEqualTo(3);
}
@Test // GH-563
@SuppressWarnings({ "unchecked", "rawtypes" })
void shouldAllowProjectionQueries() throws SecurityException, NoSuchMethodException {
when(metadataMock.getDomainType()).thenReturn((Class) Person.class);
when(metadataMock.getDomainTypeInformation()).thenReturn((TypeInformation) TypeInformation.of(Person.class));
when(metadataMock.getReturnType(any(Method.class))).thenReturn((TypeInformation) TypeInformation.of(List.class));
when(metadataMock.getReturnedDomainClass(any(Method.class))).thenReturn((Class) Person.class);
QueryMethod qm = new QueryMethod(Repo.class.getMethod("findProjectionByFirstname",String.class), metadataMock,
projectionFactoryMock);
KeyValuePartTreeQuery partTreeQuery = new KeyValuePartTreeQuery(qm, QueryMethodEvaluationContextProvider.DEFAULT,
kvOpsMock, SpelQueryCreator.class);
KeyValueQuery<?> query = partTreeQuery.prepareQuery(new Object[] { "firstname" });
partTreeQuery.doExecute(new Object[] { "firstname" }, query);
verify(kvOpsMock).find(eq(query), eq(Person.class));
}
@Test // DATAKV-142
@SuppressWarnings({ "unchecked", "rawtypes" })
void shouldApplyDerivedMaxResultsToQuery() throws SecurityException, NoSuchMethodException {
@@ -166,5 +187,11 @@ class KeyValuePartTreeQueryUnitTests {
List<Person> findTop3By();
List<Person> findTop3ByFirstname(String firstname);
PersonProjection findProjectionByFirstname(String firstname);
}
interface PersonProjection {
String getFirstname();
}
}