Merge pull request #44 from kinglozzer/generators

Update SQLite3Query to use generators
silverstripe · Mar 19, 2018 · 9c4de41 · 9c4de41
2 parents dbc0fb7 + 139c2b0
commit 9c4de41
Showing 1 changed file with 3 additions and 15 deletions.
diff --git a/code/SQLite3Query.php b/code/SQLite3Query.php
@@ -43,14 +43,11 @@ public function __destruct()
         }
     }
 
-    public function seek($row)
+    public function getIterator()
     {
-        $this->handle->reset();
-        $i=0;
-        while ($i <= $row && $result = @$this->handle->fetchArray(SQLITE3_ASSOC)) {
-            $i++;
+        while ($data = $this->handle->fetchArray(SQLITE3_ASSOC)) {
+            yield $data;
         }
-        return $result;
     }
 
     /**
@@ -66,13 +63,4 @@ public function numRecords()
         $this->handle->reset();
         return $c;
     }
-
-    public function nextRecord()
-    {
-        if ($data = $this->handle->fetchArray(SQLITE3_ASSOC)) {
-            return $data;
-        } else {
-            return false;
-        }
-    }
 }