103 lines
3.7 KiB
PHP
103 lines
3.7 KiB
PHP
|
<?php // phpcs:ignore SlevomatCodingStandard.TypeHints.DeclareStrictTypes.DeclareStrictTypesMissing
|
||
|
|
|
||
|
|
namespace MailPoet\Doctrine;
|
||
|
|
|
||
|
|
if (!defined('ABSPATH')) exit;
|
||
|
|
|
||
|
|
|
||
|
|
use MailPoet\Config\Env;
|
||
|
|
use MailPoet\Doctrine\Types\BigIntType;
|
||
|
|
use MailPoet\Doctrine\Types\DateTimeTzToStringType;
|
||
|
|
use MailPoet\Doctrine\Types\JsonOrSerializedType;
|
||
|
|
use MailPoet\Doctrine\Types\JsonType;
|
||
|
|
use MailPoet\Doctrine\Types\SerializedArrayType;
|
||
|
|
use MailPoetVendor\Doctrine\DBAL\Driver\PDO\MySQL\Driver;
|
||
|
|
use MailPoetVendor\Doctrine\DBAL\DriverManager;
|
||
|
|
use MailPoetVendor\Doctrine\DBAL\Platforms\MySqlPlatform;
|
||
|
|
use MailPoetVendor\Doctrine\DBAL\Types\Type;
|
||
|
|
use PDO;
|
||
|
|
|
||
|
|
class ConnectionFactory {
|
||
|
|
const DRIVER = 'pdo_mysql';
|
||
|
|
const PLATFORM_CLASS = MySqlPlatform::class;
|
||
|
|
|
||
|
|
private $minWaitTimeout = 60;
|
||
|
|
|
||
|
|
private $types = [
|
||
|
|
BigIntType::NAME => BigIntType::class,
|
||
|
|
DateTimeTzToStringType::NAME => DateTimeTzToStringType::class,
|
||
|
|
JsonType::NAME => JsonType::class,
|
||
|
|
JsonOrSerializedType::NAME => JsonOrSerializedType::class,
|
||
|
|
SerializedArrayType::NAME => SerializedArrayType::class,
|
||
|
|
];
|
||
|
|
|
||
|
|
public function createConnection() {
|
||
|
|
global $wpdb;
|
||
|
|
$platformClass = self::PLATFORM_CLASS;
|
||
|
|
$connectionParams = [
|
||
|
|
'wrapperClass' => SerializableConnection::class,
|
||
|
|
'driver' => self::DRIVER,
|
||
|
|
'driverClass' => Driver::class,
|
||
|
|
'platform' => new $platformClass,
|
||
|
|
'user' => Env::$dbUsername,
|
||
|
|
'password' => Env::$dbPassword,
|
||
|
|
'dbname' => Env::$dbName,
|
||
|
|
'driverOptions' => $this->getDriverOptions(Env::$dbTimezoneOffset, Env::$dbCharset, Env::$dbCollation),
|
||
|
|
];
|
||
|
|
|
||
|
|
if (!empty(Env::$dbCharset)) {
|
||
|
|
$connectionParams['charset'] = Env::$dbCharset;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!empty(Env::$dbSocket)) {
|
||
|
|
$connectionParams['unix_socket'] = Env::$dbSocket;
|
||
|
|
} else {
|
||
|
|
$connectionParams['host'] = Env::$dbIsIpv6 ? ('[' . Env::$dbHost . ']') : Env::$dbHost;
|
||
|
|
if (!empty(Env::$dbPort)) {
|
||
|
|
$connectionParams['port'] = Env::$dbPort;
|
||
|
|
} else {
|
||
|
|
$connectionParams['port'] = $wpdb->get_var('SELECT @@port');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->setupTypes();
|
||
|
|
return DriverManager::getConnection($connectionParams);
|
||
|
|
}
|
||
|
|
|
||
|
|
private function getDriverOptions($timezoneOffset, $charset, $collation) {
|
||
|
|
$driverOptions = [
|
||
|
|
"@@session.time_zone = '$timezoneOffset'",
|
||
|
|
"@@session.sql_mode = REPLACE(
|
||
|
|
REPLACE(
|
||
|
|
REPLACE(@@sql_mode, 'ONLY_FULL_GROUP_BY', ''),
|
||
|
|
'_QUOTES', ''
|
||
|
|
), 'ANSI', '')", // This is needed because ONLY_FULL_GROUP_BY mode in MariaDB is much more restrictive than in MySQL
|
||
|
|
// We need to use CONVERT for MySQL 8, Maria DB bug which triggers #1232 - Incorrect argument type to variable 'wait_timeout`
|
||
|
|
// https://stackoverflow.com/questions/35187378/mariadb-type-error-when-setting-session-variable
|
||
|
|
"@@session.wait_timeout = GREATEST(CONVERT(COALESCE(@@wait_timeout, 0), SIGNED), $this->minWaitTimeout)",
|
||
|
|
];
|
||
|
|
|
||
|
|
if (!empty(Env::$dbCharset)) {
|
||
|
|
$driverOptions[] = "NAMES $charset" . (empty($collation) ? '' : " COLLATE $collation");
|
||
|
|
}
|
||
|
|
|
||
|
|
return [
|
||
|
|
PDO::MYSQL_ATTR_INIT_COMMAND => 'SET ' . implode(', ', $driverOptions),
|
||
|
|
// In PHP 8.1 was changed MySQL behavior that numbers are returned as native PHP types instead of strings. https://www.php.net/manual/en/migration81.incompatible.php
|
||
|
|
// We force the previous state due to the backward compatibility of the Idiorm.
|
||
|
|
// This can be removed when we drop Idiorm.
|
||
|
|
PDO::ATTR_STRINGIFY_FETCHES => true,
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
private function setupTypes() {
|
||
|
|
foreach ($this->types as $name => $class) {
|
||
|
|
if (Type::hasType($name)) {
|
||
|
|
Type::overrideType($name, $class);
|
||
|
|
} else {
|
||
|
|
Type::addType($name, $class);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|