-
Notifications
You must be signed in to change notification settings - Fork 143
/
QuerySubscriber.php
147 lines (137 loc) · 6.55 KB
/
QuerySubscriber.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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
<?php
namespace Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;
use Doctrine\ORM\Query\Parameter;
use Knp\Component\Pager\Event\ItemsEvent;
use Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\Helper as QueryHelper;
use Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\CountWalker;
use Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\WhereInWalker;
use Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\LimitSubqueryWalker;
use Doctrine\ORM\Query;
use Doctrine\ORM\Tools\Pagination\CountWalker as DoctrineCountWalker;
use Doctrine\ORM\Tools\Pagination\WhereInWalker as DoctrineWhereInWalker;
use Doctrine\ORM\Tools\Pagination\LimitSubqueryWalker as DoctrineLimitSubqueryWalker;
/**
* @deprecated see UsesPaginator
**/
class QuerySubscriber implements EventSubscriberInterface
{
/**
* Used if user set the count manually
*/
const HINT_COUNT = 'knp_paginator.count';
public function items(ItemsEvent $event)
{
if ($event->target instanceof Query) {
// process count
$useDoctrineWalkers = false;
$useDoctrineOutputWalker = false;
if (version_compare(\Doctrine\ORM\Version::VERSION, '2.3.0', '>=')) {
$useDoctrineWalkers = true;
$useDoctrineOutputWalker = true;
} else if (version_compare(\Doctrine\ORM\Version::VERSION, '2.2.0', '>=')) {
$useDoctrineWalkers = true;
}
if (($count = $event->target->getHint(self::HINT_COUNT)) !== false) {
$event->count = intval($count);
} else {
$countQuery = QueryHelper::cloneQuery($event->target);
if ($useDoctrineOutputWalker) {
$treeWalker = 'Doctrine\ORM\Tools\Pagination\CountOutputWalker';
$countQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, $treeWalker);
} else if ($useDoctrineWalkers) {
QueryHelper::addCustomTreeWalker($countQuery, 'Doctrine\ORM\Tools\Pagination\CountWalker');
} else {
$treeWalker = 'Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\CountWalker';
QueryHelper::addCustomTreeWalker($countQuery, $treeWalker);
}
if ($useDoctrineWalkers) {
$countQuery->setHint(
DoctrineCountWalker::HINT_DISTINCT,
$event->options[PaginatorInterface::DISTINCT]
);
} else {
$countQuery->setHint(
CountWalker::HINT_DISTINCT,
$event->options[PaginatorInterface::DISTINCT]
);
}
$countQuery
->setFirstResult(null)
->setMaxResults(null)
;
$countQuery->getEntityManager()->getConfiguration()->addCustomHydrationMode('asIs',
'Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\AsIsHydrator');
$countResult = $countQuery->getResult('asIs');
$event->count = intval(current(current($countResult)));
}
// process items
$result = null;
if ($event->count) {
if ($event->options[PaginatorInterface::DISTINCT]) {
$limitSubQuery = QueryHelper::cloneQuery($event->target);
$limitSubQuery
->setFirstResult($event->getOffset())
->setMaxResults($event->getLimit())
->useQueryCache(false)
;
QueryHelper::addCustomTreeWalker($limitSubQuery, $useDoctrineWalkers ?
'Doctrine\ORM\Tools\Pagination\LimitSubqueryWalker' :
'Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\LimitSubqueryWalker'
);
$ids = array_map('current', $limitSubQuery->getScalarResult());
// create where-in query
$whereInQuery = QueryHelper::cloneQuery($event->target);
QueryHelper::addCustomTreeWalker($whereInQuery, $useDoctrineWalkers ?
'Doctrine\ORM\Tools\Pagination\WhereInWalker' :
'Knp\Component\Pager\Event\Subscriber\Paginate\Doctrine\ORM\Query\WhereInWalker'
);
$whereInQuery
->setHint($useDoctrineWalkers ?
DoctrineWhereInWalker::HINT_PAGINATOR_ID_COUNT :
WhereInWalker::HINT_PAGINATOR_ID_COUNT, count($ids)
)
->setFirstResult(null)
->setMaxResults(null)
;
if (version_compare(\Doctrine\ORM\Version::VERSION, '2.3.0', '>=') && count($ids) > 0) {
$whereInQuery->setParameter(WhereInWalker::PAGINATOR_ID_ALIAS, $ids);
} else {
$type = $limitSubQuery->getHint($useDoctrineWalkers ?
DoctrineLimitSubqueryWalker::IDENTIFIER_TYPE :
LimitSubqueryWalker::IDENTIFIER_TYPE
);
$idAlias = $useDoctrineWalkers ?
DoctrineWhereInWalker::PAGINATOR_ID_ALIAS :
WhereInWalker::PAGINATOR_ID_ALIAS
;
foreach ($ids as $i => $id) {
$whereInQuery->setParameter(
$idAlias . '_' . ++$i,
$id,
$type->getName()
);
}
}
$result = $whereInQuery->execute();
} else {
$event->target
->setFirstResult($event->getOffset())
->setMaxResults($event->getLimit())
;
$result = $event->target->execute();
}
} else {
$result = array(); // count is 0
}
$event->items = $result;
$event->stopPropagation();
}
}
public static function getSubscribedEvents()
{
return array(
'knp_pager.items' => array('items', 0)
);
}
}