|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace DoctrineElastic\Query; |
|
4
|
|
|
|
|
5
|
|
|
use Doctrine\ORM\Query\AST\OrderByClause; |
|
6
|
|
|
use Doctrine\ORM\Query\AST\OrderByItem; |
|
7
|
|
|
use Doctrine\ORM\Query\AST\SelectStatement; |
|
8
|
|
|
use Doctrine\ORM\Query\AST\WhereClause; |
|
9
|
|
|
use DoctrineElastic\Elastic\ElasticQuery; |
|
10
|
|
|
use DoctrineElastic\Elastic\SearchParams; |
|
11
|
|
|
use DoctrineElastic\Hydrate\AnnotationEntityHydrator; |
|
12
|
|
|
use DoctrineElastic\Mapping\Field; |
|
13
|
|
|
use DoctrineElastic\Query\Walker\Helper\WalkerHelper; |
|
14
|
|
|
use DoctrineElastic\Query\Walker\WhereWalker; |
|
15
|
|
|
|
|
16
|
|
|
/** |
|
17
|
|
|
* Main walker for queries |
|
18
|
|
|
* |
|
19
|
|
|
* @uahtor Ands |
|
20
|
|
|
*/ |
|
21
|
|
|
class ElasticWalker { |
|
22
|
|
|
|
|
23
|
|
|
/** @var ElasticQuery */ |
|
24
|
|
|
protected $query; |
|
25
|
|
|
|
|
26
|
|
|
/** @var string */ |
|
27
|
|
|
private $_className; |
|
28
|
|
|
|
|
29
|
|
|
/** @var WalkerHelper */ |
|
30
|
|
|
private $walkerHelper; |
|
31
|
|
|
|
|
32
|
|
|
/** @var SelectStatement */ |
|
33
|
|
|
private $_ast; |
|
34
|
|
|
|
|
35
|
|
|
public function __construct(ElasticQuery $query, SelectStatement $AST, $className) { |
|
36
|
|
|
$this->query = $query; |
|
37
|
|
|
$this->walkerHelper = new WalkerHelper(); |
|
38
|
|
|
$this->_ast = $AST; |
|
39
|
|
|
$this->_className = $className; |
|
40
|
|
|
} |
|
41
|
|
|
|
|
42
|
|
|
/** |
|
43
|
|
|
* @return SearchParams |
|
44
|
|
|
*/ |
|
45
|
|
|
public function walkSelectStatement() { |
|
46
|
|
|
$searchParams = new SearchParams(); |
|
47
|
|
|
$size = $this->query->getMaxResults(); |
|
48
|
|
|
$offset = $this->query->getFirstResult(); |
|
49
|
|
|
|
|
50
|
|
|
$persister = $this->query->getEntityManager()->getUnitOfWork()->getEntityPersister($this->_className); |
|
51
|
|
|
$type = $persister->getEntityType(); |
|
52
|
|
|
|
|
53
|
|
|
$searchParams->setType($type->getName()); |
|
54
|
|
|
$searchParams->setIndex($type->getIndex()); |
|
55
|
|
|
|
|
56
|
|
|
if ($this->_ast->whereClause) { |
|
57
|
|
|
$this->walkWhereClause($this->_ast->whereClause, $searchParams); |
|
58
|
|
|
} |
|
59
|
|
|
|
|
60
|
|
|
if ($this->_ast->orderByClause) { |
|
61
|
|
|
$this->walkOrderByClause($this->_ast->orderByClause, $searchParams); |
|
62
|
|
|
} |
|
63
|
|
|
|
|
64
|
|
|
$searchParams->setSize($size); |
|
65
|
|
|
$searchParams->setFrom($offset); |
|
66
|
|
|
|
|
67
|
|
|
return $searchParams; |
|
68
|
|
|
} |
|
69
|
|
|
|
|
70
|
|
|
private function walkWhereClause(WhereClause $whereClause, SearchParams $searchParams) { |
|
71
|
|
|
$whereWalker = new WhereWalker($this->query, $this->_className, $this->walkerHelper); |
|
72
|
|
|
$whereWalker->walk($whereClause, $searchParams); |
|
73
|
|
|
} |
|
74
|
|
|
|
|
75
|
|
|
private function walkOrderByClause(OrderByClause $orderByClause, SearchParams $searchParams) { |
|
76
|
|
|
$sort = []; |
|
77
|
|
|
|
|
78
|
|
|
/** @var OrderByItem $item */ |
|
79
|
|
|
foreach ($orderByClause->orderByItems as $item) { |
|
80
|
|
|
$order = $item->type; |
|
81
|
|
|
$propertyName = $item->expression->field; |
|
82
|
|
|
$ESField = $this->getEntityElasticField($propertyName, $this->_className); |
|
83
|
|
|
|
|
84
|
|
|
if (is_null($ESField)) { |
|
85
|
|
|
continue; |
|
86
|
|
|
} |
|
87
|
|
|
|
|
88
|
|
|
$sort[$ESField->name] = strtolower($order); |
|
89
|
|
|
} |
|
90
|
|
|
|
|
91
|
|
|
if (!empty($sort)) { |
|
92
|
|
|
$searchParams->setSort($sort); |
|
93
|
|
|
} |
|
94
|
|
|
} |
|
95
|
|
|
|
|
96
|
|
|
/** |
|
97
|
|
|
* @param $propertyName |
|
98
|
|
|
* @param $className |
|
99
|
|
|
* @return Field |
|
100
|
|
|
*/ |
|
101
|
|
|
private function getEntityElasticField($propertyName, $className) { |
|
102
|
|
|
/** @var Field[] $fields */ |
|
103
|
|
|
$fields = (new AnnotationEntityHydrator())->extractSpecAnnotations($className, Field::class); |
|
104
|
|
|
|
|
105
|
|
|
return isset($fields[$propertyName]) ? $fields[$propertyName]: null; |
|
106
|
|
|
} |
|
107
|
|
|
} |
|
108
|
|
|
|
andsalves /
doctrine-elastic
