File: SqliteResultWrapper.php

package info (click to toggle)
mediawiki 1%3A1.43.3%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 417,464 kB
  • sloc: php: 1,062,949; javascript: 664,290; sql: 9,714; python: 5,458; xml: 3,489; sh: 1,131; makefile: 64
file content (65 lines) | stat: -rw-r--r-- 1,385 bytes parent folder | download | duplicates (3)
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 [];
		}
	}
}