cleanup

Jeroen-G · Apr 7, 2024 · bdc2787 · bdc2787
1 parent 51ea063
commit bdc2787
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/Application/Results.php b/src/Application/Results.php
@@ -5,14 +5,15 @@
 namespace JeroenG\Explorer\Application;
 
 use Countable;
+use Elastic\Elasticsearch\Response\Elasticsearch;
 
 class Results implements Countable
 {
     private array $rawResults;
 
-    public function __construct(array $rawResults)
+    public function __construct(array|Elasticsearch $rawResults)
     {
-        $this->rawResults = $rawResults;
+        $this->rawResults = $rawResults instanceof Elasticsearch ? $rawResults->asArray() : $rawResults;
     }
 
     public function hits(): array

diff --git a/src/Infrastructure/Elastic/Finder.php b/src/Infrastructure/Elastic/Finder.php
@@ -25,6 +25,6 @@ public function find(): Results
 
         $rawResults = $this->client->search($query);
 
-        return new Results($rawResults->asArray());
+        return new Results($rawResults);
     }
 }