Error202
/
yii2-bootstrap
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

...

tags/2.0.0-beta
Qiang Xue 13 years ago
parent
1dd651df51
commit
eddf9937e1
6 changed files with 64 additions and 57 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 74
      framework/db/dao/mysql/Driver.php
  2. 17
      framework/db/dao/mysql/QueryBuilder.php

74
framework/db/dao/mysql/Driver.php
Unescape View File

@ -24,7 +24,7 @@ class Driver extends \yii\db\dao\Driver
/** /**
* @var array mapping from physical column types (keys) to abstract column types (values) * @var array mapping from physical column types (keys) to abstract column types (values)
*/ */
public $typeMap = array( // dbType => type public $typeMap = array(
'tinyint' => self::TYPE_SMALLINT, 'tinyint' => self::TYPE_SMALLINT,
'bit' => self::TYPE_SMALLINT, 'bit' => self::TYPE_SMALLINT,
'smallint' => self::TYPE_SMALLINT, 'smallint' => self::TYPE_SMALLINT,
@ -75,6 +75,16 @@ class Driver extends \yii\db\dao\Driver
} }
/** /**
* Creates a query builder for the database.
* This method may be overridden by child classes to create a DBMS-specific query builder.
* @return QueryBuilder query builder instance
*/
public function createQueryBuilder()
{
return new QueryBuilder($this->connection);
}
/**
* Loads the metadata for the specified table. * Loads the metadata for the specified table.
* @param string $name table name * @param string $name table name
* @return \yii\db\dao\TableSchema driver dependent table metadata. Null if the table does not exist. * @return \yii\db\dao\TableSchema driver dependent table metadata. Null if the table does not exist.
@ -91,9 +101,9 @@ class Driver extends \yii\db\dao\Driver
} }
/** /**
* Generates various kinds of table names. * Resolves the table name and schema name (if any).
* @param \yii\db\dao\TableSchema $table the table instance * @param \yii\db\dao\TableSchema $table the table metadata object
* @param string $name the unquoted table name * @param string $name the table name
*/ */
protected function resolveTableNames($table, $name) protected function resolveTableNames($table, $name)
{ {
@ -127,16 +137,17 @@ class Driver extends \yii\db\dao\Driver
$this->resolveColumnType($c); $this->resolveColumnType($c);
$c->resolvePhpType(); $c->resolvePhpType();
$this->resolveDefaultValue($c, $column['Default']); $this->resolveColumnDefault($c, $column['Default']);
return $c; return $c;
} }
/** /**
* @param \yii\db\dao\ColumnSchema $column * Resolves the default value for the column.
* @param string $value * @param \yii\db\dao\ColumnSchema $column the column metadata object
* @param string $value the default value fetched from database
*/ */
protected function resolveDefaultValue($column, $value) protected function resolveColumnDefault($column, $value)
{ {
if ($column->type !== 'timestamp' || $value !== 'CURRENT_TIMESTAMP') { if ($column->type !== 'timestamp' || $value !== 'CURRENT_TIMESTAMP') {
$column->defaultValue = $column->typecast($value); $column->defaultValue = $column->typecast($value);
@ -144,8 +155,8 @@ class Driver extends \yii\db\dao\Driver
} }
/** /**
* Extracts the PHP type from DB type. * Resolves the abstract data type for the column.
* @param \yii\db\dao\ColumnSchema $column the column * @param \yii\db\dao\ColumnSchema $column the column metadata object
*/ */
public function resolveColumnType($column) public function resolveColumnType($column)
{ {
@ -186,7 +197,7 @@ class Driver extends \yii\db\dao\Driver
} }
/** /**
* Collects the table column metadata. * Collects the metadata of table columns.
* @param \yii\db\dao\TableSchema $table the table metadata * @param \yii\db\dao\TableSchema $table the table metadata
* @return boolean whether the table exists in the database * @return boolean whether the table exists in the database
*/ */
@ -195,21 +206,21 @@ class Driver extends \yii\db\dao\Driver
$sql = 'SHOW COLUMNS FROM ' . $table->quotedName; $sql = 'SHOW COLUMNS FROM ' . $table->quotedName;
try { try {
$columns = $this->connection->createCommand($sql)->queryAll(); $columns = $this->connection->createCommand($sql)->queryAll();
} } catch (\Exception $e) {
catch (\Exception $e) {
return false; return false;
} }
foreach ($columns as $column) { foreach ($columns as $column) {
$table->columns[$c->name] = $c = $this->createColumn($column); $column = $this->createColumn($column);
if ($c->isPrimaryKey) { $table->columns[$column->name] = $column;
if ($column->isPrimaryKey) {
if ($table->primaryKey === null) { if ($table->primaryKey === null) {
$table->primaryKey = $c->name; $table->primaryKey = $column->name;
} elseif (is_string($table->primaryKey)) { } elseif (is_string($table->primaryKey)) {
$table->primaryKey = array($table->primaryKey, $c->name); $table->primaryKey = array($table->primaryKey, $column->name);
} else { } else {
$table->primaryKey[] = $c->name; $table->primaryKey[] = $column->name;
} }
if ($c->autoIncrement) { if ($column->autoIncrement) {
$table->sequenceName = ''; $table->sequenceName = '';
} }
} }
@ -224,9 +235,14 @@ class Driver extends \yii\db\dao\Driver
protected function findConstraints($table) protected function findConstraints($table)
{ {
$row = $this->connection->createCommand('SHOW CREATE TABLE ' . $table->quotedName)->queryRow(); $row = $this->connection->createCommand('SHOW CREATE TABLE ' . $table->quotedName)->queryRow();
$matches = array(); if (isset($row['Create Table'])) {
$sql = $row['Create Table'];
} else {
$row = array_values($row);
$sql = $row[1];
}
$regexp = '/FOREIGN KEY\s+\(([^\)]+)\)\s+REFERENCES\s+([^\(^\s]+)\s*\(([^\)]+)\)/mi'; $regexp = '/FOREIGN KEY\s+\(([^\)]+)\)\s+REFERENCES\s+([^\(^\s]+)\s*\(([^\)]+)\)/mi';
foreach ($row as $sql) {
if (preg_match_all($regexp, $sql, $matches, PREG_SET_ORDER)) { if (preg_match_all($regexp, $sql, $matches, PREG_SET_ORDER)) {
foreach ($matches as $match) { foreach ($matches as $match) {
$fks = array_map('trim', explode(',', str_replace('`', '', $match[1]))); $fks = array_map('trim', explode(',', str_replace('`', '', $match[1])));
@ -237,8 +253,6 @@ class Driver extends \yii\db\dao\Driver
} }
$table->foreignKeys[] = $constraint; $table->foreignKeys[] = $constraint;
} }
break;
}
} }
} }
@ -255,19 +269,9 @@ class Driver extends \yii\db\dao\Driver
} }
$sql = 'SHOW TABLES FROM ' . $this->quoteSimpleTableName($schema); $sql = 'SHOW TABLES FROM ' . $this->quoteSimpleTableName($schema);
$names = $this->connection->createCommand($sql)->queryColumn(); $names = $this->connection->createCommand($sql)->queryColumn();
foreach ($names as &$name) { foreach ($names as $i => $name) {
$name = $schema . '.' . $name; $names[$i] = $schema . '.' . $name;
} }
return $names; return $names;
} }
/**
* Creates a query builder for the database.
* This method may be overridden by child classes to create a DBMS-specific query builder.
* @return QueryBuilder query builder instance
*/
public function createQueryBuilder()
{
return new QueryBuilder($this->connection);
}
} }

17
framework/db/dao/mysql/QueryBuilder.php
Unescape View File

@ -10,6 +10,8 @@
namespace yii\db\dao\mysql; namespace yii\db\dao\mysql;
use yii\db\Exception;
/** /**
* QueryBuilder builds a SQL statement based on the specification given as a [[Query]] object. * QueryBuilder builds a SQL statement based on the specification given as a [[Query]] object.
* *
@ -42,16 +44,17 @@ class QueryBuilder extends \yii\db\dao\QueryBuilder
/** /**
* Builds a SQL statement for renaming a 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 $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 $oldName 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. * @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. * @return string the SQL statement for renaming a DB column.
*/ */
public function renameColumn($table, $name, $newName) public function renameColumn($table, $oldName, $newName)
{ {
$quotedTable = $this->driver->quoteTableName($table); $quotedTable = $this->driver->quoteTableName($table);
$row = $this->connection->createCommand('SHOW CREATE TABLE ' . $quotedTable)->queryRow(); $row = $this->connection->createCommand('SHOW CREATE TABLE ' . $quotedTable)->queryRow();
if ($row === false) if ($row === false) {
throw new CDbException(Yii::t('yii', 'Unable to find "{column}" in table "{table}".', array('{column}' => $name, '{table}' => $table))); throw new Exception("Unable to find '$oldName' in table '$table'.");
}
if (isset($row['Create Table'])) { if (isset($row['Create Table'])) {
$sql = $row['Create Table']; $sql = $row['Create Table'];
} else { } else {
@ -60,14 +63,14 @@ class QueryBuilder extends \yii\db\dao\QueryBuilder
} }
if (preg_match_all('/^\s*`(.*?)`\s+(.*?),?$/m', $sql, $matches)) { if (preg_match_all('/^\s*`(.*?)`\s+(.*?),?$/m', $sql, $matches)) {
foreach ($matches[1] as $i => $c) { foreach ($matches[1] as $i => $c) {
if ($c === $name) { if ($c === $oldName) {
return "ALTER TABLE $quotedTable CHANGE " . $this->driver->quoteColumnName($name) return "ALTER TABLE $quotedTable CHANGE " . $this->driver->quoteColumnName($oldName)
. ' ' . $this->driver->quoteColumnName($newName) . ' ' . $matches[2][$i]; . ' ' . $this->driver->quoteColumnName($newName) . ' ' . $matches[2][$i];
} }
} }
} }
// try to give back a SQL anyway // try to give back a SQL anyway
return "ALTER TABLE $quotedTable CHANGE " . $this->driver->quoteColumnName($name) . ' ' . $newName; return "ALTER TABLE $quotedTable CHANGE " . $this->driver->quoteColumnName($oldName) . ' ' . $newName;
} }
/** /**

Write Preview
Loading…
Cancel
Save