Function queries

* QunctionScoreQuery + FunctionsQuery

* lint

* composer.json

* composer.json

* lin+stan

* composer.json

* composer.json

* composer.json original

* clean up

* setters

* lint

Co-authored-by: iana.culenova <jana.culenova@websupport.sk>

Author: janaculenova

composer.json

@@ -50,4 +50,4 @@
             "symfony/thanks": false
         }
     }
-}
+}

src/Query/FunctionScoreQuery.php

@@ -0,0 +1,73 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Erichard\ElasticQueryBuilder\Query;
+
+use Erichard\ElasticQueryBuilder\Contracts\QueryInterface;
+
+class FunctionScoreQuery implements QueryInterface
+{
+    protected ?string $boost = null;
+
+    protected ?string $boostMode = null;
+
+    private ?array $functions = null;
+
+    /**
+     * @param array[]|string[] $fields
+     */
+    public function __construct(
+        protected array $fields,
+        protected string $query,
+    ) {
+    }
+
+    public function setBoost(string $boost): self
+    {
+        $this->boost = $boost;
+
+        return $this;
+    }
+
+    public function setBoostMode(string $boostMode): self
+    {
+        $this->boostMode = $boostMode;
+
+        return $this;
+    }
+
+    public function setFunctions(array $functions): self
+    {
+        $this->functions = $functions;
+
+        return $this;
+    }
+
+    public function build(): array
+    {
+        $build = [];
+        if (null !== $this->boostMode) {
+            $build['boost_mode'] = $this->boostMode;
+        }
+
+        if (null !== $this->functions) {
+            $build['functions'] = $this->functions;
+        }
+
+        $build['query'] = [
+            'query_string' => [
+                'query' => $this->query,
+                'fields' => $this->fields,
+            ],
+        ];
+
+        if (null !== $this->boost) {
+            $build['query']['query_string'] = $this->boost;
+        }
+
+        return [
+            'function_score' => $build,
+        ];
+    }
+}

src/Query/FunctionsQuery.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Erichard\ElasticQueryBuilder\Query;
+
+use Erichard\ElasticQueryBuilder\Contracts\QueryInterface;
+
+class FunctionsQuery implements QueryInterface
+{
+    protected ?float $weight = null;
+
+    public function __construct(
+        protected string $field,
+    ) {
+    }
+
+    public function setWeight(float $weight): self
+    {
+        $this->weight = $weight;
+
+        return $this;
+    }
+
+    public function build(): array
+    {
+        $functions = [];
+        $functions['filter'] = [
+            'term' => [
+                '_index' => $this->field,
+            ],
+        ];
+
+        if (null !== $this->weight) {
+            $functions['weight'] = $this->weight;
+        }
+
+        return $functions;
+    }
+}

src/Query/Query.php

@@ -76,6 +76,20 @@ public static function multiMatch(array $fields, string $query): MultiMatchQuery
         return new MultiMatchQuery($fields, $query);
     }
 
+    public static function functionScoreQuery(
+        array $fields,
+        string $query,
+        string $boost = null,
+        string $boostMode = null
+    ): FunctionScoreQuery {
+        return new FunctionScoreQuery($fields, $query, $boost, $boostMode);
+    }
+
+    public static function functionsQuery(?string $field = null, ?float $weight = null): FunctionsQuery
+    {
+        return new FunctionsQuery($field, $weight);
+    }
+
     /**
      * @param float[]|int[] $position
      */