mirror of
https://github.com/elyby/accounts.git
synced 2024-12-23 22:00:06 +05:30
61 lines
1.8 KiB
PHP
61 lines
1.8 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace console\db;
|
|
|
|
use yii\db\Exception;
|
|
use yii\db\Migration as YiiMigration;
|
|
|
|
/**
|
|
* @property string $tableOptions
|
|
* @method \SamIT\Yii2\MariaDb\ColumnSchemaBuilder json()
|
|
*/
|
|
class Migration extends YiiMigration {
|
|
|
|
public function getTableOptions(string $engine = 'InnoDB'): string {
|
|
$tableOptions = null;
|
|
if ($this->db->driverName === 'mysql') {
|
|
$tableOptions = 'CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci ENGINE=' . $engine;
|
|
}
|
|
|
|
return $tableOptions;
|
|
}
|
|
|
|
/**
|
|
* @param array<string|\yii\db\ColumnSchemaBuilder>|null $columns
|
|
*/
|
|
public function createTable($table, $columns, $options = null): void {
|
|
if ($options === null) {
|
|
$options = $this->getTableOptions();
|
|
}
|
|
|
|
parent::createTable($table, $columns, $options);
|
|
}
|
|
|
|
protected function primary(string ...$columns): string {
|
|
foreach ($columns as &$column) {
|
|
$column = $this->db->quoteColumnName($column);
|
|
}
|
|
|
|
return ' PRIMARY KEY (' . implode(', ', $columns) . ') ';
|
|
}
|
|
|
|
protected function getPrimaryKeyType(string $table, bool $nullable = false): string {
|
|
$primaryKeys = $this->db->getTableSchema($table)->primaryKey;
|
|
if (count($primaryKeys) === 0) {
|
|
throw new Exception("The table \"{$table}\" have no primary keys.");
|
|
}
|
|
|
|
if (count($primaryKeys) > 1) {
|
|
throw new Exception("The table \"{$table}\" have more than one primary key.");
|
|
}
|
|
|
|
return $this->getColumnType($table, $primaryKeys[0], $nullable);
|
|
}
|
|
|
|
protected function getColumnType(string $table, string $column, bool $nullable = false): string {
|
|
return $this->db->getTableSchema($table)->getColumn($column)->dbType . ($nullable ? '' : ' NOT NULL');
|
|
}
|
|
|
|
}
|