add jsonb basic functions support

Author: pfilsx

src/ORM/Query/AST/Functions/JsonGetArrayElement.php

@@ -18,18 +18,15 @@ public function parse(Parser $parser): void
     {
         $parser->match(Lexer::T_IDENTIFIER);
         $parser->match(Lexer::T_OPEN_PARENTHESIS);
-
         $this->field = $parser->StringPrimary();
         $parser->match(Lexer::T_COMMA);
         $this->path[] = $parser->ArithmeticPrimary();
-
         if (!$parser->getLexer()->isNextToken(Lexer::T_CLOSE_PARENTHESIS)) {
             while ($parser->getLexer()->isNextToken(Lexer::T_COMMA)) {
                 $parser->match(Lexer::T_COMMA);
                 $this->path[] = $parser->ArithmeticPrimary();
             }
         }
-
         $parser->match(Lexer::T_CLOSE_PARENTHESIS);
     }
 }

src/ORM/Query/AST/Functions/JsonbConcat.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Pfilsx\PostgreSQLDoctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\AST\Node;
+use Doctrine\ORM\Query\Lexer;
+use Doctrine\ORM\Query\Parser;
+use Doctrine\ORM\Query\SqlWalker;
+
+
+/**
+ * Implementation of PostgreSql jsonb concatenation function
+ * @see https://www.postgresql.org/docs/current/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+ * @example JSONB_CONCAT(entity.field, entity2.field)
+ */
+final class JsonbConcat extends FunctionNode
+{
+    protected Node $json1;
+
+    protected Node $json2;
+    public function parse(Parser $parser): void
+    {
+        $parser->match(Lexer::T_IDENTIFIER);
+        $parser->match(Lexer::T_OPEN_PARENTHESIS);
+        $this->json1 = $parser->StringPrimary();
+        $parser->match(Lexer::T_COMMA);
+        $this->json2 = $parser->StringPrimary();
+        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
+    }
+
+
+    public function getSql(SqlWalker $sqlWalker): string
+    {
+        return "{$this->json1->dispatch($sqlWalker)} || {$this->json2->dispatch($sqlWalker)}";
+    }
+}

src/ORM/Query/AST/Functions/JsonbContains.php

@@ -0,0 +1,45 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Pfilsx\PostgreSQLDoctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\AST\InputParameter;
+use Doctrine\ORM\Query\AST\Literal;
+use Doctrine\ORM\Query\AST\Node;
+use Doctrine\ORM\Query\Lexer;
+use Doctrine\ORM\Query\Parser;
+use Doctrine\ORM\Query\SqlWalker;
+
+/**
+ * Implementation of PostgreSql jsonb contains function
+ * @see https://www.postgresql.org/docs/current/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+ * @example JSONB_CONTAINS(entity.field, '{"a": 1}')
+ */
+final class JsonbContains extends FunctionNode
+{
+    protected Node $field;
+
+    protected Node $value;
+
+    public function parse(Parser $parser): void
+    {
+        $parser->match(Lexer::T_IDENTIFIER);
+        $parser->match(Lexer::T_OPEN_PARENTHESIS);
+        $this->field = $parser->StringPrimary();
+        $parser->match(Lexer::T_COMMA);
+        $this->value = $parser->StringPrimary();
+        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
+    }
+
+    public function getSql(SqlWalker $sqlWalker): string
+    {
+        $value = $this->value->dispatch($sqlWalker);
+        if ($this->value instanceof Literal || $this->value instanceof InputParameter) {
+            $value .= '::jsonb';
+        }
+
+        return "{$this->field->dispatch($sqlWalker)} @> $value";
+    }
+}

src/ORM/Query/AST/Functions/JsonbKeyExists.php

@@ -0,0 +1,38 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Pfilsx\PostgreSQLDoctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\AST\Node;
+use Doctrine\ORM\Query\Lexer;
+use Doctrine\ORM\Query\Parser;
+use Doctrine\ORM\Query\SqlWalker;
+
+/**
+ * Implementation of PostgreSql jsonb contains function
+ * @see https://www.postgresql.org/docs/current/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+ * @example JSONB_KEY_EXISTS(entity.field, 'a')
+ */
+final class JsonbKeyExists extends FunctionNode
+{
+    protected Node $field;
+
+    protected Node $key;
+
+    public function parse(Parser $parser): void
+    {
+        $parser->match(Lexer::T_IDENTIFIER);
+        $parser->match(Lexer::T_OPEN_PARENTHESIS);
+        $this->field = $parser->StringPrimary();
+        $parser->match(Lexer::T_COMMA);
+        $this->key = $parser->StringPrimary();
+        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
+    }
+
+    public function getSql(SqlWalker $sqlWalker): string
+    {
+        return "{$this->field->dispatch($sqlWalker)} ? {$this->key->dispatch($sqlWalker)}";
+    }
+}

src/ORM/Query/AST/Functions/JsonbRemove.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Pfilsx\PostgreSQLDoctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Functions\FunctionNode;
+use Doctrine\ORM\Query\AST\Node;
+use Doctrine\ORM\Query\Lexer;
+use Doctrine\ORM\Query\Parser;
+use Doctrine\ORM\Query\SqlWalker;
+
+
+/**
+ * Implementation of PostgreSql jsonb remove by key function
+ * @see https://www.postgresql.org/docs/current/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+ * @example JSONB_REMOVE(entity.field, 'a')
+ */
+final class JsonbRemove extends FunctionNode
+{
+    protected Node $jsonb;
+
+    protected Node $key;
+    public function parse(Parser $parser): void
+    {
+        $parser->match(Lexer::T_IDENTIFIER);
+        $parser->match(Lexer::T_OPEN_PARENTHESIS);
+        $this->jsonb = $parser->StringPrimary();
+        $parser->match(Lexer::T_COMMA);
+        $this->key = $parser->StringPrimary();
+        $parser->match(Lexer::T_CLOSE_PARENTHESIS);
+    }
+
+
+    public function getSql(SqlWalker $sqlWalker): string
+    {
+        return "{$this->jsonb->dispatch($sqlWalker)} - {$this->key->dispatch($sqlWalker)}";
+    }
+}