commit

Author: degami

app/base/tools/Search/Manager.php

@@ -254,6 +254,21 @@ public function ensureIndex(): bool
         return true;
     }
 
+    /**
+     * Drops index
+     * 
+     * @return static
+     */
+    public function dropIndex() : static
+    {
+        try {
+            $client = $this->getClient();
+            $client->indices()->delete(['index' => $this->getIndexName()]);
+        } catch (\Throwable $e) {}
+
+        return $this;
+    }
+
     /**
      * Prepares data from a frontend model to be indexed in Elasticsearch.
      *
@@ -449,18 +464,24 @@ public function searchData($page = 0, $pageSize = self::RESULTS_PER_PAGE, bool $
 
         if ($onlyAggregations) {
             $searchParams['body']['size'] = 0;
+            unset ($http_response_header['body']['from']);
+            $searchParams['body']['aggs'] = $this->getAggregationsArray();
         } else {
-            if (is_array($this->source)) {
-                $searchParams['body']['_source'] = $this->source;            
+            if (is_array($this->getSource())) {
+                $searchParams['body']['_source'] = $this->getSource();            
             }
 
-            if (is_array($this->sort)) {
-                $searchParams['body']['sort'] = $this->sort;
+            if (is_array($this->getSort())) {
+                $searchParams['body']['sort'] = $this->getSort();
             }    
         }
 
         $search_result = $this->getClient()->search($searchParams);
 
+        if ($onlyAggregations) {
+            return $search_result['aggregations'];
+        }
+
         $total = $search_result['hits']['total']['value'] ?? 0;
         $hits = $search_result['hits']['hits'] ?? [];
         $docs = array_map(function ($el) {
@@ -470,6 +491,16 @@ public function searchData($page = 0, $pageSize = self::RESULTS_PER_PAGE, bool $
         return ['total' => $total, 'docs' => $docs];
     }
 
+    /**
+     * return aggregated data
+     * 
+     * @return array
+     */
+    public function searchAggregatedData() : array
+    {
+        return $this->searchData(onlyAggregations: true);
+    }
+
     /**
      * Adds a condition for the "filter" clause of the Elasticsearch query.
      *
@@ -540,6 +571,28 @@ public function addGenericCondition(array $condition, string $boolType) : static
         return $this;
     }
 
+    /**
+     * Sets sorting array
+     * 
+     * @param array $sort Sort array
+     * @return static
+     */
+    public function setSort(array $sort) : static
+    {
+        $this->sort = $sort;
+        return $this;
+    }
+
+    /**
+     * Gets sorting array
+     * 
+     * @return array|null
+     */
+    public function getSort() : ?array
+    {
+        return $this->sort;
+    }
+
     /**
      * Resets the sorting for the query.
      *