Skip to content

Commit 519a2a1

Browse files
committed
DATAES-328 - Adapt to API changes in Spring Data Commons.
1 parent 5f3bacf commit 519a2a1

File tree

8 files changed

+17
-17
lines changed

8 files changed

+17
-17
lines changed

src/main/java/org/springframework/data/elasticsearch/core/ElasticsearchTemplate.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ public boolean hasNext() {
418418
SearchResponse response = getSearchResponse(
419419
client.prepareSearchScroll(scrollId).setScroll(TimeValue.timeValueMillis(scrollTimeInMillis)).execute());
420420
// Save hits and scroll id
421-
currentHits = mapper.mapResults(response, clazz, Pageable.NONE).iterator();
421+
currentHits = mapper.mapResults(response, clazz, Pageable.unpaged()).iterator();
422422
finished = !currentHits.hasNext();
423423
scrollId = response.getScrollId();
424424
}
@@ -774,7 +774,7 @@ private SearchRequestBuilder prepareScan(Query query, long scrollTimeInMillis, b
774774
SearchRequestBuilder requestBuilder = client.prepareSearch(toArray(query.getIndices())).setSearchType(SCAN)
775775
.setTypes(toArray(query.getTypes())).setScroll(TimeValue.timeValueMillis(scrollTimeInMillis)).setFrom(0);
776776

777-
if(query.getPageable() != Pageable.NONE){
777+
if(query.getPageable().isPaged()){
778778
requestBuilder.setSize(query.getPageable().getPageSize());
779779
}
780780

@@ -826,14 +826,14 @@ private String doScan(SearchRequestBuilder requestBuilder, SearchQuery searchQue
826826
public <T> Page<T> scroll(String scrollId, long scrollTimeInMillis, Class<T> clazz) {
827827
SearchResponse response = getSearchResponse(
828828
client.prepareSearchScroll(scrollId).setScroll(TimeValue.timeValueMillis(scrollTimeInMillis)).execute());
829-
return resultsMapper.mapResults(response, clazz, Pageable.NONE);
829+
return resultsMapper.mapResults(response, clazz, Pageable.unpaged());
830830
}
831831

832832
@Override
833833
public <T> Page<T> scroll(String scrollId, long scrollTimeInMillis, SearchResultMapper mapper) {
834834
SearchResponse response = getSearchResponse(
835835
client.prepareSearchScroll(scrollId).setScroll(TimeValue.timeValueMillis(scrollTimeInMillis)).execute());
836-
return mapper.mapResults(response, null, Pageable.NONE);
836+
return mapper.mapResults(response, null, Pageable.unpaged());
837837
}
838838

839839
@Override
@@ -1009,7 +1009,7 @@ private SearchRequestBuilder prepareSearch(Query query) {
10091009
searchRequestBuilder.setFetchSource(sourceFilter.getIncludes(), sourceFilter.getExcludes());
10101010
}
10111011

1012-
if (query.getPageable() != null && query.getPageable() != Pageable.NONE) {
1012+
if (query.getPageable().isPaged()) {
10131013
startRecord = query.getPageable().getPageNumber() * query.getPageable().getPageSize();
10141014
searchRequestBuilder.setSize(query.getPageable().getPageSize());
10151015
}

src/main/java/org/springframework/data/elasticsearch/core/query/CriteriaQuery.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ public class CriteriaQuery extends AbstractQuery {
3232
private CriteriaQuery() {}
3333

3434
public CriteriaQuery(Criteria criteria) {
35-
this(criteria, Pageable.NONE);
35+
this(criteria, Pageable.unpaged());
3636
}
3737

3838
public CriteriaQuery(Criteria criteria, Pageable pageable) {

src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQueryBuilder.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public class NativeSearchQueryBuilder {
4444
private List<FacetRequest> facetRequests = new ArrayList<>();
4545
private List<AbstractAggregationBuilder> aggregationBuilders = new ArrayList<>();
4646
private HighlightBuilder.Field[] highlightFields;
47-
private Pageable pageable = Pageable.NONE;
47+
private Pageable pageable = Pageable.unpaged();
4848
private String[] indices;
4949
private String[] types;
5050
private String[] fields;

src/main/java/org/springframework/data/elasticsearch/repository/cdi/ElasticsearchRepositoryBean.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,9 @@ public class ElasticsearchRepositoryBean<T> extends CdiRepositoryBean<T> {
5252
* {@link CustomRepositoryImplementationDetector}, can be {@literal null}.
5353
*/
5454
public ElasticsearchRepositoryBean(Bean<ElasticsearchOperations> operations, Set<Annotation> qualifiers,
55-
Class<T> repositoryType, BeanManager beanManager, Optional<CustomRepositoryImplementationDetector> detector) {
56-
super(qualifiers, repositoryType, beanManager, detector);
55+
Class<T> repositoryType, BeanManager beanManager, CustomRepositoryImplementationDetector detector) {
56+
57+
super(qualifiers, repositoryType, beanManager, Optional.of(detector));
5758

5859
Assert.notNull(operations, "Cannot create repository with 'null' for ElasticsearchOperations.");
5960
this.elasticsearchOperationsBean = operations;

src/main/java/org/springframework/data/elasticsearch/repository/cdi/ElasticsearchRepositoryExtension.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
import java.util.HashMap;
2121
import java.util.Map;
2222
import java.util.Map.Entry;
23-
import java.util.Optional;
2423
import java.util.Set;
2524

2625
import javax.enterprise.event.Observes;
@@ -79,6 +78,6 @@ private <T> CdiRepositoryBean<T> createRepositoryBean(Class<T> repositoryType, S
7978
Bean<ElasticsearchOperations> elasticsearchOperationsBean = this.elasticsearchOperationsMap.get(qualifiers);
8079

8180
return new ElasticsearchRepositoryBean<>(elasticsearchOperationsBean, qualifiers, repositoryType, beanManager,
82-
Optional.ofNullable(getCustomImplementationDetector()));
81+
getCustomImplementationDetector());
8382
}
8483
}

src/main/java/org/springframework/data/elasticsearch/repository/query/ElasticsearchPartQuery.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ public Object execute(Object[] parameters) {
6060
return elasticsearchOperations.queryForPage(query, queryMethod.getEntityInformation().getJavaType());
6161
} else if (queryMethod.isStreamQuery()) {
6262
Class<?> entityType = queryMethod.getEntityInformation().getJavaType();
63-
if (query.getPageable() == Pageable.NONE) {
63+
if (query.getPageable().isUnpaged()) {
6464
query.setPageable(PageRequest.of(0, 20));
6565
}
6666

@@ -85,7 +85,7 @@ private Object countOrGetDocumentsForDelete(CriteriaQuery query, ParametersParam
8585
Object result = null;
8686

8787
if (queryMethod.isCollectionQuery()) {
88-
if (accessor.getPageable() == Pageable.NONE) {
88+
if (accessor.getPageable().isUnpaged()) {
8989
int itemCount = (int) elasticsearchOperations.count(query, queryMethod.getEntityInformation().getJavaType());
9090
query.setPageable(PageRequest.of(0, Math.max(1, itemCount)));
9191
} else {

src/main/java/org/springframework/data/elasticsearch/repository/query/ElasticsearchStringQuery.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ public Object execute(Object[] parameters) {
6767
stringQuery.setPageable(accessor.getPageable());
6868
return elasticsearchOperations.queryForPage(stringQuery, queryMethod.getEntityInformation().getJavaType());
6969
} else if (queryMethod.isCollectionQuery()) {
70-
if (accessor.getPageable() != Pageable.NONE) {
70+
if (accessor.getPageable().isPaged()) {
7171
stringQuery.setPageable(accessor.getPageable());
7272
}
7373
return elasticsearchOperations.queryForList(stringQuery, queryMethod.getEntityInformation().getJavaType());

src/test/java/org/springframework/data/elasticsearch/core/DefaultResultMapperTests.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ public void shouldMapAggregationsToPage() {
8484
when(response.getAggregations()).thenReturn(aggregations);
8585

8686
//When
87-
AggregatedPage<Car> page = (AggregatedPage<Car>) resultMapper.mapResults(response, Car.class, Pageable.NONE);
87+
AggregatedPage<Car> page = (AggregatedPage<Car>) resultMapper.mapResults(response, Car.class, Pageable.unpaged());
8888

8989
//Then
9090
page.hasFacets();
@@ -102,7 +102,7 @@ public void shouldMapSearchRequestToPage() {
102102
when(response.getHits()).thenReturn(searchHits);
103103

104104
//When
105-
Page<Car> page = resultMapper.mapResults(response, Car.class, Pageable.NONE);
105+
Page<Car> page = resultMapper.mapResults(response, Car.class, Pageable.unpaged());
106106

107107
//Then
108108
assertThat(page.hasContent(), is(true));
@@ -120,7 +120,7 @@ public void shouldMapPartialSearchRequestToObject() {
120120
when(response.getHits()).thenReturn(searchHits);
121121

122122
//When
123-
Page<Car> page = resultMapper.mapResults(response, Car.class, Pageable.NONE);
123+
Page<Car> page = resultMapper.mapResults(response, Car.class, Pageable.unpaged());
124124

125125
//Then
126126
assertThat(page.hasContent(), is(true));

0 commit comments

Comments
 (0)