added Registry::getEntityManagerForObject() to conveniently get the e…

…ntity manager associated with a given Entity
symfony · Jul 6, 2011 · 090a51a · 090a51a
1 parent d3b7807
commit 090a51a
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/Symfony/Bridge/Doctrine/RegistryInterface.php b/src/Symfony/Bridge/Doctrine/RegistryInterface.php
@@ -123,4 +123,13 @@ function getEntityManagerNames();
      * @return Doctrine\ORM\EntityRepository
      */
     function getRepository($entityName, $entityManagerName = null);
+
+    /**
+     * Gets the entity manager associated with a given object.
+     *
+     * @param object $object A Doctrine Entity
+     *
+     * @return EntityManager|null
+     */
+    function getEntityManagerForObject($object);
 }
diff --git a/src/Symfony/Bundle/DoctrineBundle/Registry.php b/src/Symfony/Bundle/DoctrineBundle/Registry.php
@@ -216,4 +216,22 @@ public function getRepository($entityName, $entityManagerName = null)
     {
         return $this->getEntityManager($entityManagerName)->getRepository($entityName);
     }
+
+    /**
+     * Gets the entity manager associated with a given object.
+     *
+     * @param object $object A Doctrine Entity
+     *
+     * @return EntityManager|null
+     */
+    public function getEntityManagerForObject($object)
+    {
+        foreach ($this->entityManagers as $id) {
+            $em = $this->container->get($id);
+
+            if ($em->getConfiguration()->getMetadataDriverImpl()->isTransient($object)) {
+                return $em;
+            }
+        }
+    }
 }