Completed
Push — 2.10.x ( 6c789b...72d908 )
by Grégoire
22s queued 15s
created

SQLSrvStatement::errorCode()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 8
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 6

Importance

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

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