tonyair
/
silverstripe-postgresql
mirror of https://github.com/silverstripe/silverstripe-postgresql
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #138 from creative-commoners/pulls/3/rescue-master-generators 

API Update PostgreSQLQuery to use generators
This commit is contained in:
Steve Boyd 2022-09-15 13:16:48 +12:00 committed by GitHub
parent 2946e07319 7962f7c919
commit c529b8375c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
code/PostgreSQLQuery.php
View File

@ -2,6 +2,7 @@
namespace SilverStripe\PostgreSQL; namespace SilverStripe\PostgreSQL;
use Iterator;
use SilverStripe\ORM\Connect\Query; 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 while ($row = pg_fetch_array($this->handle, null, PGSQL_NUM)) {
$result = pg_fetch_array($this->handle, $row, PGSQL_NUM); yield $this->parseResult($row);
}
return $this->parseResult($result);
} }
public function numRecords() public function numRecords()
@ -69,18 +69,6 @@ class PostgreSQLQuery extends Query
return pg_num_rows($this->handle); 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 * @param array $row
* @return array * @return array