Passed
Push — master ( acba68...4a876c )
by Wilmer
28:04 queued 13:05
created

Schema::getTableNameParts()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 3
Code Lines 1

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 1
eloc 1
nc 1
nop 1
dl 0
loc 3
ccs 0
cts 0
cp 0
crap 2
rs 10
c 1
b 0
f 0
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Yiisoft\Db\Schema;
6
7
use Yiisoft\Db\Connection\Connection;
8
use Yiisoft\Db\Exception\Exception;
9
use Yiisoft\Db\Exception\IntegrityException;
10
use Yiisoft\Db\Exception\InvalidCallException;
11
use Yiisoft\Db\Exception\NotSupportedException;
12
use Yiisoft\Db\Query\QueryBuilder;
13
use Yiisoft\Cache\CacheInterface;
14
use Yiisoft\Cache\Dependency\TagDependency;
15
16
/**
17
 * Schema is the base class for concrete DBMS-specific schema classes.
18
 *
19
 * Schema represents the database schema information that is DBMS specific.
20
 *
21
 * @property string $lastInsertID The row ID of the last row inserted, or the last value retrieved from the sequence
22
 * object. This property is read-only.
23
 * @property QueryBuilder $queryBuilder The query builder for this connection. This property is read-only.
24
 * @property string[] $schemaNames All schema names in the database, except system schemas. This property is read-only.
25
 * @property string $serverVersion Server version as a string. This property is read-only.
26
 * @property string[] $tableNames All table names in the database. This property is read-only.
27
 * @property TableSchema[] $tableSchemas The metadata for all tables in the database. Each array element is an instance
28
 * of {@see TableSchema} or its child class. This property is read-only.
29
 * @property string $transactionIsolationLevel The transaction isolation level to use for this transaction. This can be
30
 * one of {@see Transaction::READ_UNCOMMITTED}, {@see Transaction::READ_COMMITTED},
31
 * {@see Transaction::REPEATABLE_READ} and {@see Transaction::SERIALIZABLE} but also a string containing DBMS specific
32
 * syntax to be used after `SET TRANSACTION ISOLATION LEVEL`. This property is write-only.
33
 */
34
abstract class Schema
35
{
36
    public const TYPE_PK = 'pk';
37
    public const TYPE_UPK = 'upk';
38
    public const TYPE_BIGPK = 'bigpk';
39
    public const TYPE_UBIGPK = 'ubigpk';
40
    public const TYPE_CHAR = 'char';
41
    public const TYPE_STRING = 'string';
42
    public const TYPE_TEXT = 'text';
43
    public const TYPE_TINYINT = 'tinyint';
44
    public const TYPE_SMALLINT = 'smallint';
45
    public const TYPE_INTEGER = 'integer';
46
    public const TYPE_BIGINT = 'bigint';
47
    public const TYPE_FLOAT = 'float';
48
    public const TYPE_DOUBLE = 'double';
49
    public const TYPE_DECIMAL = 'decimal';
50
    public const TYPE_DATETIME = 'datetime';
51
    public const TYPE_TIMESTAMP = 'timestamp';
52
    public const TYPE_TIME = 'time';
53
    public const TYPE_DATE = 'date';
54
    public const TYPE_BINARY = 'binary';
55
    public const TYPE_BOOLEAN = 'boolean';
56
    public const TYPE_MONEY = 'money';
57
    public const TYPE_JSON = 'json';
58
59
    /**
60
     * Schema cache version, to detect incompatibilities in cached values when the data format of the cache changes.
61
     */
62
    protected const SCHEMA_CACHE_VERSION = 1;
63
64
    /**
65
     * @var string|null the default schema name used for the current session.
66
     */
67
    protected ?string $defaultSchema = null;
68
69
    /**
70
     * @var array map of DB errors and corresponding exceptions. If left part is found in DB error message exception
71
     * class from the right part is used.
72
     */
73
    protected array $exceptionMap = [
74
        'SQLSTATE[23' => IntegrityException::class,
75
    ];
76
77
    /**
78
     * @var string|string[] character used to quote schema, table, etc. names. An array of 2 characters can be used in
79
     * case starting and ending characters are different.
80
     */
81
    protected $tableQuoteCharacter = "'";
82
83
    /**
84
     * @var string|string[] character used to quote column names. An array of 2 characters can be used in case starting
85
     * and ending characters are different.
86
     */
87
    protected $columnQuoteCharacter = '"';
88
89
    private array $schemaNames = [];
90
    private array $tableNames = [];
91
    private array $tableMetadata = [];
92
    private ?QueryBuilder $builder = null;
93
    private ?string $serverVersion = null;
94
    private CacheInterface $cache;
95
    private ?Connection $db = null;
96
97 900
    public function __construct(Connection $db)
98
    {
99 900
        $this->db = $db;
100 900
        $this->cache = $this->db->getSchemaCache();
101 900
    }
102
103
    /**
104
     * Resolves the table name and schema name (if any).
105
     *
106
     * @param string $name the table name.
107
     *
108
     * @return void with resolved table, schema, etc. names.
109
     *
110
     * @throws NotSupportedException if this method is not supported by the DBMS.
111
     *
112
     * {@see \Yiisoft\Db\Schema\TableSchema}
113
     */
114
    protected function resolveTableName(string $name): TableSchema
115
    {
116
        throw new NotSupportedException(get_class($this) . ' does not support resolving table names.');
117
    }
118
119
    /**
120
     * Returns all schema names in the database, including the default one but not system schemas.
121
     *
122
     * This method should be overridden by child classes in order to support this feature because the default
123
     * implementation simply throws an exception.
124
     *
125
     * @return void all schema names in the database, except system schemas.
126
     *
127
     * @throws NotSupportedException if this method is not supported by the DBMS.
128
     */
129
    protected function findSchemaNames()
130
    {
131
        throw new NotSupportedException(get_class($this) . ' does not support fetching all schema names.');
132
    }
133
134
    /**
135
     * Returns all table names in the database.
136
     *
137
     * This method should be overridden by child classes in order to support this feature because the default
138
     * implementation simply throws an exception.
139
     *
140
     * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema.
141
     *
142
     * @return void all table names in the database. The names have NO schema name prefix.
143
     *
144
     * @throws NotSupportedException if this method is not supported by the DBMS.
145
     */
146
    protected function findTableNames(string $schema = ''): array
147
    {
148
        throw new NotSupportedException(get_class($this) . ' does not support fetching all table names.');
149
    }
150
151
    /**
152
     * Loads the metadata for the specified table.
153
     *
154
     * @param string $name table name.
155
     *
156
     * @return TableSchema|null DBMS-dependent table metadata, `null` if the table does not exist.
157
     */
158
    abstract protected function loadTableSchema(string $name): ?TableSchema;
159
160
    /**
161
     * Creates a column schema for the database.
162
     *
163
     * This method may be overridden by child classes to create a DBMS-specific column schema.
164
     *
165
     * @return ColumnSchema column schema instance.
166
     */
167 75
    protected function createColumnSchema(): ColumnSchema
168
    {
169 75
        return new ColumnSchema();
170
    }
171
172
    /**
173
     * Obtains the metadata for the named table.
174
     *
175
     * @param string $name table name. The table name may contain schema name if any. Do not quote the table name.
176
     * @param bool $refresh whether to reload the table schema even if it is found in the cache.
177
     *
178
     * @return TableSchema|null table metadata. `null` if the named table does not exist.
179
     */
180 263
    public function getTableSchema(string $name, bool $refresh = false): ?TableSchema
181
    {
182 263
        return $this->getTableMetadata($name, 'schema', $refresh);
183
    }
184
185
    /**
186
     * Returns the metadata for all tables in the database.
187
     *
188
     * @param string $schema  the schema of the tables. Defaults to empty string, meaning the current or default schema
189
     * name.
190
     * @param bool $refresh whether to fetch the latest available table schemas. If this is `false`, cached data may be
191
     * returned if available.
192
     *
193
     * @return TableSchema[] the metadata for all tables in the database. Each array element is an instance of
194
     * {@see TableSchema} or its child class.
195
     */
196 9
    public function getTableSchemas(string $schema = '', bool $refresh = false): array
197
    {
198 9
        return $this->getSchemaMetadata($schema, 'schema', $refresh);
199
    }
200
201
    /**
202
     * Returns all schema names in the database, except system schemas.
203
     *
204
     * @param bool $refresh whether to fetch the latest available schema names. If this is false, schema names fetched
205
     * previously (if available) will be returned.
206
     *
207
     * @return string[] all schema names in the database, except system schemas.
208
     */
209 2
    public function getSchemaNames(bool $refresh = false): array
210
    {
211 2
        if (empty($this->schemaNames) || $refresh) {
212 2
            $this->schemaNames = $this->findSchemaNames();
213
        }
214
215 2
        return $this->schemaNames;
216
    }
217
218
    /**
219
     * Returns all table names in the database.
220
     *
221
     * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema
222
     * name.
223
     * If not empty, the returned table names will be prefixed with the schema name.
224
     * @param bool $refresh whether to fetch the latest available table names. If this is false, table names fetched
225
     * previously (if available) will be returned.
226
     *
227
     * @return string[] all table names in the database.
228
     */
229 15
    public function getTableNames(string $schema = '', bool $refresh = false): array
230
    {
231 15
        if (!isset($this->tableNames[$schema]) || $refresh) {
232 15
            $this->tableNames[$schema] = $this->findTableNames($schema);
233
        }
234
235 15
        return $this->tableNames[$schema];
0 ignored issues
show
Bug Best Practice introduced by
The expression return $this->tableNames[$schema] returns the type string which is incompatible with the type-hinted return array.
Loading history...
236
    }
237
238
    /**
239
     * @return QueryBuilder the query builder for this connection.
240
     */
241 179
    public function getQueryBuilder(): QueryBuilder
242
    {
243 179
        if ($this->builder === null) {
244 179
            $this->builder = $this->createQueryBuilder();
245
        }
246
247 179
        return $this->builder;
0 ignored issues
show
Bug Best Practice introduced by
The expression return $this->builder could return the type null which is incompatible with the type-hinted return Yiisoft\Db\Query\QueryBuilder. Consider adding an additional type-check to rule them out.
Loading history...
248
    }
249
250
    /**
251
     * Determines the PDO type for the given PHP data value.
252
     *
253
     * @param mixed $data the data whose PDO type is to be determined
254
     *
255
     * @return int the PDO type
256
     *
257
     * {@see http://www.php.net/manual/en/pdo.constants.php}
258
     */
259 308
    public function getPdoType($data): int
260
    {
261 308
        static $typeMap = [
262
            // php type => PDO type
263
            'boolean'  => \PDO::PARAM_BOOL,
264
            'integer'  => \PDO::PARAM_INT,
265
            'string'   => \PDO::PARAM_STR,
266
            'resource' => \PDO::PARAM_LOB,
267
            'NULL'     => \PDO::PARAM_NULL,
268
        ];
269 308
        $type = gettype($data);
270
271 308
        return $typeMap[$type] ?? \PDO::PARAM_STR;
272
    }
273
274
    /**
275
     * Refreshes the schema.
276
     *
277
     * This method cleans up all cached table schemas so that they can be re-created later to reflect the database
278
     * schema change.
279
     */
280
    public function refresh(): void
281
    {
282
        /* @var $cache CacheInterface */
283
        $cache = \is_string($this->db->getSchemaCache()) ? $this->cache : $this->db->getSchemaCache();
0 ignored issues
show
introduced by
The condition is_string($this->db->getSchemaCache()) is always false.
Loading history...
Bug introduced by
The method getSchemaCache() does not exist on null. ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-call  annotation

283
        $cache = \is_string($this->db->/** @scrutinizer ignore-call */ getSchemaCache()) ? $this->cache : $this->db->getSchemaCache();

This check looks for calls to methods that do not seem to exist on a given type. It looks for the method on the type itself as well as in inherited classes or implemented interfaces.

This is most likely a typographical error or the method has been renamed.

Loading history...
284
285
        if ($this->db->isSchemaCacheEnabled() && $cache instanceof CacheInterface) {
286
            TagDependency::invalidate($cache, $this->getCacheTag());
287
        }
288
289
        $this->tableNames = [];
290
        $this->tableMetadata = [];
291
    }
292
293
    /**
294
     * Refreshes the particular table schema.
295
     *
296
     * This method cleans up cached table schema so that it can be re-created later to reflect the database schema
297
     * change.
298
     *
299
     * @param string $name table name.
300
     */
301 65
    public function refreshTableSchema(string $name): void
302
    {
303 65
        $rawName = $this->getRawTableName($name);
304
305 65
        unset($this->tableMetadata[$rawName]);
306
307 65
        $this->tableNames = [];
308
309 65
        if ($this->db->isSchemaCacheEnabled() && $this->cache instanceof CacheInterface) {
310 65
            $this->cache->delete($this->getCacheKey($rawName));
0 ignored issues
show
Bug introduced by
$this->getCacheKey($rawName) of type array<integer,null|string> is incompatible with the type string expected by parameter $key of Psr\SimpleCache\CacheInterface::delete(). ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

310
            $this->cache->delete(/** @scrutinizer ignore-type */ $this->getCacheKey($rawName));
Loading history...
311
        }
312 65
    }
313
314
    /**
315
     * Creates a query builder for the database.
316
     *
317
     * This method may be overridden by child classes to create a DBMS-specific query builder.
318
     *
319
     * @return QueryBuilder query builder instance.
320
     */
321
    public function createQueryBuilder()
322
    {
323
        return new QueryBuilder($this->db);
0 ignored issues
show
Bug introduced by
It seems like $this->db can also be of type null; however, parameter $db of Yiisoft\Db\Query\QueryBuilder::__construct() does only seem to accept Yiisoft\Db\Connection\Connection, maybe add an additional type check? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

323
        return new QueryBuilder(/** @scrutinizer ignore-type */ $this->db);
Loading history...
324
    }
325
326
    /**
327
     * Create a column schema builder instance giving the type and value precision.
328
     *
329
     * This method may be overridden by child classes to create a DBMS-specific column schema builder.
330
     *
331
     * @param string $type type of the column. See {@see ColumnSchemaBuilder::$type}.
332
     * @param int|string|array $length length or precision of the column. See {@see ColumnSchemaBuilder::$length}.
333
     *
334
     * @return ColumnSchemaBuilder column schema builder instance
335
     */
336 4
    public function createColumnSchemaBuilder(string $type, $length = null)
337
    {
338 4
        return new ColumnSchemaBuilder($type, $length);
339
    }
340
341
    /**
342
     * Returns all unique indexes for the given table.
343
     *
344
     * Each array element is of the following structure:
345
     *
346
     * ```php
347
     * [
348
     *  'IndexName1' => ['col1' [, ...]],
349
     *  'IndexName2' => ['col2' [, ...]],
350
     * ]
351
     * ```
352
     *
353
     * This method should be overridden by child classes in order to support this feature because the default
354
     * implementation simply throws an exception
355
     *
356
     * @param TableSchema $table the table metadata
357
     *
358
     * @throws NotSupportedException if this method is called
359
     *
360
     * @return array all unique indexes for the given table.
361
     */
362
    public function findUniqueIndexes($table): array
363
    {
364
        throw new NotSupportedException(get_class($this) . ' does not support getting unique indexes information.');
365
    }
366
367
    /**
368
     * Returns the ID of the last inserted row or sequence value.
369
     *
370
     * @param string $sequenceName name of the sequence object (required by some DBMS)
371
     *
372
     * @throws InvalidCallException if the DB connection is not active
373
     *
374
     * @return string the row ID of the last row inserted, or the last value retrieved from the sequence object
375
     *
376
     * @see http://www.php.net/manual/en/function.PDO-lastInsertId.php
377
     */
378 3
    public function getLastInsertID(string $sequenceName = ''): string
379
    {
380 3
        if ($this->db->isActive()) {
381 3
            return $this->db->getPDO()->lastInsertId(
382 3
                $sequenceName === '' ? null : $this->quoteTableName($sequenceName)
383
            );
384
        }
385
386
        throw new InvalidCallException('DB Connection is not active.');
387
    }
388
389
    /**
390
     * @return bool whether this DBMS supports [savepoint](http://en.wikipedia.org/wiki/Savepoint).
391
     */
392 6
    public function supportsSavepoint(): bool
393
    {
394 6
        return $this->db->isSavepointEnabled();
395
    }
396
397
    /**
398
     * Creates a new savepoint.
399
     *
400
     * @param string $name the savepoint name
401
     */
402 3
    public function createSavepoint(string $name): void
403
    {
404 3
        $this->db->createCommand("SAVEPOINT $name")->execute();
405 3
    }
406
407
    /**
408
     * Releases an existing savepoint.
409
     *
410
     * @param string $name the savepoint name
411
     */
412
    public function releaseSavepoint(string $name): void
413
    {
414
        $this->db->createCommand("RELEASE SAVEPOINT $name")->execute();
415
    }
416
417
    /**
418
     * Rolls back to a previously created savepoint.
419
     *
420
     * @param string $name the savepoint name
421
     */
422 3
    public function rollBackSavepoint(string $name): void
423
    {
424 3
        $this->db->createCommand("ROLLBACK TO SAVEPOINT $name")->execute();
425 3
    }
426
427
    /**
428
     * Sets the isolation level of the current transaction.
429
     *
430
     * @param string $level The transaction isolation level to use for this transaction.
431
     *
432
     * This can be one of {@see Transaction::READ_UNCOMMITTED}, {@see Transaction::READ_COMMITTED},
433
     * {@see Transaction::REPEATABLE_READ} and {@see Transaction::SERIALIZABLE} but also a string containing DBMS
434
     * specific syntax to be used after `SET TRANSACTION ISOLATION LEVEL`.
435
     *
436
     * {@see http://en.wikipedia.org/wiki/Isolation_%28database_systems%29#Isolation_levels}
437
     */
438 6
    public function setTransactionIsolationLevel(string $level): void
439
    {
440 6
        $this->db->createCommand("SET TRANSACTION ISOLATION LEVEL $level")->execute();
441 6
    }
442
443
    /**
444
     * Executes the INSERT command, returning primary key values.
445
     *
446
     * @param string $table the table that new rows will be inserted into.
447
     * @param array $columns the column data (name => value) to be inserted into the table.
448
     *
449
     * @return array|false primary key values or false if the command fails.
450
     */
451
    public function insert(string $table, array $columns)
452
    {
453
        $command = $this->db->createCommand()->insert($table, $columns);
454
455
        if (!$command->execute()) {
456
            return false;
457
        }
458
459
        $tableSchema = $this->getTableSchema($table);
460
        $result = [];
461
462
        foreach ($tableSchema->getPrimaryKey() as $name) {
463
            if ($tableSchema->getColumn($name)->isAutoIncrement()) {
464
                $result[$name] = $this->getLastInsertID($tableSchema->getSequenceName());
0 ignored issues
show
Bug introduced by
It seems like $tableSchema->getSequenceName() can also be of type null; however, parameter $sequenceName of Yiisoft\Db\Schema\Schema::getLastInsertID() does only seem to accept string, maybe add an additional type check? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

464
                $result[$name] = $this->getLastInsertID(/** @scrutinizer ignore-type */ $tableSchema->getSequenceName());
Loading history...
465
                break;
466
            }
467
468
            $result[$name] = $columns[$name] ?? $tableSchema->getColumn($name)->getDefaultValue();
469
        }
470
471
        return $result;
472
    }
473
474
    /**
475
     * Quotes a string value for use in a query.
476
     *
477
     * Note that if the parameter is not a string, it will be returned without change.
478
     *
479
     * @param string|int $str string to be quoted.
480
     *
481
     * @return string|int the properly quoted string.
482
     *
483
     * {@see http://www.php.net/manual/en/function.PDO-quote.php}
484
     */
485 376
    public function quoteValue($str)
486
    {
487 376
        if (!is_string($str)) {
488 3
            return $str;
489
        }
490
491 376
        if (($value = $this->db->getSlavePdo()->quote($str)) !== false) {
492 376
            return $value;
493
        }
494
495
        /** the driver doesn't support quote (e.g. oci) */
496
        return "'" . addcslashes(str_replace("'", "''", $str), "\000\n\r\\\032") . "'";
497
    }
498
499
    /**
500
     * Quotes a table name for use in a query.
501
     *
502
     * If the table name contains schema prefix, the prefix will also be properly quoted. If the table name is already
503
     * quoted or contains '(' or '{{', then this method will do nothing.
504
     *
505
     * @param string $name table name.
506
     *
507
     * @return string the properly quoted table name.
508
     *
509
     * {@see quoteSimpleTableName()}
510
     */
511 446
    public function quoteTableName(string $name): string
512
    {
513 446
        if (strpos($name, '(') !== false || strpos($name, '{{') !== false) {
514 55
            return $name;
515
        }
516
517 430
        if (strpos($name, '.') === false) {
518 426
            return $this->quoteSimpleTableName($name);
519
        }
520
521 10
        $parts = $this->getTableNameParts($name);
522
523 10
        foreach ($parts as $i => $part) {
524 10
            $parts[$i] = $this->quoteSimpleTableName($part);
525
        }
526
527 10
        return implode('.', $parts);
528
    }
529
530
    /**
531
     * Splits full table name into parts
532
     *
533
     * @param string $name
534
     *
535
     * @return array
536
     */
537
    protected function getTableNameParts(string $name): array
538
    {
539
        return explode('.', $name);
540
    }
541
542 562
    /**
543
     * Quotes a column name for use in a query.
544 562
     *
545 39
     * If the column name contains prefix, the prefix will also be properly quoted. If the column name is already quoted
546
     * or contains '(', '[[' or '{{', then this method will do nothing.
547
     *
548 553
     * @param string $name column name.
549 23
     *
550 23
     * @return string the properly quoted column name.
551
     *
552 550
     * {@see quoteSimpleColumnName()}
553
     */
554
    public function quoteColumnName(string $name): string
555 553
    {
556 3
        if (strpos($name, '(') !== false || strpos($name, '[[') !== false) {
557
            return $name;
558
        }
559 553
560
        if (($pos = strrpos($name, '.')) !== false) {
561
            $prefix = $this->quoteTableName(substr($name, 0, $pos)) . '.';
562
            $name = substr($name, $pos + 1);
563
        } else {
564
            $prefix = '';
565
        }
566
567
        if (strpos($name, '{{') !== false) {
568
            return $name;
569
        }
570
571
        return $prefix . $this->quoteSimpleColumnName($name);
572 456
    }
573
574 456
    /**
575 456
     * Quotes a simple table name for use in a query.
576
     *
577
     * A simple table name should contain the table name only without any schema prefix. If the table name is already
578
     * quoted, this method will do nothing.
579
     *
580 456
     * @param string $name table name.
581
     *
582
     * @return string the properly quoted table name.
583
     */
584
    public function quoteSimpleTableName(string $name): string
585
    {
586
        if (is_string($this->tableQuoteCharacter)) {
587
            $startingCharacter = $endingCharacter = $this->tableQuoteCharacter;
588
        } else {
589
            [$startingCharacter, $endingCharacter] = $this->tableQuoteCharacter;
590
        }
591
592
        return strpos($name, $startingCharacter) !== false ? $name : $startingCharacter . $name . $endingCharacter;
593 553
    }
594
595 553
    /**
596 553
     * Quotes a simple column name for use in a query.
597
     *
598
     * A simple column name should contain the column name only without any prefix. If the column name is already quoted
599
     * or is the asterisk character '*', this method will do nothing.
600
     *
601 553
     * @param string $name column name.
602 553
     *
603
     * @return string the properly quoted column name.
604
     */
605
    public function quoteSimpleColumnName(string $name): string
606
    {
607
        if (is_string($this->tableQuoteCharacter)) {
608
            $startingCharacter = $endingCharacter = $this->columnQuoteCharacter;
609
        } else {
610
            [$startingCharacter, $endingCharacter] = $this->columnQuoteCharacter;
611
        }
612
613
        return $name === '*' || strpos($name, $startingCharacter) !== false ? $name : $startingCharacter . $name
614
            . $endingCharacter;
615 2
    }
616
617 2
    /**
618 2
     * Unquotes a simple table name.
619
     *
620
     * A simple table name should contain the table name only without any schema prefix. If the table name is not
621
     * quoted, this method will do nothing.
622
     *
623 2
     * @param string $name table name.
624
     *
625
     * @return string unquoted table name.
626
     */
627
    public function unquoteSimpleTableName(string $name): string
628
    {
629
        if (\is_string($this->tableQuoteCharacter)) {
630
            $startingCharacter = $this->tableQuoteCharacter;
631
        } else {
632
            $startingCharacter = $this->tableQuoteCharacter[0];
633
        }
634
635
        return strpos($name, $startingCharacter) === false ? $name : substr($name, 1, -1);
636
    }
637
638
    /**
639
     * Unquotes a simple column name.
640
     *
641
     * A simple column name should contain the column name only without any prefix. If the column name is not quoted or
642
     * is the asterisk character '*', this method will do nothing.
643
     *
644
     * @param string $name column name.
645
     *
646
     * @return string unquoted column name.
647
     */
648
    public function unquoteSimpleColumnName(string $name): string
649
    {
650
        if (\is_string($this->columnQuoteCharacter)) {
651
            $startingCharacter = $this->columnQuoteCharacter;
652
        } else {
653
            $startingCharacter = $this->columnQuoteCharacter[0];
654
        }
655
656
        return strpos($name, $startingCharacter) === false ? $name : substr($name, 1, -1);
657 443
    }
658
659 443
    /**
660 69
     * Returns the actual name of a given table name.
661
     *
662 69
     * This method will strip off curly brackets from the given table name and replace the percentage character '%' with
663
     * {@see Connection::tablePrefix}.
664
     *
665 443
     * @param string $name the table name to be converted.
666
     *
667
     * @return string the real name of the given table name.
668
     */
669
    public function getRawTableName(string $name): string
670
    {
671
        if (strpos($name, '{{') !== false) {
672
            $name = preg_replace('/\\{\\{(.*?)\\}\\}/', '\1', $name);
673
674
            return str_replace('%', $this->db->getTablePrefix(), $name);
675 246
        }
676
677 246
        return $name;
678
    }
679
680
    /**
681
     * Extracts the PHP type from abstract DB type.
682
     *
683
     * @param ColumnSchema $column the column schema information.
684
     *
685
     * @return string PHP type name.
686
     */
687
    protected function getColumnPhpType(ColumnSchema $column): string
688
    {
689
        static $typeMap = [
690 246
            // abstract type => php type
691 238
            self::TYPE_TINYINT  => 'integer',
692 36
            self::TYPE_SMALLINT => 'integer',
693
            self::TYPE_INTEGER  => 'integer',
694
            self::TYPE_BIGINT   => 'integer',
695 238
            self::TYPE_BOOLEAN  => 'boolean',
696 238
            self::TYPE_FLOAT    => 'double',
697
            self::TYPE_DOUBLE   => 'double',
698
            self::TYPE_BINARY   => 'resource',
699 150
            self::TYPE_JSON     => 'array',
700
        ];
701
702 230
        if (isset($typeMap[$column->getType()])) {
703
            if ($column->getType() === 'bigint') {
704
                return PHP_INT_SIZE === 8 && !$column->isUnsigned() ? 'integer' : 'string';
705
            }
706
707
            if ($column->getType() === 'integer') {
708
                return PHP_INT_SIZE === 4 && $column->isUnsigned() ? 'string' : 'integer';
709
            }
710
711
            return $typeMap[$column->getType()];
712
        }
713 28
714
        return 'string';
715 28
    }
716
717
    /**
718
     * Converts a DB exception to a more concrete one if possible.
719 28
     *
720
     * @param \Exception $e
721 28
     * @param string $rawSql SQL that produced exception.
722 28
     *
723 7
     * @return Exception
724
     */
725
    public function convertException(\Exception $e, string $rawSql): Exception
726
    {
727 28
        if ($e instanceof Exception) {
728 28
            return $e;
729
        }
730 28
731
        $exceptionClass = Exception::class;
732
733
        foreach ($this->exceptionMap as $error => $class) {
734
            if (strpos($e->getMessage(), $error) !== false) {
735
                $exceptionClass = $class;
736
            }
737
        }
738
739
        $message = $e->getMessage() . "\nThe SQL being executed was: $rawSql";
740 9
        $errorInfo = $e instanceof \PDOException ? $e->errorInfo : null;
741
742 9
        return new $exceptionClass($message, $errorInfo, (int) $e->getCode(), $e);
743
    }
744 9
745
    /**
746
     * Returns a value indicating whether a SQL statement is for read purpose.
747
     *
748
     * @param string $sql the SQL statement.
749
     *
750
     * @return bool whether a SQL statement is for read purpose.
751
     */
752 116
    public function isReadQuery($sql): bool
753
    {
754 116
        $pattern = '/^\s*(SELECT|SHOW|DESCRIBE)\b/i';
755 116
756
        return preg_match($pattern, $sql) > 0;
757
    }
758 116
759
    /**
760
     * Returns a server version as a string comparable by {@see \version_compare()}.
761
     *
762
     * @return string server version as a string.
763
     */
764
    public function getServerVersion(): string
765
    {
766
        if ($this->serverVersion === null) {
767
            $this->serverVersion = $this->db->getSlavePdo()->getAttribute(\PDO::ATTR_SERVER_VERSION);
768 443
        }
769
770
        return $this->serverVersion;
771 443
    }
772 443
773 443
    /**
774 443
     * Returns the cache key for the specified table name.
775
     *
776
     * @param string $name the table name.
777
     *
778
     * @return mixed the cache key.
779
     */
780
    protected function getCacheKey($name)
781
    {
782
        return [
783
            __CLASS__,
784
            $this->db->getDsn(),
785 395
            $this->db->getUsername(),
786
            $this->getRawTableName($name),
787 395
        ];
788 395
    }
789 395
790 395
    /**
791
     * Returns the cache tag name.
792
     *
793
     * This allows {@see refresh()} to invalidate all cached table schemas.
794
     *
795
     * @return string the cache tag name.
796
     */
797
    protected function getCacheTag(): string
798
    {
799
        return md5(serialize([
800
            __CLASS__,
801
            $this->db->getDsn(),
802
            $this->db->getUsername(),
803
        ]));
804
    }
805
806 443
    /**
807
     * Returns the metadata of the given type for the given table.
808 443
     *
809 443
     * If there's no metadata in the cache, this method will call a `'loadTable' . ucfirst($type)` named method with the
810
     * table name to obtain the metadata.
811
     *
812 443
     * @param string $name table name. The table name may contain schema name if any. Do not quote the table name.
813
     * @param string $type metadata type.
814 443
     * @param bool $refresh whether to reload the table metadata even if it is found in the cache.
815 443
     *
816
     * @return mixed metadata.
817
     */
818 443
    protected function getTableMetadata(string $name, string $type, bool $refresh = false)
819 443
    {
820 395
        if ($this->db->isSchemaCacheEnabled() && !\in_array($name, $this->db->getSchemaCacheExclude(), true)) {
821
            $schemaCache = $this->cache;
822
        }
823 395
824
        $rawName = $this->getRawTableName($name);
825
826
        if (!isset($this->tableMetadata[$rawName])) {
827
            $this->loadTableMetadataFromCache($schemaCache, $rawName);
0 ignored issues
show
Comprehensibility Best Practice introduced by
The variable $schemaCache does not seem to be defined for all execution paths leading up to this point.
Loading history...
828
        }
829
830
        if ($refresh || !array_key_exists($type, $this->tableMetadata[$rawName])) {
831
            $this->tableMetadata[$rawName][$type] = $this->{'loadTable' . ucfirst($type)}($rawName);
832
            $this->saveTableMetadataToCache($schemaCache, $rawName);
833
        }
834
835
        return $this->tableMetadata[$rawName][$type];
836
    }
837
838
    /**
839
     * Returns the metadata of the given type for all tables in the given schema.
840 9
     *
841
     * This method will call a `'getTable' . ucfirst($type)` named method with the table name and the refresh flag to
842 9
     * obtain the metadata.
843 9
     *
844
     * @param string $schema the schema of the metadata. Defaults to empty string, meaning the current or default schema
845 9
     * name.
846 9
     * @param string $type metadata type.
847
     * @param bool $refresh whether to fetch the latest available table metadata. If this is `false`, cached data may be
848
     * returned if available.
849 9
     *
850 9
     * @return array array of metadata.
851 9
     */
852
    protected function getSchemaMetadata(string $schema, string $type, bool $refresh): array
853
    {
854
        $metadata = [];
855 9
        $methodName = 'getTable' . ucfirst($type);
856
857
        foreach ($this->getTableNames($schema, $refresh) as $name) {
858
            if ($schema !== '') {
859
                $name = $schema . '.' . $name;
860
            }
861
862
            $tableMetadata = $this->$methodName($name, $refresh);
863
864
            if ($tableMetadata !== null) {
865 162
                $metadata[] = $tableMetadata;
866
            }
867 162
        }
868 162
869
        return $metadata;
870
    }
871
872
    /**
873
     * Sets the metadata of the given type for the given table.
874
     *
875
     * @param string $name table name.
876
     * @param string $type metadata type.
877
     * @param mixed  $data metadata.
878 186
     */
879
    protected function setTableMetadata(string $name, string $type, $data): void
880 186
    {
881 146
        $this->tableMetadata[$this->getRawTableName($name)][$type] = $data;
882
    }
883
884 40
    /**
885 40
     * Changes row's array key case to lower if PDO's one is set to uppercase.
886 39
     *
887 40
     * @param array $row row's array or an array of row's arrays.
888
     * @param bool $multiple whether multiple rows or a single row passed.
889
     *
890
     * @return array normalized row or rows.
891
     */
892
    protected function normalizePdoRowKeyCase(array $row, bool $multiple): array
893
    {
894
        if ($this->db->getSlavePdo()->getAttribute(\PDO::ATTR_CASE) !== \PDO::CASE_UPPER) {
895
            return $row;
896
        }
897
898
        if ($multiple) {
899 443
            return \array_map(function (array $row) {
900
                return \array_change_key_case($row, CASE_LOWER);
901 443
            }, $row);
902
        }
903
904
        return \array_change_key_case($row, CASE_LOWER);
905
    }
906
907 443
    /**
908
     * Tries to load and populate table metadata from cache.
909 443
     *
910 443
     * @param CacheInterface|null $cache
911
     * @param string $name
912 443
     */
913
    private function loadTableMetadataFromCache(?CacheInterface $cache, string $name): void
914
    {
915
        if ($cache === null) {
916
            $this->tableMetadata[$name] = [];
917
918
            return;
919
        }
920
921
        $metadata = $cache->get($this->getCacheKey($name));
0 ignored issues
show
Bug introduced by
$this->getCacheKey($name) of type array<integer,null|string> is incompatible with the type string expected by parameter $key of Psr\SimpleCache\CacheInterface::get(). ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

921
        $metadata = $cache->get(/** @scrutinizer ignore-type */ $this->getCacheKey($name));
Loading history...
922
923
        if (!\is_array($metadata) || !isset($metadata['cacheVersion']) || $metadata['cacheVersion'] !== static::SCHEMA_CACHE_VERSION) {
924
            $this->tableMetadata[$name] = [];
925 395
926
            return;
927 395
        }
928
929
        unset($metadata['cacheVersion']);
930
        $this->tableMetadata[$name] = $metadata;
931 395
    }
932
933 395
    /**
934
     * Saves table metadata to cache.
935 395
     *
936 395
     * @param CacheInterface|null $cache
937
     * @param string $name
938 395
     */
939 395
    private function saveTableMetadataToCache(?CacheInterface $cache, string $name): void
940
    {
941 395
        if ($cache === null) {
942
            return;
943 445
        }
944
945 445
        $metadata = $this->tableMetadata[$name];
946
947
        $metadata['cacheVersion'] = static::SCHEMA_CACHE_VERSION;
948
949
        $cache->set(
950
            $this->getCacheKey($name),
951
            $metadata,
952
            $this->db->getSchemaCacheDuration(),
953
            new TagDependency(['tags' => $this->getCacheTag()])
954
        );
955
    }
956
957
    public function getDb(): ?Connection
958
    {
959
        return $this->db;
960
    }
961
962
    public function getDefaultSchema(): ?string
963
    {
964
        return $this->defaultSchema;
965
    }
966
}
967