Code Coverage for /workspace/src/includes/libs/rdbms/database/resultwrapper/SqliteResultWrapper.php

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