123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- <?php
- /*
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * This software consists of voluntary contributions made by many individuals
- * and is licensed under the MIT license. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\DBAL\Cache;
- use Doctrine\DBAL\Driver\ResultStatement;
- use PDO;
- class ArrayStatement implements \IteratorAggregate, ResultStatement
- {
- /**
- * @var array
- */
- private $data;
- /**
- * @var integer
- */
- private $columnCount = 0;
- /**
- * @var integer
- */
- private $num = 0;
- /**
- * @var integer
- */
- private $defaultFetchMode = PDO::FETCH_BOTH;
- /**
- * @param array $data
- */
- public function __construct(array $data)
- {
- $this->data = $data;
- if (count($data)) {
- $this->columnCount = count($data[0]);
- }
- }
- /**
- * {@inheritdoc}
- */
- public function closeCursor()
- {
- unset ($this->data);
- }
- /**
- * {@inheritdoc}
- */
- public function columnCount()
- {
- return $this->columnCount;
- }
- /**
- * {@inheritdoc}
- */
- public function setFetchMode($fetchMode, $arg2 = null, $arg3 = null)
- {
- if ($arg2 !== null || $arg3 !== null) {
- throw new \InvalidArgumentException("Caching layer does not support 2nd/3rd argument to setFetchMode()");
- }
- $this->defaultFetchMode = $fetchMode;
- return true;
- }
- /**
- * {@inheritdoc}
- */
- public function getIterator()
- {
- $data = $this->fetchAll();
- return new \ArrayIterator($data);
- }
- /**
- * {@inheritdoc}
- */
- public function fetch($fetchMode = null)
- {
- if (isset($this->data[$this->num])) {
- $row = $this->data[$this->num++];
- $fetchMode = $fetchMode ?: $this->defaultFetchMode;
- if ($fetchMode === PDO::FETCH_ASSOC) {
- return $row;
- } else if ($fetchMode === PDO::FETCH_NUM) {
- return array_values($row);
- } else if ($fetchMode === PDO::FETCH_BOTH) {
- return array_merge($row, array_values($row));
- } else if ($fetchMode === PDO::FETCH_COLUMN) {
- return reset($row);
- } else {
- throw new \InvalidArgumentException("Invalid fetch-style given for fetching result.");
- }
- }
- return false;
- }
- /**
- * {@inheritdoc}
- */
- public function fetchAll($fetchMode = null)
- {
- $rows = array();
- while ($row = $this->fetch($fetchMode)) {
- $rows[] = $row;
- }
- return $rows;
- }
- /**
- * {@inheritdoc}
- */
- public function fetchColumn($columnIndex = 0)
- {
- $row = $this->fetch(PDO::FETCH_NUM);
- if (!isset($row[$columnIndex])) {
- // TODO: verify this is correct behavior
- return false;
- }
- return $row[$columnIndex];
- }
- }
|