123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186 |
- <?php
-
- namespace Illuminate\Database\Connectors;
-
- use PDO;
-
- class MySqlConnector extends Connector implements ConnectorInterface
- {
- /**
- * Establish a database connection.
- *
- * @param array $config
- * @return \PDO
- */
- public function connect(array $config)
- {
- $dsn = $this->getDsn($config);
-
- $options = $this->getOptions($config);
-
- // We need to grab the PDO options that should be used while making the brand
- // new connection instance. The PDO options control various aspects of the
- // connection's behavior, and some might be specified by the developers.
- $connection = $this->createConnection($dsn, $config, $options);
-
- if (! empty($config['database'])) {
- $connection->exec("use `{$config['database']}`;");
- }
-
- $this->configureEncoding($connection, $config);
-
- // Next, we will check to see if a timezone has been specified in this config
- // and if it has we will issue a statement to modify the timezone with the
- // database. Setting this DB timezone is an optional configuration item.
- $this->configureTimezone($connection, $config);
-
- $this->setModes($connection, $config);
-
- return $connection;
- }
-
- /**
- * Set the connection character set and collation.
- *
- * @param \PDO $connection
- * @param array $config
- * @return void
- */
- protected function configureEncoding($connection, array $config)
- {
- if (! isset($config['charset'])) {
- return $connection;
- }
-
- $connection->prepare(
- "set names '{$config['charset']}'".$this->getCollation($config)
- )->execute();
- }
-
- /**
- * Get the collation for the configuration.
- *
- * @param array $config
- * @return string
- */
- protected function getCollation(array $config)
- {
- return isset($config['collation']) ? " collate '{$config['collation']}'" : '';
- }
-
- /**
- * Set the timezone on the connection.
- *
- * @param \PDO $connection
- * @param array $config
- * @return void
- */
- protected function configureTimezone($connection, array $config)
- {
- if (isset($config['timezone'])) {
- $connection->prepare('set time_zone="'.$config['timezone'].'"')->execute();
- }
- }
-
- /**
- * Create a DSN string from a configuration.
- *
- * Chooses socket or host/port based on the 'unix_socket' config value.
- *
- * @param array $config
- * @return string
- */
- protected function getDsn(array $config)
- {
- return $this->hasSocket($config)
- ? $this->getSocketDsn($config)
- : $this->getHostDsn($config);
- }
-
- /**
- * Determine if the given configuration array has a UNIX socket value.
- *
- * @param array $config
- * @return bool
- */
- protected function hasSocket(array $config)
- {
- return isset($config['unix_socket']) && ! empty($config['unix_socket']);
- }
-
- /**
- * Get the DSN string for a socket configuration.
- *
- * @param array $config
- * @return string
- */
- protected function getSocketDsn(array $config)
- {
- return "mysql:unix_socket={$config['unix_socket']};dbname={$config['database']}";
- }
-
- /**
- * Get the DSN string for a host / port configuration.
- *
- * @param array $config
- * @return string
- */
- protected function getHostDsn(array $config)
- {
- extract($config, EXTR_SKIP);
-
- return isset($port)
- ? "mysql:host={$host};port={$port};dbname={$database}"
- : "mysql:host={$host};dbname={$database}";
- }
-
- /**
- * Set the modes for the connection.
- *
- * @param \PDO $connection
- * @param array $config
- * @return void
- */
- protected function setModes(PDO $connection, array $config)
- {
- if (isset($config['modes'])) {
- $this->setCustomModes($connection, $config);
- } elseif (isset($config['strict'])) {
- if ($config['strict']) {
- $connection->prepare($this->strictMode($connection))->execute();
- } else {
- $connection->prepare("set session sql_mode='NO_ENGINE_SUBSTITUTION'")->execute();
- }
- }
- }
-
- /**
- * Set the custom modes on the connection.
- *
- * @param \PDO $connection
- * @param array $config
- * @return void
- */
- protected function setCustomModes(PDO $connection, array $config)
- {
- $modes = implode(',', $config['modes']);
-
- $connection->prepare("set session sql_mode='{$modes}'")->execute();
- }
-
- /**
- * Get the query to enable strict mode.
- *
- * @param \PDO $connection
- * @return string
- */
- protected function strictMode(PDO $connection)
- {
- if (version_compare($connection->getAttribute(PDO::ATTR_SERVER_VERSION), '8.0.11') >= 0) {
- return "set session sql_mode='ONLY_FULL_GROUP_BY,STRICT_TRANS_TABLES,NO_ZERO_IN_DATE,NO_ZERO_DATE,ERROR_FOR_DIVISION_BY_ZERO,NO_ENGINE_SUBSTITUTION'";
- }
-
- return "set session sql_mode='ONLY_FULL_GROUP_BY,STRICT_TRANS_TABLES,NO_ZERO_IN_DATE,NO_ZERO_DATE,ERROR_FOR_DIVISION_BY_ZERO,NO_AUTO_CREATE_USER,NO_ENGINE_SUBSTITUTION'";
- }
- }
|