blob: df354af8ec7295181bf2efff7ed9d315676d6a45 (
plain)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
<?php
namespace Wikimedia\Rdbms;
use stdClass;
use RuntimeException;
/**
* Result wrapper for grabbing data queried from an IDatabase object
*
* Note that using the Iterator methods in combination with the non-Iterator
* DB result iteration functions may cause rows to be skipped or repeated.
*
* By default, this will use the iteration methods of the IDatabase handle if provided.
* Subclasses can override methods to make it solely work on the result resource instead.
* If no database is provided, and the subclass does not override the DB iteration methods,
* then a RuntimeException will be thrown when iteration is attempted.
*
* The result resource field should not be accessed from non-Database related classes.
* It is database class specific and is stored here to associate iterators with queries.
*
* @ingroup Database
*/
class ResultWrapper implements IResultWrapper {
/** @var resource|array|null Optional underlying result handle for subclass usage */
public $result;
/** @var IDatabase|null */
protected $db;
/** @var int */
protected $pos = 0;
/** @var stdClass|null */
protected $currentRow = null;
/**
* Create a row iterator from a result resource and an optional Database object
*
* Only Database-related classes should construct ResultWrapper. Other code may
* use the FakeResultWrapper subclass for convenience or compatibility shims, however.
*
* @param IDatabase|null $db Optional database handle
* @param ResultWrapper|array|resource $result Optional underlying result handle
*/
public function __construct( IDatabase $db = null, $result ) {
$this->db = $db;
if ( $result instanceof ResultWrapper ) {
$this->result = $result->result;
} else {
$this->result = $result;
}
}
public function numRows() {
return $this->getDB()->numRows( $this );
}
public function fetchObject() {
return $this->getDB()->fetchObject( $this );
}
public function fetchRow() {
return $this->getDB()->fetchRow( $this );
}
public function seek( $row ) {
$this->getDB()->dataSeek( $this, $row );
}
public function free() {
if ( $this->db ) {
$this->db->freeResult( $this );
$this->db = null;
}
$this->result = null;
}
/**
* @return IDatabase
* @throws RuntimeException
*/
private function getDB() {
if ( !$this->db ) {
throw new RuntimeException( static::class . ' needs a DB handle for iteration.' );
}
return $this->db;
}
function rewind() {
if ( $this->numRows() ) {
$this->getDB()->dataSeek( $this, 0 );
}
$this->pos = 0;
$this->currentRow = null;
}
function current() {
if ( is_null( $this->currentRow ) ) {
$this->next();
}
return $this->currentRow;
}
function key() {
return $this->pos;
}
function next() {
$this->pos++;
$this->currentRow = $this->fetchObject();
return $this->currentRow;
}
function valid() {
return $this->current() !== false;
}
}
class_alias( ResultWrapper::class, 'ResultWrapper' );
|