- entity manager single argument

- entity manager docs

Author: mdiyakov

Manager/EntityManager.php

@@ -30,53 +30,47 @@ public function __construct(
     }
 
     /**
-     * @param object[]|object $entity
+     * @param object $entity
      * @throws \Doctrine\ORM\ORMException
      * @throws \InvalidArgumentException
      * @throws \LogicException
      * @throws \Exception
      */
     public function flush($entity)
     {
-        if (!is_object($entity) && !is_array($entity)) {
-            throw new \InvalidArgumentException('Entity must be an object or array of objects');
+        if (!is_object($entity)) {
+            throw new \InvalidArgumentException('Entity must be an object');
         }
 
-        if (!is_array($entity)) {
-            $entity = [$entity];
+        if (!method_exists($entity, 'getId')) {
+            throw new \LogicException('Entity must have method "getId" to handle rollback');
         }
 
-        foreach ($entity as $object) {
-            if (!method_exists($object, 'getId')) {
-                throw new \LogicException('Entity must have method "getId" to handle rollback');
+        /** @var \Doctrine\ORM\EntityManager $em */
+        $em = $this->getEm($entity);
+        try {
+            $em->persist($entity);
+            $em->flush($entity);
+        } catch (\Exception $e) {
+            if (!$em->isOpen()) {
+                $em = $em->create(
+                    $em->getConnection(), $em->getConfiguration()
+                );
             }
 
-            /** @var \Doctrine\ORM\EntityManager $em */
-            $em = $this->getEm($object);
-            try {
-                $em->persist($object);
-                $em->flush($object);
-            } catch (\Exception $e) {
-                if (!$em->isOpen()) {
-                    $em = $em->create(
-                        $em->getConnection(), $em->getConfiguration()
-                    );
-                }
-
-                $previousObject = $em->getRepository(get_class($object))->find($object->getId());
-                if ($previousObject) {
-                    $this->indexProcessManager->reindex($previousObject);
-                } else {
-                    $this->indexProcessManager->remove($object);
-                }
-
-                throw $e;
+            $previousEntity = $em->getRepository(get_class($entity))->find($entity->getId());
+            if ($previousEntity) {
+                $this->indexProcessManager->reindex($previousEntity);
+            } else {
+                $this->indexProcessManager->remove($entity);
             }
+
+            throw $e;
         }
     }
 
     /**
-     * @param $entity
+     * @param object $entity
      * @throws \InvalidArgumentException
      * @return EntityManagerInterface
      */

Resources/doc/entity_manager.md

@@ -2,7 +2,7 @@
 
 **DoctrineSolrBundle** provides an entity manager (**\Mdiyakov\DoctrineSolrBundle\Manager\EntityManager**) to handle the case when an exception was triggered during flushing entity in database.
 
-During flushing an entity in database there is probability the exception will be thrown after the changes were successfully saved in solr. It can be caused by other entity listeners or something issues with database. In this case inconsistency can be broken beetwen database and solr because in database the changes will be rollback but in solr will be kept.
+During flushing an entity in database there is probability the exception will be thrown after the changes were successfully saved in solr. It can be caused by other entity listeners or something issues with database. In this case consistency can be broken beetwen database and solr because in database the changes will be rollback but in solr will be kept.
 
 To rollback the changes in solr in this case you can use the following approach:
 
@@ -14,18 +14,8 @@ $myEntity = new MyEntity();
  
 $dsEm ->flush($myEntity);
 ```
-In this case if any exception will be thrown in entity listener chain for example the solr changes also will be rollback.
+In this case if any exception will be thrown (for example in entity listener chain) the solr changes also will be rollback.
 
-You can use it with a few entities:
-```
-$dsEm = $this->get('ds.entity_manager');
-$myEntity = new MyEntity();
-$mySecondEntity = new MySecondEntity();
- ...
- 
-$dsEm ->flush([$myEntity,$mySecondEntity]);
-
-```
 
-> Pay attention 'ds.entity_manager' doesn't have interface like \Doctrine\ORM\EntityManager. It has only "flush"" method with mandatory argument
+> Pay attention 'ds.entity_manager' doesn't have interface like \Doctrine\ORM\EntityManager. It has only "flush"" method with mandatory argument. 
 > Also you don't need to call 'persist' method