241 lines
7.1 KiB
PHP
241 lines
7.1 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This file is part of the Propel package.
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*
|
|
* @license MIT License
|
|
*/
|
|
|
|
require_once 'reverse/BaseSchemaParser.php';
|
|
|
|
/**
|
|
* Microsoft SQL Server database schema parser.
|
|
*
|
|
* @author Hans Lellelid <hans@xmpl.org>
|
|
* @version $Revision: 1612 $
|
|
* @package propel.generator.reverse.mssql
|
|
*/
|
|
class MssqlSchemaParser extends BaseSchemaParser
|
|
{
|
|
|
|
/**
|
|
* Map MSSQL native types to Propel types.
|
|
* @var array
|
|
*/
|
|
private static $mssqlTypeMap = array(
|
|
"binary" => PropelTypes::BINARY,
|
|
"bit" => PropelTypes::BOOLEAN,
|
|
"char" => PropelTypes::CHAR,
|
|
"datetime" => PropelTypes::TIMESTAMP,
|
|
"decimal() identity" => PropelTypes::DECIMAL,
|
|
"decimal" => PropelTypes::DECIMAL,
|
|
"image" => PropelTypes::LONGVARBINARY,
|
|
"int" => PropelTypes::INTEGER,
|
|
"int identity" => PropelTypes::INTEGER,
|
|
"integer" => PropelTypes::INTEGER,
|
|
"money" => PropelTypes::DECIMAL,
|
|
"nchar" => PropelTypes::CHAR,
|
|
"ntext" => PropelTypes::LONGVARCHAR,
|
|
"numeric() identity" => PropelTypes::NUMERIC,
|
|
"numeric" => PropelTypes::NUMERIC,
|
|
"nvarchar" => PropelTypes::VARCHAR,
|
|
"real" => PropelTypes::REAL,
|
|
"float" => PropelTypes::FLOAT,
|
|
"smalldatetime" => PropelTypes::TIMESTAMP,
|
|
"smallint" => PropelTypes::SMALLINT,
|
|
"smallint identity" => PropelTypes::SMALLINT,
|
|
"smallmoney" => PropelTypes::DECIMAL,
|
|
"sysname" => PropelTypes::VARCHAR,
|
|
"text" => PropelTypes::LONGVARCHAR,
|
|
"timestamp" => PropelTypes::BINARY,
|
|
"tinyint identity" => PropelTypes::TINYINT,
|
|
"tinyint" => PropelTypes::TINYINT,
|
|
"uniqueidentifier" => PropelTypes::CHAR,
|
|
"varbinary" => PropelTypes::VARBINARY,
|
|
"varchar" => PropelTypes::VARCHAR,
|
|
"uniqueidentifier" => PropelTypes::CHAR,
|
|
// SQL Server 2000 only
|
|
"bigint identity" => PropelTypes::BIGINT,
|
|
"bigint" => PropelTypes::BIGINT,
|
|
"sql_variant" => PropelTypes::VARCHAR,
|
|
);
|
|
|
|
/**
|
|
* Gets a type mapping from native types to Propel types
|
|
*
|
|
* @return array
|
|
*/
|
|
protected function getTypeMapping()
|
|
{
|
|
return self::$mssqlTypeMap;
|
|
}
|
|
|
|
/**
|
|
*
|
|
*/
|
|
public function parse(Database $database, PDOTask $task = null)
|
|
{
|
|
$stmt = $this->dbh->query("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE = 'BASE TABLE' AND TABLE_NAME <> 'dtproperties'");
|
|
|
|
// First load the tables (important that this happen before filling out details of tables)
|
|
$tables = array();
|
|
while ($row = $stmt->fetch(PDO::FETCH_NUM)) {
|
|
$name = $row[0];
|
|
$table = new Table($name);
|
|
$database->addTable($table);
|
|
$tables[] = $table;
|
|
}
|
|
|
|
// Now populate only columns.
|
|
foreach ($tables as $table) {
|
|
$this->addColumns($table);
|
|
}
|
|
|
|
// Now add indexes and constraints.
|
|
foreach ($tables as $table) {
|
|
$this->addForeignKeys($table);
|
|
$this->addIndexes($table);
|
|
$this->addPrimaryKey($table);
|
|
}
|
|
|
|
return count($tables);
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* Adds Columns to the specified table.
|
|
*
|
|
* @param Table $table The Table model class to add columns to.
|
|
*/
|
|
protected function addColumns(Table $table)
|
|
{
|
|
$stmt = $this->dbh->query("sp_columns '" . $table->getName() . "'");
|
|
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
|
|
$name = $row['COLUMN_NAME'];
|
|
$type = $row['TYPE_NAME'];
|
|
$size = $row['LENGTH'];
|
|
$is_nullable = $row['NULLABLE'];
|
|
$default = $row['COLUMN_DEF'];
|
|
$precision = $row['PRECISION'];
|
|
$scale = $row['SCALE'];
|
|
$autoincrement = false;
|
|
if (strtolower($type) == "int identity") {
|
|
$autoincrement = true;
|
|
}
|
|
|
|
$propelType = $this->getMappedPropelType($type);
|
|
if (!$propelType) {
|
|
$propelType = Column::DEFAULT_TYPE;
|
|
$this->warn("Column [" . $table->getName() . "." . $name. "] has a column type (".$type.") that Propel does not support.");
|
|
}
|
|
|
|
$column = new Column($name);
|
|
$column->setTable($table);
|
|
$column->setDomainForType($propelType);
|
|
// We may want to provide an option to include this:
|
|
// $column->getDomain()->replaceSqlType($type);
|
|
$column->getDomain()->replaceSize($size);
|
|
$column->getDomain()->replaceScale($scale);
|
|
if ($default !== null) {
|
|
$column->getDomain()->setDefaultValue(new ColumnDefaultValue($default, ColumnDefaultValue::TYPE_VALUE));
|
|
}
|
|
$column->setAutoIncrement($autoincrement);
|
|
$column->setNotNull(!$is_nullable);
|
|
|
|
$table->addColumn($column);
|
|
}
|
|
|
|
|
|
} // addColumn()
|
|
|
|
/**
|
|
* Load foreign keys for this table.
|
|
*/
|
|
protected function addForeignKeys(Table $table)
|
|
{
|
|
$database = $table->getDatabase();
|
|
|
|
$stmt = $this->dbh->query("SELECT ccu1.TABLE_NAME, ccu1.COLUMN_NAME, ccu2.TABLE_NAME AS FK_TABLE_NAME, ccu2.COLUMN_NAME AS FK_COLUMN_NAME
|
|
FROM INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE ccu1 INNER JOIN
|
|
INFORMATION_SCHEMA.TABLE_CONSTRAINTS tc1 ON tc1.CONSTRAINT_NAME = ccu1.CONSTRAINT_NAME AND
|
|
CONSTRAINT_TYPE = 'Foreign Key' INNER JOIN
|
|
INFORMATION_SCHEMA.REFERENTIAL_CONSTRAINTS rc1 ON rc1.CONSTRAINT_NAME = tc1.CONSTRAINT_NAME INNER JOIN
|
|
INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE ccu2 ON ccu2.CONSTRAINT_NAME = rc1.UNIQUE_CONSTRAINT_NAME
|
|
WHERE (ccu1.table_name = '".$table->getName()."')");
|
|
|
|
$row = $stmt->fetch(PDO::FETCH_NUM);
|
|
|
|
$foreignKeys = array(); // local store to avoid duplicates
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
|
|
$lcol = $row['COLUMN_NAME'];
|
|
$ftbl = $row['FK_TABLE_NAME'];
|
|
$fcol = $row['FK_COLUMN_NAME'];
|
|
|
|
|
|
$foreignTable = $database->getTable($ftbl);
|
|
$foreignColumn = $foreignTable->getColumn($fcol);
|
|
$localColumn = $table->getColumn($lcol);
|
|
|
|
if (!isset($foreignKeys[$name])) {
|
|
$fk = new ForeignKey($name);
|
|
$fk->setForeignTableName($foreignTable->getName());
|
|
//$fk->setOnDelete($fkactions['ON DELETE']);
|
|
//$fk->setOnUpdate($fkactions['ON UPDATE']);
|
|
$table->addForeignKey($fk);
|
|
$foreignKeys[$name] = $fk;
|
|
}
|
|
$foreignKeys[$name]->addReference($localColumn, $foreignColumn);
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* Load indexes for this table
|
|
*/
|
|
protected function addIndexes(Table $table)
|
|
{
|
|
$stmt = $this->dbh->query("sp_indexes_rowset " . $table->getName());
|
|
|
|
$indexes = array();
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
$colName = $row["COLUMN_NAME"];
|
|
$name = $row['INDEX_NAME'];
|
|
|
|
// FIXME -- Add UNIQUE support
|
|
if (!isset($indexes[$name])) {
|
|
$indexes[$name] = new Index($name);
|
|
$table->addIndex($indexes[$name]);
|
|
}
|
|
|
|
$indexes[$name]->addColumn($table->getColumn($colName));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Loads the primary key for this table.
|
|
*/
|
|
protected function addPrimaryKey(Table $table)
|
|
{
|
|
$stmt = $this->dbh->query("SELECT COLUMN_NAME
|
|
FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS
|
|
INNER JOIN INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE ON
|
|
INFORMATION_SCHEMA.TABLE_CONSTRAINTS.CONSTRAINT_NAME = INFORMATION_SCHEMA.CONSTRAINT_COLUMN_USAGE.constraint_name
|
|
WHERE (INFORMATION_SCHEMA.TABLE_CONSTRAINTS.CONSTRAINT_TYPE = 'PRIMARY KEY') AND
|
|
(INFORMATION_SCHEMA.TABLE_CONSTRAINTS.TABLE_NAME = '".$table->getName()."')");
|
|
|
|
// Loop through the returned results, grouping the same key_name together
|
|
// adding each column for that key.
|
|
while ($row = $stmt->fetch(PDO::FETCH_NUM)) {
|
|
$name = $row[0];
|
|
$table->getColumn($name)->setPrimaryKey(true);
|
|
}
|
|
}
|
|
|
|
}
|