Ability to customize the clear method

ddeboer · Mar 26, 2015 · 20e44dd · 20e44dd
1 parent 57c19a2
commit 20e44dd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Ddeboer/DataImport/Writer/DoctrineWriter.php b/src/Ddeboer/DataImport/Writer/DoctrineWriter.php
@@ -197,8 +197,7 @@ public function writeItem(array $item)
         $this->entityManager->persist($entity);
 
         if (($this->counter % $this->batchSize) == 0) {
-            $this->entityManager->flush();
-            $this->entityManager->clear($this->entityName);
+            $this->flushAndClear();
         }
 
         return $this;
@@ -318,4 +317,10 @@ protected function findOrCreateItem(array $item)
 
         return $entity;
     }
+
+    protected function flushAndClear()
+    {
+        $this->entityManager->flush();
+        $this->entityManager->clear($this->entityName);
+    }
 }