91 lines
2.3 KiB
PHP
91 lines
2.3 KiB
PHP
|
<?php declare(strict_types = 1);
|
||
|
|
|
||
|
|
namespace MailPoet\Migrator;
|
||
|
|
|
||
|
|
if (!defined('ABSPATH')) exit;
|
||
|
|
|
||
|
|
|
||
|
|
use MailPoet\Config\Env;
|
||
|
|
use MailPoetVendor\Doctrine\DBAL\Connection;
|
||
|
|
|
||
|
|
class Store {
|
||
|
|
/** @var Connection */
|
||
|
|
private $connection;
|
||
|
|
|
||
|
|
/** @var string */
|
||
|
|
private $table;
|
||
|
|
|
||
|
|
public function __construct(
|
||
|
|
Connection $connection
|
||
|
|
) {
|
||
|
|
$this->connection = $connection;
|
||
|
|
$this->table = Env::$dbPrefix . 'migrations';
|
||
|
|
}
|
||
|
|
|
||
|
|
public function getMigrationsTable(): string {
|
||
|
|
return $this->table;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function startMigration(string $name): void {
|
||
|
|
$this->connection->executeStatement("
|
||
|
|
INSERT INTO {$this->table} (name, started_at)
|
||
|
|
VALUES (?, current_timestamp())
|
||
|
|
ON DUPLICATE KEY UPDATE
|
||
|
|
started_at = current_timestamp(),
|
||
|
|
completed_at = NULL,
|
||
|
|
retries = retries + 1,
|
||
|
|
error = NULL
|
||
|
|
", [$name]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function completeMigration(string $name): void {
|
||
|
|
$this->connection->executeStatement("
|
||
|
|
UPDATE {$this->table}
|
||
|
|
SET completed_at = current_timestamp()
|
||
|
|
WHERE name = ?
|
||
|
|
", [$name]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function failMigration(string $name, string $error): void {
|
||
|
|
$this->connection->executeStatement("
|
||
|
|
UPDATE {$this->table}
|
||
|
|
SET
|
||
|
|
completed_at = current_timestamp(),
|
||
|
|
error = ?
|
||
|
|
WHERE name = ?
|
||
|
|
", [$error ?: 'Unknown error', $name]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function getAll(): array {
|
||
|
|
// Some backup plugins may convert NULL values to empty strings,
|
||
|
|
// in which case we need to cast the error column value to NULL.
|
||
|
|
return $this->connection->fetchAllAssociative("
|
||
|
|
SELECT
|
||
|
|
id,
|
||
|
|
name,
|
||
|
|
started_at,
|
||
|
|
completed_at,
|
||
|
|
retries,
|
||
|
|
IF(error = '', NULL, error) AS error
|
||
|
|
FROM {$this->table}
|
||
|
|
ORDER BY id ASC
|
||
|
|
");
|
||
|
|
}
|
||
|
|
|
||
|
|
public function ensureMigrationsTable(): void {
|
||
|
|
$collate = Env::$dbCharsetCollate;
|
||
|
|
$this->connection->executeStatement("
|
||
|
|
CREATE TABLE IF NOT EXISTS {$this->table} (
|
||
|
|
id int(11) unsigned NOT NULL AUTO_INCREMENT,
|
||
|
|
name varchar(191) NOT NULL,
|
||
|
|
started_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||
|
|
completed_at timestamp NULL,
|
||
|
|
retries int(11) unsigned NOT NULL DEFAULT 0,
|
||
|
|
error text NULL,
|
||
|
|
PRIMARY KEY (id),
|
||
|
|
UNIQUE KEY (name)
|
||
|
|
) {$collate};
|
||
|
|
");
|
||
|
|
}
|
||
|
|
}
|