Skip to content

Y
yii2
Commit 3335f09d by Qiang Xue
Options

w

parent 05c7915d
Showing with 441 additions and 24 deletions
framework/db/dao/ColumnSchema.php
<?php
/**
* CDbColumnSchema class file.
* ColumnSchema class file.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @link http://www.yiiframework.com/
* @copyright Copyright &copy; 2008-2011 Yii Software LLC
* @copyright Copyright &copy; 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 <qiang.xue@gmail.com>
* @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;
......
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))
......
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;
/**
......
framework/db/dao/TableSchema.php
<?php
/**
* CDbTableSchema class file.
* TableSchema class file.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @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:
* <ul>
* <li>{@link name}</li>
* <li>{@link rawName}</li>
......@@ -24,20 +26,22 @@
* </ul>
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @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();
......
framework/db/dao/mysql/ColumnSchema.php 0 → 100644
<?php
/**
* ColumnSchema class file.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @link http://www.yiiframework.com/
* @copyright Copyright &copy; 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 <qiang.xue@gmail.com>
* @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
framework/db/dao/mysql/Schema.php 0 → 100644
<?php
/**
* CMysqlSchema class file.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @link http://www.yiiframework.com/
* @copyright Copyright &copy; 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 <qiang.xue@gmail.com>
* @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);
}
}
framework/db/dao/mysql/TableSchema.php 0 → 100644
<?php
/**
* TableSchema class file.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @link http://www.yiiframework.com/
* @copyright Copyright &copy; 2008-2012 Yii Software LLC
* @license http://www.yiiframework.com/license/
*/
/**
* TableSchema represents the metadata for a MySQL table.
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @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;
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment