Skip to content

Commit d25f7fe

Browse files
authored
DATAES-985 - Add builder method for track_total_hits to NativeSearchQueryBuilder.
Original PR: spring-projects#569
1 parent fcb8a9b commit d25f7fe

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

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

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ public class NativeSearchQueryBuilder {
6767
@Nullable private IndicesOptions indicesOptions;
6868
@Nullable private String preference;
6969
@Nullable private Integer maxResults;
70+
@Nullable private Boolean trackTotalHits;
7071

7172
public NativeSearchQueryBuilder withQuery(QueryBuilder queryBuilder) {
7273
this.queryBuilder = queryBuilder;
@@ -173,6 +174,14 @@ public NativeSearchQueryBuilder withMaxResults(Integer maxResults) {
173174
return this;
174175
}
175176

177+
/**
178+
* @since 4.2
179+
*/
180+
public NativeSearchQueryBuilder withTrackTotalHits(Boolean trackTotalHits) {
181+
this.trackTotalHits = trackTotalHits;
182+
return this;
183+
}
184+
176185
public NativeSearchQuery build() {
177186

178187
NativeSearchQuery nativeSearchQuery = new NativeSearchQuery(queryBuilder, filterBuilder, sortBuilders,
@@ -233,6 +242,8 @@ public NativeSearchQuery build() {
233242
nativeSearchQuery.setMaxResults(maxResults);
234243
}
235244

245+
nativeSearchQuery.setTrackTotalHits(trackTotalHits);
246+
236247
return nativeSearchQuery;
237248
}
238249
}

0 commit comments

Comments
 (0)