Completed
Pull Request — master (#3738)
by
unknown
61:48 queued 11s
created

SQLSrvStatement::columnCount()   A

Complexity

Conditions 3
Paths 2

Size

Total Lines 7
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 3
CRAP Score 4.125

Importance

Changes 0
Metric Value
eloc 3
dl 0
loc 7
ccs 3
cts 6
cp 0.5
rs 10
c 0
b 0
f 0
cc 3
nc 2
nop 0
crap 4.125
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Doctrine\DBAL\Driver\SQLSrv;
6
7
use Doctrine\DBAL\Driver\Statement;
8
use Doctrine\DBAL\Driver\StatementIterator;
9
use Doctrine\DBAL\Exception\InvalidColumnIndex;
10
use Doctrine\DBAL\FetchMode;
11
use Doctrine\DBAL\ParameterType;
12
use IteratorAggregate;
13
use const SQLSRV_ENC_BINARY;
14
use const SQLSRV_FETCH_ASSOC;
15
use const SQLSRV_FETCH_BOTH;
16
use const SQLSRV_FETCH_NUMERIC;
17
use const SQLSRV_PARAM_IN;
18
use function array_key_exists;
19
use function assert;
20
use function count;
21
use function in_array;
22
use function is_int;
23
use function sqlsrv_execute;
24
use function sqlsrv_fetch;
25
use function sqlsrv_fetch_array;
26
use function sqlsrv_fetch_object;
27
use function sqlsrv_get_field;
28
use function sqlsrv_next_result;
29
use function sqlsrv_num_fields;
30
use function SQLSRV_PHPTYPE_STREAM;
31
use function SQLSRV_PHPTYPE_STRING;
32
use function sqlsrv_prepare;
33
use function sqlsrv_rows_affected;
34
use function SQLSRV_SQLTYPE_VARBINARY;
35
use function stripos;
36
37
/**
38
 * SQL Server Statement.
39
 */
40
class SQLSrvStatement implements IteratorAggregate, Statement
41
{
42
    /**
43
     * The SQLSRV Resource.
44
     *
45
     * @var resource
46
     */
47
    private $conn;
48
49
    /**
50
     * The SQL statement to execute.
51
     *
52
     * @var string
53
     */
54
    private $sql;
55
56
    /**
57
     * The SQLSRV statement resource.
58
     *
59
     * @var resource|null
60
     */
61
    private $stmt;
62
63
    /**
64
     * References to the variables bound as statement parameters.
65
     *
66
     * @var mixed
67
     */
68
    private $variables = [];
69
70
    /**
71
     * Bound parameter types.
72
     *
73
     * @var int[]
74
     */
75
    private $types = [];
76
77
    /**
78
     * Translations.
79
     *
80
     * @var int[]
81
     */
82
    private static $fetchMap = [
83
        FetchMode::MIXED       => SQLSRV_FETCH_BOTH,
84
        FetchMode::ASSOCIATIVE => SQLSRV_FETCH_ASSOC,
85
        FetchMode::NUMERIC     => SQLSRV_FETCH_NUMERIC,
86
    ];
87
88
    /**
89
     * The name of the default class to instantiate when fetching class instances.
90
     *
91
     * @var string
92
     */
93
    private $defaultFetchClass = '\stdClass';
94
95
    /**
96
     * The constructor arguments for the default class to instantiate when fetching class instances.
97
     *
98
     * @var mixed[]
99
     */
100
    private $defaultFetchClassCtorArgs = [];
101
102
    /**
103
     * The fetch style.
104
     *
105
     * @var int
106
     */
107
    private $defaultFetchMode = FetchMode::MIXED;
108
109
    /**
110
     * The last insert ID.
111
     *
112
     * @var LastInsertId|null
113
     */
114
    private $lastInsertId;
115
116
    /**
117
     * Indicates whether the statement is in the state when fetching results is possible
118
     *
119
     * @var bool
120
     */
121
    private $result = false;
122
123
    /**
124
     * Append to any INSERT query to retrieve the last insert id.
125
     *
126
     * @deprecated This constant has been deprecated and will be made private in 3.0
127
     */
128
    public const LAST_INSERT_ID_SQL = ';SELECT SCOPE_IDENTITY() AS LastInsertId;';
129
130
    /**
131
     * @param resource $conn
132
     */
133 251
    public function __construct($conn, string $sql, ?LastInsertId $lastInsertId = null)
134
    {
135 251
        $this->conn = $conn;
136 251
        $this->sql  = $sql;
137
138 251
        if (stripos($sql, 'INSERT INTO ') !== 0) {
139 250
            return;
140
        }
141
142 251
        $this->sql         .= self::LAST_INSERT_ID_SQL;
0 ignored issues
show
Deprecated Code introduced by
The constant Doctrine\DBAL\Driver\SQL...ent::LAST_INSERT_ID_SQL has been deprecated: This constant has been deprecated and will be made private in 3.0 ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-deprecated  annotation

142
        $this->sql         .= /** @scrutinizer ignore-deprecated */ self::LAST_INSERT_ID_SQL;

This class constant has been deprecated. The supplier of the class has supplied an explanatory message.

The explanatory message should give you some clue as to whether and when the constant will be removed from the class and what other constant to use instead.

Loading history...
143 251
        $this->lastInsertId = $lastInsertId;
144 251
    }
145
146
    /**
147
     * {@inheritdoc}
148
     */
149 251
    public function bindValue($param, $value, int $type = ParameterType::STRING) : void
150
    {
151 251
        assert(is_int($param));
152
153 251
        $this->variables[$param] = $value;
154 251
        $this->types[$param]     = $type;
155 251
    }
156
157
    /**
158
     * {@inheritdoc}
159
     */
160 246
    public function bindParam($param, &$variable, int $type = ParameterType::STRING, ?int $length = null) : void
161
    {
162 246
        assert(is_int($param));
163
164 246
        $this->variables[$param] =& $variable;
165 246
        $this->types[$param]     = $type;
166
167
        // unset the statement resource if it exists as the new one will need to be bound to the new variable
168 246
        $this->stmt = null;
169 246
    }
170
171
    /**
172
     * {@inheritdoc}
173
     */
174 126
    public function closeCursor() : void
175
    {
176
        // not having the result means there's nothing to close
177 126
        if ($this->stmt === null || ! $this->result) {
178 43
            return;
179
        }
180
181
        // emulate it by fetching and discarding rows, similarly to what PDO does in this case
182
        // @link http://php.net/manual/en/pdostatement.closecursor.php
183
        // @link https://github.com/php/php-src/blob/php-7.0.11/ext/pdo/pdo_stmt.c#L2075
184
        // deliberately do not consider multiple result sets, since doctrine/dbal doesn't support them
185 126
        while (sqlsrv_fetch($this->stmt) !== false) {
186
        }
187
188 126
        $this->result = false;
189 126
    }
190
191
    /**
192
     * {@inheritdoc}
193
     */
194 126
    public function columnCount() : int
195
    {
196 126
        if ($this->stmt === null) {
197
            return 0;
198
        }
199
200 126
        return sqlsrv_num_fields($this->stmt) ?: 0;
201
    }
202
203
    /**
204
     * {@inheritdoc}
205
     */
206 251
    public function execute(?array $params = null) : void
207
    {
208 251
        if ($params) {
209 241
            $hasZeroIndex = array_key_exists(0, $params);
210
211 241
            foreach ($params as $key => $val) {
212 241
                if ($hasZeroIndex && is_int($key)) {
213 241
                    $this->bindValue($key + 1, $val);
214
                } else {
215
                    $this->bindValue($key, $val);
216
                }
217
            }
218
        }
219
220 251
        if (! $this->stmt) {
221 251
            $this->stmt = $this->prepare();
222
        }
223
224 251
        if (! sqlsrv_execute($this->stmt)) {
225
            throw SQLSrvException::fromSqlSrvErrors();
226
        }
227
228 251
        if ($this->lastInsertId) {
229 251
            sqlsrv_next_result($this->stmt);
230 251
            sqlsrv_fetch($this->stmt);
231 251
            $this->lastInsertId->setId(sqlsrv_get_field($this->stmt, 0));
232
        }
233
234 251
        $this->result = true;
235 251
    }
236
237
    /**
238
     * Prepares SQL Server statement resource
239
     *
240
     * @return resource
241
     *
242
     * @throws SQLSrvException
243
     */
244 251
    private function prepare()
245
    {
246 251
        $params = [];
247
248 251
        foreach ($this->variables as $column => &$variable) {
249 251
            switch ($this->types[$column]) {
250
                case ParameterType::LARGE_OBJECT:
251 251
                    $params[$column - 1] = [
252 251
                        &$variable,
253
                        SQLSRV_PARAM_IN,
254 251
                        SQLSRV_PHPTYPE_STREAM(SQLSRV_ENC_BINARY),
255 251
                        SQLSRV_SQLTYPE_VARBINARY('max'),
256
                    ];
257 251
                    break;
258
259
                case ParameterType::BINARY:
260 16
                    $params[$column - 1] = [
261 16
                        &$variable,
262
                        SQLSRV_PARAM_IN,
263 16
                        SQLSRV_PHPTYPE_STRING(SQLSRV_ENC_BINARY),
264
                    ];
265 16
                    break;
266
267
                default:
268 251
                    $params[$column - 1] =& $variable;
269 251
                    break;
270
            }
271
        }
272
273 251
        $stmt = sqlsrv_prepare($this->conn, $this->sql, $params);
274
275 251
        if (! $stmt) {
0 ignored issues
show
introduced by
$stmt is of type false|resource, thus it always evaluated to false.
Loading history...
276 155
            throw SQLSrvException::fromSqlSrvErrors();
277
        }
278
279 251
        return $stmt;
280
    }
281
282
    /**
283
     * {@inheritdoc}
284
     */
285 250
    public function setFetchMode(int $fetchMode, ...$args) : void
286
    {
287 250
        $this->defaultFetchMode = $fetchMode;
288
289 250
        if (isset($args[0])) {
290 163
            $this->defaultFetchClass = $args[0];
291
        }
292
293 250
        if (! isset($args[1])) {
294 250
            return;
295
        }
296
297
        $this->defaultFetchClassCtorArgs = (array) $args[1];
298
    }
299
300
    /**
301
     * {@inheritdoc}
302
     */
303 237
    public function getIterator()
304
    {
305 237
        return new StatementIterator($this);
306
    }
307
308
    /**
309
     * {@inheritdoc}
310
     *
311
     * @throws SQLSrvException
312
     */
313 250
    public function fetch(?int $fetchMode = null, ...$args)
314
    {
315
        // do not try fetching from the statement if it's not expected to contain result
316
        // in order to prevent exceptional situation
317 250
        if ($this->stmt === null || ! $this->result) {
318 46
            return false;
319
        }
320
321 250
        $fetchMode = $fetchMode ?: $this->defaultFetchMode;
322
323 250
        if ($fetchMode === FetchMode::COLUMN) {
324 37
            return $this->fetchColumn();
325
        }
326
327 250
        if (isset(self::$fetchMap[$fetchMode])) {
328 250
            return sqlsrv_fetch_array($this->stmt, self::$fetchMap[$fetchMode]) ?: false;
329
        }
330
331 166
        if (in_array($fetchMode, [FetchMode::STANDARD_OBJECT, FetchMode::CUSTOM_OBJECT], true)) {
332 166
            $className = $this->defaultFetchClass;
333 166
            $ctorArgs  = $this->defaultFetchClassCtorArgs;
334
335 166
            if (count($args) > 0) {
336 165
                $className = $args[0];
337 165
                $ctorArgs  = $args[1] ?? [];
338
            }
339
340 166
            return sqlsrv_fetch_object($this->stmt, $className, $ctorArgs) ?: false;
341
        }
342
343
        throw new SQLSrvException('Fetch mode is not supported.');
344
    }
345
346
    /**
347
     * {@inheritdoc}
348
     */
349 250
    public function fetchAll(?int $fetchMode = null, ...$args) : array
350
    {
351 250
        $rows = [];
352
353 250
        switch ($fetchMode) {
354
            case FetchMode::CUSTOM_OBJECT:
355 165
                while (($row = $this->fetch($fetchMode, ...$args)) !== false) {
356 165
                    $rows[] = $row;
357
                }
358 165
                break;
359
360
            case FetchMode::COLUMN:
361 250
                while (($row = $this->fetchColumn()) !== false) {
362 250
                    $rows[] = $row;
363
                }
364 250
                break;
365
366
            default:
367 239
                while (($row = $this->fetch($fetchMode)) !== false) {
368 239
                    $rows[] = $row;
369
                }
370
        }
371
372 250
        return $rows;
373
    }
374
375
    /**
376
     * {@inheritdoc}
377
     */
378 250
    public function fetchColumn(int $columnIndex = 0)
379
    {
380 250
        $row = $this->fetch(FetchMode::NUMERIC);
381
382 250
        if ($row === false) {
383 250
            return false;
384
        }
385
386 250
        if (! array_key_exists($columnIndex, $row)) {
0 ignored issues
show
Bug introduced by
It seems like $row can also be of type object; however, parameter $search of array_key_exists() does only seem to accept array, maybe add an additional type check? ( Ignorable by Annotation )

If this is a false-positive, you can also ignore this issue in your code via the ignore-type  annotation

386
        if (! array_key_exists($columnIndex, /** @scrutinizer ignore-type */ $row)) {
Loading history...
387 36
            throw InvalidColumnIndex::new($columnIndex, count($row));
388
        }
389
390 250
        return $row[$columnIndex];
391
    }
392
393
    /**
394
     * {@inheritdoc}
395
     */
396 251
    public function rowCount() : int
397
    {
398 251
        if ($this->stmt === null) {
399
            return 0;
400
        }
401
402 251
        return sqlsrv_rows_affected($this->stmt) ?: 0;
403
    }
404
}
405