120 lines
3 KiB
PHP
120 lines
3 KiB
PHP
<?php
|
|
|
|
namespace Appwrite\Migration;
|
|
|
|
use Appwrite\Database\Document;
|
|
use Appwrite\Database\Database;
|
|
use PDO;
|
|
use Swoole\Runtime;
|
|
use Utopia\CLI\Console;
|
|
use Utopia\Exception;
|
|
|
|
abstract class Migration
|
|
{
|
|
/**
|
|
* @var PDO
|
|
*/
|
|
protected PDO $db;
|
|
|
|
/**
|
|
* @var int
|
|
*/
|
|
protected int $limit = 50;
|
|
|
|
/**
|
|
* @var Document
|
|
*/
|
|
protected Document $project;
|
|
|
|
/**
|
|
* @var Database
|
|
*/
|
|
protected Database $projectDB;
|
|
|
|
/**
|
|
* Migration constructor.
|
|
*
|
|
* @param PDO $pdo
|
|
*/
|
|
public function __construct(PDO $db)
|
|
{
|
|
$this->db = $db;
|
|
}
|
|
|
|
/**
|
|
* Set project for migration.
|
|
*
|
|
* @param Document $project
|
|
* @param Database $projectDB
|
|
*
|
|
* @return Migration
|
|
*/
|
|
public function setProject(Document $project, Database $projectDB): Migration
|
|
{
|
|
$this->project = $project;
|
|
$this->projectDB = $projectDB;
|
|
$this->projectDB->setNamespace('app_' . $project->getId());
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Iterates through every document.
|
|
*
|
|
* @param callable $callback
|
|
*/
|
|
public function forEachDocument(callable $callback): void
|
|
{
|
|
$sum = $this->limit;
|
|
$offset = 0;
|
|
|
|
while ($sum >= $this->limit) {
|
|
$all = $this->projectDB->getCollection([
|
|
'limit' => $this->limit,
|
|
'offset' => $offset,
|
|
'orderType' => 'DESC',
|
|
]);
|
|
|
|
$sum = \count($all);
|
|
Runtime::setHookFlags(SWOOLE_HOOK_ALL);
|
|
|
|
Console::log('Migrating: ' . $offset . ' / ' . $this->projectDB->getSum());
|
|
\Co\run(function () use ($all, $callback) {
|
|
Runtime::enableCoroutine(SWOOLE_HOOK_ALL);
|
|
|
|
foreach ($all as $document) {
|
|
go(function () use ($document, $callback) {
|
|
|
|
$old = $document->getArrayCopy();
|
|
$new = call_user_func($callback, $document);
|
|
|
|
if (empty($new->getId())) {
|
|
throw new Exception('Missing ID');
|
|
}
|
|
if (!array_diff($new->getArrayCopy(), $old)) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
$new = $this->projectDB->overwriteDocument($document->getArrayCopy());
|
|
} catch (\Throwable $th) {
|
|
Console::error('Failed to update document: ' . $th->getMessage());
|
|
return;
|
|
|
|
if ($document && $new->getId() !== $document->getId()) {
|
|
throw new Exception('Duplication Error');
|
|
}
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
$offset += $this->limit;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Executes migration for set project.
|
|
*/
|
|
abstract public function execute(): void;
|
|
}
|