1
0
mirror of synced 2025-01-06 00:57:10 +03:00

Better code generation when association is nullable

This commit is contained in:
Asmir Mustafic 2011-10-21 09:38:37 +02:00
parent 596ba3d5b1
commit fe84a61d0b

View File

@ -634,7 +634,8 @@ public function <methodName>()
foreach ($metadata->associationMappings as $associationMapping) { foreach ($metadata->associationMappings as $associationMapping) {
if ($associationMapping['type'] & ClassMetadataInfo::TO_ONE) { if ($associationMapping['type'] & ClassMetadataInfo::TO_ONE) {
if ($code = $this->_generateEntityStubMethod($metadata, 'set', $associationMapping['fieldName'], $associationMapping['targetEntity'], 'null')) { $nullable = $this->_associationIsNullable($associationMapping);
if ($code = $this->_generateEntityStubMethod($metadata, 'set', $associationMapping['fieldName'], $associationMapping['targetEntity'], ($nullable?'null':null))) {
$methods[] = $code; $methods[] = $code;
} }
if ($code = $this->_generateEntityStubMethod($metadata, 'get', $associationMapping['fieldName'], $associationMapping['targetEntity'])) { if ($code = $this->_generateEntityStubMethod($metadata, 'get', $associationMapping['fieldName'], $associationMapping['targetEntity'])) {
@ -653,6 +654,24 @@ public function <methodName>()
return implode("\n\n", $methods); return implode("\n\n", $methods);
} }
private function _associationIsNullable($associationMapping)
{
if(isset($associationMapping['joinColumns'])){
$joinColumns = $associationMapping['joinColumns'];
}else{
//@todo thereis no way to retreive targetEntity metadata
//$targetMetadata = $this->getClassMetadata($associationMapping['targetEntity']);
//$joinColumns = $targetMetadata->associationMappings[$associationMapping["mappedBy"]]['joinColumns'];
$joinColumns = array();
}
foreach ($joinColumns as $joinColumn) {
if(!isset($joinColumn['nullable']) || !$joinColumn['nullable']){
return false;
}
}
return true;
}
private function _generateEntityLifecycleCallbackMethods(ClassMetadataInfo $metadata) private function _generateEntityLifecycleCallbackMethods(ClassMetadataInfo $metadata)
{ {
if (isset($metadata->lifecycleCallbacks) && $metadata->lifecycleCallbacks) { if (isset($metadata->lifecycleCallbacks) && $metadata->lifecycleCallbacks) {
@ -808,7 +827,7 @@ public function <methodName>()
$lines[] = $this->_spaces . '/**'; $lines[] = $this->_spaces . '/**';
if ($associationMapping['type'] & ClassMetadataInfo::TO_MANY) { if ($associationMapping['type'] & ClassMetadataInfo::TO_MANY) {
$lines[] = $this->_spaces . ' * @var \Doctrine\Common\Collections\Collection'; $lines[] = $this->_spaces . ' * @var \Doctrine\Common\Collections\ArrayCollection';
}else{ }else{
$lines[] = $this->_spaces . ' * @var ' . $associationMapping['targetEntity']; $lines[] = $this->_spaces . ' * @var ' . $associationMapping['targetEntity'];
} }