/
BufferedStatement.php
99 lines (80 loc) · 2.29 KB
/
BufferedStatement.php
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
<?php
/**
* PHP Version 5.4
*
* CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
* Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
*
* Licensed under The MIT License
* For full copyright and license information, please see the LICENSE.txt
* Redistributions of files must retain the above copyright notice.
*
* @copyright Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
* @link http://cakephp.org CakePHP(tm) Project
* @since CakePHP(tm) v 3.0.0
* @license MIT License (http://www.opensource.org/licenses/mit-license.php)
*/
namespace Cake\Database\Statement;
class BufferedStatement extends StatementDecorator {
protected $_count = 0;
protected $_records = [];
protected $_allFetched = true;
protected $_counter = 0;
/**
* Constructor
*
* @param Statement implementation such as PDOStatement
* @return void
*/
public function __construct($statement = null, $driver = null) {
parent::__construct($statement, $driver);
$this->_reset();
}
public function execute($params = null) {
$this->_reset();
return parent::execute($params);
}
public function fetch($type = 'num') {
if ($this->_allFetched) {
$row = ($this->_counter < $this->_count) ? $this->_records[$this->_counter++] : false;
$row = ($row && $type === 'num') ? array_values($row) : $row;
return $row;
}
$this->_fetchType = $type;
$record = parent::fetch($type);
if ($record === false) {
$this->_allFetched = true;
$this->_counter = $this->_count + 1;
$this->_statement->closeCursor();
return false;
}
$this->_count++;
return $this->_records[] = $record;
}
public function fetchAll($type = 'num') {
if ($this->_allFetched) {
return $this->_records;
}
$this->_records = parent::fetchAll($type);
$this->_count = count($this->_records);
$this->_allFetched = true;
$this->_statement->closeCursor();
return $this->_records;
}
public function rowCount() {
if (!$this->_allFetched) {
$counter = $this->_counter;
while($this->fetch('assoc'));
$this->_counter = $counter;
}
return $this->_count;
}
public function rewind() {
$this->_counter = 0;
}
protected function _reset() {
$this->_count = $this->_counter = 0;
$this->_records = [];
$this->_allFetched = false;
}
}