close #2976 (#2977)

Philippus · Dec 18, 2023 · 83162ce · 83162ce
1 parent 67ff0c2
commit 83162ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...4s-core/src/main/scala/com/sksamuel/elastic4s/requests/searches/SearchBodyBuilderFn.scala b/...4s-core/src/main/scala/com/sksamuel/elastic4s/requests/searches/SearchBodyBuilderFn.scala
@@ -170,7 +170,7 @@ object SearchBodyBuilderFn {
       builder.endObject()
     }
 
-    request.trackHits.map(builder.field("track_total_hits", _))
+    request.trackHits.map(builder.autofield("track_total_hits", _))
 
     if (request.ext.nonEmpty) {
       builder.autofield("ext", request.ext)

diff --git a/elastic4s-domain/src/main/scala/com/sksamuel/elastic4s/requests/searches/SearchRequest.scala b/elastic4s-domain/src/main/scala/com/sksamuel/elastic4s/requests/searches/SearchRequest.scala
@@ -53,7 +53,7 @@ case class SearchRequest(indexes: Indexes,
                          seqNoPrimaryTerm: Option[Boolean] = None,
                          profile: Option[Boolean] = None,
                          source: Option[String] = None,
-                         trackHits: Option[Boolean] = None,
+                         trackHits: Option[Any] = None,
                          allowPartialSearchResults: Option[Boolean] = None,
                          batchedReduceSize: Option[Int] = None,
                          typedKeys: Option[Boolean] = None,
@@ -105,6 +105,8 @@ case class SearchRequest(indexes: Indexes,
 
   def trackTotalHits(value: Boolean): SearchRequest = copy(trackHits = Some(value))
 
+  def trackTotalHits(value: Long): SearchRequest = copy(trackHits = Some(value))
+
   /** This method introduces zero or more script field definitions into the search construction
     *
     * @param fields zero or more [[ScriptField]] instances