Support for setting _source

digiaonline · Jan 19, 2018 · aeb8413 · aeb8413
1 parent 368bbff
commit aeb8413
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 0 deletions.
diff --git a/src/Search/Search.php b/src/Search/Search.php
@@ -21,6 +21,11 @@ class Search
      */
     private $query;
 
+    /**
+     * @var array
+     */
+    private $source;
+
     /**
      * @var Sort
      */
@@ -135,6 +140,24 @@ public function getSort()
         return $this->sort;
     }
 
+    /**
+     * @param array $source
+     *
+     * @return $this
+     */
+    public function setSource(array $source)
+    {
+        $this->source = $source;
+        return $this;
+    }
+
+    /**
+     * @return array
+     */
+    public function getSource()
+    {
+        return $this->source;
+    }
 
     /**
      * @return AggregationCollection
@@ -253,6 +276,10 @@ public function buildBody()
             }
         }
 
+        if (!empty($this->getSource())) {
+            $body['_source'] = $this->getSource();
+        }
+
         $aggregations = $this->getAggregations();
         if ($aggregations->count() > 0) {
             $body['aggs'] = $aggregations->toArray();

diff --git a/tests/Search/SearchTest.php b/tests/Search/SearchTest.php
@@ -86,6 +86,9 @@ public function testSetterGetter()
         $this->search->setSize(100);
         $this->assertEquals(100, $this->search->getSize());
 
+        $this->search->setSource(['*']);
+        $this->assertEquals(['*'], $this->search->getSource());
+
         $this->search->setSort($this->sort);
         $this->assertInstanceOf(Sort::class, $this->search->getSort());
 
@@ -173,6 +176,26 @@ public function testToArray()
         ], $this->search->buildBody());
     }
 
+    public function testToArrayWithSource()
+    {
+        $this->search = $this->service->createSearch();
+        $this->search->setPage(1);
+        $this->search->setSize(100);
+        $this->search->setSource(['id', 'title', 'description']);
+
+        $this->assertEquals([
+            'query' => ['match_all' => new \stdClass()],
+            'size' => 100,
+            'from' => 0,
+            '_source' => [
+                'id',
+                'title',
+                'description'
+            ],
+        ], $this->search->buildBody());
+    }
+
+
     /**
      * Test adding an array of aggregation to Search
      */