Add custom Doctrine Driver

[MAILPOET-3150]
2020-11-11 09:51:17 +01:00
parent ce27fd08fe
commit a751a2a459
8 changed files with 348 additions and 4 deletions
--- a/lib/Doctrine/Driver/PDOConnection.php
+++ b/lib/Doctrine/Driver/PDOConnection.php
@@ -0,0 +1,131 @@
+<?php
+
+namespace MailPoet\Doctrine\Driver;
+
+use MailPoet\InvalidStateException;
+
+class PDOConnection implements \MailPoetVendor\Doctrine\DBAL\Driver\Connection, \MailPoetVendor\Doctrine\DBAL\Driver\ServerInfoAwareConnection {
+  /** @var \PDO */
+  private $connection;
+
+  /**
+   * @param string $dsn
+   * @param string|null $user
+   * @param string|null $password
+   * @param array|null $options
+   */
+  public function __construct($dsn, $user = null, $password = null, array $options = null) {
+    try {
+      $this->connection = new \PDO($dsn, (string)$user, (string)$password, (array)$options);
+      $this->connection->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
+    } catch (\PDOException $exception) {
+      throw new \MailPoetVendor\Doctrine\DBAL\Driver\PDOException($exception);
+    }
+  }
+
+  public function getConnection(): \PDO {
+    return $this->connection;
+  }
+
+  /**
+   * {@inheritdoc}
+   * @return int|false
+   */
+  public function exec($statement) {
+    try {
+      return $this->connection->exec($statement);
+    } catch (\PDOException $exception) {
+      throw new \MailPoetVendor\Doctrine\DBAL\Driver\PDOException($exception);
+    }
+  }
+
+  public function getServerVersion() {
+    return $this->connection->getAttribute(\PDO::ATTR_SERVER_VERSION);
+  }
+
+  public function prepare($prepareString, $driverOptions = []) {
+    try {
+      return $this->createStatement(
+        $this->connection->prepare($prepareString, $driverOptions)
+      );
+    } catch (\PDOException $exception) {
+      throw new \MailPoetVendor\Doctrine\DBAL\Driver\PDOException($exception);
+    }
+  }
+
+  /**
+   * Creates a wrapped statement
+   */
+  protected function createStatement(\PDOStatement $stmt): PDOStatement {
+    return new PDOStatement($stmt);
+  }
+
+  public function query() {
+    $args = func_get_args();
+    $argsCount = count($args);
+
+    try {
+      if ($argsCount == 4) {
+        $stmt = $this->connection->query($args[0], $args[1], $args[2], $args[3]);
+      } elseif ($argsCount == 3) {
+        $stmt = $this->connection->query($args[0], $args[1], $args[2]);
+      } elseif ($argsCount == 2) {
+        $stmt = $this->connection->query($args[0], $args[1]);
+      } else {
+        $stmt = $this->connection->query($args[0]);
+      }
+
+      if ($stmt !== false) {
+        return $this->createStatement($stmt);
+      } else {
+        throw new InvalidStateException('Statement is missing');
+      }
+    } catch (\PDOException $exception) {
+      throw new \MailPoetVendor\Doctrine\DBAL\Driver\PDOException($exception);
+    }
+  }
+
+  public function quote($input, $type = \PDO::PARAM_STR) {
+    return $this->connection->quote($input, $type);
+  }
+
+  public function lastInsertId($name = null) {
+    try {
+      if ($name === null) {
+        return $this->connection->lastInsertId();
+      }
+
+      return $this->connection->lastInsertId($name);
+    } catch (\PDOException $exception) {
+      throw new \MailPoetVendor\Doctrine\DBAL\Driver\PDOException($exception);
+    }
+  }
+
+  public function getAttribute($attribute) {
+    return $this->connection->getAttribute($attribute);
+  }
+
+  public function requiresQueryForServerVersion() {
+    return false;
+  }
+
+  public function beginTransaction() {
+    return $this->connection->beginTransaction();
+  }
+
+  public function commit() {
+    return $this->connection->commit();
+  }
+
+  public function rollBack() {
+    return $this->connection->rollBack();
+  }
+
+  public function errorCode() {
+    return $this->connection->errorCode();
+  }
+
+  public function errorInfo() {
+    return $this->connection->errorInfo();
+  }
+}