Your IP : 216.73.216.95
<?php declare(strict_types = 1);
namespace MailPoet\Migrator;
if (!defined('ABSPATH')) exit;
use MailPoet\Config\Env;
use MailPoet\DI\ContainerWrapper;
use MailPoetVendor\Doctrine\DBAL\Connection;
use MailPoetVendor\Doctrine\ORM\EntityManager;
abstract class Migration {
/** @var ContainerWrapper */
protected $container;
/** @var Connection */
protected $connection;
/** @var EntityManager */
protected $entityManager;
public function __construct(
ContainerWrapper $container
) {
$this->container = $container;
$this->connection = $container->get(Connection::class);
$this->entityManager = $container->get(EntityManager::class);
}
abstract public function run(): void;
protected function getTableName(string $entityClass): string {
return $this->entityManager->getClassMetadata($entityClass)->getTableName();
}
protected function createTable(string $tableName, array $attributes): void {
$prefix = Env::$dbPrefix;
$charsetCollate = Env::$dbCharsetCollate;
$sql = implode(",\n", $attributes);
$this->connection->executeStatement("
CREATE TABLE IF NOT EXISTS {$prefix}{$tableName} (
$sql
) {$charsetCollate};
");
}
protected function columnExists(string $tableName, string $columnName): bool {
// We had a problem with the dbName value in ENV for some customers, because it doesn't match DB name in information schema.
// So we decided to use the DATABASE() value instead.
return $this->connection->executeQuery("
SELECT 1
FROM information_schema.columns
WHERE table_schema = COALESCE(DATABASE(), ?)
AND table_name = ?
AND column_name = ?
", [Env::$dbName, $tableName, $columnName])->fetchOne() !== false;
}
}