DATACMNS-1482 - Properly convert collections when collection type matches but element type doesn't.
Various fast returns and the use of Class instead of TypeDescriptor led to e.g. List<BigDecimal> not getting properly converted to List<Integer> leading to unexpected ClassCastExceptions when the collection elements where accessed.
This commit is contained in:
committed by
Oliver Drotbohm
parent
f65fd511ac
commit
6dc0ced366
@@ -34,6 +34,7 @@ import org.springframework.lang.Nullable;
|
||||
*
|
||||
* @author Oliver Gierke
|
||||
* @author Mark Paluch
|
||||
* @author Jens Schauder
|
||||
*/
|
||||
class QueryExecutionResultHandler {
|
||||
|
||||
@@ -63,10 +64,6 @@ class QueryExecutionResultHandler {
|
||||
@Nullable
|
||||
public Object postProcessInvocationResult(@Nullable Object result, Method method) {
|
||||
|
||||
if (method.getReturnType().isInstance(result)) {
|
||||
return result;
|
||||
}
|
||||
|
||||
MethodParameter parameter = new MethodParameter(method, -1);
|
||||
|
||||
return postProcessInvocationResult(result, 0, parameter);
|
||||
@@ -91,20 +88,8 @@ class QueryExecutionResultHandler {
|
||||
|
||||
Class<?> expectedReturnType = returnTypeDescriptor.getType();
|
||||
|
||||
// Early return if the raw value matches
|
||||
|
||||
if (result != null && expectedReturnType.isInstance(result)) {
|
||||
return result;
|
||||
}
|
||||
|
||||
result = unwrapOptional(result);
|
||||
|
||||
// Early return if the unrwapped value matches
|
||||
|
||||
if (result != null && expectedReturnType.isInstance(result)) {
|
||||
return result;
|
||||
}
|
||||
|
||||
if (QueryExecutionConverters.supports(expectedReturnType)) {
|
||||
|
||||
// For a wrapper type, try nested resolution first
|
||||
@@ -132,8 +117,8 @@ class QueryExecutionResultHandler {
|
||||
return ReactiveWrapperConverters.toWrapper(result, expectedReturnType);
|
||||
}
|
||||
|
||||
return conversionService.canConvert(result.getClass(), expectedReturnType)
|
||||
? conversionService.convert(result, expectedReturnType)
|
||||
return conversionService.canConvert(TypeDescriptor.forObject(result), returnTypeDescriptor)
|
||||
? conversionService.convert(result, returnTypeDescriptor)
|
||||
: result;
|
||||
}
|
||||
|
||||
|
||||
@@ -15,6 +15,7 @@
|
||||
*/
|
||||
package org.springframework.data.repository.core.support;
|
||||
|
||||
import static java.util.Arrays.*;
|
||||
import static org.assertj.core.api.Assertions.*;
|
||||
|
||||
import io.vavr.control.Option;
|
||||
@@ -27,6 +28,7 @@ import rx.Observable;
|
||||
import rx.Single;
|
||||
|
||||
import java.lang.reflect.Method;
|
||||
import java.math.BigDecimal;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.Iterator;
|
||||
@@ -36,6 +38,7 @@ import java.util.Optional;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import org.assertj.core.api.SoftAssertions;
|
||||
import org.junit.Test;
|
||||
import org.reactivestreams.Publisher;
|
||||
import org.springframework.dao.InvalidDataAccessApiUsageException;
|
||||
@@ -47,6 +50,7 @@ import org.springframework.data.util.Streamable;
|
||||
*
|
||||
* @author Oliver Gierke
|
||||
* @author Mark Paluch
|
||||
* @author Jens Schauder
|
||||
*/
|
||||
public class QueryExecutionResultHandlerUnitTests {
|
||||
|
||||
@@ -356,7 +360,7 @@ public class QueryExecutionResultHandlerUnitTests {
|
||||
@SuppressWarnings("unchecked")
|
||||
public void convertsIterableIntoStreamable() throws Exception {
|
||||
|
||||
Iterable<?> source = Arrays.asList(new Object());
|
||||
Iterable<?> source = asList(new Object());
|
||||
|
||||
Object result = handler.postProcessInvocationResult(source, getMethod("streamable"));
|
||||
|
||||
@@ -384,6 +388,22 @@ public class QueryExecutionResultHandlerUnitTests {
|
||||
});
|
||||
}
|
||||
|
||||
@Test // DATACMNS-1482
|
||||
public void nestedConversion() throws Exception {
|
||||
|
||||
Object result = handler.postProcessInvocationResult(asList(BigDecimal.ZERO, BigDecimal.ONE),
|
||||
getMethod("listOfInteger"));
|
||||
|
||||
assertThat(result).isInstanceOf(List.class);
|
||||
|
||||
List list = (List) result;
|
||||
SoftAssertions.assertSoftly(s -> {
|
||||
// for making the test failure more obvious:
|
||||
(list).forEach(v -> s.assertThat(v).isInstanceOf(Integer.class));
|
||||
s.assertThat(list).containsExactly(0, 1);
|
||||
});
|
||||
}
|
||||
|
||||
private static Method getMethod(String methodName) throws Exception {
|
||||
return Sample.class.getMethod(methodName);
|
||||
}
|
||||
@@ -421,6 +441,8 @@ public class QueryExecutionResultHandlerUnitTests {
|
||||
|
||||
// DATACMNS-1430
|
||||
CustomStreamableWrapper<String> customStreamable();
|
||||
|
||||
List<Integer> listOfInteger();
|
||||
}
|
||||
|
||||
static class Entity {}
|
||||
|
||||
Reference in New Issue
Block a user