Completed
Pull Request — 2.10.x (#3936)
by Asmir
24:12
created

SQLSrvStatement::bindParam()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 13
Code Lines 6

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 6
CRAP Score 2.1481

Importance

Changes 1
Bugs 0 Features 0
Metric Value
eloc 6
c 1
b 0
f 0
dl 0
loc 13
ccs 6
cts 9
cp 0.6667
rs 10
cc 2
nc 2
nop 4
crap 2.1481
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 227
    public function __construct($conn, $sql, ?LastInsertId $lastInsertId = null)
136
    {
137 227
        $this->conn = $conn;
138 227
        $this->sql  = $sql;
139
140 227
        if (stripos($sql, 'INSERT INTO ') !== 0) {
141 226
            return;
142
        }
143
144 227
        $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 227
        $this->lastInsertId = $lastInsertId;
146 227
    }
147
148
    /**
149
     * {@inheritdoc}
150
     */
151 227
    public function bindValue($param, $value, $type = ParameterType::STRING)
152
    {
153 227
        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 227
        $this->variables[$param] = $value;
160 227
        $this->types[$param]     = $type;
161
162 227
        return true;
163
    }
164
165
    /**
166
     * {@inheritdoc}
167
     */
168 222
    public function bindParam($column, &$variable, $type = ParameterType::STRING, $length = null)
169
    {
170 222
        if (! is_numeric($column)) {
171
            throw new SQLSrvException('sqlsrv does not support named parameters to queries, use question mark (?) placeholders instead.');
172
        }
173
174 222
        $this->variables[$column] =& $variable;
175 222
        $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 222
        $this->stmt = null;
179
180 222
        return true;
181
    }
182
183
    /**
184
     * {@inheritdoc}
185
     */
186 135
    public function closeCursor()
187
    {
188
        // not having the result means there's nothing to close
189 135
        if ($this->stmt === null || ! $this->result) {
190 46
            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 135
        while (sqlsrv_fetch($this->stmt)) {
198
        }
199
200 135
        $this->result = false;
201
202 135
        return true;
203
    }
204
205
    /**
206
     * {@inheritdoc}
207
     */
208 135
    public function columnCount()
209
    {
210 135
        if ($this->stmt === null) {
211
            return 0;
212
        }
213
214 135
        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 227
    public function execute($params = null)
242
    {
243 227
        if ($params) {
244 211
            $hasZeroIndex = array_key_exists(0, $params);
245
246 211
            foreach ($params as $key => $val) {
247 211
                if ($hasZeroIndex && is_int($key)) {
248 211
                    $this->bindValue($key + 1, $val);
249
                } else {
250
                    $this->bindValue($key, $val);
251
                }
252
            }
253
        }
254
255 227
        if (! $this->stmt) {
256 227
            $this->stmt = $this->prepare();
257
        }
258
259 227
        if (! sqlsrv_execute($this->stmt)) {
260
            throw SQLSrvException::fromSqlSrvErrors();
261
        }
262
263 227
        if ($this->lastInsertId) {
264 227
            sqlsrv_next_result($this->stmt);
265 227
            sqlsrv_fetch($this->stmt);
266 227
            $this->lastInsertId->setId(sqlsrv_get_field($this->stmt, 0));
267
        }
268
269 227
        $this->result = true;
270
271 227
        return true;
272
    }
273
274
    /**
275
     * Prepares SQL Server statement resource
276
     *
277
     * @return resource
278
     *
279
     * @throws SQLSrvException
280
     */
281 227
    private function prepare()
282
    {
283 227
        $params = [];
284
285 227
        foreach ($this->variables as $column => &$variable) {
286 227
            switch ($this->types[$column]) {
287
                case ParameterType::LARGE_OBJECT:
288 227
                    $params[$column - 1] = [
289 227
                        &$variable,
290
                        SQLSRV_PARAM_IN,
291 227
                        SQLSRV_PHPTYPE_STREAM(SQLSRV_ENC_BINARY),
292 227
                        SQLSRV_SQLTYPE_VARBINARY('max'),
293
                    ];
294 227
                    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 227
                    $params[$column - 1] =& $variable;
306 227
                    break;
307
            }
308
        }
309
310 227
        $stmt = sqlsrv_prepare($this->conn, $this->sql, $params);
311
312 227
        if (! $stmt) {
0 ignored issues
show
introduced by
$stmt is of type false|resource, thus it always evaluated to false.
Loading history...
313 164
            throw SQLSrvException::fromSqlSrvErrors();
314
        }
315
316 227
        return $stmt;
317
    }
318
319
    /**
320
     * {@inheritdoc}
321
     */
322 226
    public function setFetchMode($fetchMode, $arg2 = null, $arg3 = null)
323
    {
324 226
        $this->defaultFetchMode          = $fetchMode;
325 226
        $this->defaultFetchClass         = $arg2 ?: $this->defaultFetchClass;
326 226
        $this->defaultFetchClassCtorArgs = $arg3 ? (array) $arg3 : $this->defaultFetchClassCtorArgs;
327
328 226
        return true;
329
    }
330
331
    /**
332
     * {@inheritdoc}
333
     */
334 213
    public function getIterator()
335
    {
336 213
        return new StatementIterator($this);
337
    }
338
339
    /**
340
     * {@inheritdoc}
341
     *
342
     * @throws SQLSrvException
343
     */
344 226
    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 226
        if ($this->stmt === null || ! $this->result) {
349 49
            return false;
350
        }
351
352 226
        $args      = func_get_args();
353 226
        $fetchMode = $fetchMode ?: $this->defaultFetchMode;
354
355 226
        if ($fetchMode === FetchMode::COLUMN) {
356 38
            return $this->fetchColumn();
357
        }
358
359 226
        if (isset(self::$fetchMap[$fetchMode])) {
360 226
            return sqlsrv_fetch_array($this->stmt, self::$fetchMap[$fetchMode]) ?: false;
361
        }
362
363 175
        if (in_array($fetchMode, [FetchMode::STANDARD_OBJECT, FetchMode::CUSTOM_OBJECT], true)) {
364 175
            $className = $this->defaultFetchClass;
365 175
            $ctorArgs  = $this->defaultFetchClassCtorArgs;
366
367 175
            if (count($args) >= 2) {
368 174
                $className = $args[1];
369 174
                $ctorArgs  = $args[2] ?? [];
370
            }
371
372 175
            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 226
    public function fetchAll($fetchMode = null, $fetchArgument = null, $ctorArgs = null)
382
    {
383 226
        $rows = [];
384
385 226
        switch ($fetchMode) {
386
            case FetchMode::CUSTOM_OBJECT:
387 174
                while (($row = $this->fetch(...func_get_args())) !== false) {
388 174
                    $rows[] = $row;
389
                }
390 174
                break;
391
392
            case FetchMode::COLUMN:
393 226
                while (($row = $this->fetchColumn()) !== false) {
394 226
                    $rows[] = $row;
395
                }
396 226
                break;
397
398
            default:
399 215
                while (($row = $this->fetch($fetchMode)) !== false) {
400 215
                    $rows[] = $row;
401
                }
402
        }
403
404 226
        return $rows;
405
    }
406
407
    /**
408
     * {@inheritdoc}
409
     */
410 226
    public function fetchColumn($columnIndex = 0)
411
    {
412 226
        $row = $this->fetch(FetchMode::NUMERIC);
413
414 226
        if ($row === false) {
415 226
            return false;
416
        }
417
418 226
        return $row[$columnIndex] ?? null;
419
    }
420
421
    /**
422
     * {@inheritdoc}
423
     */
424 227
    public function rowCount()
425
    {
426 227
        if ($this->stmt === null) {
427
            return 0;
428
        }
429
430 227
        return sqlsrv_rows_affected($this->stmt) ?: 0;
431
    }
432
}
433