@@ -15,252 +15,252 @@ |
||
15 | 15 | */ |
16 | 16 | class FindObjectsFromRawSqlQueryFactory implements QueryFactory |
17 | 17 | { |
18 | - /** |
|
19 | - * @var array[] |
|
20 | - */ |
|
21 | - protected $columnDescriptors; |
|
22 | - /** |
|
23 | - * @var Schema |
|
24 | - */ |
|
25 | - private $schema; |
|
26 | - /** |
|
27 | - * @var string |
|
28 | - */ |
|
29 | - private $sql; |
|
30 | - /** |
|
31 | - * @var string |
|
32 | - */ |
|
33 | - private $sqlCount; |
|
34 | - /** |
|
35 | - * @var string |
|
36 | - */ |
|
37 | - private $processedSql; |
|
38 | - /** |
|
39 | - * @var string |
|
40 | - */ |
|
41 | - private $processedSqlCount; |
|
42 | - /** |
|
43 | - * @var TDBMService |
|
44 | - */ |
|
45 | - private $tdbmService; |
|
46 | - /** |
|
47 | - * @var string |
|
48 | - */ |
|
49 | - private $mainTable; |
|
18 | + /** |
|
19 | + * @var array[] |
|
20 | + */ |
|
21 | + protected $columnDescriptors; |
|
22 | + /** |
|
23 | + * @var Schema |
|
24 | + */ |
|
25 | + private $schema; |
|
26 | + /** |
|
27 | + * @var string |
|
28 | + */ |
|
29 | + private $sql; |
|
30 | + /** |
|
31 | + * @var string |
|
32 | + */ |
|
33 | + private $sqlCount; |
|
34 | + /** |
|
35 | + * @var string |
|
36 | + */ |
|
37 | + private $processedSql; |
|
38 | + /** |
|
39 | + * @var string |
|
40 | + */ |
|
41 | + private $processedSqlCount; |
|
42 | + /** |
|
43 | + * @var TDBMService |
|
44 | + */ |
|
45 | + private $tdbmService; |
|
46 | + /** |
|
47 | + * @var string |
|
48 | + */ |
|
49 | + private $mainTable; |
|
50 | 50 | |
51 | - /** |
|
52 | - * FindObjectsFromRawSqlQueryFactory constructor. |
|
53 | - * @param TDBMService $tdbmService |
|
54 | - * @param Schema $schema |
|
55 | - * @param string $mainTable |
|
56 | - * @param string $sql |
|
57 | - * @param string $sqlCount |
|
58 | - */ |
|
59 | - public function __construct(TDBMService $tdbmService, Schema $schema, string $mainTable, string $sql, string $sqlCount = null) |
|
60 | - { |
|
61 | - $this->sql = $sql; |
|
62 | - $this->sqlCount = $sqlCount; |
|
63 | - $this->tdbmService = $tdbmService; |
|
64 | - $this->schema = $schema; |
|
65 | - $this->mainTable = $mainTable; |
|
51 | + /** |
|
52 | + * FindObjectsFromRawSqlQueryFactory constructor. |
|
53 | + * @param TDBMService $tdbmService |
|
54 | + * @param Schema $schema |
|
55 | + * @param string $mainTable |
|
56 | + * @param string $sql |
|
57 | + * @param string $sqlCount |
|
58 | + */ |
|
59 | + public function __construct(TDBMService $tdbmService, Schema $schema, string $mainTable, string $sql, string $sqlCount = null) |
|
60 | + { |
|
61 | + $this->sql = $sql; |
|
62 | + $this->sqlCount = $sqlCount; |
|
63 | + $this->tdbmService = $tdbmService; |
|
64 | + $this->schema = $schema; |
|
65 | + $this->mainTable = $mainTable; |
|
66 | 66 | |
67 | - $this->compute(); |
|
68 | - } |
|
67 | + $this->compute(); |
|
68 | + } |
|
69 | 69 | |
70 | - public function sort($orderBy) |
|
71 | - { |
|
72 | - throw new TDBMException('sort not supported for raw sql queries'); |
|
73 | - } |
|
70 | + public function sort($orderBy) |
|
71 | + { |
|
72 | + throw new TDBMException('sort not supported for raw sql queries'); |
|
73 | + } |
|
74 | 74 | |
75 | - public function getMagicSql(): string |
|
76 | - { |
|
77 | - return $this->processedSql; |
|
78 | - } |
|
75 | + public function getMagicSql(): string |
|
76 | + { |
|
77 | + return $this->processedSql; |
|
78 | + } |
|
79 | 79 | |
80 | - public function getMagicSqlCount(): string |
|
81 | - { |
|
82 | - return $this->processedSqlCount; |
|
83 | - } |
|
80 | + public function getMagicSqlCount(): string |
|
81 | + { |
|
82 | + return $this->processedSqlCount; |
|
83 | + } |
|
84 | 84 | |
85 | - public function getColumnDescriptors(): array |
|
86 | - { |
|
87 | - return $this->columnDescriptors; |
|
88 | - } |
|
85 | + public function getColumnDescriptors(): array |
|
86 | + { |
|
87 | + return $this->columnDescriptors; |
|
88 | + } |
|
89 | 89 | |
90 | - protected function compute() |
|
91 | - { |
|
92 | - $parser = new PHPSQLParser(); |
|
93 | - $generator = new PHPSQLCreator(); |
|
94 | - $parsedSql = $parser->parse($this->sql); |
|
90 | + protected function compute() |
|
91 | + { |
|
92 | + $parser = new PHPSQLParser(); |
|
93 | + $generator = new PHPSQLCreator(); |
|
94 | + $parsedSql = $parser->parse($this->sql); |
|
95 | 95 | |
96 | - // 1: let's reformat the SELECT and construct our columns |
|
97 | - list($select, $columnDescriptors) = $this->formatSelect($parsedSql['SELECT']); |
|
98 | - $parsedSql['SELECT'] = $select; |
|
99 | - $this->processedSql = $generator->create($parsedSql); |
|
100 | - $this->columnDescriptors = $columnDescriptors; |
|
96 | + // 1: let's reformat the SELECT and construct our columns |
|
97 | + list($select, $columnDescriptors) = $this->formatSelect($parsedSql['SELECT']); |
|
98 | + $parsedSql['SELECT'] = $select; |
|
99 | + $this->processedSql = $generator->create($parsedSql); |
|
100 | + $this->columnDescriptors = $columnDescriptors; |
|
101 | 101 | |
102 | - // 2: let's compute the count query if needed |
|
103 | - if ($this->sqlCount === null) { |
|
104 | - $parsedSqlCount = $this->generateParsedSqlCount($parsedSql); |
|
105 | - $this->processedSqlCount = $generator->create($parsedSqlCount); |
|
106 | - } else { |
|
107 | - $this->processedSqlCount = $this->sqlCount; |
|
108 | - } |
|
109 | - } |
|
102 | + // 2: let's compute the count query if needed |
|
103 | + if ($this->sqlCount === null) { |
|
104 | + $parsedSqlCount = $this->generateParsedSqlCount($parsedSql); |
|
105 | + $this->processedSqlCount = $generator->create($parsedSqlCount); |
|
106 | + } else { |
|
107 | + $this->processedSqlCount = $this->sqlCount; |
|
108 | + } |
|
109 | + } |
|
110 | 110 | |
111 | - private function formatSelect($baseSelect) |
|
112 | - { |
|
113 | - $relatedTables = $this->tdbmService->_getRelatedTablesByInheritance($this->mainTable); |
|
114 | - $tableGroup = $this->getTableGroupName($relatedTables); |
|
111 | + private function formatSelect($baseSelect) |
|
112 | + { |
|
113 | + $relatedTables = $this->tdbmService->_getRelatedTablesByInheritance($this->mainTable); |
|
114 | + $tableGroup = $this->getTableGroupName($relatedTables); |
|
115 | 115 | |
116 | - $connection = $this->tdbmService->getConnection(); |
|
117 | - $formattedSelect = []; |
|
118 | - $columnDescritors = []; |
|
119 | - $fetchedTables = []; |
|
116 | + $connection = $this->tdbmService->getConnection(); |
|
117 | + $formattedSelect = []; |
|
118 | + $columnDescritors = []; |
|
119 | + $fetchedTables = []; |
|
120 | 120 | |
121 | - foreach ($baseSelect as $entry) { |
|
122 | - if ($entry['expr_type'] !== 'colref') { |
|
123 | - $formattedSelect[] = $entry; |
|
124 | - continue; |
|
125 | - } |
|
121 | + foreach ($baseSelect as $entry) { |
|
122 | + if ($entry['expr_type'] !== 'colref') { |
|
123 | + $formattedSelect[] = $entry; |
|
124 | + continue; |
|
125 | + } |
|
126 | 126 | |
127 | - $noQuotes = $entry['no_quotes']; |
|
128 | - if ($noQuotes['delim'] != '.' || count($noQuotes['parts']) !== 2) { |
|
129 | - $formattedSelect[] = $entry; |
|
130 | - continue; |
|
131 | - } |
|
127 | + $noQuotes = $entry['no_quotes']; |
|
128 | + if ($noQuotes['delim'] != '.' || count($noQuotes['parts']) !== 2) { |
|
129 | + $formattedSelect[] = $entry; |
|
130 | + continue; |
|
131 | + } |
|
132 | 132 | |
133 | - $tableName = $noQuotes['parts'][0]; |
|
134 | - if (!in_array($tableName, $relatedTables)) { |
|
135 | - $formattedSelect[] = $entry; |
|
136 | - continue; |
|
137 | - } |
|
133 | + $tableName = $noQuotes['parts'][0]; |
|
134 | + if (!in_array($tableName, $relatedTables)) { |
|
135 | + $formattedSelect[] = $entry; |
|
136 | + continue; |
|
137 | + } |
|
138 | 138 | |
139 | - $columnName = $noQuotes['parts'][1]; |
|
140 | - if ($columnName !== '*') { |
|
141 | - $formattedSelect[] = $entry; |
|
142 | - continue; |
|
143 | - } |
|
139 | + $columnName = $noQuotes['parts'][1]; |
|
140 | + if ($columnName !== '*') { |
|
141 | + $formattedSelect[] = $entry; |
|
142 | + continue; |
|
143 | + } |
|
144 | 144 | |
145 | - $table = $this->schema->getTable($tableName); |
|
146 | - foreach ($table->getColumns() as $column) { |
|
147 | - $columnName = $column->getName(); |
|
148 | - $alias = "{$tableName}____{$columnName}"; |
|
149 | - $formattedSelect[] = [ |
|
150 | - 'expr_type' => 'colref', |
|
151 | - 'base_expr' => $connection->quoteIdentifier($tableName).'.'.$connection->quoteIdentifier($columnName), |
|
152 | - 'no_quotes' => [ |
|
153 | - 'delim' => '.', |
|
154 | - 'parts' => [ |
|
155 | - $tableName, |
|
156 | - $columnName |
|
157 | - ] |
|
158 | - ], |
|
159 | - 'alias' => [ |
|
160 | - 'as' => true, |
|
161 | - 'name' => $alias, |
|
162 | - ] |
|
163 | - ]; |
|
145 | + $table = $this->schema->getTable($tableName); |
|
146 | + foreach ($table->getColumns() as $column) { |
|
147 | + $columnName = $column->getName(); |
|
148 | + $alias = "{$tableName}____{$columnName}"; |
|
149 | + $formattedSelect[] = [ |
|
150 | + 'expr_type' => 'colref', |
|
151 | + 'base_expr' => $connection->quoteIdentifier($tableName).'.'.$connection->quoteIdentifier($columnName), |
|
152 | + 'no_quotes' => [ |
|
153 | + 'delim' => '.', |
|
154 | + 'parts' => [ |
|
155 | + $tableName, |
|
156 | + $columnName |
|
157 | + ] |
|
158 | + ], |
|
159 | + 'alias' => [ |
|
160 | + 'as' => true, |
|
161 | + 'name' => $alias, |
|
162 | + ] |
|
163 | + ]; |
|
164 | 164 | |
165 | - $columnDescritors[$alias] = [ |
|
166 | - 'as' => $alias, |
|
167 | - 'table' => $tableName, |
|
168 | - 'column' => $columnName, |
|
169 | - 'type' => $column->getType(), |
|
170 | - 'tableGroup' => $tableGroup, |
|
171 | - ]; |
|
172 | - } |
|
173 | - $fetchedTables[] = $tableName; |
|
174 | - } |
|
165 | + $columnDescritors[$alias] = [ |
|
166 | + 'as' => $alias, |
|
167 | + 'table' => $tableName, |
|
168 | + 'column' => $columnName, |
|
169 | + 'type' => $column->getType(), |
|
170 | + 'tableGroup' => $tableGroup, |
|
171 | + ]; |
|
172 | + } |
|
173 | + $fetchedTables[] = $tableName; |
|
174 | + } |
|
175 | 175 | |
176 | - $missingTables = array_diff($relatedTables, $fetchedTables); |
|
177 | - if (!empty($missingTables)) { |
|
178 | - throw new TDBMException('Missing tables '.implode(', ', $missingTables).' in SELECT statement'); |
|
179 | - } |
|
176 | + $missingTables = array_diff($relatedTables, $fetchedTables); |
|
177 | + if (!empty($missingTables)) { |
|
178 | + throw new TDBMException('Missing tables '.implode(', ', $missingTables).' in SELECT statement'); |
|
179 | + } |
|
180 | 180 | |
181 | - for ($i = 0; $i < count($formattedSelect) - 1; $i++) { |
|
182 | - $formattedSelect[$i]['delim'] = ','; |
|
183 | - } |
|
184 | - return [$formattedSelect, $columnDescritors]; |
|
185 | - } |
|
181 | + for ($i = 0; $i < count($formattedSelect) - 1; $i++) { |
|
182 | + $formattedSelect[$i]['delim'] = ','; |
|
183 | + } |
|
184 | + return [$formattedSelect, $columnDescritors]; |
|
185 | + } |
|
186 | 186 | |
187 | - private function generateParsedSqlCount($parsedSql) |
|
188 | - { |
|
189 | - if (isset($parsedSql['ORDER'])) { |
|
190 | - unset($parsedSql['ORDER']); |
|
191 | - } |
|
187 | + private function generateParsedSqlCount($parsedSql) |
|
188 | + { |
|
189 | + if (isset($parsedSql['ORDER'])) { |
|
190 | + unset($parsedSql['ORDER']); |
|
191 | + } |
|
192 | 192 | |
193 | - if (!isset($parsedSql['GROUP'])) { |
|
194 | - // most simple case:no GROUP BY in query |
|
195 | - return $this->generateSimpleSqlCount($parsedSql); |
|
196 | - } elseif (!isset($parsedSql['HAVING'])) { |
|
197 | - // GROUP BY without HAVING statement: let's COUNT the DISTINCT grouped columns |
|
198 | - return $this->generateGroupedSqlCount($parsedSql); |
|
199 | - } else { |
|
200 | - // GROUP BY with a HAVING statement: we'll have to wrap the query |
|
201 | - return $this->generateWrappedSqlCount($parsedSql); |
|
202 | - } |
|
203 | - } |
|
193 | + if (!isset($parsedSql['GROUP'])) { |
|
194 | + // most simple case:no GROUP BY in query |
|
195 | + return $this->generateSimpleSqlCount($parsedSql); |
|
196 | + } elseif (!isset($parsedSql['HAVING'])) { |
|
197 | + // GROUP BY without HAVING statement: let's COUNT the DISTINCT grouped columns |
|
198 | + return $this->generateGroupedSqlCount($parsedSql); |
|
199 | + } else { |
|
200 | + // GROUP BY with a HAVING statement: we'll have to wrap the query |
|
201 | + return $this->generateWrappedSqlCount($parsedSql); |
|
202 | + } |
|
203 | + } |
|
204 | 204 | |
205 | - private function generateSimpleSqlCount($parsedSql) |
|
206 | - { |
|
207 | - return [[ |
|
208 | - 'expr_type' => 'aggregate_function', |
|
209 | - 'alias' => false, |
|
210 | - 'base_expr' => 'COUNT', |
|
211 | - 'sub_tree' => $parsedSql['SELECT'], |
|
212 | - 'delim' => false, |
|
213 | - ]]; |
|
214 | - } |
|
205 | + private function generateSimpleSqlCount($parsedSql) |
|
206 | + { |
|
207 | + return [[ |
|
208 | + 'expr_type' => 'aggregate_function', |
|
209 | + 'alias' => false, |
|
210 | + 'base_expr' => 'COUNT', |
|
211 | + 'sub_tree' => $parsedSql['SELECT'], |
|
212 | + 'delim' => false, |
|
213 | + ]]; |
|
214 | + } |
|
215 | 215 | |
216 | - private function generateGroupedSqlCount($parsedSql) |
|
217 | - { |
|
218 | - $group = $parsedSql['GROUP']; |
|
219 | - unset($parsedSql['GROUP']); |
|
220 | - $parsedSql['SELECT'] = [[ |
|
221 | - 'expr_type' => 'aggregate_function', |
|
222 | - 'alias' => false, |
|
223 | - 'base_expr' => 'COUNT', |
|
224 | - 'sub_tree' => array_merge([[ |
|
225 | - 'expr_type' => 'reserved', |
|
226 | - 'base_expr' => 'DISTINCT', |
|
227 | - 'delim' => ',' |
|
228 | - ]], $group), |
|
229 | - 'delim' => false, |
|
230 | - ]]; |
|
231 | - return $parsedSql; |
|
232 | - } |
|
216 | + private function generateGroupedSqlCount($parsedSql) |
|
217 | + { |
|
218 | + $group = $parsedSql['GROUP']; |
|
219 | + unset($parsedSql['GROUP']); |
|
220 | + $parsedSql['SELECT'] = [[ |
|
221 | + 'expr_type' => 'aggregate_function', |
|
222 | + 'alias' => false, |
|
223 | + 'base_expr' => 'COUNT', |
|
224 | + 'sub_tree' => array_merge([[ |
|
225 | + 'expr_type' => 'reserved', |
|
226 | + 'base_expr' => 'DISTINCT', |
|
227 | + 'delim' => ',' |
|
228 | + ]], $group), |
|
229 | + 'delim' => false, |
|
230 | + ]]; |
|
231 | + return $parsedSql; |
|
232 | + } |
|
233 | 233 | |
234 | - private function generateWrappedSqlCount($parsedSql) |
|
235 | - { |
|
236 | - return [ |
|
237 | - 'SELECT' => [[ |
|
238 | - 'expr_type' => 'aggregate_function', |
|
239 | - 'alias' => false, |
|
240 | - 'base_expr' => 'COUNT', |
|
241 | - 'sub_tree' => [ |
|
242 | - [ |
|
243 | - 'expr_type' => 'colref', |
|
244 | - 'base_expr' => '*', |
|
245 | - 'sub_tree' => false |
|
246 | - ] |
|
247 | - ], |
|
248 | - 'delim' => false, |
|
249 | - ]], |
|
250 | - 'FROM' => [[ |
|
251 | - 'expr_type' => 'subquery', |
|
252 | - 'alias' => [ |
|
253 | - 'as' => true, |
|
254 | - 'name' => '____query' |
|
255 | - ], |
|
256 | - 'sub_tree' => $parsedSql, |
|
257 | - ]] |
|
258 | - ]; |
|
259 | - } |
|
234 | + private function generateWrappedSqlCount($parsedSql) |
|
235 | + { |
|
236 | + return [ |
|
237 | + 'SELECT' => [[ |
|
238 | + 'expr_type' => 'aggregate_function', |
|
239 | + 'alias' => false, |
|
240 | + 'base_expr' => 'COUNT', |
|
241 | + 'sub_tree' => [ |
|
242 | + [ |
|
243 | + 'expr_type' => 'colref', |
|
244 | + 'base_expr' => '*', |
|
245 | + 'sub_tree' => false |
|
246 | + ] |
|
247 | + ], |
|
248 | + 'delim' => false, |
|
249 | + ]], |
|
250 | + 'FROM' => [[ |
|
251 | + 'expr_type' => 'subquery', |
|
252 | + 'alias' => [ |
|
253 | + 'as' => true, |
|
254 | + 'name' => '____query' |
|
255 | + ], |
|
256 | + 'sub_tree' => $parsedSql, |
|
257 | + ]] |
|
258 | + ]; |
|
259 | + } |
|
260 | 260 | |
261 | - protected function getTableGroupName(array $relatedTables) |
|
262 | - { |
|
263 | - sort($relatedTables); |
|
264 | - return implode('_``_', $relatedTables); |
|
265 | - } |
|
261 | + protected function getTableGroupName(array $relatedTables) |
|
262 | + { |
|
263 | + sort($relatedTables); |
|
264 | + return implode('_``_', $relatedTables); |
|
265 | + } |
|
266 | 266 | } |
@@ -178,7 +178,7 @@ |
||
178 | 178 | throw new TDBMException('Missing tables '.implode(', ', $missingTables).' in SELECT statement'); |
179 | 179 | } |
180 | 180 | |
181 | - for ($i = 0; $i < count($formattedSelect) - 1; $i++) { |
|
181 | + for ($i = 0; $i<count($formattedSelect)-1; $i++) { |
|
182 | 182 | $formattedSelect[$i]['delim'] = ','; |
183 | 183 | } |
184 | 184 | return [$formattedSelect, $columnDescritors]; |