vendor/doctrine/dbal/src/Driver/PDO/PgSQL/Driver.php line 28

Open in your IDE?
  1. <?php
  3. namespace Doctrine\DBAL\Driver\PDO\PgSQL;
  5. use Doctrine\DBAL\Driver\AbstractPostgreSQLDriver;
  6. use Doctrine\DBAL\Driver\PDO\Connection;
  7. use Doctrine\DBAL\Driver\PDO\Exception;
  8. use PDO;
  9. use PDOException;
  11. final class Driver extends AbstractPostgreSQLDriver
  12. {
  13.     /**
  14.      * {@inheritdoc}
  15.      *
  16.      * @return Connection
  17.      */
  18.     public function connect(array $params)
  19.     {
  20.         $driverOptions = $params['driverOptions'] ?? [];
  22.         if (! empty($params['persistent'])) {
  23.             $driverOptions[PDO::ATTR_PERSISTENT] = true;
  24.         }
  26.         try {
  27.             $pdo = new PDO(
  28.                 $this->constructPdoDsn($params),
  29.                 $params['user'] ?? '',
  30.                 $params['password'] ?? '',
  31.                 $driverOptions,
  32.             );
  33.         } catch (PDOException $exception) {
  34.             throw Exception::new($exception);
  35.         }
  37.         if (
  38.             ! isset($driverOptions[PDO::PGSQL_ATTR_DISABLE_PREPARES])
  39.             || $driverOptions[PDO::PGSQL_ATTR_DISABLE_PREPARES] === true
  40.         ) {
  41.             $pdo->setAttribute(PDO::PGSQL_ATTR_DISABLE_PREPARES, true);
  42.         }
  44.         $connection = new Connection($pdo);
  46.         /* defining client_encoding via SET NAMES to avoid inconsistent DSN support
  47.          * - passing client_encoding via the 'options' param breaks pgbouncer support
  48.          */
  49.         if (isset($params['charset'])) {
  50.             $connection->exec('SET NAMES \'' . $params['charset'] . '\'');
  51.         }
  53.         return $connection;
  54.     }
  56.     /**
  57.      * Constructs the Postgres PDO DSN.
  58.      *
  59.      * @param mixed[] $params
  60.      */
  61.     private function constructPdoDsn(array $params): string
  62.     {
  63.         $dsn = 'pgsql:';
  65.         if (isset($params['host']) && $params['host'] !== '') {
  66.             $dsn .= 'host=' . $params['host'] . ';';
  67.         }
  69.         if (isset($params['port']) && $params['port'] !== '') {
  70.             $dsn .= 'port=' . $params['port'] . ';';
  71.         }
  73.         if (isset($params['dbname'])) {
  74.             $dsn .= 'dbname=' . $params['dbname'] . ';';
  75.         } elseif (isset($params['default_dbname'])) {
  76.             $dsn .= 'dbname=' . $params['default_dbname'] . ';';
  77.         } else {
  78.             // Used for temporary connections to allow operations like dropping the database currently connected to.
  79.             // Connecting without an explicit database does not work, therefore "postgres" database is used
  80.             // as it is mostly present in every server setup.
  81.             $dsn .= 'dbname=postgres;';
  82.         }
  84.         if (isset($params['sslmode'])) {
  85.             $dsn .= 'sslmode=' . $params['sslmode'] . ';';
  86.         }
  88.         if (isset($params['sslrootcert'])) {
  89.             $dsn .= 'sslrootcert=' . $params['sslrootcert'] . ';';
  90.         }
  92.         if (isset($params['sslcert'])) {
  93.             $dsn .= 'sslcert=' . $params['sslcert'] . ';';
  94.         }
  96.         if (isset($params['sslkey'])) {
  97.             $dsn .= 'sslkey=' . $params['sslkey'] . ';';
  98.         }
  100.         if (isset($params['sslcrl'])) {
  101.             $dsn .= 'sslcrl=' . $params['sslcrl'] . ';';
  102.         }
  104.         if (isset($params['application_name'])) {
  105.             $dsn .= 'application_name=' . $params['application_name'] . ';';
  106.         }
  108.         return $dsn;
  109.     }
  110. }