-
Notifications
You must be signed in to change notification settings - Fork 0
/
TsHeadline.php
61 lines (51 loc) · 1.92 KB
/
TsHeadline.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
<?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 TS_HEADLINE() function.
*
* @see https://www.postgresql.org/docs/current/textsearch-controls.html#TEXTSEARCH-HEADLINE
*
* @example TS_HEADLINE(entity.field, TO_TSQUERY('text'))
* @example TS_HEADLINE('some text', TO_TSQUERY('text'))
* @example TS_HEADLINE('english', entity.field, TO_TSQUERY('text'))
* @example TS_HEADLINE('english', entity.field, TO_TSQUERY('text'), 'MaxWords=7, MinWords=3')
*/
class TsHeadline extends FunctionNode
{
private Node $node1;
private Node $node2;
private ?Node $node3 = null;
private ?Node $node4 = null;
public function parse(Parser $parser): void
{
$parser->match(Lexer::T_IDENTIFIER);
$parser->match(Lexer::T_OPEN_PARENTHESIS);
$this->node1 = $parser->StringPrimary();
$parser->match(Lexer::T_COMMA);
$this->node2 = $parser->StringPrimary();
if ($parser->getLexer()->isNextToken(Lexer::T_COMMA)) {
$parser->match(Lexer::T_COMMA);
$this->node3 = $parser->StringPrimary();
}
if ($parser->getLexer()->isNextToken(Lexer::T_COMMA)) {
$parser->match(Lexer::T_COMMA);
$this->node4 = $parser->StringPrimary();
}
$parser->match(Lexer::T_CLOSE_PARENTHESIS);
}
public function getSql(SqlWalker $sqlWalker): string
{
return sprintf('TS_HEADLINE(%s, %s%s%s)',
$this->node1->dispatch($sqlWalker),
$this->node2->dispatch($sqlWalker),
$this->node3 !== null ? ', ' . $this->node3->dispatch($sqlWalker) : '',
$this->node4 !== null ? ', ' . $this->node4->dispatch($sqlWalker) : ''
);
}
}