CCET-768: add track total hits to os-query lib (#15)

Author: mikesmitharoo

search.go

@@ -15,18 +15,19 @@ import (
 // Not all features of the search API are currently supported, but a request can
 // currently include a query, aggregations, and more.
 type SearchRequest struct {
-	aggs        []Aggregation
-	explain     *bool
-	from        *uint64
-	highlight   Mappable
-	searchAfter []interface{}
-	postFilter  Mappable
-	query       Mappable
-	size        *uint64
-	sorts       Sorts
-	source      Source
-	timeout     *time.Duration
-	collapse    *CollapseRequest
+	aggs           []Aggregation
+	explain        *bool
+	from           *uint64
+	highlight      Mappable
+	searchAfter    []interface{}
+	postFilter     Mappable
+	query          Mappable
+	size           *uint64
+	sorts          Sorts
+	source         Source
+	timeout        *time.Duration
+	collapse       *CollapseRequest
+	trackTotalHits any
 }
 
 // Search creates a new SearchRequest object, to be filled via method chaining.
@@ -84,6 +85,12 @@ func (req *SearchRequest) Explain(b bool) *SearchRequest {
 	return req
 }
 
+// TrackTotalHits sets whether the search should count the total number of hits past the limit. Supported values is a number [range] or a bool
+func (req *SearchRequest) TrackTotalHits(b any) *SearchRequest {
+	req.trackTotalHits = b
+	return req
+}
+
 // Timeout sets a timeout for the request.
 func (req *SearchRequest) Timeout(dur time.Duration) *SearchRequest {
 	req.timeout = &dur
@@ -163,6 +170,9 @@ func (req *SearchRequest) Map() map[string]interface{} {
 	if req.collapse != nil {
 		m["collapse"] = req.collapse.Map()
 	}
+	if req.trackTotalHits != nil {
+		m["track_total_hits"] = req.trackTotalHits
+	}
 
 	if req.source.disabled {
 		m["_source"] = false

search_test.go

@@ -77,7 +77,8 @@ func TestSearchMaps(t *testing.T) {
 				).
 				SourceIncludes("field_1", "field_2").
 				SourceExcludes("field_3").
-				Timeout(time.Duration(20000000000)),
+				Timeout(time.Duration(20000000000)).
+				TrackTotalHits(false),
 			map[string]interface{}{
 				"query": map[string]interface{}{
 					"bool": map[string]interface{}{
@@ -138,6 +139,7 @@ func TestSearchMaps(t *testing.T) {
 					"includes": []string{"field_1", "field_2"},
 					"excludes": []string{"field_3"},
 				},
+				"track_total_hits": false,
 			},
 		},
 		{