diff options
Diffstat (limited to 'framework/Testing/Data/Schema/mssql')
5 files changed, 781 insertions, 0 deletions
diff --git a/framework/Testing/Data/Schema/mssql/TMssqlColumnSchema.php b/framework/Testing/Data/Schema/mssql/TMssqlColumnSchema.php new file mode 100755 index 00000000..0123f183 --- /dev/null +++ b/framework/Testing/Data/Schema/mssql/TMssqlColumnSchema.php @@ -0,0 +1,57 @@ +<?php +/** + * TMssqlColumnSchema class file. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2009 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +prado::using('System.Testing.Data.Schema.TDbColumnSchema'); + +/** + * TMssqlColumnSchema class describes the column meta data of a MSSQL table. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @version $Id: TMssqlColumnSchema.php 2679 2009-06-15 07:49:42Z Christophe.Boulain $ + * @package System.Testing.Data.Schema.mssql + * @since 1.0.4 + */ +class TMssqlColumnSchema extends TDbColumnSchema +{ +	/** +	 * Extracts the PHP type from DB type. +	 * @param string DB type +	 */ +	protected function extractType($dbType) +	{ +		if(strpos($dbType,'bigint')!==false || strpos($dbType,'float')!==false || strpos($dbType,'real')!==false) +			$this->type='double'; +		else if(strpos($dbType,'int')!==false || strpos($dbType,'smallint')!==false || strpos($dbType,'tinyint')) +			$this->type='integer'; +		else if(strpos($dbType,'bit')!==false) +			$this->type='boolean'; +		else +			$this->type='string'; +	} + +	protected function extractDefault($defaultValue) +	{ +		if($this->dbType==='timestamp' ) +			$this->defaultValue=null; +		else +			parent::extractDefault(str_replace(array('(',')',"'"), '', $defaultValue)); +	} + +	/** +	 * Extracts size, precision and scale information from column's DB type. +	 * We do nothing here, since sizes and precisions have been computed before. +	 * @param string the column's DB type +	 */ +	protected function extractLimit($dbType) +	{ +	} +} diff --git a/framework/Testing/Data/Schema/mssql/TMssqlCommandBuilder.php b/framework/Testing/Data/Schema/mssql/TMssqlCommandBuilder.php new file mode 100755 index 00000000..17a6f6ab --- /dev/null +++ b/framework/Testing/Data/Schema/mssql/TMssqlCommandBuilder.php @@ -0,0 +1,303 @@ +<?php +/** + * CMsCommandBuilder class file. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @author Wei Zhuo <weizhuo[at]gmail[dot]com> + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2009 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +prado::using ('System.Testing.Data.schame.TDbCommandBuilder'); + +/** + * TMssqlCommandBuilder provides basic methods to create query commands for tables for Mssql Servers. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @author Wei Zhuo <weizhuo[at]gmail[dot]com> + * @version $Id: TMssqlCommandBuilder.php 2679 2009-06-15 07:49:42Z Christophe.Boulain $ + * @package System.Testing.Data.schema.mssql + * @since 1.0.4 + */ +class TMssqlCommandBuilder extends TDbCommandBuilder +{ +   	/** +	 * Returns the last insertion ID for the specified table. +	 * Override parent implemantation since PDO mssql driver does not provide this method +	 * @param TDbTableSchema the table metadata +	 * @return mixed last insertion id. Null is returned if no sequence name. +	 */ +	public function getLastInsertID($table) +	{ +		if($table->sequenceName!==null) +			return $this->getDbConnection()->createCommand('SELECT SCOPE_IDENTITY()')->queryScalar(); +		else +			return null; +	} + +	/** +	 * Creates a COUNT(*) command for a single table. +	 * Override parent implementation to remove the order clause of criteria if it exists +	 * @param TDbTableSchema the table metadata +	 * @param TDbCriteria the query criteria +	 * @return TDbCommand query command. +	 */ +	public function createCountCommand($table,$criteria) +	{ +		$criteria->order=''; +		return parent::createCountCommand($table, $criteria); +	} + +	/** +	 * Creates a SELECT command for a single table. +	 * Override parent implementation to check if an orderby clause if specified when querying with an offset +	 * @param TDbTableSchema the table metadata +	 * @param TDbCriteria the query criteria +	 * @return TDbCommand query command. +	 */ +	public function createFindCommand($table,$criteria) +	{ +		$criteria=$this->checkCriteria($table,$criteria); +		return parent::createFindCommand($table,$criteria); + +	} + +	/** +	 * Creates an UPDATE command. +	 * Override parent implementation because mssql don't want to update an identity column +	 * @param TDbTableSchema the table metadata +	 * @param array list of columns to be updated (name=>value) +	 * @param TDbCriteria the query criteria +	 * @return TDbCommand update command. +	 */ +	public function createUpdateCommand($table,$data,$criteria) +	{ +		$criteria=$this->checkCriteria($table,$criteria); +		$fields=array(); +		$values=array(); +		$bindByPosition=isset($criteria->params[0]); +		foreach($data as $name=>$value) +		{ +			if(($column=$table->getColumn($name))!==null) +			{ +				if ($table->sequenceName !== null && $column->isPrimaryKey === true) continue; +				if($value instanceof TDbExpression) +					$fields[]=$column->rawName.'='.(string)$value; +				else if($bindByPosition) +				{ +					$fields[]=$column->rawName.'=?'; +					$values[]=$column->typecast($value); +				} +				else +				{ +					$fields[]=$column->rawName.'=:'.$name; +					$values[':'.$name]=$column->typecast($value); +				} +			} +		} +		if($fields===array()) +			throw new TDbException('No columns are being updated for table "{0}".', +				$table->name); +		$sql="UPDATE {$table->rawName} SET ".implode(', ',$fields); +		$sql=$this->applyJoin($sql,$criteria->join); +		$sql=$this->applyCondition($sql,$criteria->condition); +		$sql=$this->applyOrder($sql,$criteria->order); +		$sql=$this->applyLimit($sql,$criteria->limit,$criteria->offset); + +		$command=$this->getDbConnection()->createCommand($sql); +		$this->bindValues($command,array_merge($values,$criteria->params)); + +		return $command; +	} + +	/** +	 * Creates a DELETE command. +	 * Override parent implementation to check if an orderby clause if specified when querying with an offset +	 * @param TDbTableSchema the table metadata +	 * @param TDbCriteria the query criteria +	 * @return TDbCommand delete command. +	 */ +	public function createDeleteCommand($table,$criteria) +	{ +		$criteria=$this->checkCriteria($table, $criteria); +		return parent::createDeleteCommand($table, $criteria); +	} + +	/** +	 * Creates an UPDATE command that increments/decrements certain columns. +	 * Override parent implementation to check if an orderby clause if specified when querying with an offset +	 * @param TDbTableSchema the table metadata +	 * @param TDbCriteria the query criteria +	 * @param array counters to be updated (counter increments/decrements indexed by column names.) +	 * @return TDbCommand the created command +	 * @throws CException if no counter is specified +	 */ +	public function createUpdateCounterCommand($table,$counters,$criteria) +	{ +		$criteria=$this->checkCriteria($table, $criteria); +		return parent::createUpdateCounterCommand($table, $counters, $criteria); +	} + +	/** +	 * This is a port from Prado Framework. +	 * +	 * Overrides parent implementation. Alters the sql to apply $limit and $offset. +	 * The idea for limit with offset is done by modifying the sql on the fly +	 * with numerous assumptions on the structure of the sql string. +	 * The modification is done with reference to the notes from +	 * http://troels.arvin.dk/db/rdbms/#select-limit-offset +	 * +	 * <code> +	 * SELECT * FROM ( +	 *  SELECT TOP n * FROM ( +	 *    SELECT TOP z columns      -- (z=n+skip) +	 *    FROM tablename +	 *    ORDER BY key ASC +	 *  ) AS FOO ORDER BY key DESC -- ('FOO' may be anything) +	 * ) AS BAR ORDER BY key ASC    -- ('BAR' may be anything) +	 * </code> +	 * +	 * <b>Regular expressions are used to alter the SQL query. The resulting SQL query +	 * may be malformed for complex queries.</b> The following restrictions apply +	 * +	 * <ul> +	 *   <li> +	 * In particular, <b>commas</b> should <b>NOT</b> +	 * be used as part of the ordering expression or identifier. Commas must only be +	 * used for separating the ordering clauses. +	 *  </li> +	 *  <li> +	 * In the ORDER BY clause, the column name should NOT be be qualified +	 * with a table name or view name. Alias the column names or use column index. +	 * </li> +	 * <li> +	 * No clauses should follow the ORDER BY clause, e.g. no COMPUTE or FOR clauses. +	 * </li> +	 * +	 * @param string SQL query string. +	 * @param integer maximum number of rows, -1 to ignore limit. +	 * @param integer row offset, -1 to ignore offset. +	 * @return string SQL with limit and offset. +	 * +	 * @author Wei Zhuo <weizhuo[at]gmail[dot]com> +	 */ +	public function applyLimit($sql, $limit, $offset) +	{ +		$limit = $limit!==null ? intval($limit) : -1; +		$offset = $offset!==null ? intval($offset) : -1; +		if ($limit > 0 && $offset <= 0) //just limit +			$sql = preg_replace('/^([\s(])*SELECT( DISTINCT)?(?!\s*TOP\s*\()/i',"\\1SELECT\\2 TOP $limit", $sql); +		else if($limit > 0 && $offset > 0) +			$sql = $this->rewriteLimitOffsetSql($sql, $limit,$offset); +		return $sql; +	} + +	/** +	 * Rewrite sql to apply $limit > and $offset > 0 for MSSQL database. +	 * See http://troels.arvin.dk/db/rdbms/#select-limit-offset +	 * @param string sql query +	 * @param integer $limit > 0 +	 * @param integer $offset > 0 +	 * @return sql modified sql query applied with limit and offset. +	 * +	 * @author Wei Zhuo <weizhuo[at]gmail[dot]com> +	 */ +	protected function rewriteLimitOffsetSql($sql, $limit, $offset) +	{ +		$fetch = $limit+$offset; +		$sql = preg_replace('/^([\s(])*SELECT( DISTINCT)?(?!\s*TOP\s*\()/i',"\\1SELECT\\2 TOP $fetch", $sql); +		$ordering = $this->findOrdering($sql); + +		$orginalOrdering = $this->joinOrdering($ordering); +		$reverseOrdering = $this->joinOrdering($this->reverseDirection($ordering)); +		$sql = "SELECT * FROM (SELECT TOP {$limit} * FROM ($sql) as [__inner top table__] {$reverseOrdering}) as [__outer top table__] {$orginalOrdering}"; +		return $sql; +	} + +	/** +	 * Base on simplified syntax http://msdn2.microsoft.com/en-us/library/aa259187(SQL.80).aspx +	 * +	 * @param string $sql +	 * @return array ordering expression as key and ordering direction as value +	 * +	 * @author Wei Zhuo <weizhuo[at]gmail[dot]com> +	 */ +	protected function findOrdering($sql) +	{ +		if(!preg_match('/ORDER BY/i', $sql)) +			return array(); +		$matches=array(); +		$ordering=array(); +		preg_match_all('/(ORDER BY)[\s"\[](.*)(ASC|DESC)?(?:[\s"\[]|$|COMPUTE|FOR)/i', $sql, $matches); +		if(count($matches)>1 && count($matches[2]) > 0) +		{ +			$parts = explode(',', $matches[2][0]); +			foreach($parts as $part) +			{ +				$subs=array(); +				if(preg_match_all('/(.*)[\s"\]](ASC|DESC)$/i', trim($part), $subs)) +				{ +					if(count($subs) > 1 && count($subs[2]) > 0) +					{ +						$ordering[$subs[1][0]] = $subs[2][0]; +					} +					//else what? +				} +				else +					$ordering[trim($part)] = 'ASC'; +			} +		} +		return $ordering; +	} + +	/** +	 * @param array ordering obtained from findOrdering() +	 * @return string concat the orderings +	 * +	 * @author Wei Zhuo <weizhuo[at]gmail[dot]com> +	 */ +	protected function joinOrdering($orders) +	{ +		if(count($orders)>0) +		{ +			$str=array(); +			foreach($orders as $column => $direction) +				$str[] = $column.' '.$direction; +			return 'ORDER BY '.implode(', ', $str); +		} +	} + +	/** +	 * @param array original ordering +	 * @return array ordering with reversed direction. +	 * +	 * @author Wei Zhuo <weizhuo[at]gmail[dot]com> +	 */ +	protected function reverseDirection($orders) +	{ +		foreach($orders as $column => $direction) +			$orders[$column] = strtolower(trim($direction))==='desc' ? 'ASC' : 'DESC'; +		return $orders; +	} + + +	/** +	 * Checks if the criteria has an order by clause when using offset/limit. +	 * Override parent implementation to check if an orderby clause if specified when querying with an offset +	 * If not, order it by pk. +	 * @param TMssqlTableSchema table schema +	 * @param TDbCriteria criteria +	 * @return TDbCrireria the modified criteria +	 */ +	protected function checkCriteria($table, $criteria) +	{ +		if ($criteria->offset > 0 && $criteria->order==='') +		{ +			$criteria->order=is_array($table->primaryKey)?implode(',',$table->primaryKey):$table->primaryKey; +		} +		return $criteria; +	} +} diff --git a/framework/Testing/Data/Schema/mssql/TMssqlPdoAdapter.php b/framework/Testing/Data/Schema/mssql/TMssqlPdoAdapter.php new file mode 100755 index 00000000..6f8777dd --- /dev/null +++ b/framework/Testing/Data/Schema/mssql/TMssqlPdoAdapter.php @@ -0,0 +1,74 @@ +<?php +/** + * TMssqlPdo class file + * + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2009 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +/** + * This is an extension of default PDO class for mssql driver only + * It provides some missing functionalities of pdo driver + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @version $Id: TMssqlPdoAdapter.php 2679 2009-06-15 07:49:42Z Christophe.Boulain $ + * @package System.Testing.Data.schema.mssql + * @since 1.0.4 + */ +class TMssqlPdoAdapter extends PDO +{ +	/** +	 * Get the last inserted id value +	 * MSSQL doesn't support sequence, so, argument is ignored +	 * +	 * @param string sequence name. Defaults to null +	 * @return int last inserted id +	 */ +	public function lastInsertId ($sequence=NULL) +	{ +		return $this->query('SELECT SCOPE_IDENTITY()')->fetchColumn(); +	} + +	/** +	 * Begin a transaction +	 * +	 * Is is necessary to override pdo's method, as mssql pdo drivers +	 * does not support transaction +	 * +	 * @return boolean +	 */ +	public function beginTransaction () +	{ +		$this->exec('BEGIN TRANSACTION'); +		return true; +	} + +	/** +	 * Commit a transaction +	 * +	 * Is is necessary to override pdo's method, as mssql pdo drivers +	 * does not support transaction +	 * +	 * @return boolean +	 */ +	public function commit () +	{ +		$this->exec('COMMIT TRANSACTION'); +		return true; +	} + +	/** +	 * Rollback a transaction +	 * +	 * Is is necessary to override pdo's method, ac mssql pdo drivers +	 * does not support transaction +	 * +	 * @return boolean +	 */ +	public function rollBack () +	{ +		$this->exec('ROLLBACK TRANSACTION'); +		return true; +	} +} diff --git a/framework/Testing/Data/Schema/mssql/TMssqlSchema.php b/framework/Testing/Data/Schema/mssql/TMssqlSchema.php new file mode 100755 index 00000000..1b3d815c --- /dev/null +++ b/framework/Testing/Data/Schema/mssql/TMssqlSchema.php @@ -0,0 +1,312 @@ +<?php +/** + * TMssqlSchema class file. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2009 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +prado::using('System.Testing.Data.Schema.TDbSchema'); + +/** + * TMssqlSchema is the class for retrieving metadata information from a MS SQL Server database. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @version $Id: TMssqlSchema.php 2679 2009-06-15 07:49:42Z Christophe.Boulain $ + * @package System.Testing.Data.Schema.mssql + * @since 1.0.4 + */ +class TMssqlSchema extends TDbSchema +{ +	const DEFAULT_SCHEMA='dbo'; + + +	/** +	 * Quotes a table name for use in a query. +	 * @param string table name +	 * @return string the properly quoted table name +	 */ +	public function quoteTableName($name) +	{ +		if (strpos($name,'.')===false) +			return '['.$name.']'; +		$names=explode('.',$name); +		foreach ($names as &$n) +			$n = '['.$n.']'; +		return implode('.',$names); +	} + +	/** +	 * Quotes a column name for use in a query. +	 * @param string column name +	 * @return string the properly quoted column name +	 */ +	public function quoteColumnName($name) +	{ +		return '['.$name.']'; +	} + +	/** +	 * Compares two table names. +	 * The table names can be either quoted or unquoted. This method +	 * will consider both cases. +	 * @param string table name 1 +	 * @param string table name 2 +	 * @return boolean whether the two table names refer to the same table. +	 */ +	public function compareTableNames($name1,$name2) +	{ +		$name1=str_replace(array('[',']'),'',$name1); +		$name1=str_replace(array('[',']'),'',$name2); +		return parent::compareTableNames(strtolower($name1),strtolower($name2)); +	} + +	/** +	 * Creates a table instance representing the metadata for the named table. +	 * @return CMysqlTableSchema driver dependent table metadata. Null if the table does not exist. +	 */ +	protected function createTable($name) +	{ +		$table=new TMssqlTableSchema; +		$this->resolveTableNames($table,$name); +		//if (!in_array($table->name, $this->tableNames)) return null; +		$table->primaryKey=$this->findPrimaryKey($table); +		$table->foreignKeys=$this->findForeignKeys($table); +		if($this->findColumns($table)) +		{ +			return $table; +		} +		else +			return null; +	} + +	/** +	 * Generates various kinds of table names. +	 * @param CMysqlTableSchema the table instance +	 * @param string the unquoted table name +	 */ +	protected function resolveTableNames($table,$name) +	{ +		$parts=explode('.',str_replace(array('[',']'),'',$name)); +		if(($c=count($parts))==3) +		{ +			// Catalog name, schema name and table name provided +			$table->catalogName=$parts[0]; +			$table->schemaName=$parts[1]; +			$table->name=$parts[2]; +			$table->rawName=$this->quoteTableName($table->catalogName).'.'.$this->quoteTableName($table->schemaName).'.'.$this->quoteTableName($table->name); +		} +		elseif ($c==2) +		{ +			// Only schema name and table name provided +			$table->name=$parts[1]; +			$table->schemaName=$parts[0]; +			$table->rawName=$this->quoteTableName($table->schemaName).'.'.$this->quoteTableName($table->name); +		} +		else +		{ +			// Only the name given, we need to get at least the schema name +			//if (empty($this->_schemaNames)) $this->findTableNames(); +			$table->name=$parts[0]; +			$table->schemaName=self::DEFAULT_SCHEMA; +			$table->rawName=$this->quoteTableName($table->schemaName).'.'.$this->quoteTableName($table->name); +		} +	} + +	/** +	 * Gets the primary key column(s) details for the given table. +	 * @param TMssqlTableSchema table +	 * @return mixed primary keys (null if no pk, string if only 1 column pk, or array if composite pk) +	 */ +	protected function findPrimaryKey($table) +	{ +		$kcu='INFORMATION_SCHEMA.KEY_COLUMN_USAGE'; +		$tc='INFORMATION_SCHEMA.TABLE_CONSTRAINTS'; +		if (isset($table->catalogName)) +		{ +			$kcu=$table->catalogName.'.'.$kcu; +			$tc=$table->catalogName.'.'.$tc; +		} + +		$sql = <<<EOD +		SELECT k.column_name field_name +			FROM {$this->quoteTableName($kcu)} k +		    LEFT JOIN {$this->quoteTableName($tc)} c +		      ON k.table_name = c.table_name +		     AND k.constraint_name = c.constraint_name +		   WHERE c.constraint_type ='PRIMARY KEY' +		   	    AND k.table_name = :table +				AND k.table_schema = :schema +EOD; +		$command = $this->getDbConnection()->createCommand($sql); +		$command->bindValue(':table', $table->name); +		$command->bindValue(':schema', $table->schemaName); +		$primary=$command->queryColumn(); +		switch (count($primary)) +		{ +			case 0: // No primary key on table +				$primary=null; +				break; +			case 1: // Only 1 primary key +				$primary=$primary[0]; +				break; +		} +		return $primary; +	} + +	/** +	 * Gets foreign relationship constraint keys and table name +	 * @param TMssqlTableSchema table +	 * @return array foreign relationship table name and keys. +	 */ +	protected function findForeignKeys($table) +	{ +		$rc='INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS'; +		$kcu='INFORMATION_SCHEMA.KEY_COLUMN_USAGE'; +		if (isset($table->catalogName)) +		{ +			$kcu=$table->catalogName.'.'.$kcu; +			$rc=$table->catalogName.'.'.$rc; +		} + +		//From http://msdn2.microsoft.com/en-us/library/aa175805(SQL.80).aspx +		$sql = <<<EOD +		SELECT +		     KCU1.CONSTRAINT_NAME AS 'FK_CONSTRAINT_NAME' +		   , KCU1.TABLE_NAME AS 'FK_TABLE_NAME' +		   , KCU1.COLUMN_NAME AS 'FK_COLUMN_NAME' +		   , KCU1.ORDINAL_POSITION AS 'FK_ORDINAL_POSITION' +		   , KCU2.CONSTRAINT_NAME AS 'UQ_CONSTRAINT_NAME' +		   , KCU2.TABLE_NAME AS 'UQ_TABLE_NAME' +		   , KCU2.COLUMN_NAME AS 'UQ_COLUMN_NAME' +		   , KCU2.ORDINAL_POSITION AS 'UQ_ORDINAL_POSITION' +		FROM {$this->quoteTableName($rc)} RC +		JOIN {$this->quoteTableName($kcu)} KCU1 +		ON KCU1.CONSTRAINT_CATALOG = RC.CONSTRAINT_CATALOG +		   AND KCU1.CONSTRAINT_SCHEMA = RC.CONSTRAINT_SCHEMA +		   AND KCU1.CONSTRAINT_NAME = RC.CONSTRAINT_NAME +		JOIN {$this->quoteTableName($kcu)} KCU2 +		ON KCU2.CONSTRAINT_CATALOG = +		RC.UNIQUE_CONSTRAINT_CATALOG +		   AND KCU2.CONSTRAINT_SCHEMA = +		RC.UNIQUE_CONSTRAINT_SCHEMA +		   AND KCU2.CONSTRAINT_NAME = +		RC.UNIQUE_CONSTRAINT_NAME +		   AND KCU2.ORDINAL_POSITION = KCU1.ORDINAL_POSITION +		WHERE KCU1.TABLE_NAME = :table +EOD; +		$command = $this->getDbConnection()->createCommand($sql); +		$command->bindValue(':table', $table->name); +		$fkeys=array(); +		foreach($command->queryAll() as $info) +		{ +			$fkeys[$info['FK_COLUMN_NAME']]=array($info['UQ_TABLE_NAME'],$info['UQ_COLUMN_NAME'],); + +		} +		return $fkeys; +	} + + +	/** +	 * Collects the table column metadata. +	 * @param CMysqlTableSchema the table metadata +	 * @return boolean whether the table exists in the database +	 */ +	protected function findColumns($table) +	{ +		$where=array(); +		$where[]="TABLE_NAME='".$table->name."'"; +		if (isset($table->catalogName)) +			$where[]="TABLE_CATALOG='".$table->catalogName."'"; +		if (isset($table->schemaName)) +			$where[]="TABLE_SCHEMA='".$table->schemaName."'"; +		$sql="SELECT *, columnproperty(object_id(table_schema+'.'+table_name), column_name, 'IsIdentity') as IsIdentity ". +			 "FROM INFORMATION_SCHEMA.COLUMNS WHERE ".join(' AND ',$where); +		if (($columns=$this->getDbConnection()->createCommand($sql)->queryAll())===array()) +			return false; + +		foreach($columns as $column) +		{ +			$c=$this->createColumn($column); +			if (is_array($table->primaryKey)) +				$c->isPrimaryKey=in_array($c->name, $table->primaryKey); +			else +				$c->isPrimaryKey=strcasecmp($c->name,$table->primaryKey)===0; + +			$c->isForeignKey=isset($table->foreignKeys[$c->name]); +			$table->columns[$c->name]=$c; +			if ($column['IsIdentity']==1 && $table->sequenceName===null) +				$table->sequenceName=''; + +		} +		return true; +	} + +	/** +	 * Creates a table column. +	 * @param array column metadata +	 * @return TDbColumnSchema normalized column metadata +	 */ +	protected function createColumn($column) +	{ +		$c=new TMssqlColumnSchema; +		$c->name=$column['COLUMN_NAME']; +		$c->rawName=$this->quoteColumnName($c->name); +		$c->allowNull=$column['IS_NULLABLE']=='YES'; +		if ($column['NUMERIC_PRECISION_RADIX']!==null) +		{ +			// We have a numeric datatype +			$c->size=$c->precision=$column['NUMERIC_PRECISION']!==null?(int)$column['NUMERIC_PRECISION']:null; +			$c->scale=$column['NUMERIC_SCALE']!==null?(int)$column['NUMERIC_SCALE']:null; +		} +		elseif ($column['DATA_TYPE']=='image' || $column['DATA_TYPE']=='text') +			$c->size=$c->precision=null; +		else +			$c->size=$c->precision=($column['CHARACTER_MAXIMUM_LENGTH']!== null)?(int)$column['CHARACTER_MAXIMUM_LENGTH']:null; + +		$c->init($column['DATA_TYPE'],$column['COLUMN_DEFAULT']); +		return $c; +	} + +	/** +	 * Returns all table names in the database. +	 * @return array all table names in the database. +	 * @since 1.0.4 +	 */ +	protected function findTableNames($schema='') +	{ +		if($schema==='') +			$schema=self::DEFAULT_SCHEMA; +		$sql=<<<EOD +SELECT TABLE_NAME, TABLE_SCHEMA FROM [INFORMATION_SCHEMA].[TABLES] +WHERE TABLE_TYPE='BASE TABLE' AND TABLE_SCHEMA=:schema +EOD; +		$command=$this->getDbConnection()->createCommand($sql); +		$command->bindParam(":schema", $schema); +		$rows=$command->queryAll(); +		$names=array(); +		foreach ($rows as $row) +		{ +			if ($schema == self::DEFAULT_SCHEMA) +				$names[]=$row['TABLE_NAME']; +			else +				$names[]=$schema.'.'.$row['TABLE_SCHEMA'].'.'.$row['TABLE_NAME']; +		} + +		return $names; +	} + +	/** +	 * Creates a command builder for the database. +	 * This method overrides parent implementation in order to create a MSSQL specific command builder +	 * @return TDbCommandBuilder command builder instance +	 */ +	protected function createCommandBuilder() +	{ +		return new TMssqlCommandBuilder($this); +	} +} diff --git a/framework/Testing/Data/Schema/mssql/TMssqlTableSchema.php b/framework/Testing/Data/Schema/mssql/TMssqlTableSchema.php new file mode 100755 index 00000000..b6584a32 --- /dev/null +++ b/framework/Testing/Data/Schema/mssql/TMssqlTableSchema.php @@ -0,0 +1,35 @@ +<?php +/** + * TMssqlTableSchema class file. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @link http://www.yiiframework.com/ + * @copyright Copyright © 2008-2009 Yii Software LLC + * @license http://www.yiiframework.com/license/ + */ + +prado::using('System.Testing.Data.TDbTableSchema'); + +/** + * TMssqlTableSchema represents the metadata for a MSSQL table. + * + * @author Qiang Xue <qiang.xue@gmail.com> + * @author Christophe Boulain <Christophe.Boulain@gmail.com> + * @version $Id: TMssqlTableSchema.php 2679 2009-06-15 07:49:42Z Christophe.Boulain $ + * @package System.Testing.Data.Schema.mssql + * @since 1.0.4 + */ +class TMssqlTableSchema extends TDbTableSchema +{ +	/** +	 * @var string name of the catalog (database) that this table belongs to. +	 * Defaults to null, meaning no schema (or the current database). +	 */ +	public $catalogName; +	/** +	 * @var string name of the schema that this table belongs to. +	 * Defaults to null, meaning no schema (or the current database owner). +	 */ +	public $schemaName; +}  | 
