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