From d25f7fe7b8c0ecad4af3aab1f06feba7da3f92b7 Mon Sep 17 00:00:00 2001
From: Peter-Josef Meisch <pj.meisch@sothawo.com>
Date: Sun, 6 Dec 2020 13:05:33 +0100
Subject: [PATCH] DATAES-985 - Add builder method for track_total_hits to
 NativeSearchQueryBuilder.

Original PR: #569
---
 .../core/query/NativeSearchQueryBuilder.java          | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQueryBuilder.java b/src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQueryBuilder.java
index c83abd00e..499c46444 100755
--- a/src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQueryBuilder.java
+++ b/src/main/java/org/springframework/data/elasticsearch/core/query/NativeSearchQueryBuilder.java
@@ -67,6 +67,7 @@ public class NativeSearchQueryBuilder {
 	@Nullable private IndicesOptions indicesOptions;
 	@Nullable private String preference;
 	@Nullable private Integer maxResults;
+	@Nullable private Boolean trackTotalHits;
 
 	public NativeSearchQueryBuilder withQuery(QueryBuilder queryBuilder) {
 		this.queryBuilder = queryBuilder;
@@ -173,6 +174,14 @@ public class NativeSearchQueryBuilder {
 		return this;
 	}
 
+	/**
+	 * @since 4.2
+	 */
+	public NativeSearchQueryBuilder withTrackTotalHits(Boolean trackTotalHits) {
+		this.trackTotalHits = trackTotalHits;
+		return this;
+	}
+
 	public NativeSearchQuery build() {
 
 		NativeSearchQuery nativeSearchQuery = new NativeSearchQuery(queryBuilder, filterBuilder, sortBuilders,
@@ -233,6 +242,8 @@ public class NativeSearchQueryBuilder {
 			nativeSearchQuery.setMaxResults(maxResults);
 		}
 
+		nativeSearchQuery.setTrackTotalHits(trackTotalHits);
+
 		return nativeSearchQuery;
 	}
 }