'track_total_hits' option in the Connection for default Query Builder

Author: pdphilip

src/Connection.php

@@ -123,6 +123,7 @@ private function sanitizeConfig(): void
                     'cert_password' => null,
                 ],
                 'options' => [
+                    'track_total_hits' => null, // null -> skips - max 10k by default; true -> full values; false -> no hit tracking, returns -1; int -> max hit tracking val, ex 20000
                     'bypass_map_validation' => false, // This skips the safety checks for Elastic Specific queries.
                     'logging' => false,
                     'ssl_verification' => true,
@@ -162,6 +163,8 @@ public function setOptions(): void
     {
         $this->allowIdSort = $this->config['options']['allow_id_sort'] ?? false;
 
+        $this->options()->add('track_total_hits', $this->config['options']['track_total_hits'] ?? null);
+
         $this->options()->add('bypass_map_validation', $this->config['options']['bypass_map_validation'] ?? null);
 
         if (isset($this->config['options']['ssl_verification'])) {

src/Query/Builder.php

@@ -13,6 +13,8 @@
 use Illuminate\Contracts\Support\Arrayable;
 use Illuminate\Database\Query\Builder as BaseBuilder;
 use Illuminate\Database\Query\Expression;
+use Illuminate\Database\Query\Grammars\Grammar;
+use Illuminate\Database\Query\Processors\Processor;
 use Illuminate\Support\Arr;
 use Illuminate\Support\Collection;
 use Illuminate\Support\Str;
@@ -108,6 +110,12 @@ class Builder extends BaseBuilder
 
     protected ?MetaDTO $metaTransfer = null;
 
+    public function __construct(...$args)
+    {
+        parent::__construct(...$args);
+        $this->applyConnectionOptions();
+    }
+
     public function __call($method, $parameters)
     {
         if (Str::startsWith($method, 'filterWhere')) {
@@ -117,7 +125,7 @@ public function __call($method, $parameters)
         return parent::__call($method, $parameters);
     }
 
-    public function toDsl(): array
+    public function toDsl(): array|string
     {
         $this->applyBeforeQueryCallbacks();
 
@@ -129,6 +137,14 @@ public function toSql(): array|string
         return $this->toDsl();
     }
 
+    private function applyConnectionOptions()
+    {
+        $trackTotalHits = $this->connection->options()->get('track_total_hits');
+        if ($trackTotalHits !== null) {
+            $this->bodyParameters['track_total_hits'] = $trackTotalHits;
+        }
+    }
+
     // ======================================================================
     // Inherited Methods
     // ======================================================================
@@ -2417,13 +2433,20 @@ public function withAnalyzer(string $analyzer): self
     public function withTrackTotalHits(bool|int|null $val = true): self
     {
         if ($val === null) {
-            return $this;
+            return $this->withoutTrackTotalHits();
         }
         $this->bodyParameters['track_total_hits'] = $val;
 
         return $this;
     }
 
+    public function withoutTrackTotalHits(): self
+    {
+        unset($this->bodyParameters['track_total_hits']);
+
+        return $this;
+    }
+
     // ----------------------------------------------------------------------
     // Internal Operations
     // ----------------------------------------------------------------------