EntityDao always flushes only it's entities (and relations, if needed)

Kdyby · Mar 7, 2013 · a142375 · a142375
1 parent b2b02f1
commit a142375
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/Kdyby/Doctrine/EntityDao.php b/src/Kdyby/Doctrine/EntityDao.php
@@ -18,6 +18,7 @@
 use Kdyby;
 use Kdyby\Persistence;
 use Nette;
+use Nette\Utils\Arrays;
 
 
 
@@ -105,7 +106,11 @@ public function delete($entity, $flush = Persistence\ObjectDao::FLUSH)
 	protected function flush($flush = Persistence\ObjectDao::FLUSH)
 	{
 		if ($flush === Persistence\ObjectDao::FLUSH) {
-			$this->getEntityManager()->flush();
+			$em = $this->getEntityManager();
+			$im = $em->getUnitOfWork()->getIdentityMap();
+			if (!empty($im[$this->_entityName])) {
+				$em->flush(Arrays::flatten($im[$this->_entityName]));
+			}
 		}
 	}
 

diff --git a/src/Kdyby/Doctrine/EntityManager.php b/src/Kdyby/Doctrine/EntityManager.php
@@ -22,6 +22,8 @@
 
 /**
  * @author Filip Procházka <filip@prochazka.su>
+ *
+ * @method flush(array $entity = NULL)
  */
 class EntityManager extends Doctrine\ORM\EntityManager
 {