1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
<?php
namespace Wikimedia\Rdbms;
use ArrayIterator;
use PDO;
use PDOStatement;
class SqliteResultWrapper extends ResultWrapper {
/** @var PDOStatement|null */
private $result;
/** @var ArrayIterator|null */
private $rows;
/**
* @internal
* @param PDOStatement $result
*/
public function __construct( PDOStatement $result ) {
$this->result = $result;
// SQLite doesn't allow buffered results or data seeking etc, so we'll
// use fetchAll. PDO has PDO::CURSOR_SCROLL but the SQLite C API doesn't
// support it, so the driver raises an error if it is used.
$this->rows = $result->fetchAll( PDO::FETCH_OBJ );
}
protected function doNumRows() {
return count( $this->rows );
}
protected function doFetchObject() {
return $this->rows[$this->currentPos] ?? false;
}
protected function doFetchRow() {
$obj = $this->doFetchObject();
if ( is_object( $obj ) ) {
$i = 0;
$row = get_object_vars( $obj );
foreach ( $row as $value ) {
$row[$i++] = $value;
}
return $row;
} else {
return $obj;
}
}
protected function doSeek( $pos ) {
// Nothing to do -- parent updates $this->currentPos
}
protected function doFree() {
$this->rows = null;
$this->result = null;
}
protected function doGetFieldNames() {
if ( $this->rows ) {
return array_keys( get_object_vars( $this->rows[0] ) );
} else {
return [];
}
}
}
|