Merge pull request #84 from kinglozzer/generators

Update PostgreSQLQuery to use generators
silverstripe · Mar 17, 2018 · bb5bda6 · bb5bda6
2 parents 5779887 + 25e57f4
commit bb5bda6
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/code/PostgreSQLQuery.php b/code/PostgreSQLQuery.php
@@ -34,19 +34,15 @@ public function __destruct()
         }
     }
 
-    public function seek($row)
+    public function getIterator()
     {
-        pg_result_seek($this->handle, $row);
-        return pg_fetch_assoc($this->handle);
+        while ($data = pg_fetch_assoc($this->handle)) {
+            yield $data;
+        }
     }
 
     public function numRecords()
     {
         return pg_num_rows($this->handle);
     }
-
-    public function nextRecord()
-    {
-        return pg_fetch_assoc($this->handle);
-    }
 }