Add ClearEntityManager

hlecorche · hlecorche · commit da52478d7482 · 2015-11-06T21:29:09.000+01:00
diff --git a/Doctrine/ClearEntityManager.php b/Doctrine/ClearEntityManager.php
@@ -0,0 +1,69 @@
+<?php
+
+/**
+ * This file is part of the EcommitUtilBundle package.
+ *
+ * (c) E-commit <contact@e-commit.fr>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Ecommit\UtilBundle\Doctrine;
+
+use Doctrine\Bundle\DoctrineBundle\Registry;
+
+class ClearEntityManager
+{
+    /**
+     * @var Registry
+     */
+    protected $doctrine;
+
+    /**
+     * @var array
+     */
+    protected $snapshots = array();
+
+    public function __construct(Registry $registry)
+    {
+        $this->doctrine = $registry;
+    }
+
+    /**
+     * EntityManager snapshot
+     * @param string|null $managerName
+     */
+    public function snapshotEntityManager($managerName = null)
+    {
+        if (null === $managerName) {
+            $managerName = $this->doctrine->getDefaultConnectionName();
+        }
+        $this->snapshots[$managerName] = $this->doctrine->getManager($managerName)->getUnitOfWork()->getIdentityMap();
+    }
+
+    /**
+     * Detach all objects in EntityManager persisted since snapshot
+     * @param string|null $managerName
+     */
+    public function clearEntityManager($managerName = null)
+    {
+        if (null === $managerName) {
+            $managerName = $this->doctrine->getDefaultConnectionName();
+        }
+        if (!array_key_exists($managerName, $this->snapshots)) {
+            throw new \Exception('The snapshot was not done');
+        }
+        $snapshot = $this->snapshots[$managerName];
+
+        $em = $this->doctrine->getManager($managerName);
+        $identityMap = $em->getUnitOfWork()->getIdentityMap();
+        foreach ($identityMap as $class => $objects) {
+            foreach ($objects as $id => $object) {
+                if (!array_key_exists($class, $snapshot) || !array_key_exists($id, $snapshot[$class])) {
+                    $em->detach($object);
+                }
+            }
+        }
+    }
+}
diff --git a/Resources/config/services.xml b/Resources/config/services.xml
@@ -13,5 +13,9 @@
     <service id="ecommit_util.cache" class="Ecommit\UtilBundle\Cache\Cache" abstract="true" public="false">
          <argument /> <!-- Options -->
     </service>
+
+    <service id="ecommit_util.clear_entity_manager" class="Ecommit\UtilBundle\Doctrine\ClearEntityManager">
+         <argument type="service" id="doctrine" />
+    </service>
   </services>
 </container>
