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