RetailCRM-Mirror/doctrine2
1
0
Fork 0
mirror of synced 2025-02-02 21:41:45 +03:00
Code Issues Projects Releases Wiki Activity

Fix issue when using notify tracking policy with multiple flush on entity

Browse Source
This commit is contained in:
xhuberty 2016-01-12 15:01:34 +01:00 committed by Marco Pivetta
parent 5770459bfc
commit 866a424963
No known key found for this signature in database
GPG Key ID: 4167D3337FD9D629
2 changed files with 67 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
lib/Doctrine/ORM/UnitOfWork.php
View File

@ -324,7 +324,7 @@ class UnitOfWork implements PropertyChangedListener
}
// Compute changes done since last commit.
if ($entity === null) {
if (null === $entity) {
$this->computeChangeSets();
} elseif (is_object($entity)) {
$this->computeSingleEntityChangeSet($entity);
@ -414,17 +414,40 @@ class UnitOfWork implements PropertyChangedListener
$this->dispatchPostFlushEvent();
$this->postCommitClear($entity);
}
/**
* @param null|object|array $entity
*/
private function postCommitClear($entity = null)
{
// Clear up
$this->entityInsertions =
$this->entityUpdates =
$this->entityDeletions =
$this->extraUpdates =
$this->entityChangeSets =
$this->collectionUpdates =
$this->collectionDeletions =
$this->visitedCollections =
$this->scheduledForSynchronization =
$this->orphanRemovals = [];
$this->orphanRemovals = array();
if (null === $entity) {
$this->entityChangeSets = $this->scheduledForSynchronization = [];
return;
}
if (is_object($entity)) {
$entity = [$entity];
}
foreach ($entity as $object) {
$oid = spl_object_hash($object);
$class = $this->em->getClassMetadata(get_class($object));
$this->clearEntityChangeSet($oid);
$this->clearScheduledForSynchronization($class, $oid);
}
}
/**
@ -3099,11 +3122,20 @@ class UnitOfWork implements PropertyChangedListener
*
* @return void
*/
public function clearEntityChangeSet($oid)
private function clearEntityChangeSet($oid)
{
$this->entityChangeSets[$oid] = [];
}
/**
* @param $class
* @param string $oid
*/
private function clearScheduledForSynchronization($class, $oid)
{
unset($this->scheduledForSynchronization[$class->rootEntityName][$oid]);
}
/* PropertyChangedListener implementation */
/**

31
tests/Doctrine/Tests/ORM/UnitOfWorkTest.php
View File

@ -163,7 +163,8 @@ class UnitOfWorkTest extends OrmTestCase
$this->_unitOfWork->persist($entity);
$this->_unitOfWork->commit();
$this->assertEquals(1, count($persister->getInserts()));
$this->assertCount(1, $persister->getInserts());
$persister->reset();
$this->assertTrue($this->_unitOfWork->isInIdentityMap($entity));
@ -360,6 +361,34 @@ class UnitOfWorkTest extends OrmTestCase
$this->assertFalse($this->_unitOfWork->isScheduledForInsert($entity2));
}
/**
* @group 5579
*/
public function testEntityChangeSetNotClearAfterFlushOnEntityOrArrayOfEntity()
{
// Create and Set first entity
$entity1 = new NotifyChangedEntity;
$entity1->setData('thedata');
$this->_unitOfWork->persist($entity1);
// Create and Set second entity
$entity2 = new NotifyChangedEntity;
$entity2->setData('thedata');
$this->_unitOfWork->persist($entity2);
$this->_unitOfWork->commit($entity1);
$this->assertCount(1, $this->_unitOfWork->getEntityChangeSet($entity2));
// Create and Set third entity
$entity3 = new NotifyChangedEntity;
$entity3->setData('thedata');
$this->_unitOfWork->persist($entity3);
$this->_unitOfWork->commit([$entity1,$entity2]);
$this->assertCount(1, $this->_unitOfWork->getEntityChangeSet($entity3));
}
/**
* Data Provider
*