Distinct search after keys & PIT ops

Author: pdphilip

src/Eloquent/Builder.php

@@ -68,7 +68,8 @@ class Builder extends BaseEloquentBuilder
         // ES
         'todsl',
         'bucket',
-        'bucketAggregation',
+        'bucketaggregation',
+        'openpit',
     ];
 
     /**

src/Query/Builder.php

@@ -2288,6 +2288,9 @@ public function processedRaw($dsl): ?array
 
     public function viaPit($pid, $afterKey): self
     {
+        if (! $pid) {
+            $pid = $this->openPit();
+        }
         $this->pitId = $pid;
         $this->searchAfter = $afterKey;
 

src/Query/Processor.php

@@ -57,6 +57,12 @@ public function processInsertGetId(Builder|BaseBuilder $query, $sql, $values, $s
 
     public function processDistinctAggregations($result, $columns, $withCount): Collection
     {
+        if (! empty($result['hits']['hits']) && is_array($result['hits']['hits'])) {
+            $last = collect($result['hits']['hits'])->last();
+            if (! empty($last['sort'])) {
+                $this->query->getMetaTransfer()->set('after_key', $last['sort']);
+            }
+        }
         $keys = [];
         foreach ($columns as $column) {
             $keys[] = 'by_'.$column;