Simplifying revert's code. (#16)

GetDKAN · Jan 11, 2020 · 44368ed · 44368ed
1 parent fc16170
commit 44368ed
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/Harvester.php b/src/Harvester.php
@@ -22,14 +22,18 @@ public function revert()
     {
         $ids = $this->factory->hashStorage->retrieveAll();
         $load = $this->factory->get("load");
+
+        if (!method_exists($load, "removeItem")) {
+            throw new \Exception("Load of class " . get_class($load) . " does not implement the removeItem method.");
+        }
+
         $counter = 0;
         foreach ($ids as $id) {
-            if (method_exists($load, "removeItem")) {
-                $load->removeItem($id);
-                $this->factory->hashStorage->remove($id);
-                $counter++;
-            }
+            $load->removeItem($id);
+            $this->factory->hashStorage->remove($id);
+            $counter++;
         }
+
         return $counter;
     }