Total Complexity | 67 |
Total Lines | 313 |
Duplicated Lines | 0 % |
Changes | 2 | ||
Bugs | 0 | Features | 0 |
Complex classes like DbDumpTrait often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use DbDumpTrait, and based on these observations, apply Extract Interface, too.
1 | <?php |
||
18 | trait DbDumpTrait |
||
19 | { |
||
20 | /** |
||
21 | * The dump options |
||
22 | * |
||
23 | * @var array |
||
24 | */ |
||
25 | private $options; |
||
26 | |||
27 | /** |
||
28 | * The queries generated by the dump |
||
29 | * |
||
30 | * @var array |
||
31 | */ |
||
32 | private $queries = []; |
||
33 | |||
34 | // Temp vars for data dumps |
||
35 | private $insert = ''; |
||
36 | private $buffer = ''; |
||
37 | private $suffix = ''; |
||
38 | private $views = []; |
||
39 | private $fkeys = []; |
||
40 | |||
41 | /** |
||
42 | * Print CSV row |
||
43 | * |
||
44 | * @param array $row |
||
45 | * |
||
46 | * @return void |
||
47 | */ |
||
48 | private function dumpCsv(array $row) |
||
49 | { |
||
50 | // From functions.inc.php |
||
51 | foreach ($row as $key => $val) { |
||
52 | if (preg_match('~["\n,;\t]|^0|\.\d*0$~', $val) || $val === '') { |
||
53 | $row[$key] = '"' . str_replace('"', '""', $val) . '"'; |
||
54 | } |
||
55 | } |
||
56 | $separator = $this->options['format'] === 'csv' ? ',' : |
||
57 | ($this->options['format'] === 'tsv' ? "\t" : ';'); |
||
58 | $this->queries[] = implode($separator, $row); |
||
59 | } |
||
60 | |||
61 | /** |
||
62 | * Convert a value to string |
||
63 | * |
||
64 | * @param mixed $value |
||
65 | * @param TableFieldEntity $field |
||
66 | * |
||
67 | * @return string |
||
68 | */ |
||
69 | private function convertToString($value, TableFieldEntity $field): string |
||
70 | { |
||
71 | // From functions.inc.php |
||
72 | if ($value === null) { |
||
73 | return 'NULL'; |
||
74 | } |
||
75 | if (!preg_match($this->driver->numberRegex(), $field->type) || |
||
76 | preg_match('~\[~', $field->fullType) && is_numeric($value)) { |
||
77 | $value = $this->driver->quote(($value === false ? 0 : $value)); |
||
78 | } |
||
79 | return $this->driver->unconvertField($field, $value); |
||
80 | } |
||
81 | |||
82 | /** |
||
83 | * @param string $table |
||
84 | * @param string $style |
||
85 | * @param int $tableType |
||
86 | * |
||
87 | * @return string |
||
88 | */ |
||
89 | private function getCreateTableQuery(string $table, string $style, int $tableType): string |
||
90 | { |
||
91 | if ($tableType !== 2) { |
||
92 | return $this->driver->sqlForCreateTable($table, $this->options['autoIncrement'], $style); |
||
93 | } |
||
94 | $fields = []; |
||
95 | foreach ($this->driver->fields($table) as $name => $field) { |
||
96 | $fields[] = $this->driver->escapeId($name) . ' ' . $field->fullType; |
||
97 | } |
||
98 | return 'CREATE TABLE ' . $this->driver->table($table) . ' (' . implode(', ', $fields) . ')'; |
||
99 | } |
||
100 | |||
101 | /** |
||
102 | * Export table structure |
||
103 | * |
||
104 | * @param string $table |
||
105 | * @param string $style |
||
106 | * @param int $tableType 0 table, 1 view, 2 temporary view table |
||
107 | * |
||
108 | * @return void |
||
109 | */ |
||
110 | private function dumpTableOrView(string $table, string $style, int $tableType = 0) |
||
111 | { |
||
112 | // From adminer.inc.php |
||
113 | if ($this->options['format'] !== 'sql') { |
||
114 | $this->queries[] = "\xef\xbb\xbf"; // UTF-8 byte order mark |
||
115 | if ($style) { |
||
116 | $this->dumpCsv(array_keys($this->driver->fields($table))); |
||
117 | } |
||
118 | return; |
||
119 | } |
||
120 | if (!$style) { |
||
121 | return; |
||
122 | } |
||
123 | |||
124 | $create = $this->getCreateTableQuery($table, $style, $tableType); |
||
125 | $this->driver->setUtf8mb4($create); |
||
126 | if (!$create) { |
||
127 | return; |
||
128 | } |
||
129 | if ($style === 'DROP+CREATE' || $tableType === 1) { |
||
130 | $this->queries[] = 'DROP ' . ($tableType === 2 ? 'VIEW' : 'TABLE') . |
||
131 | ' IF EXISTS ' . $this->driver->table($table) . ';'; |
||
132 | } |
||
133 | if ($tableType === 1) { |
||
134 | $create = $this->admin->removeDefiner($create); |
||
135 | } |
||
136 | $this->queries[] = $create . ';'; |
||
137 | } |
||
138 | |||
139 | /** |
||
140 | * @param array $row |
||
141 | * @param StatementInterface $statement |
||
142 | * |
||
143 | * @return array |
||
144 | */ |
||
145 | private function getDataRowKeys(array $row, StatementInterface $statement): array |
||
146 | { |
||
147 | $values = []; |
||
148 | $keys = []; |
||
149 | // For is preferred to foreach because the values are not used. |
||
150 | // foreach ($row as $val) { |
||
151 | // } |
||
152 | for ($i = 0; $i < count($row); $i++) { |
||
|
|||
153 | $field = $statement->fetchField(); |
||
154 | $keys[] = $field->name(); |
||
155 | $key = $this->driver->escapeId($field->name()); |
||
156 | $values[] = "$key = VALUES($key)"; |
||
157 | } |
||
158 | $this->suffix = ";\n"; |
||
159 | if ($this->options['data_style'] === 'INSERT+UPDATE') { |
||
160 | $this->suffix = "\nON DUPLICATE KEY UPDATE " . implode(', ', $values) . ";\n"; |
||
161 | } |
||
162 | return $keys; |
||
163 | } |
||
164 | |||
165 | /** |
||
166 | * @param array $row |
||
167 | * |
||
168 | * @return void |
||
169 | */ |
||
170 | private function saveRowInBuffer(array $row) |
||
184 | } |
||
185 | |||
186 | /** |
||
187 | * @param string $table |
||
188 | * @param array $fields |
||
189 | * @param array $row |
||
190 | * @param array $keys |
||
191 | * |
||
192 | * @return void |
||
193 | */ |
||
194 | private function dumpRow(string $table, array $fields, array $row, array $keys) |
||
195 | { |
||
196 | if ($this->options['format'] !== 'sql') { |
||
197 | if ($this->options['data_style'] === 'table') { |
||
198 | $this->dumpCsv($keys); |
||
199 | $this->options['data_style'] = 'INSERT'; |
||
200 | } |
||
201 | $this->dumpCsv($row); |
||
202 | return; |
||
203 | } |
||
204 | if (!$this->insert) { |
||
205 | $this->insert = 'INSERT INTO ' . $this->driver->table($table) . ' (' . |
||
206 | implode(', ', array_map(function ($key) { |
||
207 | return $this->driver->escapeId($key); |
||
208 | }, $keys)) . ') VALUES'; |
||
209 | } |
||
210 | foreach ($row as $key => $val) { |
||
211 | $field = $fields[$key]; |
||
212 | $row[$key] = $this->convertToString($val, $field); |
||
213 | } |
||
214 | $this->saveRowInBuffer($row); |
||
215 | } |
||
216 | |||
217 | /** Export table data |
||
218 | * |
||
219 | * @param string $table |
||
220 | * @param string $query |
||
221 | * |
||
222 | * @return void |
||
223 | */ |
||
224 | private function dumpData(string $table, string $query) |
||
225 | { |
||
226 | if (!$this->options['data_style']) { |
||
227 | return; |
||
228 | } |
||
229 | $fields = []; |
||
230 | if ($this->options['format'] === 'sql') { |
||
231 | if ($this->options['data_style'] === 'TRUNCATE+INSERT') { |
||
232 | $this->queries[] = $this->driver->sqlForTruncateTable($table) . ";\n"; |
||
233 | } |
||
234 | $fields = $this->driver->fields($table); |
||
235 | } |
||
236 | $statement = $this->driver->query($query); // 1 - MYSQLI_USE_RESULT //! enum and set as numbers |
||
237 | if (!$statement) { |
||
238 | if ($this->options['format'] === 'sql') { |
||
239 | $this->queries[] = '-- ' . str_replace("\n", ' ', $this->driver->error()) . "\n"; |
||
240 | } |
||
241 | return; |
||
242 | } |
||
243 | $this->insert = ''; |
||
244 | $this->buffer = ''; |
||
245 | $this->suffix = ''; |
||
246 | $keys = []; |
||
247 | $fetch_function = ($table !== '' ? 'fetchAssoc' : 'fetchRow'); |
||
248 | while ($row = $statement->$fetch_function()) { |
||
249 | if (empty($keys)) { |
||
250 | $keys = $this->getDataRowKeys($row, $statement); |
||
251 | } |
||
252 | $this->dumpRow($table, $fields, $row, $keys); |
||
253 | } |
||
254 | if (($this->buffer)) { |
||
255 | $this->queries[] = $this->buffer . $this->suffix; |
||
256 | } |
||
257 | } |
||
258 | |||
259 | /** |
||
260 | * @param string $table |
||
261 | * @param bool $dumpTable |
||
262 | * @param bool $dumpData |
||
263 | * |
||
264 | * @return void |
||
265 | */ |
||
266 | private function dumpTable(string $table, bool $dumpTable, bool $dumpData) |
||
267 | { |
||
268 | if (!$dumpTable && !$dumpData) { |
||
269 | return; |
||
270 | } |
||
271 | $this->dumpTableOrView($table, ($dumpTable ? $this->options['table_style'] : '')); |
||
272 | if ($dumpData) { |
||
273 | $fields = $this->driver->fields($table); |
||
274 | $query = 'SELECT *' . $this->driver->convertFields($fields, $fields) . |
||
275 | ' FROM ' . $this->driver->table($table); |
||
276 | $this->dumpData($table, $query); |
||
277 | } |
||
278 | if ($this->options['is_sql'] && $this->options['triggers'] && $dumpTable && |
||
279 | ($triggers = $this->driver->sqlForCreateTrigger($table))) { |
||
280 | $this->queries[] = 'DELIMITER ;'; |
||
281 | $this->queries[] = $triggers; |
||
282 | $this->queries[] = 'DELIMITER ;'; |
||
283 | } |
||
284 | if ($this->options['is_sql']) { |
||
285 | $this->queries[] = ''; |
||
286 | } |
||
287 | } |
||
288 | |||
289 | /** |
||
290 | * Dump tables |
||
291 | * |
||
292 | * @param string $database The database name |
||
293 | * |
||
294 | * @return void |
||
295 | */ |
||
296 | private function dumpTables(string $database) |
||
297 | { |
||
298 | $dbDumpTable = $this->tables['list'] === '*' && in_array($database, $this->databases['list']); |
||
299 | $dbDumpData = in_array($database, $this->databases['data']); |
||
300 | $this->views = []; // View names |
||
301 | $this->fkeys = []; // Table names for foreign keys |
||
302 | $dbTables = $this->driver->tableStatuses(true); |
||
303 | foreach ($dbTables as $table => $tableStatus) { |
||
304 | $isView = $this->driver->isView($tableStatus); |
||
305 | if ($isView) { |
||
306 | // The views will be dumped after the tables |
||
307 | $this->views[] = $table; |
||
308 | continue; |
||
309 | } |
||
310 | $this->fkeys[] = $table; |
||
311 | $dumpTable = $dbDumpTable || in_array($table, $this->tables['list']); |
||
312 | $dumpData = $dbDumpData || in_array($table, $this->tables['data']); |
||
313 | $this->dumpTable($table, $dumpTable, $dumpData); |
||
314 | } |
||
315 | } |
||
316 | |||
317 | /** |
||
318 | * @return void |
||
319 | */ |
||
320 | private function dumpViewsAndFKeys() |
||
331 | } |
||
332 | } |
||
333 | } |
||
334 |
If the size of the collection does not change during the iteration, it is generally a good practice to compute it beforehand, and not on each iteration: