appwrite/app/tasks/migrate.php

96 lines
3.1 KiB
PHP
Raw Normal View History

2019-12-31 10:55:39 +00:00
<?php
global $cli, $register;
2019-12-31 10:55:39 +00:00
use Utopia\CLI\Console;
2021-07-02 09:09:02 +00:00
use Appwrite\Migration\Migration;
2022-01-18 11:05:04 +00:00
use Utopia\App;
use Utopia\Cache\Cache;
use Utopia\Cache\Adapter\Redis as RedisCache;
use Utopia\Database\Adapter\MariaDB;
use Utopia\Database\Database;
2022-08-11 23:53:52 +00:00
use Utopia\Database\Query;
2022-01-18 11:05:04 +00:00
use Utopia\Database\Validator\Authorization;
2021-07-02 09:03:01 +00:00
use Utopia\Validator\Text;
2020-02-17 07:16:11 +00:00
2019-12-31 10:55:39 +00:00
$cli
2020-07-28 19:48:51 +00:00
->task('migrate')
2022-09-07 08:43:05 +00:00
->param('version', APP_VERSION_STABLE, new Text(32), 'Version to migrate to.', true)
2021-07-02 09:03:01 +00:00
->action(function ($version) use ($register) {
2021-12-08 17:49:44 +00:00
Authorization::disable();
2021-07-02 09:09:02 +00:00
if (!array_key_exists($version, Migration::$versions)) {
2021-07-02 09:03:01 +00:00
Console::error("Version {$version} not found.");
Console::exit(1);
return;
}
2021-12-08 17:49:44 +00:00
$app = new App('UTC');
2021-12-09 10:42:49 +00:00
Console::success('Starting Data Migration to version ' . $version);
2021-12-08 17:49:44 +00:00
2021-07-01 13:35:54 +00:00
$db = $register->get('db', true);
2022-06-22 13:52:21 +00:00
$redis = $register->get('cache', true);
$redis->flushAll();
$cache = new Cache(new RedisCache($redis));
2021-01-13 16:51:02 +00:00
2022-01-18 11:05:04 +00:00
$projectDB = new Database(new MariaDB($db), $cache);
$projectDB->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
2020-10-29 11:22:46 +00:00
2022-01-18 11:05:04 +00:00
$consoleDB = new Database(new MariaDB($db), $cache);
$consoleDB->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
$consoleDB->setNamespace('_project_console');
$console = $app->getResource('console');
2020-10-29 11:22:46 +00:00
2020-02-10 16:16:02 +00:00
$limit = 30;
$sum = 30;
$offset = 0;
2022-09-09 17:50:05 +00:00
/**
* @var \Utopia\Database\Document[] $projects
*/
2020-02-10 16:16:02 +00:00
$projects = [$console];
2020-10-29 11:22:46 +00:00
$count = 0;
2022-02-17 18:24:50 +00:00
try {
$totalProjects = $consoleDB->count('projects') + 1;
} catch (\Throwable $th) {
$consoleDB->setNamespace('_console');
$totalProjects = $consoleDB->count('projects') + 1;
}
2020-02-10 16:16:02 +00:00
2021-12-09 10:42:49 +00:00
$class = 'Appwrite\\Migration\\Version\\' . Migration::$versions[$version];
2022-01-18 11:05:04 +00:00
$migration = new $class();
2020-02-17 07:16:11 +00:00
2022-01-19 12:03:11 +00:00
while (!empty($projects)) {
2021-01-13 16:51:02 +00:00
foreach ($projects as $project) {
2022-09-09 17:50:05 +00:00
/**
* Skip user projects with id 'console'
*/
if ($project->getId() === 'console' && $project->getInternalId() !== 'console') {
continue;
}
2020-02-17 07:16:11 +00:00
try {
2021-01-13 16:51:02 +00:00
$migration
2021-12-08 17:49:44 +00:00
->setProject($project, $projectDB, $consoleDB)
2021-01-13 16:51:02 +00:00
->execute();
2020-02-17 07:16:11 +00:00
} catch (\Throwable $th) {
2020-10-29 11:22:46 +00:00
throw $th;
2021-12-09 10:42:49 +00:00
Console::error('Failed to update project ("' . $project->getId() . '") version with error: ' . $th->getMessage());
2020-02-17 07:16:11 +00:00
}
2020-02-10 16:16:02 +00:00
}
2020-06-20 11:20:49 +00:00
$sum = \count($projects);
2022-08-11 23:53:52 +00:00
$projects = $consoleDB->find('projects', [Query::limit($limit), Query::offset($offset)]);
2020-02-10 16:16:02 +00:00
$offset = $offset + $limit;
2020-10-29 11:22:46 +00:00
$count = $count + $sum;
2021-08-30 12:25:23 +00:00
2022-01-19 12:03:11 +00:00
Console::log('Migrated ' . $count . '/' . $totalProjects . ' projects...');
2020-02-10 16:16:02 +00:00
}
2020-05-21 06:25:29 +00:00
Swoole\Event::wait(); // Wait for Coroutines to finish
2022-06-22 13:52:21 +00:00
$redis->flushAll();
2020-05-21 06:25:29 +00:00
Console::success('Data Migration Completed');
2021-01-13 16:51:02 +00:00
});