resurtm
12 years ago
5 changed files with 219 additions and 4 deletions
@ -0,0 +1,68 @@
|
||||
<?php |
||||
/** |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright (c) 2008 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\mssql; |
||||
|
||||
/** |
||||
* This is an extension of default PDO class for MSSQL and DBLIB drivers. It provides workaround for improperly |
||||
* implemented functionalities of the drivers. |
||||
* |
||||
* @author Timur Ruziev <qiang.xue@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class PDO extends \PDO |
||||
{ |
||||
/** |
||||
* Returns last inserted ID value. |
||||
* |
||||
* @param string|null sequence the sequence name. Defaults to null. |
||||
* @return integer last inserted ID value. |
||||
*/ |
||||
public function lastInsertId($sequence = null) |
||||
{ |
||||
return $this->query('SELECT CAST(COALESCE(SCOPE_IDENTITY(), @@IDENTITY) AS bigint)')->fetchColumn(); |
||||
} |
||||
|
||||
/** |
||||
* Begin a transaction. |
||||
* |
||||
* Is is necessary to override PDO's method as MSSQL PDO drivers does not support transactions. |
||||
* |
||||
* @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 transactions. |
||||
* |
||||
* @return boolean |
||||
*/ |
||||
public function commit() |
||||
{ |
||||
$this->exec('COMMIT TRANSACTION'); |
||||
return true; |
||||
} |
||||
|
||||
/** |
||||
* Rollback a transaction. |
||||
* |
||||
* Is is necessary to override PDO's method as MSSQL PDO drivers does not support transaction. |
||||
* |
||||
* @return boolean |
||||
*/ |
||||
public function rollBack() |
||||
{ |
||||
$this->exec('ROLLBACK TRANSACTION'); |
||||
return true; |
||||
} |
||||
} |
@ -0,0 +1,19 @@
|
||||
<?php |
||||
/** |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright (c) 2008 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\mssql; |
||||
|
||||
/** |
||||
* QueryBuilder is the query builder for MS SQL databases. |
||||
* |
||||
* @author Timur Ruziev <resurtm@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class QueryBuilder extends \yii\db\QueryBuilder |
||||
{ |
||||
// TODO: mssql driver |
||||
} |
@ -0,0 +1,93 @@
|
||||
<?php |
||||
/** |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright (c) 2008 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\mssql; |
||||
|
||||
use yii\db\TableSchema; |
||||
|
||||
/** |
||||
* Schema is the class for retrieving metadata from a MS SQL database (version 2008 and above). |
||||
* |
||||
* @author Timur Ruziev <qiang.xue@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class Schema extends \yii\db\Schema |
||||
{ |
||||
/** |
||||
* Default schema name to be used. |
||||
*/ |
||||
const DEFAULT_SCHEMA = 'dbo'; |
||||
|
||||
/** |
||||
* @var array mapping from physical column types (keys) to abstract column types (values) |
||||
*/ |
||||
public $typeMap = array( |
||||
// TODO: mssql driver |
||||
); |
||||
|
||||
/** |
||||
* @param string $name |
||||
* @return TableSchema |
||||
*/ |
||||
public function loadTableSchema($name) |
||||
{ |
||||
return null; |
||||
} |
||||
|
||||
/** |
||||
* Quotes a table name for use in a query. |
||||
* A simple table name has no schema prefix. |
||||
* @param string $name table name. |
||||
* @return string the properly quoted table name. |
||||
*/ |
||||
public function quoteSimpleTableName($name) |
||||
{ |
||||
return strpos($name, '[') !== false ? $name : '[' . $name . ']'; |
||||
} |
||||
|
||||
/** |
||||
* Quotes a column name for use in a query. |
||||
* A simple column name has no prefix. |
||||
* @param string $name column name. |
||||
* @return string the properly quoted column name. |
||||
*/ |
||||
public function quoteSimpleColumnName($name) |
||||
{ |
||||
return strpos($name, '[') !== false || $name === '*' ? $name : '[' . $name . ']'; |
||||
} |
||||
|
||||
/** |
||||
* Creates a query builder for the MSSQL database. |
||||
* @return QueryBuilder query builder interface. |
||||
*/ |
||||
public function createQueryBuilder() |
||||
{ |
||||
return new QueryBuilder($this->db); |
||||
} |
||||
|
||||
/** |
||||
* Returns all table names in the database. |
||||
* This method should be overridden by child classes in order to support this feature |
||||
* because the default implementation simply throws an exception. |
||||
* @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default schema. |
||||
* @return array all table names in the database. The names have NO the schema name prefix. |
||||
*/ |
||||
protected function findTableNames($schema = '') |
||||
{ |
||||
if ('' === $schema) { |
||||
$schema = self::DEFAULT_SCHEMA; |
||||
} |
||||
$sql = "SELECT TABLE_NAME FROM [INFORMATION_SCHEMA].[TABLES] WHERE TABLE_SCHEMA = :schema AND TABLE_TYPE = 'BASE TABLE'"; |
||||
$names = $this->db->createCommand($sql, array(':schema' => $schema))->queryColumn(); |
||||
if (self::DEFAULT_SCHEMA !== $schema) { |
||||
foreach ($names as $index => $name) { |
||||
$names[$index] = $schema . '.' . $name; |
||||
} |
||||
} |
||||
return $names; |
||||
} |
||||
} |
@ -0,0 +1,33 @@
|
||||
<?php |
||||
/** |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright (c) 2008 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\mssql; |
||||
|
||||
/** |
||||
* This is an extension of default PDO class for MSSQL SQLSRV driver. It provides workaround for improperly |
||||
* implemented functionalities of the PDO SQLSRV driver. |
||||
* |
||||
* @author Timur Ruziev <resurtm@gmail.com> |
||||
* @since 2.0 |
||||
*/ |
||||
class SqlsrvPDO extends \PDO |
||||
{ |
||||
/** |
||||
* Returns last inserted ID value. |
||||
* |
||||
* SQLSRV driver supports PDO::lastInsertId() with one peculiarity: when $sequence value is null |
||||
* or empty string it returns empty string. But when parameter is not specified it's working |
||||
* as expected and returns actual last inserted ID (like the other PDO drivers). |
||||
* |
||||
* @param string|null $sequence the sequence name. Defaults to null. |
||||
* @return integer last inserted ID value. |
||||
*/ |
||||
public function lastInsertId($sequence = null) |
||||
{ |
||||
return !$sequence ? parent::lastInsertId() : parent::lastInsertId($sequence); |
||||
} |
||||
} |
Loading…
Reference in new issue