Qiang Xue
13 years ago
5 changed files with 384 additions and 8 deletions
@ -0,0 +1,210 @@
|
||||
<?php |
||||
/** |
||||
* Driver class file. |
||||
* |
||||
* @author Qiang Xue <qiang.xue@gmail.com> |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright © 2008-2012 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\dao\sqlite; |
||||
|
||||
use yii\db\dao\TableSchema; |
||||
use yii\db\dao\ColumnSchema; |
||||
|
||||
/** |
||||
* Driver is the class for retrieving metadata from a SQLite (2/3) database. |
||||
* |
||||
* @author Qiang Xue <qiang.xue@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class Driver extends \yii\db\dao\Driver |
||||
{ |
||||
/** |
||||
* @var array mapping from physical column types (keys) to abstract column types (values) |
||||
*/ |
||||
public $typeMap = array( |
||||
'tinyint' => self::TYPE_SMALLINT, |
||||
'bit' => self::TYPE_SMALLINT, |
||||
'smallint' => self::TYPE_SMALLINT, |
||||
'mediumint' => self::TYPE_INTEGER, |
||||
'int' => self::TYPE_INTEGER, |
||||
'integer' => self::TYPE_INTEGER, |
||||
'bigint' => self::TYPE_BIGINT, |
||||
'float' => self::TYPE_FLOAT, |
||||
'double' => self::TYPE_FLOAT, |
||||
'real' => self::TYPE_FLOAT, |
||||
'decimal' => self::TYPE_DECIMAL, |
||||
'numeric' => self::TYPE_DECIMAL, |
||||
'tinytext' => self::TYPE_TEXT, |
||||
'mediumtext' => self::TYPE_TEXT, |
||||
'longtext' => self::TYPE_TEXT, |
||||
'text' => self::TYPE_TEXT, |
||||
'varchar' => self::TYPE_STRING, |
||||
'string' => self::TYPE_STRING, |
||||
'char' => self::TYPE_STRING, |
||||
'datetime' => self::TYPE_DATETIME, |
||||
'year' => self::TYPE_DATE, |
||||
'date' => self::TYPE_DATE, |
||||
'time' => self::TYPE_TIME, |
||||
'timestamp' => self::TYPE_TIMESTAMP, |
||||
'enum' => self::TYPE_STRING, |
||||
); |
||||
|
||||
/** |
||||
* Creates a query builder for the MySQL 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); |
||||
} |
||||
|
||||
/** |
||||
* 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. |
||||
*/ |
||||
protected function findTableNames($schema = '') |
||||
{ |
||||
$sql = "SELECT DISTINCT tbl_name FROM sqlite_master WHERE tbl_name<>'sqlite_sequence'"; |
||||
return $this->connection->createCommand($sql)->queryColumn(); |
||||
} |
||||
|
||||
/** |
||||
* Loads the metadata for the specified table. |
||||
* @param string $name table name |
||||
* @return \yii\db\dao\TableSchema driver dependent table metadata. Null if the table does not exist. |
||||
*/ |
||||
protected function loadTableSchema($name) |
||||
{ |
||||
$table = new TableSchema; |
||||
$table->name = $name; |
||||
$table->quotedName = $this->quoteTableName($name); |
||||
|
||||
if ($this->findColumns($table)) { |
||||
$this->findConstraints($table); |
||||
return $table; |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Collects the table column metadata. |
||||
* @param \yii\db\dao\TableSchema $table the table metadata |
||||
* @return boolean whether the table exists in the database |
||||
*/ |
||||
protected function findColumns($table) |
||||
{ |
||||
$sql = "PRAGMA table_info({$table->quotedName})"; |
||||
$columns = $this->connection->createCommand($sql)->queryAll(); |
||||
if (empty($columns)) { |
||||
return false; |
||||
} |
||||
|
||||
foreach ($columns as $column) { |
||||
$column = $this->createColumn($column); |
||||
$table->columns[$column->name] = $column; |
||||
if ($column->isPrimaryKey) { |
||||
if ($table->primaryKey === null) { |
||||
$table->primaryKey = $column->name; |
||||
} elseif (is_string($table->primaryKey)) { |
||||
$table->primaryKey = array($table->primaryKey, $column->name); |
||||
} else { |
||||
$table->primaryKey[] = $column->name; |
||||
} |
||||
} |
||||
} |
||||
if (is_string($table->primaryKey) && !strncasecmp($table->columns[$table->primaryKey]->dbType, 'int', 3)) { |
||||
$table->sequenceName = ''; |
||||
$table->columns[$table->primaryKey]->autoIncrement = true; |
||||
} |
||||
|
||||
return true; |
||||
} |
||||
|
||||
/** |
||||
* Collects the foreign key column details for the given table. |
||||
* @param \yii\db\dao\TableSchema $table the table metadata |
||||
*/ |
||||
protected function findConstraints($table) |
||||
{ |
||||
$sql = "PRAGMA foreign_key_list({$table->quotedName})"; |
||||
$keys = $this->connection->createCommand($sql)->queryAll(); |
||||
foreach ($keys as $key) { |
||||
$table->foreignKeys[] = array($key['table'], $key['from'] => $key['to']); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Creates a table column. |
||||
* @param array $column column metadata |
||||
* @return ColumnSchema normalized column metadata |
||||
*/ |
||||
protected function createColumn($column) |
||||
{ |
||||
$c = new ColumnSchema; |
||||
$c->name = $column['name']; |
||||
$c->quotedName = $this->quoteSimpleColumnName($c->name); |
||||
$c->allowNull = !$column['notnull']; |
||||
$c->isPrimaryKey = $column['pk'] != 0; |
||||
|
||||
$c->dbType = $column['type']; |
||||
$this->resolveColumnType($c); |
||||
$c->resolvePhpType(); |
||||
|
||||
$this->resolveColumnDefault($c, $column['dflt_value']); |
||||
|
||||
return $c; |
||||
} |
||||
|
||||
/** |
||||
* Resolves the abstract data type for the column. |
||||
* @param \yii\db\dao\ColumnSchema $column the column metadata object |
||||
*/ |
||||
public function resolveColumnType($column) |
||||
{ |
||||
$column->type = self::TYPE_STRING; |
||||
$column->unsigned = strpos($column->dbType, 'unsigned') !== false; |
||||
|
||||
if (preg_match('/^(\w+)(?:\(([^\)]+)\))?/', $column->dbType, $matches)) { |
||||
$type = $matches[1]; |
||||
if (isset($this->typeMap[$type])) { |
||||
$column->type = $this->typeMap[$type]; |
||||
} |
||||
|
||||
if (!empty($matches[2])) { |
||||
$values = explode(',', $matches[2]); |
||||
$column->size = $column->precision = (int)$values[0]; |
||||
if (isset($values[1])) { |
||||
$column->scale = (int)$values[1]; |
||||
} |
||||
if ($column->size === 1 && ($type === 'tinyint' || $type === 'bit')) { |
||||
$column->type = 'boolean'; |
||||
} elseif ($type === 'bit') { |
||||
if ($column->size > 32) { |
||||
$column->type = 'bigint'; |
||||
} elseif ($column->size === 32) { |
||||
$column->type = 'integer'; |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Resolves the default value for the column. |
||||
* @param \yii\db\dao\ColumnSchema $column the column metadata object |
||||
* @param string $value the default value fetched from database |
||||
*/ |
||||
protected function resolveColumnDefault($column, $value) |
||||
{ |
||||
if ($column->type === 'string') { |
||||
$column->defaultValue = trim($value, "'\""); |
||||
} else { |
||||
$column->defaultValue = $column->typecast(strcasecmp($value, 'null') ? $value : null); |
||||
} |
||||
} |
||||
} |
@ -0,0 +1,167 @@
|
||||
<?php |
||||
/** |
||||
* QueryBuilder class file. |
||||
* |
||||
* @author Qiang Xue <qiang.xue@gmail.com> |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright © 2008-2012 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\dao\sqlite; |
||||
|
||||
use yii\db\Exception; |
||||
|
||||
/** |
||||
* QueryBuilder is the query builder for MySQL databases. |
||||
* |
||||
* @author Qiang Xue <qiang.xue@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class QueryBuilder extends \yii\db\dao\QueryBuilder |
||||
{ |
||||
/** |
||||
* @var array mapping from abstract column types (keys) to physical column types (values). |
||||
*/ |
||||
public $typeMap = array( |
||||
Driver::TYPE_PK => 'integer PRIMARY KEY AUTOINCREMENT NOT NULL', |
||||
Driver::TYPE_STRING => 'varchar(255)', |
||||
Driver::TYPE_TEXT => 'text', |
||||
Driver::TYPE_SMALLINT => 'smallint', |
||||
Driver::TYPE_INTEGER => 'integer', |
||||
Driver::TYPE_BIGINT => 'bigint', |
||||
Driver::TYPE_FLOAT => 'float', |
||||
Driver::TYPE_DECIMAL => 'decimal', |
||||
Driver::TYPE_DATETIME => 'datetime', |
||||
Driver::TYPE_TIMESTAMP => 'timestamp', |
||||
Driver::TYPE_TIME => 'time', |
||||
Driver::TYPE_DATE => 'date', |
||||
Driver::TYPE_BINARY => 'blob', |
||||
Driver::TYPE_BOOLEAN => 'tinyint(1)', |
||||
Driver::TYPE_MONEY => 'decimal(19,4)', |
||||
); |
||||
|
||||
/** |
||||
* 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 string $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. |
||||
*/ |
||||
public function resetSequence($table, $value = null) |
||||
{ |
||||
if ($table->sequenceName !== null) { |
||||
if ($value === null) { |
||||
$value = $this->connection->createCommand("SELECT MAX(`{$table->primaryKey}`) FROM {$table->quotedName}")->queryScalar(); |
||||
} |
||||
else { |
||||
$value = (int)$value - 1; |
||||
} |
||||
try { |
||||
// it's possible sqlite_sequence does not exist |
||||
$this->connection->createCommand("UPDATE sqlite_sequence SET seq='$value' WHERE name='{$table->name}'")->execute(); |
||||
} catch (Exception $e) { |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 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. |
||||
*/ |
||||
public function checkIntegrity($check = true, $schema = '') |
||||
{ |
||||
// SQLite doesn't enforce integrity |
||||
} |
||||
|
||||
/** |
||||
* 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) |
||||
{ |
||||
return "DELETE FROM " . $this->driver->quoteTableName($table); |
||||
} |
||||
|
||||
/** |
||||
* 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->driver->quoteTableName($name); |
||||
} |
||||
|
||||
/** |
||||
* 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. |
||||
*/ |
||||
public function dropColumn($table, $column) |
||||
{ |
||||
throw new Exception('Dropping DB column is not supported by SQLite.'); |
||||
} |
||||
|
||||
/** |
||||
* 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 $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. |
||||
* @return string the SQL statement for renaming a DB column. |
||||
*/ |
||||
public function renameColumn($table, $oldName, $newName) |
||||
{ |
||||
throw new Exception('Renaming a DB column is not supported by SQLite.'); |
||||
} |
||||
|
||||
/** |
||||
* 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|array $columns the name of the column to that the constraint will be added on. |
||||
* If there are multiple columns, separate them with commas or use an array to represent them. |
||||
* @param string $refTable the table that the foreign key references to. |
||||
* @param string|array $refColumns the name of the column that the foreign key references to. |
||||
* If there are multiple columns, separate them with commas or use an array to represent them. |
||||
* @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) |
||||
{ |
||||
throw new Exception('Adding a foreign key constraint to an existing table is not supported by SQLite.'); |
||||
} |
||||
|
||||
/** |
||||
* 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) |
||||
{ |
||||
throw new Exception('Dropping a foreign key constraint is not supported by SQLite.'); |
||||
} |
||||
|
||||
/** |
||||
* 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 [[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) |
||||
{ |
||||
throw new Exception('Altering a DB column is not supported by SQLite.'); |
||||
} |
||||
} |
Loading…
Reference in new issue