Cwikius-Spring/spring-data-elasticsearch
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-data-elasticsearch.git synced 2025-06-30 15:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

DATAES-661 - Added track_total_hits request parameter to Query.

Browse Source 
Original PR: #378
This commit is contained in:
Peter-Josef Meisch 2020-01-09 23:29:34 +01:00 committed by GitHub
parent 0843481cc5
commit 7294accaca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 38 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/org/springframework/data/elasticsearch/core/RequestFactory.java
View File

@ -552,6 +552,10 @@ class RequestFactory {
} }
if (query.getTrackTotalHits()) {
sourceBuilder.trackTotalHits(query.getTrackTotalHits());
}
request.source(sourceBuilder); request.source(sourceBuilder);
return request; return request;
} }
@ -714,6 +718,10 @@ class RequestFactory {
prepareNativeSearch(searchRequestBuilder, (NativeSearchQuery) query); prepareNativeSearch(searchRequestBuilder, (NativeSearchQuery) query);
} }
if (query.getTrackTotalHits()) {
searchRequestBuilder.setTrackTotalHits(query.getTrackTotalHits());
}
return searchRequestBuilder; return searchRequestBuilder;
} }

10
src/main/java/org/springframework/data/elasticsearch/core/query/AbstractQuery.java
View File

@ -54,6 +54,7 @@ abstract class AbstractQuery implements Query {
protected String preference; protected String preference;
protected Integer maxResults; protected Integer maxResults;
protected HighlightQuery highlightQuery; protected HighlightQuery highlightQuery;
private boolean trackTotalHits = false;
@Override @Override
public Sort getSort() { public Sort getSort() {
@ -208,4 +209,13 @@ abstract class AbstractQuery implements Query {
return Optional.ofNullable(highlightQuery); return Optional.ofNullable(highlightQuery);
} }
@Override
public void setTrackTotalHits(boolean trackTotalHits) {
this.trackTotalHits = trackTotalHits;
}
@Override
public boolean getTrackTotalHits() {
return trackTotalHits;
}
} }

20
src/main/java/org/springframework/data/elasticsearch/core/query/Query.java
View File

@ -202,4 +202,24 @@ public interface Query {
default Optional<HighlightQuery> getHighlightQuery() { default Optional<HighlightQuery> getHighlightQuery() {
return Optional.empty(); return Optional.empty();
} }
/**
* Sets the flag whether to set the Track_total_hits parameter on queries {@see <a href=
* "https://www.elastic.co/guide/en/elasticsearch/reference/7.0/search-request-track-total-hits.html">Elasticseacrh
* documentation</>}
*
* @param trackTotalHits the value to set.
* @since 4.0
*/
void setTrackTotalHits(boolean trackTotalHits);
/**
* Sets the flag whether to set the Track_total_hits parameter on queries {@see <a href=
* "https://www.elastic.co/guide/en/elasticsearch/reference/7.0/search-request-track-total-hits.html">Elasticseacrh
* documentation</>}
*
* @return the set value.
* @since 4.0
*/
boolean getTrackTotalHits();
} }