|
|
@ -12,7 +12,7 @@ use yii\db\TableSchema; |
|
|
|
use yii\db\ColumnSchema; |
|
|
|
use yii\db\ColumnSchema; |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* Schema is the class for retrieving metadata from a PostgreSQL database |
|
|
|
* Schema is the class for retrieving metadata from a PostgreSQL database |
|
|
|
* (version 9.x and above). |
|
|
|
* (version 9.x and above). |
|
|
|
* |
|
|
|
* |
|
|
|
* @author Gevik Babakhani <gevikb@gmail.com> |
|
|
|
* @author Gevik Babakhani <gevikb@gmail.com> |
|
|
@ -23,60 +23,61 @@ class Schema extends \yii\db\Schema |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* The default schema used for the current session. |
|
|
|
* The default schema used for the current session. |
|
|
|
* @var string |
|
|
|
* @var string |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public $defaultSchema = 'public'; |
|
|
|
public $defaultSchema = 'public'; |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* @var array mapping from physical column types (keys) to abstract |
|
|
|
* @var array mapping from physical column types (keys) to abstract |
|
|
|
* column types (values) |
|
|
|
* column types (values) |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public $typeMap = array( |
|
|
|
public $typeMap = array( |
|
|
|
'abstime' => self::TYPE_TIMESTAMP, |
|
|
|
'abstime' => self::TYPE_TIMESTAMP, |
|
|
|
'bit' => self::TYPE_STRING, |
|
|
|
'bit' => self::TYPE_STRING, |
|
|
|
'boolean' => self::TYPE_BOOLEAN, |
|
|
|
'boolean' => self::TYPE_BOOLEAN, |
|
|
|
'box' => self::TYPE_STRING, |
|
|
|
'box' => self::TYPE_STRING, |
|
|
|
'character' => self::TYPE_STRING, |
|
|
|
'character' => self::TYPE_STRING, |
|
|
|
'bytea' => self::TYPE_BINARY, |
|
|
|
'bytea' => self::TYPE_BINARY, |
|
|
|
'char' => self::TYPE_STRING, |
|
|
|
'char' => self::TYPE_STRING, |
|
|
|
'cidr' => self::TYPE_STRING, |
|
|
|
'cidr' => self::TYPE_STRING, |
|
|
|
'circle' => self::TYPE_STRING, |
|
|
|
'circle' => self::TYPE_STRING, |
|
|
|
'date' => self::TYPE_DATE, |
|
|
|
'date' => self::TYPE_DATE, |
|
|
|
'real' => self::TYPE_FLOAT, |
|
|
|
'real' => self::TYPE_FLOAT, |
|
|
|
'double precision' => self::TYPE_DECIMAL, |
|
|
|
'double precision' => self::TYPE_DECIMAL, |
|
|
|
'inet' => self::TYPE_STRING, |
|
|
|
'inet' => self::TYPE_STRING, |
|
|
|
'smallint' => self::TYPE_SMALLINT, |
|
|
|
'smallint' => self::TYPE_SMALLINT, |
|
|
|
'integer' => self::TYPE_INTEGER, |
|
|
|
'integer' => self::TYPE_INTEGER, |
|
|
|
'bigint' => self::TYPE_BIGINT, |
|
|
|
'bigint' => self::TYPE_BIGINT, |
|
|
|
'interval' => self::TYPE_STRING, |
|
|
|
'interval' => self::TYPE_STRING, |
|
|
|
'json' => self::TYPE_STRING, |
|
|
|
'json' => self::TYPE_STRING, |
|
|
|
'line' => self::TYPE_STRING, |
|
|
|
'line' => self::TYPE_STRING, |
|
|
|
'macaddr' => self::TYPE_STRING, |
|
|
|
'macaddr' => self::TYPE_STRING, |
|
|
|
'money' => self::TYPE_MONEY, |
|
|
|
'money' => self::TYPE_MONEY, |
|
|
|
'name' => self::TYPE_STRING, |
|
|
|
'name' => self::TYPE_STRING, |
|
|
|
'numeric' => self::TYPE_STRING, |
|
|
|
'numeric' => self::TYPE_STRING, |
|
|
|
'numrange' => self::TYPE_DECIMAL, |
|
|
|
'numrange' => self::TYPE_DECIMAL, |
|
|
|
'oid' => self::TYPE_BIGINT, // should not be used. it's pg internal! |
|
|
|
'oid' => self::TYPE_BIGINT, // should not be used. it's pg internal! |
|
|
|
'path' => self::TYPE_STRING, |
|
|
|
'path' => self::TYPE_STRING, |
|
|
|
'point' => self::TYPE_STRING, |
|
|
|
'point' => self::TYPE_STRING, |
|
|
|
'polygon' => self::TYPE_STRING, |
|
|
|
'polygon' => self::TYPE_STRING, |
|
|
|
'text' => self::TYPE_TEXT, |
|
|
|
'text' => self::TYPE_TEXT, |
|
|
|
'time without time zone' => self::TYPE_TIME, |
|
|
|
'time without time zone' => self::TYPE_TIME, |
|
|
|
'timestamp without time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'timestamp without time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'timestamp with time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'timestamp with time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'time with time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'time with time zone' => self::TYPE_TIMESTAMP, |
|
|
|
'unknown' => self::TYPE_STRING, |
|
|
|
'unknown' => self::TYPE_STRING, |
|
|
|
'uuid' => self::TYPE_STRING, |
|
|
|
'uuid' => self::TYPE_STRING, |
|
|
|
'bit varying' => self::TYPE_STRING, |
|
|
|
'bit varying' => self::TYPE_STRING, |
|
|
|
'character varying' => self::TYPE_STRING, |
|
|
|
'character varying' => self::TYPE_STRING, |
|
|
|
'xml' => self::TYPE_STRING |
|
|
|
'xml' => self::TYPE_STRING |
|
|
|
); |
|
|
|
); |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
/** |
|
|
|
* Creates a query builder for the MySQL database. |
|
|
|
* Creates a query builder for the MySQL database. |
|
|
|
* @return QueryBuilder query builder instance |
|
|
|
* @return QueryBuilder query builder instance |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public function createQueryBuilder() { |
|
|
|
public function createQueryBuilder() |
|
|
|
|
|
|
|
{ |
|
|
|
return new QueryBuilder($this->db); |
|
|
|
return new QueryBuilder($this->db); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -85,7 +86,8 @@ class Schema extends \yii\db\Schema |
|
|
|
* @param TableSchema $table the table metadata object |
|
|
|
* @param TableSchema $table the table metadata object |
|
|
|
* @param string $name the table name |
|
|
|
* @param string $name the table name |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
protected function resolveTableNames($table, $name) { |
|
|
|
protected function resolveTableNames($table, $name) |
|
|
|
|
|
|
|
{ |
|
|
|
$parts = explode('.', str_replace('"', '', $name)); |
|
|
|
$parts = explode('.', str_replace('"', '', $name)); |
|
|
|
if (isset($parts[1])) { |
|
|
|
if (isset($parts[1])) { |
|
|
|
$table->schemaName = $parts[0]; |
|
|
|
$table->schemaName = $parts[0]; |
|
|
@ -104,7 +106,8 @@ class Schema extends \yii\db\Schema |
|
|
|
* @param string $name table name |
|
|
|
* @param string $name table name |
|
|
|
* @return string the properly quoted table name |
|
|
|
* @return string the properly quoted table name |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public function quoteSimpleTableName($name) { |
|
|
|
public function quoteSimpleTableName($name) |
|
|
|
|
|
|
|
{ |
|
|
|
return strpos($name, '"') !== false ? $name : '"' . $name . '"'; |
|
|
|
return strpos($name, '"') !== false ? $name : '"' . $name . '"'; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -113,12 +116,15 @@ class Schema extends \yii\db\Schema |
|
|
|
* @param string $name table name |
|
|
|
* @param string $name table name |
|
|
|
* @return TableSchema|null driver dependent table metadata. Null if the table does not exist. |
|
|
|
* @return TableSchema|null driver dependent table metadata. Null if the table does not exist. |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
public function loadTableSchema($name) { |
|
|
|
public function loadTableSchema($name) |
|
|
|
|
|
|
|
{ |
|
|
|
$table = new TableSchema(); |
|
|
|
$table = new TableSchema(); |
|
|
|
$this->resolveTableNames($table, $name); |
|
|
|
$this->resolveTableNames($table, $name); |
|
|
|
if ($this->findColumns($table)) { |
|
|
|
if ($this->findColumns($table)) { |
|
|
|
$this->findConstraints($table); |
|
|
|
$this->findConstraints($table); |
|
|
|
return $table; |
|
|
|
return $table; |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
return null; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -126,7 +132,8 @@ class Schema extends \yii\db\Schema |
|
|
|
* Collects the foreign key column details for the given table. |
|
|
|
* Collects the foreign key column details for the given table. |
|
|
|
* @param TableSchema $table the table metadata |
|
|
|
* @param TableSchema $table the table metadata |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
protected function findConstraints($table) { |
|
|
|
protected function findConstraints($table) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
|
|
|
|
$tableName = $this->quoteValue($table->name); |
|
|
|
$tableName = $this->quoteValue($table->name); |
|
|
|
$tableSchema = $this->quoteValue($table->schemaName); |
|
|
|
$tableSchema = $this->quoteValue($table->schemaName); |
|
|
@ -175,7 +182,8 @@ SQL; |
|
|
|
* @param TableSchema $table the table metadata |
|
|
|
* @param TableSchema $table the table metadata |
|
|
|
* @return boolean whether the table exists in the database |
|
|
|
* @return boolean whether the table exists in the database |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
protected function findColumns($table) { |
|
|
|
protected function findColumns($table) |
|
|
|
|
|
|
|
{ |
|
|
|
$tableName = $this->db->quoteValue($table->name); |
|
|
|
$tableName = $this->db->quoteValue($table->name); |
|
|
|
$schemaName = $this->db->quoteValue($table->schemaName); |
|
|
|
$schemaName = $this->db->quoteValue($table->schemaName); |
|
|
|
$sql = <<<SQL |
|
|
|
$sql = <<<SQL |
|
|
@ -234,11 +242,11 @@ ORDER BY |
|
|
|
a.attnum; |
|
|
|
a.attnum; |
|
|
|
SQL; |
|
|
|
SQL; |
|
|
|
|
|
|
|
|
|
|
|
try { |
|
|
|
$columns = $this->db->createCommand($sql)->queryAll(); |
|
|
|
$columns = $this->db->createCommand($sql)->queryAll(); |
|
|
|
if (empty($columns)) { |
|
|
|
} catch (\Exception $e) { |
|
|
|
|
|
|
|
return false; |
|
|
|
return false; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
foreach ($columns as $column) { |
|
|
|
foreach ($columns as $column) { |
|
|
|
$column = $this->loadColumnSchema($column); |
|
|
|
$column = $this->loadColumnSchema($column); |
|
|
|
$table->columns[$column->name] = $column; |
|
|
|
$table->columns[$column->name] = $column; |
|
|
@ -257,7 +265,8 @@ SQL; |
|
|
|
* @param array $info column information |
|
|
|
* @param array $info column information |
|
|
|
* @return ColumnSchema the column schema object |
|
|
|
* @return ColumnSchema the column schema object |
|
|
|
*/ |
|
|
|
*/ |
|
|
|
protected function loadColumnSchema($info) { |
|
|
|
protected function loadColumnSchema($info) |
|
|
|
|
|
|
|
{ |
|
|
|
$column = new ColumnSchema(); |
|
|
|
$column = new ColumnSchema(); |
|
|
|
$column->allowNull = $info['is_nullable']; |
|
|
|
$column->allowNull = $info['is_nullable']; |
|
|
|
$column->autoIncrement = $info['is_autoinc']; |
|
|
|
$column->autoIncrement = $info['is_autoinc']; |
|
|
@ -280,5 +289,4 @@ SQL; |
|
|
|
$column->phpType = $this->getColumnPhpType($column); |
|
|
|
$column->phpType = $this->getColumnPhpType($column); |
|
|
|
return $column; |
|
|
|
return $column; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|