Compare commits

...

2 Commits

Author SHA1 Message Date
Steve Boyd c529b8375c
Merge pull request #138 from creative-commoners/pulls/3/rescue-master-generators
API Update PostgreSQLQuery to use generators
2022-09-15 13:16:48 +12:00
Loz Calver 7962f7c919
API Update PostgreSQLQuery to use generators 2022-09-02 13:37:04 +12:00
1 changed files with 5 additions and 17 deletions

View File

@ -2,6 +2,7 @@
namespace SilverStripe\PostgreSQL;
use Iterator;
use SilverStripe\ORM\Connect\Query;
/**
@ -56,12 +57,11 @@ class PostgreSQLQuery extends Query
}
}
public function seek($row)
public function getIterator(): Iterator
{
// Specifying the zero-th record here will reset the pointer
$result = pg_fetch_array($this->handle, $row, PGSQL_NUM);
return $this->parseResult($result);
while ($row = pg_fetch_array($this->handle, null, PGSQL_NUM)) {
yield $this->parseResult($row);
}
}
public function numRecords()
@ -69,18 +69,6 @@ class PostgreSQLQuery extends Query
return pg_num_rows($this->handle);
}
public function nextRecord()
{
$row = pg_fetch_array($this->handle, null, PGSQL_NUM);
// Correct non-string types
if ($row) {
return $this->parseResult($row);
}
return false;
}
/**
* @param array $row
* @return array