ArrayStatement.php 2.88 KB
Newer Older
1 2 3 4
<?php

namespace Doctrine\DBAL\Cache;

5
use ArrayIterator;
6
use Doctrine\DBAL\Driver\ResultStatement;
7
use Doctrine\DBAL\FetchMode;
8 9 10
use InvalidArgumentException;
use IteratorAggregate;
use PDO;
11

12 13 14 15
use function array_merge;
use function array_values;
use function count;
use function reset;
16

17
class ArrayStatement implements IteratorAggregate, ResultStatement
18
{
19
    /** @var mixed[] */
20
    private $data;
Benjamin Morel's avatar
Benjamin Morel committed
21

22
    /** @var int */
23
    private $columnCount = 0;
Benjamin Morel's avatar
Benjamin Morel committed
24

25
    /** @var int */
26
    private $num = 0;
Benjamin Morel's avatar
Benjamin Morel committed
27

28
    /** @var int */
29
    private $defaultFetchMode = FetchMode::MIXED;
30

Benjamin Morel's avatar
Benjamin Morel committed
31
    /**
32
     * @param mixed[] $data
Benjamin Morel's avatar
Benjamin Morel committed
33
     */
34 35 36
    public function __construct(array $data)
    {
        $this->data = $data;
37 38
        if (! count($data)) {
            return;
39
        }
40 41

        $this->columnCount = count($data[0]);
42 43
    }

Benjamin Morel's avatar
Benjamin Morel committed
44 45 46
    /**
     * {@inheritdoc}
     */
47 48
    public function closeCursor()
    {
49
        unset($this->data);
50 51

        return true;
52 53
    }

Benjamin Morel's avatar
Benjamin Morel committed
54 55 56
    /**
     * {@inheritdoc}
     */
57 58 59 60 61
    public function columnCount()
    {
        return $this->columnCount;
    }

Benjamin Morel's avatar
Benjamin Morel committed
62 63 64
    /**
     * {@inheritdoc}
     */
65
    public function setFetchMode($fetchMode, $arg2 = null, $arg3 = null)
66
    {
67
        if ($arg2 !== null || $arg3 !== null) {
68
            throw new InvalidArgumentException('Caching layer does not support 2nd/3rd argument to setFetchMode()');
69 70
        }

71
        $this->defaultFetchMode = $fetchMode;
Benjamin Morel's avatar
Benjamin Morel committed
72 73

        return true;
74 75
    }

Benjamin Morel's avatar
Benjamin Morel committed
76 77 78
    /**
     * {@inheritdoc}
     */
79 80
    public function getIterator()
    {
81
        $data = $this->fetchAll();
Benjamin Morel's avatar
Benjamin Morel committed
82

83
        return new ArrayIterator($data);
84 85
    }

Benjamin Morel's avatar
Benjamin Morel committed
86 87 88
    /**
     * {@inheritdoc}
     */
89
    public function fetch($fetchMode = null, $cursorOrientation = PDO::FETCH_ORI_NEXT, $cursorOffset = 0)
90
    {
Sergei Morozov's avatar
Sergei Morozov committed
91
        if (! isset($this->data[$this->num])) {
92
            return false;
93
        }
Benjamin Morel's avatar
Benjamin Morel committed
94

95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
        $row       = $this->data[$this->num++];
        $fetchMode = $fetchMode ?: $this->defaultFetchMode;

        if ($fetchMode === FetchMode::ASSOCIATIVE) {
            return $row;
        }

        if ($fetchMode === FetchMode::NUMERIC) {
            return array_values($row);
        }

        if ($fetchMode === FetchMode::MIXED) {
            return array_merge($row, array_values($row));
        }

        if ($fetchMode === FetchMode::COLUMN) {
            return reset($row);
        }

114
        throw new InvalidArgumentException('Invalid fetch-style given for fetching result.');
115 116
    }

Benjamin Morel's avatar
Benjamin Morel committed
117 118 119
    /**
     * {@inheritdoc}
     */
120
    public function fetchAll($fetchMode = null, $fetchArgument = null, $ctorArgs = null)
121
    {
122
        $rows = [];
123
        while ($row = $this->fetch($fetchMode)) {
124 125
            $rows[] = $row;
        }
Benjamin Morel's avatar
Benjamin Morel committed
126

127 128 129
        return $rows;
    }

Benjamin Morel's avatar
Benjamin Morel committed
130 131 132
    /**
     * {@inheritdoc}
     */
133 134
    public function fetchColumn($columnIndex = 0)
    {
135
        $row = $this->fetch(FetchMode::NUMERIC);
Benjamin Morel's avatar
Benjamin Morel committed
136

137
        // TODO: verify that return false is the correct behavior
138
        return $row[$columnIndex] ?? false;
139
    }
140
}