migrationRegister = $migrationRegister;
$this->migrationLogFactory = $migrationLogFactory;
$this->migrationLogRepository = $migrationLogRepository;
$this->completedMigrations = $this->migrationLogRepository->getCompletedMigrationsIDs();
}
/**
* Run database migrations.
*
* @since 4.0.0 add support for batch processing
* @since 2.9.0
*/
public function run()
{
if ( ! $this->hasMigrationToRun()) {
return;
}
// Stop Migration Runner if there are failed migrations
if ($this->migrationLogRepository->getFailedMigrationsCountByIds(
$this->migrationRegister->getRegisteredIds()
)) {
return;
}
$migrations = $this->migrationRegister->getMigrations();
foreach ($migrations as $migrationClass) {
$migrationId = $migrationClass::id();
if (in_array($migrationId, $this->completedMigrations, true)) {
continue;
}
$migrationLog = $this->migrationLogFactory->make($migrationId);
try {
/**
* @var Migration|BatchMigration $migration
*/
$migration = give($migrationClass);
if ($migration instanceof BatchMigration) {
$status = (new BatchMigrationRunner($migration))->run();
if ($status === MigrationLogStatus::RUNNING) {
give()->notices->register_notice(
[
'id' => $migrationId,
'description' => esc_html__('GiveWP is running database updates in the background. You will be notified as soon as it completes.',
'give'),
]
);
// Update status to RUNNING
if (MigrationLogStatus::RUNNING !== $migrationLog->getStatus()) {
$migrationLog->setStatus(MigrationLogStatus::RUNNING);
$migrationLog->save();
}
break;
}
if ($status === MigrationLogStatus::INCOMPLETE) {
$listTableLink = sprintf(
'%s',
admin_url('edit.php?post_type=give_forms&page=give-tools&tab=data'),
esc_html__('Resume update', 'give')
);
give()->notices->register_notice(
[
'id' => $migrationId,
'type' => 'warning',
'description' => sprintf(
__('Incomplete database update: "%s". %s', 'give'),
$migration::title(),
$listTableLink
),
]
);
}
$migrationLog->setStatus($status);
} else {
$migration->run();
$migrationLog->setStatus(MigrationLogStatus::SUCCESS);
}
} catch (Exception $exception) {
DB::rollback();
$migrationLog
->setStatus(MigrationLogStatus::FAILED)
->setError([
'status' => __('Migration failed', 'give'),
'error' => [
'message' => $exception->getMessage(),
'code' => $exception->getCode(),
'file' => $exception->getFile(),
'line' => $exception->getLine(),
],
]);
give()->notices->register_notice(
[
'id' => 'migration-failure',
'description' => sprintf(
'%1$s https://givewp.com/support',
esc_html__(
'There was a problem running the migrations. Please reach out to GiveWP support for assistance:',
'give'
)
),
]
);
}
try {
$migrationLog->save();
} catch (DatabaseQueryException $e) {
Log::error(
'Failed to save migration log',
[
'Error Message' => $e->getMessage(),
'Query Errors' => $e->getQueryErrors(),
]
);
}
// Stop Migration Runner if migration has failed
if ($migrationLog->getStatus() === MigrationLogStatus::FAILED) {
break;
}
// Commit transaction if successful
DB::commit();
}
}
/**
* Return whether or not all migrations completed.
*
* @since 2.9.0
*
* @return bool
*/
public function hasMigrationToRun()
{
return (bool)array_diff($this->migrationRegister->getRegisteredIds(), $this->completedMigrations);
}
}