DATAJPA-809 - ParameterBinder now uses ParameterAccessor.

We now use a ParameterAccessor to obtain query method parameters in ParameterBinder to benefit from value post-processing (e.g. unwrapping of Optional etc.).

Related tickets: DATACMNS-768.
This commit is contained in:
Oliver Gierke
2015-10-07 19:39:37 +02:00
parent 7fa997085d
commit 7fb58c3c47
2 changed files with 33 additions and 34 deletions

View File

@@ -22,7 +22,9 @@ import javax.persistence.Query;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.repository.query.JpaParameters.JpaParameter;
import org.springframework.data.repository.query.ParameterAccessor;
import org.springframework.data.repository.query.Parameters;
import org.springframework.data.repository.query.ParametersParameterAccessor;
import org.springframework.util.Assert;
/**
@@ -35,6 +37,7 @@ import org.springframework.util.Assert;
public class ParameterBinder {
private final JpaParameters parameters;
private final ParameterAccessor accessor;
private final Object[] values;
/**
@@ -52,6 +55,7 @@ public class ParameterBinder {
this.parameters = parameters;
this.values = values.clone();
this.accessor = new ParametersParameterAccessor(parameters, this.values);
}
ParameterBinder(JpaParameters parameters) {
@@ -64,12 +68,7 @@ public class ParameterBinder {
* @return
*/
public Pageable getPageable() {
if (!parameters.hasPageableParameter()) {
return null;
}
return (Pageable) values[parameters.getPageableIndex()];
return accessor.getPageable();
}
/**
@@ -79,16 +78,7 @@ public class ParameterBinder {
* @return
*/
public Sort getSort() {
if (parameters.hasSortParameter()) {
return (Sort) values[parameters.getSortIndex()];
}
if (parameters.hasPageableParameter() && getPageable() != null) {
return getPageable().getSort();
}
return null;
return accessor.getSort();
}
/**
@@ -99,19 +89,17 @@ public class ParameterBinder {
*/
public <T extends Query> T bind(T query) {
int methodParameterPosition = 0;
int bindableParameterIndex = 0;
int queryParameterPosition = 1;
for (JpaParameter parameter : parameters) {
if (canBindParameter(parameter)) {
Object value = values[methodParameterPosition];
Object value = accessor.getBindableValue(bindableParameterIndex);
bind(query, parameter, value, queryParameterPosition++);
bindableParameterIndex++;
}
methodParameterPosition++;
}
return query;
@@ -181,15 +169,6 @@ public class ParameterBinder {
return result;
}
/**
* Returns the values to bind.
*
* @return
*/
Object[] getValues() {
return values;
}
/**
* Returns the parameters.
*
@@ -198,4 +177,8 @@ public class ParameterBinder {
JpaParameters getParameters() {
return parameters;
}
protected Object[] getValues() {
return values;
}
}

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2008-2014 the original author or authors.
* Copyright 2008-2015 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.
@@ -24,6 +24,7 @@ import static org.mockito.Mockito.*;
import java.lang.reflect.Method;
import java.util.Date;
import java.util.List;
import java.util.Optional;
import javax.persistence.Embeddable;
import javax.persistence.Query;
@@ -86,6 +87,8 @@ public class ParameterBinderUnitTests {
User invalidWithTemporalTypeParameter(@Temporal String registerDate);
List<User> validWithVarArgs(Integer... ids);
User optionalParameter(Optional<String> name);
}
@Test(expected = IllegalArgumentException.class)
@@ -174,8 +177,8 @@ public class ParameterBinderUnitTests {
public void bindsSortForIndexedParameters() throws Exception {
Sort sort = new Sort("name");
ParameterBinder binder = new ParameterBinder(new JpaParameters(indexedParametersWithSort), new Object[] { "name",
sort });
ParameterBinder binder = new ParameterBinder(new JpaParameters(indexedParametersWithSort),
new Object[] { "name", sort });
assertThat(binder.getSort(), is(sort));
}
@@ -223,7 +226,6 @@ public class ParameterBinderUnitTests {
/**
* @see DATAJPA-461
* @throws Exception
*/
@Test
public void shouldAllowBindingOfVarArgsAsIs() throws Exception {
@@ -236,6 +238,20 @@ public class ParameterBinderUnitTests {
verify(query).setParameter(eq(1), eq(ids));
}
/**
* @see DATAJPA-809
*/
@Test
public void unwrapsOptionalParameter() throws Exception {
Method method = SampleRepository.class.getMethod("optionalParameter", Optional.class);
JpaParameters parameters = new JpaParameters(method);
new ParameterBinder(parameters, new Object[] { Optional.of("Foo") }).bind(query);
verify(query).setParameter(eq(1), eq("Foo"));
}
public SampleEntity findByEmbeddable(SampleEmbeddable embeddable) {
return null;