vendor/doctrine/dbal/lib/Doctrine/DBAL/Driver/PDOSqlite/Driver.php line 45

Open in your IDE?
  1. <?php
  3. namespace Doctrine\DBAL\Driver\PDOSqlite;
  5. use Doctrine\DBAL\DBALException;
  6. use Doctrine\DBAL\Driver\AbstractSQLiteDriver;
  7. use Doctrine\DBAL\Driver\PDOConnection;
  8. use Doctrine\DBAL\Platforms\SqlitePlatform;
  9. use PDOException;
  10. use function array_merge;
  12. /**
  13.  * The PDO Sqlite driver.
  14.  */
  15. class Driver extends AbstractSQLiteDriver
  16. {
  17.     /** @var mixed[] */
  18.     protected $_userDefinedFunctions = [
  19.         'sqrt' => ['callback' => [SqlitePlatform::class, 'udfSqrt'], 'numArgs' => 1],
  20.         'mod'  => ['callback' => [SqlitePlatform::class, 'udfMod'], 'numArgs' => 2],
  21.         'locate'  => ['callback' => [SqlitePlatform::class, 'udfLocate'], 'numArgs' => -1],
  22.     ];
  24.     /**
  25.      * {@inheritdoc}
  26.      */
  27.     public function connect(array $params, $username = null, $password = null, array $driverOptions = [])
  28.     {
  29.         if (isset($driverOptions['userDefinedFunctions'])) {
  30.             $this->_userDefinedFunctions = array_merge(
  31.                 $this->_userDefinedFunctions,
  32.                 $driverOptions['userDefinedFunctions']
  33.             );
  34.             unset($driverOptions['userDefinedFunctions']);
  35.         }
  37.         try {
  38.             $pdo = new PDOConnection(
  39.                 $this->_constructPdoDsn($params),
  40.                 $username,
  41.                 $password,
  42.                 $driverOptions
  43.             );
  44.         } catch (PDOException $ex) {
  45.             throw DBALException::driverException($this, $ex);
  46.         }
  48.         foreach ($this->_userDefinedFunctions as $fn => $data) {
  49.             $pdo->sqliteCreateFunction($fn, $data['callback'], $data['numArgs']);
  50.         }
  52.         return $pdo;
  53.     }
  55.     /**
  56.      * Constructs the Sqlite PDO DSN.
  57.      *
  58.      * @param mixed[] $params
  59.      *
  60.      * @return string The DSN.
  61.      */
  62.     protected function _constructPdoDsn(array $params)
  63.     {
  64.         $dsn = 'sqlite:';
  65.         if (isset($params['path'])) {
  66.             $dsn .= $params['path'];
  67.         } elseif (isset($params['memory'])) {
  68.             $dsn .= ':memory:';
  69.         }
  71.         return $dsn;
  72.     }
  74.     /**
  75.      * {@inheritdoc}
  76.      */
  77.     public function getName()
  78.     {
  79.         return 'pdo_sqlite';
  80.     }
  81. }