Merge pull request #10 from olekjs/dev-release-v1.8.0

Add ability to select specific fields

Author: olekjs

src/Builder/Builder.php

@@ -47,6 +47,8 @@ class Builder implements BuilderInterface
 
     private array $sort;
 
+    private array $select;
+
     private array $body = [];
 
     public function __construct(private readonly ClientInterface $client)
@@ -248,6 +250,17 @@ public function rawSort(array $sort): self
         return $this;
     }
 
+    public function select(mixed $fields): self
+    {
+        $fields = is_array($fields) ? $fields : func_get_args();
+
+        foreach ($fields as $field) {
+            $this->select[] = $field;
+        }
+
+        return $this;
+    }
+
     /**
      * @throws IndexNotFoundResponseException
      * @throws FindResponseException
@@ -394,12 +407,21 @@ public function getSort(): array
         return $this->sort;
     }
 
-    private function performSearchBody(): void
+    public function getSelect(): array
+    {
+        return $this->select;
+    }
+
+    public function performSearchBody(): void
     {
         if (isset($this->query)) {
             $this->body['query'] = $this->query;
         }
 
+        if (isset($this->select)) {
+            $this->body['_source'] = $this->select;
+        }
+
         if (empty($this->body)) {
             $this->body['query'] = [
                 'match_all' => (object)[]

src/Contracts/BuilderInterface.php

@@ -68,6 +68,8 @@ public function rawQuery(array $query): self;
 
     public function rawSort(array $sort): self;
 
+    public function select(mixed $fields): self;
+
     /**
      * @throws IndexNotFoundResponseException
      * @throws FindResponseException
@@ -135,4 +137,8 @@ public function getBody(): array;
     public function getQuery(): array;
 
     public function getSort(): array;
+
+    public function getSelect(): array;
+
+    public function performSearchBody(): void;
 }

tests/Unit/BuilderTest.php

@@ -543,4 +543,20 @@ public function testDeleteMethod(): void
 
         $this->assertInstanceOf(IndexResponseDto::class, $result);
     }
+
+    public function testSelectMethod(): void
+    {
+        $expected = ['test', 'test1', 'test2', 'test3', 'test4'];
+
+        $builder = Builder::query()
+            ->select('test')
+            ->select(['test1', 'test2'])
+            ->select('test3', 'test4');
+
+        $this->assertSame($expected, $builder->getSelect());
+
+        $builder->performSearchBody();
+
+        $this->assertSame(['_source' => $expected], $builder->getBody());
+    }
 }