From 25e57f4598d0d201431cff8e1fe3c6fda29a669d Mon Sep 17 00:00:00 2001 From: Loz Calver Date: Fri, 16 Mar 2018 16:43:07 +0000 Subject: [PATCH] Update PostgreSQLQuery to use generators --- code/PostgreSQLQuery.php | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/code/PostgreSQLQuery.php b/code/PostgreSQLQuery.php index fc7748f..8cf9696 100644 --- a/code/PostgreSQLQuery.php +++ b/code/PostgreSQLQuery.php @@ -34,19 +34,15 @@ class PostgreSQLQuery extends Query } } - 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); - } }