DB2Connection.php 4.24 KB
Newer Older
1 2
<?php

Benjamin Eberlei's avatar
Benjamin Eberlei committed
3
namespace Doctrine\DBAL\Driver\IBMDB2;
4

5
use Doctrine\DBAL\Driver\ResultStatement;
6
use Doctrine\DBAL\Driver\ServerInfoAwareConnection;
7
use Doctrine\DBAL\Driver\Statement as DriverStatement;
8
use Doctrine\DBAL\ParameterType;
9
use stdClass;
10 11
use const DB2_AUTOCOMMIT_OFF;
use const DB2_AUTOCOMMIT_ON;
12
use function assert;
13 14 15 16 17 18 19 20 21 22 23 24 25 26
use function db2_autocommit;
use function db2_commit;
use function db2_conn_error;
use function db2_conn_errormsg;
use function db2_connect;
use function db2_escape_string;
use function db2_exec;
use function db2_last_insert_id;
use function db2_num_rows;
use function db2_pconnect;
use function db2_prepare;
use function db2_rollback;
use function db2_server_info;
use function db2_stmt_errormsg;
27
use function is_bool;
Steve Müller's avatar
Steve Müller committed
28

29
class DB2Connection implements ServerInfoAwareConnection
30
{
31
    /** @var resource */
32
    private $conn = null;
33

Benjamin Morel's avatar
Benjamin Morel committed
34
    /**
35 36 37 38
     * @param mixed[] $params
     * @param string  $username
     * @param string  $password
     * @param mixed[] $driverOptions
Benjamin Morel's avatar
Benjamin Morel committed
39
     *
40
     * @throws DB2Exception
Benjamin Morel's avatar
Benjamin Morel committed
41
     */
42
    public function __construct(array $params, $username, $password, $driverOptions = [])
43
    {
44
        $isPersistent = (isset($params['persistent']) && $params['persistent'] === true);
45

Possum's avatar
Possum committed
46
        if ($isPersistent) {
Sergei Morozov's avatar
Sergei Morozov committed
47
            $conn = db2_pconnect($params['dbname'], $username, $password, $driverOptions);
48
        } else {
Sergei Morozov's avatar
Sergei Morozov committed
49
            $conn = db2_connect($params['dbname'], $username, $password, $driverOptions);
50
        }
Sergei Morozov's avatar
Sergei Morozov committed
51 52

        if ($conn === false) {
Benjamin Eberlei's avatar
Benjamin Eberlei committed
53
            throw new DB2Exception(db2_conn_errormsg());
54
        }
Sergei Morozov's avatar
Sergei Morozov committed
55 56

        $this->conn = $conn;
57 58
    }

59 60 61 62 63
    /**
     * {@inheritdoc}
     */
    public function getServerVersion()
    {
64
        /** @var stdClass $serverInfo */
65
        $serverInfo = db2_server_info($this->conn);
66 67 68 69 70 71 72 73 74 75 76 77

        return $serverInfo->DBMS_VER;
    }

    /**
     * {@inheritdoc}
     */
    public function requiresQueryForServerVersion()
    {
        return false;
    }

Benjamin Morel's avatar
Benjamin Morel committed
78 79 80
    /**
     * {@inheritdoc}
     */
81
    public function prepare(string $sql) : DriverStatement
82
    {
83
        $stmt = @db2_prepare($this->conn, $sql);
84
        if ($stmt === false) {
Benjamin Eberlei's avatar
Benjamin Eberlei committed
85
            throw new DB2Exception(db2_stmt_errormsg());
86
        }
87

Benjamin Eberlei's avatar
Benjamin Eberlei committed
88
        return new DB2Statement($stmt);
89
    }
90

Benjamin Morel's avatar
Benjamin Morel committed
91 92 93
    /**
     * {@inheritdoc}
     */
94
    public function query(string $sql) : ResultStatement
95 96 97
    {
        $stmt = $this->prepare($sql);
        $stmt->execute();
98

99 100 101
        return $stmt;
    }

Benjamin Morel's avatar
Benjamin Morel committed
102 103 104
    /**
     * {@inheritdoc}
     */
105
    public function quote($input, $type = ParameterType::STRING)
106 107
    {
        $input = db2_escape_string($input);
108

Sergei Morozov's avatar
Sergei Morozov committed
109
        if ($type === ParameterType::INTEGER) {
110 111
            return $input;
        }
Gabriel Caruso's avatar
Gabriel Caruso committed
112

113
        return "'" . $input . "'";
114 115
    }

Benjamin Morel's avatar
Benjamin Morel committed
116 117 118
    /**
     * {@inheritdoc}
     */
119
    public function exec(string $statement) : int
120
    {
121
        $stmt = @db2_exec($this->conn, $statement);
122

123
        if ($stmt === false) {
124 125
            throw new DB2Exception(db2_stmt_errormsg());
        }
126

127
        return db2_num_rows($stmt);
128 129
    }

Benjamin Morel's avatar
Benjamin Morel committed
130 131 132
    /**
     * {@inheritdoc}
     */
133
    public function lastInsertId($name = null)
134
    {
135
        return db2_last_insert_id($this->conn);
136 137
    }

Benjamin Morel's avatar
Benjamin Morel committed
138 139 140
    /**
     * {@inheritdoc}
     */
141
    public function beginTransaction()
142
    {
143 144 145 146
        $result = db2_autocommit($this->conn, DB2_AUTOCOMMIT_OFF);
        assert(is_bool($result));

        return $result;
147 148
    }

Benjamin Morel's avatar
Benjamin Morel committed
149 150 151
    /**
     * {@inheritdoc}
     */
152
    public function commit()
153
    {
154 155
        if (! db2_commit($this->conn)) {
            throw new DB2Exception(db2_conn_errormsg($this->conn));
156
        }
157 158 159 160 161

        $result = db2_autocommit($this->conn, DB2_AUTOCOMMIT_ON);
        assert(is_bool($result));

        return $result;
162 163
    }

Benjamin Morel's avatar
Benjamin Morel committed
164 165 166
    /**
     * {@inheritdoc}
     */
167
    public function rollBack()
168
    {
169 170
        if (! db2_rollback($this->conn)) {
            throw new DB2Exception(db2_conn_errormsg($this->conn));
171
        }
172 173 174 175 176

        $result = db2_autocommit($this->conn, DB2_AUTOCOMMIT_ON);
        assert(is_bool($result));

        return $result;
177 178
    }

Benjamin Morel's avatar
Benjamin Morel committed
179 180 181
    /**
     * {@inheritdoc}
     */
182
    public function errorCode()
183
    {
184
        return db2_conn_error($this->conn);
185 186
    }

Benjamin Morel's avatar
Benjamin Morel committed
187 188 189
    /**
     * {@inheritdoc}
     */
190
    public function errorInfo()
191
    {
192
        return [
193
            0 => db2_conn_errormsg($this->conn),
194
            1 => $this->errorCode(),
195
        ];
196
    }
197
}