Connection.php 1.28 KB
Newer Older
Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
1 2 3 4
<?php

namespace Doctrine\DBAL\Driver\PDOSqlsrv;

Steve Müller's avatar
Steve Müller committed
5
use Doctrine\DBAL\Driver\PDOConnection;
6
use Doctrine\DBAL\ParameterType;
7
use PDO;
8 9
use function strpos;
use function substr;
Steve Müller's avatar
Steve Müller committed
10

Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
11 12 13
/**
 * Sqlsrv Connection implementation.
 */
Steve Müller's avatar
Steve Müller committed
14
class Connection extends PDOConnection implements \Doctrine\DBAL\Driver\Connection
Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
15
{
16 17 18
    /**
     * {@inheritdoc}
     */
19
    public function __construct($dsn, $user = null, $password = null, ?array $options = null)
20 21
    {
        parent::__construct($dsn, $user, $password, $options);
22
        $this->setAttribute(PDO::ATTR_STATEMENT_CLASS, [Statement::class, []]);
23 24
    }

25 26 27 28 29
    /**
     * {@inheritDoc}
     */
    public function lastInsertId($name = null)
    {
30
        if ($name === null) {
31 32 33 34
            return parent::lastInsertId($name);
        }

        $stmt = $this->prepare('SELECT CONVERT(VARCHAR(MAX), current_value) FROM sys.sequences WHERE name = ?');
35
        $stmt->execute([$name]);
36 37 38 39

        return $stmt->fetchColumn();
    }

Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
40
    /**
41
     * {@inheritDoc}
Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
42
     */
43
    public function quote($value, $type = ParameterType::STRING)
Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
44 45
    {
        $val = parent::quote($value, $type);
46

Steve Müller's avatar
Steve Müller committed
47 48 49 50
        // Fix for a driver version terminating all values with null byte
        if (strpos($val, "\0") !== false) {
            $val = substr($val, 0, -1);
        }
51

Steve Müller's avatar
Steve Müller committed
52
        return $val;
Juozas Kaziukenas's avatar
Juozas Kaziukenas committed
53
    }
Benjamin Eberlei's avatar
Benjamin Eberlei committed
54
}