tonyair/silverstripe-sqlite3
1
0
Fork 0
mirror of https://github.com/silverstripe/silverstripe-sqlite3 synced 2024-10-22 15:05:37 +00:00
Code Issues Projects Releases Wiki Activity

Counting should not interfere with iterator

Browse Source
This commit is contained in:
Thomas Portelange 2023-10-24 17:54:01 +02:00 committed by GitHub
parent 2e61981c95
commit 2a85e21e35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
code/SQLite3Query.php
View File

@ -26,6 +26,8 @@ class SQLite3Query extends Query
*/
protected $handle;
protected int $count = 0;
/**
* Hook the result-set given into a Query class, suitable for use by framework.
* @param SQLite3Connector $database The database object that created this query.
@ -35,6 +37,8 @@ class SQLite3Query extends Query
{
$this->database = $database;
$this->handle = $handle;
// Count early to make sure we don't interfere with the generator and rewind operation
$this->count = $this->countRecords();
}
public function __destruct()
@ -44,10 +48,15 @@ class SQLite3Query extends Query
}
}
public function numRecords()
{
return $this->count;
}
/**
* @todo This looks terrible but there is no SQLite3::get_num_rows() implementation
*/
public function numRecords()
private function countRecords()
{
// Some queries are not iterable using fetchArray like CREATE statement
if (!$this->handle->numColumns()) {
@ -55,7 +64,7 @@ class SQLite3Query extends Query
}
$this->handle->reset();
$c=0;
$c = 0;
while ($this->handle->fetchArray()) {
$c++;
}