From 3335f09dae23a27003f7e661d63d76380d69fa92 Mon Sep 17 00:00:00 2001 From: Qiang Xue Date: Sun, 21 Aug 2011 08:50:29 -0400 Subject: [PATCH] w --- framework/db/dao/ColumnSchema.php | 29 +- framework/db/dao/Command.php | 2 +- framework/db/dao/Query.php | 5 + framework/db/dao/QueryBuilder.php | 608 +++++++++++++++++++------------- framework/db/dao/Schema.php | 229 +----------- framework/db/dao/TableSchema.php | 24 +- framework/db/dao/mysql/ColumnSchema.php | 70 ++++ framework/db/dao/mysql/Schema.php | 310 ++++++++++++++++ framework/db/dao/mysql/TableSchema.php | 24 ++ 9 files changed, 810 insertions(+), 491 deletions(-) create mode 100644 framework/db/dao/mysql/ColumnSchema.php create mode 100644 framework/db/dao/mysql/Schema.php create mode 100644 framework/db/dao/mysql/TableSchema.php diff --git a/framework/db/dao/ColumnSchema.php b/framework/db/dao/ColumnSchema.php index a0e48e3..3b4e413 100644 --- a/framework/db/dao/ColumnSchema.php +++ b/framework/db/dao/ColumnSchema.php @@ -1,22 +1,20 @@ * @link http://www.yiiframework.com/ - * @copyright Copyright © 2008-2011 Yii Software LLC + * @copyright Copyright © 2008-2012 Yii Software LLC * @license http://www.yiiframework.com/license/ */ /** - * CDbColumnSchema class describes the column meta data of a database table. + * ColumnSchema class describes the column meta data of a database table. * * @author Qiang Xue - * @version $Id: CDbColumnSchema.php 3099 2011-03-19 01:26:47Z qiang.xue $ - * @package system.db.schema - * @since 1.0 + * @since 2.0 */ -class CDbColumnSchema extends CComponent +class ColumnSchema extends CComponent { /** * @var string name of this column (without quotes). @@ -25,19 +23,25 @@ class CDbColumnSchema extends CComponent /** * @var string raw name of this column. This is the quoted name that can be used in SQL queries. */ - public $rawName; + public $quotedName; /** * @var boolean whether this column can be null. */ public $allowNull; /** - * @var string the DB type of this column. + * @var string logical type of this column. Possible logic types include: + * string, text, boolean, integer, float, decimal, datetime, timestamp, time, date, binary */ - public $dbType; + public $type; /** - * @var string the PHP type of this column. + * @var string the PHP type of this column. Possible PHP types include: + * string, boolean, integer, double. */ - public $type; + public $phpType; + /** + * @var string the DB type of this column. + */ + public $dbType; /** * @var mixed default value of this column */ @@ -64,7 +68,6 @@ class CDbColumnSchema extends CComponent public $isForeignKey; /** * @var boolean whether this column is auto-incremental - * @since 1.1.7 */ public $autoIncrement = false; diff --git a/framework/db/dao/Command.php b/framework/db/dao/Command.php index f275677..ecb10fb 100644 --- a/framework/db/dao/Command.php +++ b/framework/db/dao/Command.php @@ -78,7 +78,7 @@ class Command extends \yii\base\Component * {@link setFetchMode FetchMode}. See {@link http://www.php.net/manual/en/function.PDOStatement-setFetchMode.php} * for more details. */ - public function __construct(CDbConnection $connection, $query = null) + public function __construct($connection, $query = null) { $this->_connection = $connection; if (is_array($query)) diff --git a/framework/db/dao/Query.php b/framework/db/dao/Query.php index 6f623c9..691dbda 100644 --- a/framework/db/dao/Query.php +++ b/framework/db/dao/Query.php @@ -24,6 +24,11 @@ class Query extends CComponent * or an array of column names. Defaults to '*', meaning all columns. */ public $select; + /** + * @var string additional option that should be appended to the 'SELECT' keyword. For example, + * in MySQL, the option 'SQL_CALC_FOUND_ROWS' can be used. + */ + public $selectOption; public $from; /** diff --git a/framework/db/dao/QueryBuilder.php b/framework/db/dao/QueryBuilder.php index 2762841..a071593 100644 --- a/framework/db/dao/QueryBuilder.php +++ b/framework/db/dao/QueryBuilder.php @@ -20,7 +20,7 @@ class QueryBuilder extends \yii\base\Component { private $_connection; - public function __construct(Connection $connection) + public function __construct($connection) { $this->_connection = $connection; } @@ -36,24 +36,228 @@ class QueryBuilder extends \yii\base\Component public function build($query) { $clauses = array( - $this->buildSelect($query->select, $query->distinct), - $this->buildFrom($query->from), - $this->buildJoin($query->join), - $this->buildWhere($query->where), - $this->buildGroupBy($query->groupBy), - $this->buildHaving($query->having), - $this->buildOrderBy($query->orderBy), - $this->buildLimit($query->offset, $query->limit), - $this->buildUnion($query->union), + $this->buildSelect($query), + $this->buildFrom($query), + $this->buildJoin($query), + $this->buildWhere($query), + $this->buildGroupBy($query), + $this->buildHaving($query), + $this->buildUnion($query), + $this->buildOrderBy($query), + $this->buildLimit($query), ); return implode("\n", array_filter($clauses)); } - protected function buildSelect($columns, $distinct) + /** + * Builds a SQL statement for creating a new DB table. + * + * The columns in the new table should be specified as name-definition pairs (e.g. 'name'=>'string'), + * where name stands for a column name which will be properly quoted by the method, and definition + * stands for the column type which can contain an abstract DB type. + * The {@link getColumnType} method will be invoked to convert any abstract type into a physical one. + * + * If a column is specified with definition only (e.g. 'PRIMARY KEY (name, type)'), it will be directly + * inserted into the generated SQL. + * + * @param string $table the name of the table to be created. The name will be properly quoted by the method. + * @param array $columns the columns (name=>definition) in the new table. + * @param string $options additional SQL fragment that will be appended to the generated SQL. + * @return string the SQL statement for creating a new DB table. + */ + public function createTable($table, $columns, $options = null) + { + $cols = array(); + foreach ($columns as $name => $type) + { + if (is_string($name)) + $cols[] = "\t" . $this->quoteColumnName($name) . ' ' . $this->getColumnType($type); + else + $cols[] = "\t" . $type; + } + $sql = "CREATE TABLE " . $this->quoteTableName($table) . " (\n" . implode(",\n", $cols) . "\n)"; + return $options === null ? $sql : $sql . ' ' . $options; + } + + /** + * Builds a SQL statement for renaming a DB table. + * @param string $table the table to be renamed. The name will be properly quoted by the method. + * @param string $newName the new table name. The name will be properly quoted by the method. + * @return string the SQL statement for renaming a DB table. + */ + public function renameTable($table, $newName) + { + return 'RENAME TABLE ' . $this->quoteTableName($table) . ' TO ' . $this->quoteTableName($newName); + } + + /** + * Builds a SQL statement for dropping a DB table. + * @param string $table the table to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping a DB table. + */ + public function dropTable($table) + { + return "DROP TABLE " . $this->quoteTableName($table); + } + + /** + * Builds a SQL statement for truncating a DB table. + * @param string $table the table to be truncated. The name will be properly quoted by the method. + * @return string the SQL statement for truncating a DB table. + */ + public function truncateTable($table) { - $select = $distinct ? 'SELECT DISTINCT' : 'SELECT'; + return "TRUNCATE TABLE " . $this->quoteTableName($table); + } + + /** + * Builds a SQL statement for adding a new DB column. + * @param string $table the table that the new column will be added to. The table name will be properly quoted by the method. + * @param string $column the name of the new column. The name will be properly quoted by the method. + * @param string $type the column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any) + * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL. + * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'. + * @return string the SQL statement for adding a new column. + * @since 1.1.6 + */ + public function addColumn($table, $column, $type) + { + return 'ALTER TABLE ' . $this->quoteTableName($table) + . ' ADD ' . $this->quoteColumnName($column) . ' ' + . $this->getColumnType($type); + } + /** + * Builds a SQL statement for dropping a DB column. + * @param string $table the table whose column is to be dropped. The name will be properly quoted by the method. + * @param string $column the name of the column to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping a DB column. + * @since 1.1.6 + */ + public function dropColumn($table, $column) + { + return "ALTER TABLE " . $this->quoteTableName($table) + . " DROP COLUMN " . $this->quoteColumnName($column); + } + + /** + * Builds a SQL statement for renaming a column. + * @param string $table the table whose column is to be renamed. The name will be properly quoted by the method. + * @param string $name the old name of the column. The name will be properly quoted by the method. + * @param string $newName the new name of the column. The name will be properly quoted by the method. + * @return string the SQL statement for renaming a DB column. + * @since 1.1.6 + */ + public function renameColumn($table, $name, $newName) + { + return "ALTER TABLE " . $this->quoteTableName($table) + . " RENAME COLUMN " . $this->quoteColumnName($name) + . " TO " . $this->quoteColumnName($newName); + } + + /** + * Builds a SQL statement for changing the definition of a column. + * @param string $table the table whose column is to be changed. The table name will be properly quoted by the method. + * @param string $column the name of the column to be changed. The name will be properly quoted by the method. + * @param string $type the new column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any) + * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL. + * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'. + * @return string the SQL statement for changing the definition of a column. + */ + public function alterColumn($table, $column, $type) + { + return 'ALTER TABLE ' . $this->quoteTableName($table) . ' CHANGE ' + . $this->quoteColumnName($column) . ' ' + . $this->quoteColumnName($column) . ' ' + . $this->getColumnType($type); + } + + /** + * Builds a SQL statement for adding a foreign key constraint to an existing table. + * The method will properly quote the table and column names. + * @param string $name the name of the foreign key constraint. + * @param string $table the table that the foreign key constraint will be added to. + * @param string $columns the name of the column to that the constraint will be added on. If there are multiple columns, separate them with commas. + * @param string $refTable the table that the foreign key references to. + * @param string $refColumns the name of the column that the foreign key references to. If there are multiple columns, separate them with commas. + * @param string $delete the ON DELETE option. Most DBMS support these options: RESTRICT, CASCADE, NO ACTION, SET DEFAULT, SET NULL + * @param string $update the ON UPDATE option. Most DBMS support these options: RESTRICT, CASCADE, NO ACTION, SET DEFAULT, SET NULL + * @return string the SQL statement for adding a foreign key constraint to an existing table. + */ + public function addForeignKey($name, $table, $columns, $refTable, $refColumns, $delete = null, $update = null) + { + $columns = preg_split('/\s*,\s*/', $columns, -1, PREG_SPLIT_NO_EMPTY); + foreach ($columns as $i => $col) + $columns[$i] = $this->quoteColumnName($col); + $refColumns = preg_split('/\s*,\s*/', $refColumns, -1, PREG_SPLIT_NO_EMPTY); + foreach ($refColumns as $i => $col) + $refColumns[$i] = $this->quoteColumnName($col); + $sql = 'ALTER TABLE ' . $this->quoteTableName($table) + . ' ADD CONSTRAINT ' . $this->quoteColumnName($name) + . ' FOREIGN KEY (' . implode(', ', $columns) . ')' + . ' REFERENCES ' . $this->quoteTableName($refTable) + . ' (' . implode(', ', $refColumns) . ')'; + if ($delete !== null) + $sql .= ' ON DELETE ' . $delete; + if ($update !== null) + $sql .= ' ON UPDATE ' . $update; + return $sql; + } + + /** + * Builds a SQL statement for dropping a foreign key constraint. + * @param string $name the name of the foreign key constraint to be dropped. The name will be properly quoted by the method. + * @param string $table the table whose foreign is to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping a foreign key constraint. + */ + public function dropForeignKey($name, $table) + { + return 'ALTER TABLE ' . $this->quoteTableName($table) + . ' DROP CONSTRAINT ' . $this->quoteColumnName($name); + } + + /** + * Builds a SQL statement for creating a new index. + * @param string $name the name of the index. The name will be properly quoted by the method. + * @param string $table the table that the new index will be created for. The table name will be properly quoted by the method. + * @param string $column the column(s) that should be included in the index. If there are multiple columns, please separate them + * by commas. Each column name will be properly quoted by the method, unless a parenthesis is found in the name. + * @param boolean $unique whether to add UNIQUE constraint on the created index. + * @return string the SQL statement for creating a new index. + */ + public function createIndex($name, $table, $column, $unique = false) + { + $cols = array(); + $columns = preg_split('/\s*,\s*/', $column, -1, PREG_SPLIT_NO_EMPTY); + foreach ($columns as $col) + { + if (strpos($col, '(') !== false) + $cols[] = $col; + else + $cols[] = $this->quoteColumnName($col); + } + return ($unique ? 'CREATE UNIQUE INDEX ' : 'CREATE INDEX ') + . $this->quoteTableName($name) . ' ON ' + . $this->quoteTableName($table) . ' (' . implode(', ', $cols) . ')'; + } + + /** + * Builds a SQL statement for dropping an index. + * @param string $name the name of the index to be dropped. The name will be properly quoted by the method. + * @param string $table the table whose index is to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping an index. + */ + public function dropIndex($name, $table) + { + return 'DROP INDEX ' . $this->quoteTableName($name) . ' ON ' . $this->quoteTableName($table); + } + + protected function buildSelect($query) + { + $select = $query->distinct ? 'SELECT DISTINCT' : 'SELECT'; + + $columns = $query->select; if (empty($columns)) { return $select . ' *'; } @@ -70,8 +274,8 @@ class QueryBuilder extends \yii\base\Component $columns[$i] = (string)$column; } elseif (strpos($column, '(') === false) { - if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)(.*)$/', $column, $matches)) { - $columns[$i] = $this->_connection->quoteColumnName($matches[1]) . ' AS ' . $this->_connection->quoteColumnName($matches[2]); + if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)([\w\-\.])$/', $column, $matches)) { + $columns[$i] = $this->_connection->quoteColumnName($matches[1]) . ' AS ' . $this->_connection->quoteSimpleColumnName($matches[2]); } else { $columns[$i] = $this->_connection->quoteColumnName($column); @@ -82,10 +286,11 @@ class QueryBuilder extends \yii\base\Component return $select . ' ' . implode(', ', $columns); } - protected function buildFrom($tables) + protected function buildFrom($query) { + $tables = $query->from; if (is_string($tables) && strpos($tables, '(') !== false) { - return $tables; + return 'FROM ' . $tables; } if (!is_array($tables)) { @@ -93,7 +298,7 @@ class QueryBuilder extends \yii\base\Component } foreach ($tables as $i => $table) { if (strpos($table, '(') === false) { - if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)(.*)$/', $table, $matches)) { // with alias + if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)(.*)$/i', $table, $matches)) { // with alias $tables[$i] = $this->_connection->quoteTableName($matches[1]) . ' ' . $this->_connection->quoteTableName($matches[2]); } else { @@ -101,302 +306,211 @@ class QueryBuilder extends \yii\base\Component } } } - return implode(', ', $tables); - } - $this->buildJoin($query->join), - $this->buildWhere($query->where), - $this->buildGroupBy($query->groupBy), - $this->buildHaving($query->having), - $this->buildOrderBy($query->orderBy), - $this->buildLimit($query->offset, $query->limit), + return 'FROM ' . implode(', ', $tables); + } + protected function buildJoin($query) + { + $joins = $query->join; + if (empty($joins)) { + return ''; + } + if (is_string($joins)) { + return $joins; + } - if (isset($query['union'])) - $sql .= "\nUNION (\n" . (is_array($query['union']) ? implode("\n) UNION (\n", $query['union']) : $query['union']) . ')'; + foreach ($joins as $i => $join) { + if (is_array($join)) { // join type, table name, on-condition + if (isset($join[0], $join[1])) { + $table = $join[1]; + if (strpos($table,'(')===false) { + if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)(.*)$/', $table, $matches)) { // with alias + $table = $this->_connection->quoteTableName($matches[1]).' '.$this->_connection->quoteTableName($matches[2]); + } + else { + $table = $this->_connection->quoteTableName($table); + } + } + $joins[$i] = strtoupper($join[0]) . ' ' . $table; + if (isset($join[2])) { // join condition + $condition = $this->processCondition($join[2]); + $joins[$i] .= ' ON ' . $condition; + } + } + else { + throw new Exception('The join clause may be specified as an array of at least two elements.'); + } + } + } - return $sql; + return implode("\n", $joins); } - - /** - * Sets the WHERE part of the query. - * - * The method requires a $conditions parameter, and optionally a $params parameter - * specifying the values to be bound to the query. - * - * The $conditions parameter should be either a string (e.g. 'id=1') or an array. - * If the latter, it must be of the format array(operator, operand1, operand2, ...), - * where the operator can be one of the followings, and the possible operands depend on the corresponding - * operator: - * - * @param mixed $conditions the conditions that should be put in the WHERE part. - * @param array $params the parameters (name=>value) to be bound to the query - * @return Command the command object itself - * @since 1.1.6 - */ - public function where($conditions, $params = array()) + protected function buildWhere($query) { - $this->_query['where'] = $this->processConditions($conditions); - foreach ($params as $name => $value) - $this->params[$name] = $value; - return $this; + $where = $this->processConditions($query->where); + return empty($where) ? '' : 'WHERE ' . $where; } - /** - * Appends an INNER JOIN part to the query. - * @param string $table the table to be joined. - * Table name can contain schema prefix (e.g. 'public.tbl_user') and/or table alias (e.g. 'tbl_user u'). - * The method will automatically quote the table name unless it contains some parenthesis - * (which means the table is given as a sub-query or DB expression). - * @param mixed $conditions the join condition that should appear in the ON part. - * Please refer to {@link where} on how to specify conditions. - * @param array $params the parameters (name=>value) to be bound to the query - * @return Command the command object itself - * @since 1.1.6 - */ - public function join($table, $conditions, $params = array()) + protected function buildGroupBy($query) { - return $this->joinInternal('join', $table, $conditions, $params); - } + $columns = $query->groupBy; + if (empty($columns)) { + return ''; + } + if (is_string($columns) && strpos($columns, '(') !== false) { + return 'GROUP BY ' . $columns; + } - /** - * Sets the GROUP BY part of the query. - * @param mixed $columns the columns to be grouped by. - * Columns can be specified in either a string (e.g. "id, name") or an array (e.g. array('id', 'name')). - * The method will automatically quote the column names unless a column contains some parenthesis - * (which means the column contains a DB expression). - * @return Command the command object itself - * @since 1.1.6 - */ - public function group($columns) - { - if (is_string($columns) && strpos($columns, '(') !== false) - $this->_query['group'] = $columns; - else - { - if (!is_array($columns)) - $columns = preg_split('/\s*,\s*/', trim($columns), -1, PREG_SPLIT_NO_EMPTY); - foreach ($columns as $i => $column) - { - if (is_object($column)) - $columns[$i] = (string)$column; - elseif (strpos($column, '(') === false) - $columns[$i] = $this->_connection->quoteColumnName($column); + if (!is_array($columns)) { + $columns = preg_split('/\s*,\s*/', trim($columns), -1, PREG_SPLIT_NO_EMPTY); + } + foreach ($columns as $i => $column) { + if (is_object($column)) { + $columns[$i] = (string)$column; + } + elseif (strpos($column, '(') === false) { + $columns[$i] = $this->_connection->quoteColumnName($column); } - $this->_query['group'] = implode(', ', $columns); } - return $this; + return 'GROUP BY ' . implode(', ', $columns); } - /** - * Sets the HAVING part of the query. - * @param mixed $conditions the conditions to be put after HAVING. - * Please refer to {@link where} on how to specify conditions. - * @param array $params the parameters (name=>value) to be bound to the query - * @return Command the command object itself - * @since 1.1.6 - */ - public function having($conditions, $params = array()) + protected function buildHaving($query) { - $this->_query['having'] = $this->processConditions($conditions); - foreach ($params as $name => $value) - $this->params[$name] = $value; - return $this; + $having = $this->processConditions($query->having); + return empty($having) ? '' : 'HAVING ' . $having; } - /** - * Sets the ORDER BY part of the query. - * @param mixed $columns the columns (and the directions) to be ordered by. - * Columns can be specified in either a string (e.g. "id ASC, name DESC") or an array (e.g. array('id ASC', 'name DESC')). - * The method will automatically quote the column names unless a column contains some parenthesis - * (which means the column contains a DB expression). - * @return Command the command object itself - * @since 1.1.6 - */ - public function order($columns) + protected function buildOrderBy($query) { - if (is_string($columns) && strpos($columns, '(') !== false) - $this->_query['order'] = $columns; - else - { - if (!is_array($columns)) - $columns = preg_split('/\s*,\s*/', trim($columns), -1, PREG_SPLIT_NO_EMPTY); - foreach ($columns as $i => $column) - { - if (is_object($column)) - $columns[$i] = (string)$column; - elseif (strpos($column, '(') === false) - { - if (preg_match('/^(.*?)\s+(asc|desc)$/i', $column, $matches)) - $columns[$i] = $this->_connection->quoteColumnName($matches[1]) . ' ' . strtoupper($matches[2]); - else - $columns[$i] = $this->_connection->quoteColumnName($column); + $columns = $query->orderBy; + if (empty($columns)) { + return ''; + } + if (is_string($columns) && strpos($columns, '(') !== false) { + return 'ORDER BY ' . $columns; + } + + if (!is_array($columns)) { + $columns = preg_split('/\s*,\s*/', trim($columns), -1, PREG_SPLIT_NO_EMPTY); + } + foreach ($columns as $i => $column) { + if (is_object($column)) { + $columns[$i] = (string)$column; + } + elseif (strpos($column, '(') === false) { + if (preg_match('/^(.*?)\s+(asc|desc)$/i', $column, $matches)) { + $columns[$i] = $this->_connection->quoteColumnName($matches[1]) . ' ' . strtoupper($matches[2]); + } + else { + $columns[$i] = $this->_connection->quoteColumnName($column); } } - $this->_query['order'] = implode(', ', $columns); } - return $this; + return 'ORDER BY ' . implode(', ', $columns); } - /** - * Sets the LIMIT part of the query. - * @param integer $limit the limit - * @param integer $offset the offset - * @return Command the command object itself - * @since 1.1.6 - */ - public function limit($limit, $offset = null) + protected function buildLimit($query) { - $this->_query['limit'] = (int)$limit; - if ($offset !== null) - $this->offset($offset); - return $this; + $sql = ''; + if ($query->limit !== null && $query->limit >= 0) { + $sql = 'LIMIT ' . (int)$query->limit; + } + if ($query->offset>0) { + $sql .= ' OFFSET '.(int)$query->offset; + } + return ltrim($sql); } - /** - * Appends a SQL statement using UNION operator. - * @param string $sql the SQL statement to be appended using UNION - * @return Command the command object itself - * @since 1.1.6 - */ - public function union($sql) + protected function buildUnion($query) { - if (isset($this->_query['union']) && is_string($this->_query['union'])) - $this->_query['union'] = array($this->_query['union']); - - $this->_query['union'][] = $sql; - - return $this; + $unions = $query->union; + if (empty($unions)) { + return ''; + } + if (!is_array($unions)) { + $unions = array($unions); + } + foreach ($unions as $i => $union) { + if ($union instanceof Query) { + $unions[$i] = $union->getSql($this->_connection); + } + } + return "UNION (\n" . implode("\n) UNION (\n", $unions) . "\n)"; } - /** - * Generates the condition string that will be put in the WHERE part - * @param mixed $conditions the conditions that will be put in the WHERE part. - * @return string the condition string to put in the WHERE part - */ - private function buildConditions($conditions) + protected function processCondition($conditions) { - if (!is_array($conditions)) + if (!is_array($conditions)) { return $conditions; - elseif ($conditions === array()) + } + elseif ($conditions === array()) { return ''; + } + $n = count($conditions); $operator = strtoupper($conditions[0]); - if ($operator === 'OR' || $operator === 'AND') - { + if ($operator === 'OR' || $operator === 'AND') { $parts = array(); - for ($i = 1;$i < $n;++$i) - { - $condition = $this->processConditions($conditions[$i]); - if ($condition !== '') + for ($i = 1; $i < $n; ++$i) { + $condition = $this->processCondition($conditions[$i]); + if ($condition !== '') { $parts[] = '(' . $condition . ')'; + } } return $parts === array() ? '' : implode(' ' . $operator . ' ', $parts); } - if (!isset($conditions[1], $conditions[2])) + if (!isset($conditions[1], $conditions[2])) { return ''; + } $column = $conditions[1]; - if (strpos($column, '(') === false) + if (strpos($column, '(') === false) { $column = $this->_connection->quoteColumnName($column); + } $values = $conditions[2]; - if (!is_array($values)) + if (!is_array($values)) { $values = array($values); + } - if ($operator === 'IN' || $operator === 'NOT IN') - { - if ($values === array()) + if ($operator === 'IN' || $operator === 'NOT IN') { + if ($values === array()) { return $operator === 'IN' ? '0=1' : ''; - foreach ($values as $i => $value) - { - if (is_string($value)) + } + foreach ($values as $i => $value) { + if (is_string($value)) { $values[$i] = $this->_connection->quoteValue($value); - else + } + else { $values[$i] = (string)$value; + } } return $column . ' ' . $operator . ' (' . implode(', ', $values) . ')'; } - if ($operator === 'LIKE' || $operator === 'NOT LIKE' || $operator === 'OR LIKE' || $operator === 'OR NOT LIKE') - { - if ($values === array()) + if ($operator === 'LIKE' || $operator === 'NOT LIKE' || $operator === 'OR LIKE' || $operator === 'OR NOT LIKE') { + if ($values === array()) { return $operator === 'LIKE' || $operator === 'OR LIKE' ? '0=1' : ''; + } - if ($operator === 'LIKE' || $operator === 'NOT LIKE') + if ($operator === 'LIKE' || $operator === 'NOT LIKE') { $andor = ' AND '; - else - { + } + else { $andor = ' OR '; $operator = $operator === 'OR LIKE' ? 'LIKE' : 'NOT LIKE'; } $expressions = array(); - foreach ($values as $value) + foreach ($values as $value) { $expressions[] = $column . ' ' . $operator . ' ' . $this->_connection->quoteValue($value); + } return implode($andor, $expressions); } - throw new CDbException(Yii::t('yii', 'Unknown operator "{operator}".', array('{operator}' => $operator))); - } - - /** - * Appends an JOIN part to the query. - * @param string $type the join type ('join', 'left join', 'right join', 'cross join', 'natural join') - * @param string $table the table to be joined. - * Table name can contain schema prefix (e.g. 'public.tbl_user') and/or table alias (e.g. 'tbl_user u'). - * The method will automatically quote the table name unless it contains some parenthesis - * (which means the table is given as a sub-query or DB expression). - * @param mixed $conditions the join condition that should appear in the ON part. - * Please refer to {@link where} on how to specify conditions. - * @param array $params the parameters (name=>value) to be bound to the query - * @return Command the command object itself - * @since 1.1.6 - */ - private function joinInternal($type, $table, $conditions = '', $params = array()) - { - if (strpos($table, '(') === false) - { - if (preg_match('/^(.*?)(?i:\s+as\s+|\s+)(.*)$/', $table, $matches)) // with alias - $table = $this->_connection->quoteTableName($matches[1]) . ' ' . $this->_connection->quoteTableName($matches[2]); - else - $table = $this->_connection->quoteTableName($table); - } - - $conditions = $this->processConditions($conditions); - if ($conditions != '') - $conditions = ' ON ' . $conditions; - - if (isset($this->_query['join']) && is_string($this->_query['join'])) - $this->_query['join'] = array($this->_query['join']); - - $this->_query['join'][] = strtoupper($type) . ' ' . $table . $conditions; - - foreach ($params as $name => $value) - $this->params[$name] = $value; - return $this; + throw new Exception('Unknown operator: ' . $operator); } } diff --git a/framework/db/dao/Schema.php b/framework/db/dao/Schema.php index 7f21a69..0ac1020 100644 --- a/framework/db/dao/Schema.php +++ b/framework/db/dao/Schema.php @@ -1,6 +1,6 @@ * @link http://www.yiiframework.com/ @@ -8,15 +8,15 @@ * @license http://www.yiiframework.com/license/ */ +namespace yii\db\dao; + /** - * CDbSchema is the base class for retrieving metadata information. + * Schema is the base class for retrieving metadata information. * * @author Qiang Xue - * @version $Id: CDbSchema.php 3359 2011-07-18 11:25:17Z qiang.xue $ - * @package system.db.schema - * @since 1.0 + * @since 2.0 */ -abstract class CDbSchema extends CComponent +abstract class Schema extends \yii\base\Component { /** * @var array the abstract column types mapped to physical column types. @@ -41,10 +41,10 @@ abstract class CDbSchema extends CComponent * Constructor. * @param CDbConnection $conn database connection. */ - public function __construct($conn) + public function __construct($connection) { - $this->_connection = $conn; - foreach ($conn->schemaCachingExclude as $name) + $this->_connection = $connection; + foreach ($connection->schemaCachingExclude as $name) $this->_cacheExclude[$name] = true; } @@ -342,215 +342,4 @@ abstract class CDbSchema extends CComponent return $type; } - /** - * Builds a SQL statement for creating a new DB table. - * - * The columns in the new table should be specified as name-definition pairs (e.g. 'name'=>'string'), - * where name stands for a column name which will be properly quoted by the method, and definition - * stands for the column type which can contain an abstract DB type. - * The {@link getColumnType} method will be invoked to convert any abstract type into a physical one. - * - * If a column is specified with definition only (e.g. 'PRIMARY KEY (name, type)'), it will be directly - * inserted into the generated SQL. - * - * @param string $table the name of the table to be created. The name will be properly quoted by the method. - * @param array $columns the columns (name=>definition) in the new table. - * @param string $options additional SQL fragment that will be appended to the generated SQL. - * @return string the SQL statement for creating a new DB table. - * @since 1.1.6 - */ - public function createTable($table, $columns, $options = null) - { - $cols = array(); - foreach ($columns as $name => $type) - { - if (is_string($name)) - $cols[] = "\t" . $this->quoteColumnName($name) . ' ' . $this->getColumnType($type); - else - $cols[] = "\t" . $type; - } - $sql = "CREATE TABLE " . $this->quoteTableName($table) . " (\n" . implode(",\n", $cols) . "\n)"; - return $options === null ? $sql : $sql . ' ' . $options; - } - - /** - * Builds a SQL statement for renaming a DB table. - * @param string $table the table to be renamed. The name will be properly quoted by the method. - * @param string $newName the new table name. The name will be properly quoted by the method. - * @return string the SQL statement for renaming a DB table. - * @since 1.1.6 - */ - public function renameTable($table, $newName) - { - return 'RENAME TABLE ' . $this->quoteTableName($table) . ' TO ' . $this->quoteTableName($newName); - } - - /** - * Builds a SQL statement for dropping a DB table. - * @param string $table the table to be dropped. The name will be properly quoted by the method. - * @return string the SQL statement for dropping a DB table. - * @since 1.1.6 - */ - public function dropTable($table) - { - return "DROP TABLE " . $this->quoteTableName($table); - } - - /** - * Builds a SQL statement for truncating a DB table. - * @param string $table the table to be truncated. The name will be properly quoted by the method. - * @return string the SQL statement for truncating a DB table. - * @since 1.1.6 - */ - public function truncateTable($table) - { - return "TRUNCATE TABLE " . $this->quoteTableName($table); - } - - /** - * Builds a SQL statement for adding a new DB column. - * @param string $table the table that the new column will be added to. The table name will be properly quoted by the method. - * @param string $column the name of the new column. The name will be properly quoted by the method. - * @param string $type the column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any) - * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL. - * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'. - * @return string the SQL statement for adding a new column. - * @since 1.1.6 - */ - public function addColumn($table, $column, $type) - { - return 'ALTER TABLE ' . $this->quoteTableName($table) - . ' ADD ' . $this->quoteColumnName($column) . ' ' - . $this->getColumnType($type); - } - - /** - * Builds a SQL statement for dropping a DB column. - * @param string $table the table whose column is to be dropped. The name will be properly quoted by the method. - * @param string $column the name of the column to be dropped. The name will be properly quoted by the method. - * @return string the SQL statement for dropping a DB column. - * @since 1.1.6 - */ - public function dropColumn($table, $column) - { - return "ALTER TABLE " . $this->quoteTableName($table) - . " DROP COLUMN " . $this->quoteColumnName($column); - } - - /** - * Builds a SQL statement for renaming a column. - * @param string $table the table whose column is to be renamed. The name will be properly quoted by the method. - * @param string $name the old name of the column. The name will be properly quoted by the method. - * @param string $newName the new name of the column. The name will be properly quoted by the method. - * @return string the SQL statement for renaming a DB column. - * @since 1.1.6 - */ - public function renameColumn($table, $name, $newName) - { - return "ALTER TABLE " . $this->quoteTableName($table) - . " RENAME COLUMN " . $this->quoteColumnName($name) - . " TO " . $this->quoteColumnName($newName); - } - - /** - * Builds a SQL statement for changing the definition of a column. - * @param string $table the table whose column is to be changed. The table name will be properly quoted by the method. - * @param string $column the name of the column to be changed. The name will be properly quoted by the method. - * @param string $type the new column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any) - * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL. - * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'. - * @return string the SQL statement for changing the definition of a column. - * @since 1.1.6 - */ - public function alterColumn($table, $column, $type) - { - return 'ALTER TABLE ' . $this->quoteTableName($table) . ' CHANGE ' - . $this->quoteColumnName($column) . ' ' - . $this->quoteColumnName($column) . ' ' - . $this->getColumnType($type); - } - - /** - * Builds a SQL statement for adding a foreign key constraint to an existing table. - * The method will properly quote the table and column names. - * @param string $name the name of the foreign key constraint. - * @param string $table the table that the foreign key constraint will be added to. - * @param string $columns the name of the column to that the constraint will be added on. If there are multiple columns, separate them with commas. - * @param string $refTable the table that the foreign key references to. - * @param string $refColumns the name of the column that the foreign key references to. If there are multiple columns, separate them with commas. - * @param string $delete the ON DELETE option. Most DBMS support these options: RESTRICT, CASCADE, NO ACTION, SET DEFAULT, SET NULL - * @param string $update the ON UPDATE option. Most DBMS support these options: RESTRICT, CASCADE, NO ACTION, SET DEFAULT, SET NULL - * @return string the SQL statement for adding a foreign key constraint to an existing table. - * @since 1.1.6 - */ - public function addForeignKey($name, $table, $columns, $refTable, $refColumns, $delete = null, $update = null) - { - $columns = preg_split('/\s*,\s*/', $columns, -1, PREG_SPLIT_NO_EMPTY); - foreach ($columns as $i => $col) - $columns[$i] = $this->quoteColumnName($col); - $refColumns = preg_split('/\s*,\s*/', $refColumns, -1, PREG_SPLIT_NO_EMPTY); - foreach ($refColumns as $i => $col) - $refColumns[$i] = $this->quoteColumnName($col); - $sql = 'ALTER TABLE ' . $this->quoteTableName($table) - . ' ADD CONSTRAINT ' . $this->quoteColumnName($name) - . ' FOREIGN KEY (' . implode(', ', $columns) . ')' - . ' REFERENCES ' . $this->quoteTableName($refTable) - . ' (' . implode(', ', $refColumns) . ')'; - if ($delete !== null) - $sql .= ' ON DELETE ' . $delete; - if ($update !== null) - $sql .= ' ON UPDATE ' . $update; - return $sql; - } - - /** - * Builds a SQL statement for dropping a foreign key constraint. - * @param string $name the name of the foreign key constraint to be dropped. The name will be properly quoted by the method. - * @param string $table the table whose foreign is to be dropped. The name will be properly quoted by the method. - * @return string the SQL statement for dropping a foreign key constraint. - * @since 1.1.6 - */ - public function dropForeignKey($name, $table) - { - return 'ALTER TABLE ' . $this->quoteTableName($table) - . ' DROP CONSTRAINT ' . $this->quoteColumnName($name); - } - - /** - * Builds a SQL statement for creating a new index. - * @param string $name the name of the index. The name will be properly quoted by the method. - * @param string $table the table that the new index will be created for. The table name will be properly quoted by the method. - * @param string $column the column(s) that should be included in the index. If there are multiple columns, please separate them - * by commas. Each column name will be properly quoted by the method, unless a parenthesis is found in the name. - * @param boolean $unique whether to add UNIQUE constraint on the created index. - * @return string the SQL statement for creating a new index. - * @since 1.1.6 - */ - public function createIndex($name, $table, $column, $unique = false) - { - $cols = array(); - $columns = preg_split('/\s*,\s*/', $column, -1, PREG_SPLIT_NO_EMPTY); - foreach ($columns as $col) - { - if (strpos($col, '(') !== false) - $cols[] = $col; - else - $cols[] = $this->quoteColumnName($col); - } - return ($unique ? 'CREATE UNIQUE INDEX ' : 'CREATE INDEX ') - . $this->quoteTableName($name) . ' ON ' - . $this->quoteTableName($table) . ' (' . implode(', ', $cols) . ')'; - } - - /** - * Builds a SQL statement for dropping an index. - * @param string $name the name of the index to be dropped. The name will be properly quoted by the method. - * @param string $table the table whose index is to be dropped. The name will be properly quoted by the method. - * @return string the SQL statement for dropping an index. - * @since 1.1.6 - */ - public function dropIndex($name, $table) - { - return 'DROP INDEX ' . $this->quoteTableName($name) . ' ON ' . $this->quoteTableName($table); - } } diff --git a/framework/db/dao/TableSchema.php b/framework/db/dao/TableSchema.php index ece39f5..be97032 100644 --- a/framework/db/dao/TableSchema.php +++ b/framework/db/dao/TableSchema.php @@ -1,6 +1,6 @@ * @link http://www.yiiframework.com/ @@ -8,12 +8,14 @@ * @license http://www.yiiframework.com/license/ */ +namespace yii\db\dao; + /** - * CDbTableSchema is the base class for representing the metadata of a database table. + * TableSchema is the base class for representing the metadata of a database table. * * It may be extended by different DBMS driver to provide DBMS-specific table metadata. * - * CDbTableSchema provides the following information about a table: + * TableSchema provides the following information about a table: *
    *
  • {@link name}
    • *
  • {@link rawName}
    • @@ -24,20 +26,22 @@ *
* * @author Qiang Xue - * @version $Id: CDbTableSchema.php 2799 2011-01-01 19:31:13Z qiang.xue $ - * @package system.db.schema - * @since 1.0 + * @since 2.0 */ -class CDbTableSchema extends CComponent +class TableSchema extends \yii\base\Component { /** + * @var string name of the schema that this table belongs to. + */ + public $schemaName; + /** * @var string name of this table. */ public $name; /** - * @var string raw name of this table. This is the quoted version of table name with optional schema name. It can be directly used in SQLs. + * @var string quoted name of this table. This may include [[schemaName]] if it is not empty. */ - public $rawName; + public $quotedName; /** * @var string|array primary key name of this table. If composite key, an array of key names is returned. */ @@ -51,7 +55,7 @@ class CDbTableSchema extends CComponent */ public $foreignKeys = array(); /** - * @var array column metadata of this table. Each array element is a CDbColumnSchema object, indexed by column names. + * @var array column metadata of this table. Each array element is a [[ColumnSchema]] object, indexed by column names. */ public $columns = array(); diff --git a/framework/db/dao/mysql/ColumnSchema.php b/framework/db/dao/mysql/ColumnSchema.php new file mode 100644 index 0000000..671dd12 --- /dev/null +++ b/framework/db/dao/mysql/ColumnSchema.php @@ -0,0 +1,70 @@ + + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2012 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +/** + * ColumnSchema class describes the column meta data of a MySQL table. + * + * @author Qiang Xue + * @since 2.0 + */ +class ColumnSchema extends \yii\db\dao\ColumnSchema +{ + /** + * Extracts the PHP type from DB type. + * @param string $dbType DB type + */ + protected function extractType($dbType) + { + if (strncmp($dbType, 'enum', 4) === 0) + $this->type = 'string'; + elseif (strpos($dbType, 'float') !== false || strpos($dbType, 'double') !== false) + $this->type = 'double'; + elseif (strpos($dbType, 'bool') !== false) + $this->type = 'boolean'; + elseif (strpos($dbType, 'int') === 0 && strpos($dbType, 'unsigned') === false || preg_match('/(bit|tinyint|smallint|mediumint)/', $dbType)) + $this->type = 'integer'; + else + $this->type = 'string'; + } + + /** + * Extracts the default value for the column. + * The value is typecasted to correct PHP type. + * @param mixed $defaultValue the default value obtained from metadata + */ + protected function extractDefault($defaultValue) + { + if ($this->dbType === 'timestamp' && $defaultValue === 'CURRENT_TIMESTAMP') + $this->defaultValue = null; + else + parent::extractDefault($defaultValue); + } + + /** + * Extracts size, precision and scale information from column's DB type. + * @param string $dbType the column's DB type + */ + protected function extractLimit($dbType) + { + if (strncmp($dbType, 'enum', 4) === 0 && preg_match('/\((.*)\)/', $dbType, $matches)) + { + $values = explode(',', $matches[1]); + $size = 0; + foreach ($values as $value) + { + if (($n = strlen($value)) > $size) + $size = $n; + } + $this->size = $this->precision = $size-2; + } + else + parent::extractLimit($dbType); + } +} \ No newline at end of file diff --git a/framework/db/dao/mysql/Schema.php b/framework/db/dao/mysql/Schema.php new file mode 100644 index 0000000..751ca5c --- /dev/null +++ b/framework/db/dao/mysql/Schema.php @@ -0,0 +1,310 @@ + + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2011 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +/** + * CMysqlSchema is the class for retrieving metadata information from a MySQL database (version 4.1.x and 5.x). + * + * @author Qiang Xue + * @version $Id: CMysqlSchema.php 3304 2011-06-23 14:53:50Z qiang.xue $ + * @package system.db.schema.mysql + * @since 1.0 + */ +class CMysqlSchema extends CDbSchema +{ + /** + * @var array the abstract column types mapped to physical column types. + * @since 1.1.6 + */ + public $columnTypes = array( + 'pk' => 'int(11) NOT NULL AUTO_INCREMENT PRIMARY KEY', + 'string' => 'varchar(255)', + 'text' => 'text', + 'integer' => 'int(11)', + 'float' => 'float', + 'decimal' => 'decimal', + 'datetime' => 'datetime', + 'timestamp' => 'timestamp', + 'time' => 'time', + 'date' => 'date', + 'binary' => 'blob', + 'boolean' => 'tinyint(1)', + 'money' => 'decimal(19,4)', + ); + + /** + * Quotes a table name for use in a query. + * A simple table name does not schema prefix. + * @param string $name table name + * @return string the properly quoted table name + * @since 1.1.6 + */ + public function quoteSimpleTableName($name) + { + return '`' . $name . '`'; + } + + /** + * Quotes a column name for use in a query. + * A simple column name does not contain prefix. + * @param string $name column name + * @return string the properly quoted column name + * @since 1.1.6 + */ + public function quoteSimpleColumnName($name) + { + return '`' . $name . '`'; + } + + /** + * Compares two table names. + * The table names can be either quoted or unquoted. This method + * will consider both cases. + * @param string $name1 table name 1 + * @param string $name2 table name 2 + * @return boolean whether the two table names refer to the same table. + */ + public function compareTableNames($name1, $name2) + { + return parent::compareTableNames(strtolower($name1), strtolower($name2)); + } + + /** + * Resets the sequence value of a table's primary key. + * The sequence will be reset such that the primary key of the next new row inserted + * will have the specified value or 1. + * @param CDbTableSchema $table the table schema whose primary key sequence will be reset + * @param mixed $value the value for the primary key of the next new row inserted. If this is not set, + * the next new row's primary key will have a value 1. + * @since 1.1 + */ + public function resetSequence($table, $value = null) + { + if ($table->sequenceName !== null) + { + if ($value === null) + $value = $this->getDbConnection()->createCommand("SELECT MAX(` {$table->primaryKey}`) FROM {$table->rawName}")->queryScalar() + 1; + else + $value = (int)$value; + $this->getDbConnection()->createCommand("ALTER TABLE {$table->rawName} AUTO_INCREMENT=$value")->execute(); + } + } + + /** + * Enables or disables integrity check. + * @param boolean $check whether to turn on or off the integrity check. + * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema. + * @since 1.1 + */ + public function checkIntegrity($check = true, $schema = '') + { + $this->getDbConnection()->createCommand('SET FOREIGN_KEY_CHECKS=' . ($check ? 1 : 0))->execute(); + } + + /** + * Loads the metadata for the specified table. + * @param string $name table name + * @return CMysqlTableSchema driver dependent table metadata. Null if the table does not exist. + */ + protected function loadTable($name) + { + $table = new CMysqlTableSchema; + $this->resolveTableNames($table, $name); + + if ($this->findColumns($table)) + { + $this->findConstraints($table); + return $table; + } + else + return null; + } + + /** + * Generates various kinds of table names. + * @param CMysqlTableSchema $table the table instance + * @param string $name the unquoted table name + */ + protected function resolveTableNames($table, $name) + { + $parts = explode('.', str_replace('`', '', $name)); + if (isset($parts[1])) + { + $table->schemaName = $parts[0]; + $table->name = $parts[1]; + $table->rawName = $this->quoteTableName($table->schemaName) . '.' . $this->quoteTableName($table->name); + } + else + { + $table->name = $parts[0]; + $table->rawName = $this->quoteTableName($table->name); + } + } + + /** + * Collects the table column metadata. + * @param CMysqlTableSchema $table the table metadata + * @return boolean whether the table exists in the database + */ + protected function findColumns($table) + { + $sql = 'SHOW COLUMNS FROM ' . $table->rawName; + try + { + $columns = $this->getDbConnection()->createCommand($sql)->queryAll(); + } + catch(Exception $e) + { + return false; + } + foreach ($columns as $column) + { + $c = $this->createColumn($column); + $table->columns[$c->name] = $c; + if ($c->isPrimaryKey) + { + if ($table->primaryKey === null) + $table->primaryKey = $c->name; + elseif (is_string($table->primaryKey)) + $table->primaryKey = array($table->primaryKey, $c->name); + else + $table->primaryKey[] = $c->name; + if ($c->autoIncrement) + $table->sequenceName = ''; + } + } + return true; + } + + /** + * Creates a table column. + * @param array $column column metadata + * @return CDbColumnSchema normalized column metadata + */ + protected function createColumn($column) + { + $c = new CMysqlColumnSchema; + $c->name = $column['Field']; + $c->rawName = $this->quoteColumnName($c->name); + $c->allowNull = $column['Null'] === 'YES'; + $c->isPrimaryKey = strpos($column['Key'], 'PRI') !== false; + $c->isForeignKey = false; + $c->init($column['Type'], $column['Default']); + $c->autoIncrement = strpos(strtolower($column['Extra']), 'auto_increment') !== false; + + return $c; + } + + /** + * @return float server version. + */ + protected function getServerVersion() + { + $version = $this->getDbConnection()->getAttribute(PDO::ATTR_SERVER_VERSION); + $digits = array(); + preg_match('/(\d+)\.(\d+)\.(\d+)/', $version, $digits); + return floatval($digits[1] . '.' . $digits[2] . $digits[3]); + } + + /** + * Collects the foreign key column details for the given table. + * @param CMysqlTableSchema $table the table metadata + */ + protected function findConstraints($table) + { + $row = $this->getDbConnection()->createCommand('SHOW CREATE TABLE ' . $table->rawName)->queryRow(); + $matches = array(); + $regexp = '/FOREIGN KEY\s+\(([^\)]+)\)\s+REFERENCES\s+([^\(^\s]+)\s*\(([^\)]+)\)/mi'; + foreach ($row as $sql) + { + if (preg_match_all($regexp, $sql, $matches, PREG_SET_ORDER)) + break; + } + foreach ($matches as $match) + { + $keys = array_map('trim', explode(',', str_replace('`', '', $match[1]))); + $fks = array_map('trim', explode(',', str_replace('`', '', $match[3]))); + foreach ($keys as $k => $name) + { + $table->foreignKeys[$name] = array(str_replace('`', '', $match[2]), $fks[$k]); + if (isset($table->columns[$name])) + $table->columns[$name]->isForeignKey = true; + } + } + } + + /** + * Returns all table names in the database. + * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema. + * If not empty, the returned table names will be prefixed with the schema name. + * @return array all table names in the database. + * @since 1.0.2 + */ + protected function findTableNames($schema = '') + { + if ($schema === '') + return $this->getDbConnection()->createCommand('SHOW TABLES')->queryColumn(); + $names = $this->getDbConnection()->createCommand('SHOW TABLES FROM ' . $this->quoteTableName($schema))->queryColumn(); + foreach ($names as &$name) + $name = $schema . '.' . $name; + return $names; + } + + /** + * Builds a SQL statement for renaming a column. + * @param string $table the table whose column is to be renamed. The name will be properly quoted by the method. + * @param string $name the old name of the column. The name will be properly quoted by the method. + * @param string $newName the new name of the column. The name will be properly quoted by the method. + * @return string the SQL statement for renaming a DB column. + * @since 1.1.6 + */ + public function renameColumn($table, $name, $newName) + { + $db = $this->getDbConnection(); + $row = $db->createCommand('SHOW CREATE TABLE ' . $db->quoteTableName($table))->queryRow(); + if ($row === false) + throw new CDbException(Yii::t('yii', 'Unable to find "{column}" in table "{table}".', array('{column}' => $name, '{table}' => $table))); + if (isset($row['Create Table'])) + $sql = $row['Create Table']; + else + { + $row = array_values($row); + $sql = $row[1]; + } + if (preg_match_all('/^\s*`(.*?)`\s+(.*?),?$/m', $sql, $matches)) + { + foreach ($matches[1] as $i => $c) + { + if ($c === $name) + { + return "ALTER TABLE " . $db->quoteTableName($table) + . " CHANGE " . $db->quoteColumnName($name) + . ' ' . $db->quoteColumnName($newName) . ' ' . $matches[2][$i]; + } + } + } + + // try to give back a SQL anyway + return "ALTER TABLE " . $db->quoteTableName($table) + . " CHANGE " . $db->quoteColumnName($name) . ' ' . $newName; + } + + /** + * Builds a SQL statement for dropping a foreign key constraint. + * @param string $name the name of the foreign key constraint to be dropped. The name will be properly quoted by the method. + * @param string $table the table whose foreign is to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping a foreign key constraint. + * @since 1.1.6 + */ + public function dropForeignKey($name, $table) + { + return 'ALTER TABLE ' . $this->quoteTableName($table) + . ' DROP FOREIGN KEY ' . $this->quoteColumnName($name); + } +} diff --git a/framework/db/dao/mysql/TableSchema.php b/framework/db/dao/mysql/TableSchema.php new file mode 100644 index 0000000..42dcb51 --- /dev/null +++ b/framework/db/dao/mysql/TableSchema.php @@ -0,0 +1,24 @@ + + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2012 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +/** + * TableSchema represents the metadata for a MySQL table. + * + * @author Qiang Xue + * @since 2.0 + */ +class TableSchema extends CDbTableSchema +{ + /** + * @var string name of the schema (database) that this table belongs to. + * Defaults to null, meaning no schema (or the current database). + */ + public $schemaName; +}