RetailCRM-Mirror/doctrine2
1
0
Fork 0
mirror of synced 2024-12-14 23:26:04 +03:00
Code Issues Projects Releases Wiki Activity

Primary Keys as Foreign Keys - reverse engineering

Browse Source
This commit is contained in:
Rivaros 2012-02-13 11:35:55 +02:00 committed by Benjamin Eberlei
parent de5e4b0fdc
commit 06eed4cfda
3 changed files with 33 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Doctrine/ORM/Mapping/ClassMetadataInfo.php
View File

@ -1224,7 +1224,8 @@ class ClassMetadataInfo implements ClassMetadata
foreach ($mapping['joinColumns'] as $key => &$joinColumn) { foreach ($mapping['joinColumns'] as $key => &$joinColumn) {
if ($mapping['type'] === self::ONE_TO_ONE && ! $this->isInheritanceTypeSingleTable()) { if ($mapping['type'] === self::ONE_TO_ONE && ! $this->isInheritanceTypeSingleTable()) {
if (count($mapping['joinColumns']) == 1) { if (count($mapping['joinColumns']) == 1) {
$joinColumn['unique'] = true; if(! isset($mapping['id']) || ! $mapping['id'])
$joinColumn['unique'] = true;
} else { } else {
$uniqueContraintColumns[] = $joinColumn['name']; $uniqueContraintColumns[] = $joinColumn['name'];
} }

24
lib/Doctrine/ORM/Mapping/Driver/DatabaseDriver.php
View File

@ -192,10 +192,12 @@ class DatabaseDriver implements Driver
$fieldMappings = array(); $fieldMappings = array();
foreach ($columns as $column) { foreach ($columns as $column) {
$fieldMapping = array(); $fieldMapping = array();
if ($primaryKeyColumns && in_array($column->getName(), $primaryKeyColumns)) {
$fieldMapping['id'] = true;
} else if (in_array($column->getName(), $allForeignKeyColumns)) { if (in_array($column->getName(), $allForeignKeyColumns)) {
continue; continue;
} else if ($primaryKeyColumns && in_array($column->getName(), $primaryKeyColumns)) {
$fieldMapping['id'] = true;
} }
$fieldMapping['fieldName'] = $this->getFieldNameForColumn($tableName, $column->getName(), false); $fieldMapping['fieldName'] = $this->getFieldNameForColumn($tableName, $column->getName(), false);
@ -298,13 +300,27 @@ class DatabaseDriver implements Driver
$associationMapping['fieldName'] = $this->getFieldNameForColumn($tableName, $localColumn, true); $associationMapping['fieldName'] = $this->getFieldNameForColumn($tableName, $localColumn, true);
$associationMapping['targetEntity'] = $this->getClassNameForTable($foreignTable); $associationMapping['targetEntity'] = $this->getClassNameForTable($foreignTable);
if ($primaryKeyColumns && in_array($localColumn, $primaryKeyColumns))
{
$associationMapping['id'] = true;
}
for ($i = 0; $i < count($cols); $i++) { for ($i = 0; $i < count($cols); $i++) {
$associationMapping['joinColumns'][] = array( $associationMapping['joinColumns'][] = array(
'name' => $cols[$i], 'name' => $cols[$i],
'referencedColumnName' => $fkCols[$i], 'referencedColumnName' => $fkCols[$i],
); );
} }
$metadata->mapManyToOne($associationMapping);
//Here we need to check if $cols are the same as $primaryKeyColums
if(!array_diff($cols,$primaryKeyColumns))
$metadata->mapOneToOne($associationMapping);
else
$metadata->mapManyToOne($associationMapping);
} }
} }

10
lib/Doctrine/ORM/Tools/EntityGenerator.php
View File

@ -890,6 +890,16 @@ public function <methodName>()
if ($this->_generateAnnotations) { if ($this->_generateAnnotations) {
$lines[] = $this->_spaces . ' *'; $lines[] = $this->_spaces . ' *';
if (isset($associationMapping['id']) && $associationMapping['id']) {
$lines[] = $this->_spaces . ' * @' . $this->_annotationsPrefix . 'Id';
if ($generatorType = $this->_getIdGeneratorTypeString($metadata->generatorType)) {
$lines[] = $this->_spaces.' * @' . $this->_annotationsPrefix . 'GeneratedValue(strategy="' . $generatorType . '")';
}
}
$type = null; $type = null;
switch ($associationMapping['type']) { switch ($associationMapping['type']) {
case ClassMetadataInfo::ONE_TO_ONE: case ClassMetadataInfo::ONE_TO_ONE: