SqliteResultWrapper.php

Go to the documentation of this file.

<?php
 
namespace Wikimedia\Rdbms;
 
use ArrayIterator;
use PDO;
use PDOStatement;
 
class SqliteResultWrapper extends ResultWrapper {
    private $result;
    private $rows;
 
    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 [];
        }
    }
}