Skip approcimation when track total hits sets to true

Signed-off-by: Prudhvi Godithi <[email protected]>

Author: prudhvigodithi

server/src/main/java/org/opensearch/search/approximate/ApproximateMatchAllQuery.java

@@ -36,6 +36,10 @@ protected boolean canApproximate(SearchContext context) {
         if (context.aggregations() != null) {
             return false;
         }
+        // When "track_total_hits": true,
+        if (context.trackTotalHitsUpTo() == SearchContext.TRACK_TOTAL_HITS_ACCURATE) {
+            return false;
+        }
 
         if (context.request() != null && context.request().source() != null && context.innerHits().getInnerHits().isEmpty()) {
             FieldSortBuilder primarySortField = FieldSortBuilder.getPrimaryFieldSortOrNull(context.request().source());

server/src/main/java/org/opensearch/search/approximate/ApproximatePointRangeQuery.java

@@ -440,6 +440,11 @@ public boolean canApproximate(SearchContext context) {
         if (context.aggregations() != null) {
             return false;
         }
+        // When "track_total_hits": true,
+        if (context.trackTotalHitsUpTo() == SearchContext.TRACK_TOTAL_HITS_ACCURATE) {
+            return false;
+        }
+
         // size 0 could be set for caching
         if (context.from() + context.size() == 0) {
             this.setSize(SearchContext.DEFAULT_TRACK_TOTAL_HITS_UP_TO);