|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace App; |
|
4
|
|
|
|
|
5
|
|
|
/** |
|
6
|
|
|
* Database connection class. |
|
7
|
|
|
* |
|
8
|
|
|
* @package App |
|
9
|
|
|
* |
|
10
|
|
|
* @copyright YetiForce S.A. |
|
11
|
|
|
* @license YetiForce Public License 6.5 (licenses/LicenseEN.txt or yetiforce.com) |
|
12
|
|
|
* @author Mariusz Krzaczkowski <[email protected]> |
|
13
|
|
|
* @author Radosław Skrzypczak <[email protected]> |
|
14
|
|
|
*/ |
|
15
|
|
|
class Db extends \yii\db\Connection |
|
16
|
|
|
{ |
|
17
|
|
|
/** |
|
18
|
|
|
* Sorting order flag. |
|
19
|
|
|
*/ |
|
20
|
|
|
public const ASC = 'ASC'; |
|
21
|
|
|
|
|
22
|
|
|
/** |
|
23
|
|
|
* Sorting order flag. |
|
24
|
|
|
*/ |
|
25
|
|
|
public const DESC = 'DESC'; |
|
26
|
|
|
|
|
27
|
|
|
/** |
|
28
|
|
|
* @var bool whether to turn on prepare emulation. Defaults to false, meaning PDO |
|
29
|
|
|
* will use the native prepare support if available. For some databases (such as MySQL), |
|
30
|
|
|
* this may need to be set true so that PDO can emulate the prepare support to bypass |
|
31
|
|
|
* the buggy native prepare support. |
|
32
|
|
|
* The default value is null, which means the PDO ATTR_EMULATE_PREPARES value will not be changed |
|
33
|
|
|
*/ |
|
34
|
|
|
public $emulatePrepare = false; |
|
35
|
|
|
|
|
36
|
|
|
/** |
|
37
|
|
|
* @var \App\Db Table of connections with database |
|
38
|
|
|
*/ |
|
39
|
|
|
private static $cache = []; |
|
40
|
|
|
|
|
41
|
|
|
/** |
|
42
|
|
|
* @var array Configuration with database |
|
43
|
|
|
*/ |
|
44
|
|
|
private static $config = []; |
|
45
|
|
|
|
|
46
|
|
|
/** |
|
47
|
|
|
* @var bool Enable caching database instance |
|
48
|
|
|
*/ |
|
49
|
|
|
public static $connectCache = false; |
|
50
|
|
|
|
|
51
|
|
|
/** |
|
52
|
|
|
* @var string Database Name |
|
53
|
|
|
*/ |
|
54
|
|
|
public $dbName; |
|
55
|
|
|
|
|
56
|
|
|
/** |
|
57
|
|
|
* @var string Database section |
|
58
|
|
|
*/ |
|
59
|
|
|
public $dbType; |
|
60
|
|
|
|
|
61
|
|
|
/** |
|
62
|
|
|
* @var string Host database server |
|
63
|
|
|
*/ |
|
64
|
|
|
public $host; |
|
65
|
|
|
|
|
66
|
|
|
/** |
|
67
|
|
|
* @var int Port database server |
|
68
|
|
|
*/ |
|
69
|
|
|
public $port; |
|
70
|
|
|
|
|
71
|
|
|
/** {@inheritdoc} */ |
|
72
|
|
|
public $schemaMap = [ |
|
73
|
|
|
'pgsql' => 'App\Db\Drivers\Pgsql\Schema', // PostgreSQL |
|
74
|
|
|
'mysqli' => 'yii\db\mysql\Schema', // MySQL |
|
75
|
|
|
'mysql' => 'App\Db\Drivers\Mysql\Schema', // MySQL |
|
76
|
|
|
'sqlite' => 'yii\db\sqlite\Schema', // sqlite 3 |
|
77
|
|
|
'sqlite2' => 'yii\db\sqlite\Schema', // sqlite 2 |
|
78
|
|
|
'sqlsrv' => 'yii\db\mssql\Schema', // newer MSSQL driver on MS Windows hosts |
|
79
|
|
|
'oci' => 'yii\db\oci\Schema', // Oracle driver |
|
80
|
|
|
'mssql' => 'yii\db\mssql\Schema', // older MSSQL driver on MS Windows hosts |
|
81
|
|
|
'dblib' => 'yii\db\mssql\Schema', // dblib drivers on GNU/Linux (and maybe other OSes) hosts |
|
82
|
|
|
'cubrid' => 'yii\db\cubrid\Schema', // CUBRID |
|
83
|
|
|
]; |
|
84
|
|
|
|
|
85
|
|
|
/** |
|
86
|
|
|
* @var string the class used to create new database [[Command]] objects. If you want to extend the [[Command]] class, |
|
87
|
|
|
* you may configure this property to use your extended version of the class |
|
88
|
|
|
*/ |
|
89
|
|
|
public $commandClass = '\App\Db\Command'; |
|
90
|
|
|
|
|
91
|
|
|
/** |
|
92
|
|
|
* @var Cache|string the cache object or the ID of the cache application component that |
|
93
|
|
|
* is used to cache the table metadata |
|
94
|
|
|
* |
|
95
|
6356 |
|
* @see enableSchemaCache |
|
96
|
|
|
*/ |
|
97
|
6356 |
|
public $schemaCache = false; |
|
98
|
6356 |
|
|
|
99
|
|
|
/** |
|
100
|
3 |
|
* Creates the Db connection instance. |
|
101
|
3 |
|
* |
|
102
|
3 |
|
* @param string $type Name of database connection |
|
103
|
3 |
|
* |
|
104
|
|
|
* @return \App\Db |
|
105
|
|
|
*/ |
|
106
|
|
|
public static function getInstance($type = 'base') |
|
107
|
|
|
{ |
|
108
|
|
|
if (isset(self::$cache[$type])) { |
|
109
|
|
|
return self::$cache[$type]; |
|
110
|
|
|
} |
|
111
|
|
|
$db = new self(self::getConfig($type)); |
|
112
|
|
|
$db->dbType = $type; |
|
113
|
4 |
|
self::$cache[$type] = $db; |
|
114
|
|
|
return $db; |
|
115
|
4 |
|
} |
|
116
|
3 |
|
|
|
117
|
|
|
/** |
|
118
|
4 |
|
* Load database connection configuration. |
|
119
|
|
|
* |
|
120
|
|
|
* @param string $type |
|
121
|
|
|
* |
|
122
|
|
|
* @return array with database configuration |
|
123
|
|
|
*/ |
|
124
|
|
|
public static function getConfig(string $type) |
|
125
|
|
|
{ |
|
126
|
|
|
if (!isset(self::$config[$type])) { |
|
127
|
|
|
self::$config[$type] = Config::db($type) ?? Config::db('base'); |
|
128
|
|
|
} |
|
129
|
|
|
return self::$config[$type]; |
|
130
|
|
|
} |
|
131
|
|
|
|
|
132
|
|
|
/** |
|
133
|
|
|
* Set database connection configuration. |
|
134
|
|
|
* |
|
135
|
|
|
* @param array $config |
|
136
|
|
|
* @param string $type |
|
137
|
2 |
|
*/ |
|
138
|
|
|
public static function setConfig($config, $type = 'base') |
|
139
|
2 |
|
{ |
|
140
|
2 |
|
self::$config[$type] = $config; |
|
141
|
2 |
|
} |
|
142
|
2 |
|
|
|
143
|
2 |
|
/** |
|
144
|
2 |
|
* Get info database server. |
|
145
|
2 |
|
* |
|
146
|
|
|
* @return array |
|
147
|
|
|
*/ |
|
148
|
2 |
|
public function getInfo() |
|
149
|
|
|
{ |
|
150
|
|
|
$pdo = $this->getSlavePdo(); |
|
151
|
2 |
|
$statement = $pdo->prepare('SHOW VARIABLES'); |
|
152
|
2 |
|
$statement->execute(); |
|
153
|
2 |
|
$conf = $statement->fetchAll(\PDO::FETCH_KEY_PAIR); |
|
154
|
2 |
|
$statement = $pdo->prepare('SHOW STATUS'); |
|
155
|
2 |
|
$statement->execute(); |
|
156
|
2 |
|
$conf = array_merge($conf, $statement->fetchAll(\PDO::FETCH_KEY_PAIR)); |
|
157
|
2 |
|
$statement = $pdo->prepare('SELECT VERSION()'); |
|
158
|
|
|
$statement->execute(); |
|
159
|
|
|
$fullVersion = $statement->fetch(\PDO::FETCH_COLUMN); |
|
160
|
|
|
[$version] = explode('-', $conf['version']); |
|
161
|
|
|
$conf['version_comment'] = $conf['version_comment'] . '|' . $fullVersion; |
|
162
|
|
|
$typeDb = 'MySQL'; |
|
163
|
|
|
if (false !== stripos($conf['version_comment'], 'MariaDb')) { |
|
164
|
|
|
$typeDb = 'MariaDb'; |
|
165
|
|
|
} |
|
166
|
|
|
$memory = $conf['key_buffer_size'] + ($conf['query_cache_size'] ?? 0) + $conf['tmp_table_size'] + $conf['innodb_buffer_pool_size'] + |
|
167
|
|
|
($conf['innodb_additional_mem_pool_size'] ?? 0) + $conf['innodb_log_buffer_size'] + ($conf['max_connections'] * ($conf['sort_buffer_size'] |
|
168
|
|
|
+ $conf['read_buffer_size'] + $conf['read_rnd_buffer_size'] + $conf['join_buffer_size'] + $conf['thread_stack'] + $conf['binlog_cache_size'])); |
|
169
|
|
|
return \array_merge($conf, [ |
|
170
|
|
|
'driver' => $this->getDriverName(), |
|
171
|
|
|
'typeDb' => $typeDb, |
|
172
|
6352 |
|
'serverVersion' => $version, |
|
173
|
|
|
'maximumMemorySize' => $memory, |
|
174
|
6352 |
|
'clientVersion' => $pdo->getAttribute(\PDO::ATTR_CLIENT_VERSION), |
|
175
|
|
|
'connectionStatus' => $pdo->getAttribute(\PDO::ATTR_CONNECTION_STATUS), |
|
176
|
|
|
'serverInfo' => $pdo->getAttribute(\PDO::ATTR_SERVER_INFO), |
|
177
|
|
|
]); |
|
178
|
|
|
} |
|
179
|
|
|
|
|
180
|
|
|
/** |
|
181
|
|
|
* Get database info. |
|
182
|
|
|
* |
|
183
|
|
|
* @return array |
|
184
|
|
|
*/ |
|
185
|
|
|
public function getDbInfo(): array |
|
186
|
75 |
|
{ |
|
187
|
|
|
$return = [ |
|
188
|
75 |
|
'isFileSize' => false, |
|
189
|
|
|
'size' => 0, |
|
190
|
|
|
'dataSize' => 0, |
|
191
|
|
|
'indexSize' => 0, |
|
192
|
|
|
'filesSize' => 0, |
|
193
|
|
|
'tables' => [], |
|
194
|
|
|
]; |
|
195
|
|
|
$statement = $this->getSlavePdo()->prepare("SHOW TABLE STATUS FROM `{$this->dbName}`"); |
|
196
|
|
|
$statement->execute(); |
|
197
|
|
|
while ($row = $statement->fetch(\PDO::FETCH_ASSOC)) { |
|
198
|
|
|
$return['tables'][$row['Name']] = [ |
|
199
|
3 |
|
'rows' => $row['Rows'], |
|
200
|
|
|
'format' => $row['Row_format'], |
|
201
|
3 |
|
'engine' => $row['Engine'], |
|
202
|
|
|
'dataSize' => $row['Data_length'], |
|
203
|
|
|
'indexSize' => $row['Index_length'], |
|
204
|
|
|
'collation' => $row['Collation'], |
|
205
|
|
|
]; |
|
206
|
|
|
$return['dataSize'] += $row['Data_length']; |
|
207
|
3 |
|
$return['indexSize'] += $row['Index_length']; |
|
208
|
|
|
$return['size'] += $row['Data_length'] += $row['Index_length']; |
|
209
|
|
|
} |
|
210
|
|
|
try { |
|
211
|
|
|
$statement = $this->getSlavePdo()->prepare("SELECT * FROM `information_schema`.`INNODB_SYS_TABLESPACES` WHERE `NAME` LIKE '{$this->dbName}/%'"); |
|
212
|
|
|
$statement->execute(); |
|
213
|
|
|
while ($row = $statement->fetch(\PDO::FETCH_ASSOC)) { |
|
214
|
|
|
$tableName = str_replace($this->dbName . '/', '', $row['NAME']); |
|
215
|
|
|
if (!empty($row['ALLOCATED_SIZE'])) { |
|
216
|
|
|
if (isset($return['tables'][$tableName])) { |
|
217
|
|
|
$return['tables'][$tableName]['fileSize'] = $row['ALLOCATED_SIZE']; |
|
218
|
|
|
$return['isFileSize'] = true; |
|
219
|
11 |
|
} |
|
220
|
|
|
$return['filesSize'] += $row['ALLOCATED_SIZE']; |
|
221
|
11 |
|
} |
|
222
|
7 |
|
} |
|
223
|
7 |
|
} catch (\Throwable $th) { |
|
|
|
|
|
|
224
|
7 |
|
} |
|
225
|
7 |
|
return $return; |
|
226
|
7 |
|
} |
|
227
|
7 |
|
|
|
228
|
|
|
/** |
|
229
|
4 |
|
* Processes a SQL statement by quoting table and column names that are enclosed within double brackets. |
|
230
|
4 |
|
* Tokens enclosed within double curly brackets are treated as table names, while |
|
231
|
4 |
|
* tokens enclosed within double square brackets are column names. They will be quoted accordingly. |
|
232
|
4 |
|
* Also, the percentage character "%" at the beginning or ending of a table name will be replaced |
|
233
|
|
|
* with [[tablePrefix]]. |
|
234
|
11 |
|
* |
|
235
|
|
|
* @param string $sql the SQL to be quoted |
|
236
|
|
|
* |
|
237
|
|
|
* @return string the quoted SQL |
|
238
|
|
|
*/ |
|
239
|
|
|
public function quoteSql($sql) |
|
240
|
|
|
{ |
|
241
|
|
|
return str_replace('#__', $this->tablePrefix, $sql); |
|
242
|
|
|
} |
|
243
|
|
|
|
|
244
|
10 |
|
/** |
|
245
|
|
|
* Returns the ID of the last inserted row or sequence value. |
|
246
|
10 |
|
* |
|
247
|
|
|
* @param string $sequenceName name of the sequence object (required by some DBMS) ex. table vtiger_picklist >>> vtiger_picklist_picklistid_seq |
|
248
|
|
|
* |
|
249
|
|
|
* @return string the row ID of the last row inserted, or the last value retrieved from the sequence object |
|
250
|
|
|
* |
|
251
|
|
|
* @see https://www.php.net/manual/en/function.PDO-lastInsertId.php |
|
252
|
|
|
*/ |
|
253
|
|
|
public function getLastInsertID($sequenceName = '') |
|
254
|
|
|
{ |
|
255
|
|
|
return parent::getLastInsertID(str_replace('#__', $this->tablePrefix, $sequenceName)); |
|
256
|
|
|
} |
|
257
|
4 |
|
|
|
258
|
|
|
/** |
|
259
|
4 |
|
* Creates the PDO instance. |
|
260
|
4 |
|
* This method is called by [[open]] to establish a DB connection. |
|
261
|
4 |
|
* The default implementation will create a PHP PDO instance. |
|
262
|
|
|
* You may override this method if the default PDO needs to be adapted for certain DBMS. |
|
263
|
4 |
|
* |
|
264
|
4 |
|
* @return PDO the pdo instance |
|
|
|
|
|
|
265
|
|
|
*/ |
|
266
|
|
|
protected function createPdoInstance() |
|
267
|
|
|
{ |
|
268
|
|
|
if (Debuger::isDebugBar()) { |
|
269
|
|
|
$bebugBar = Debuger::getDebugBar(); |
|
270
|
|
|
$pdo = new Debug\DebugBar\TraceablePDO(parent::createPdoInstance()); |
|
271
|
|
|
if ($bebugBar->hasCollector('pdo')) { |
|
272
|
|
|
$pdoCollector = $bebugBar->getCollector('pdo'); |
|
273
|
|
|
$pdoCollector->addConnection($pdo, $this->dbType); |
|
|
|
|
|
|
274
|
|
|
} else { |
|
275
|
|
|
$pdoCollector = new \DebugBar\DataCollector\PDO\PDOCollector(); |
|
276
|
|
|
$pdoCollector->addConnection($pdo, $this->dbType); |
|
277
|
|
|
$bebugBar->addCollector($pdoCollector); |
|
278
|
|
|
} |
|
279
|
|
|
return $pdo; |
|
|
|
|
|
|
280
|
|
|
} |
|
281
|
|
|
return parent::createPdoInstance(); |
|
|
|
|
|
|
282
|
|
|
} |
|
283
|
|
|
|
|
284
|
|
|
/** |
|
285
|
|
|
* Get table unique ID. Temporary function. |
|
286
|
|
|
* |
|
287
|
|
|
* @param string $tableName |
|
288
|
|
|
* @param false|string $columnName |
|
289
|
|
|
* @param bool $seq |
|
290
|
|
|
* |
|
291
|
|
|
* @return int |
|
292
|
|
|
*/ |
|
293
|
|
|
public function getUniqueID($tableName, $columnName = false, $seq = true) |
|
294
|
|
|
{ |
|
295
|
|
|
if ($seq) { |
|
296
|
|
|
$tableName .= '_seq'; |
|
297
|
|
|
$id = (new \App\Db\Query())->from($tableName)->scalar($this); |
|
298
|
|
|
++$id; |
|
299
|
|
|
$this->createCommand()->update($tableName, [ |
|
300
|
|
|
'id' => $id, |
|
301
|
|
|
])->execute(); |
|
302
|
|
|
} else { |
|
303
|
|
|
$id = (new \App\Db\Query()) |
|
304
|
|
|
->from($tableName) |
|
305
|
|
|
->max($columnName, $this); |
|
306
|
|
|
++$id; |
|
307
|
|
|
} |
|
308
|
|
|
return $id; |
|
|
|
|
|
|
309
|
|
|
} |
|
310
|
|
|
|
|
311
|
|
|
/** |
|
312
|
|
|
* Check if table is present in database. |
|
313
|
|
|
* |
|
314
|
|
|
* @param string $tableName |
|
315
|
|
|
* |
|
316
|
|
|
* @return bool |
|
317
|
|
|
*/ |
|
318
|
|
|
public function isTableExists($tableName) |
|
319
|
|
|
{ |
|
320
|
|
|
return \in_array(str_replace('#__', $this->tablePrefix, $tableName), $this->getSchema()->getTableNames()); |
|
321
|
|
|
} |
|
322
|
|
|
|
|
323
|
|
|
/** |
|
324
|
|
|
* Creating a new DB table. |
|
325
|
|
|
* |
|
326
|
|
|
* @param string $tableName |
|
327
|
|
|
* @param mixed $columns |
|
328
|
|
|
* |
|
329
|
|
|
* @return bool |
|
330
|
|
|
*/ |
|
331
|
|
|
public function createTable($tableName, $columns) |
|
332
|
|
|
{ |
|
333
|
|
|
$tableOptions = null; |
|
334
|
|
|
if ('mysql' === $this->getDriverName()) { |
|
335
|
|
|
$tableOptions = 'CHARACTER SET utf8 ENGINE=InnoDB'; |
|
336
|
|
|
} |
|
337
|
|
|
$this->createCommand()->createTable($tableName, $columns, $tableOptions)->execute(); |
|
338
|
|
|
} |
|
339
|
|
|
|
|
340
|
|
|
/** |
|
341
|
|
|
* Get table keys. |
|
342
|
|
|
* |
|
343
|
|
|
* @param string $tableName |
|
344
|
|
|
* |
|
345
|
|
|
* @return array |
|
346
|
|
|
*/ |
|
347
|
|
|
public function getTableKeys($tableName) |
|
348
|
|
|
{ |
|
349
|
|
|
if (Cache::has('getTableKeys', $tableName)) { |
|
350
|
|
|
return Cache::get('getTableKeys', $tableName); |
|
351
|
|
|
} |
|
352
|
|
|
if (!$this->isTableExists($tableName)) { |
|
353
|
|
|
return []; |
|
354
|
|
|
} |
|
355
|
|
|
$tableName = $this->quoteTableName(str_replace('#__', $this->tablePrefix, $tableName)); |
|
356
|
|
|
$keys = []; |
|
357
|
|
|
if ('mysql' === $this->getDriverName()) { |
|
358
|
|
|
$dataReader = $this->createCommand()->setSql('SHOW KEYS FROM ' . $tableName)->query(); |
|
359
|
|
|
while ($row = $dataReader->read()) { |
|
360
|
|
|
$keys[$row['Key_name']][$row['Column_name']] = ['columnName' => $row['Column_name'], 'unique' => empty($row['Non_unique'])]; |
|
361
|
|
|
} |
|
362
|
|
|
} |
|
363
|
|
|
Cache::save('getTableKeys', $tableName, $keys, Cache::LONG); |
|
364
|
|
|
return $keys; |
|
365
|
|
|
} |
|
366
|
|
|
|
|
367
|
|
|
/** |
|
368
|
|
|
* Get table primary keys. |
|
369
|
|
|
* |
|
370
|
|
|
* @param type $tableName |
|
|
|
|
|
|
371
|
|
|
* |
|
372
|
|
|
* @return type |
|
373
|
|
|
*/ |
|
374
|
|
|
public function getPrimaryKey($tableName) |
|
375
|
|
|
{ |
|
376
|
|
|
if (Cache::has('getPrimaryKey', $tableName)) { |
|
377
|
|
|
return Cache::get('getPrimaryKey', $tableName); |
|
378
|
|
|
} |
|
379
|
|
|
$key = []; |
|
380
|
|
|
if ('mysql' === $this->getDriverName()) { |
|
381
|
|
|
$tableKeys = $this->getTableKeys($tableName); |
|
382
|
|
|
$key = isset($tableKeys['PRIMARY']) ? ['PRIMARY' => array_keys($tableKeys['PRIMARY'])] : []; |
|
383
|
|
|
} |
|
384
|
|
|
Cache::save('getPrimaryKey', $tableName, $key, Cache::LONG); |
|
385
|
|
|
return $key; |
|
|
|
|
|
|
386
|
|
|
} |
|
387
|
|
|
} |
|
388
|
|
|
|