1
0
Fork 0
mirror of synced 2024-09-20 19:42:42 +12:00

get project db key by region

This commit is contained in:
shimon 2024-07-30 18:41:15 +03:00
parent 998a017781
commit 67da5a8a4b

View file

@ -4,26 +4,18 @@ namespace Appwrite\Platform\Tasks;
use Appwrite\Migration\Migration; use Appwrite\Migration\Migration;
use Utopia\App; use Utopia\App;
use Utopia\Cache\Cache;
use Utopia\CLI\Console; use Utopia\CLI\Console;
use Utopia\Database\Database; use Utopia\Database\Database;
use Utopia\Database\Document; use Utopia\Database\Document;
use Utopia\Database\Query; use Utopia\Database\Query;
use Utopia\Database\Validator\Authorization; use Utopia\Database\Validator\Authorization;
use Utopia\Platform\Action; use Utopia\Platform\Action;
use Utopia\Pools\Group;
use Utopia\Registry\Registry; use Utopia\Registry\Registry;
use Utopia\Validator\Text; use Utopia\Validator\Text;
use Utopia\System\System;
use Redis;
class Migrate extends Action class Migrate extends Action
{ {
/**
* @var Redis
*/
protected Redis $redis;
public static function getName(): string public static function getName(): string
{ {
return 'migrate'; return 'migrate';
@ -35,44 +27,23 @@ class Migrate extends Action
->desc('Migrate Appwrite to new version') ->desc('Migrate Appwrite to new version')
/** @TODO APP_VERSION_STABLE needs to be defined */ /** @TODO APP_VERSION_STABLE needs to be defined */
->param('version', APP_VERSION_STABLE, new Text(8), 'Version to migrate to.', true) ->param('version', APP_VERSION_STABLE, new Text(8), 'Version to migrate to.', true)
->inject('cache')
->inject('dbForConsole') ->inject('dbForConsole')
->inject('getProjectDB') ->inject('getProjectDB')
->inject('register') ->inject('register')
->callback(fn($version, $dbForConsole, $getProjectDB, Registry $register) => $this->action($version, $dbForConsole, $getProjectDB, $register)); ->callback(fn ($version, $cache, $dbForConsole, $getProjectDB, Registry $register) => $this->action($version, $cache, $dbForConsole, $getProjectDB, $register));
$this->redis = new Redis();
$this->redis->connect(
System::getEnv('_APP_REDIS_HOST', null) ,
System::getEnv('_APP_REDIS_PORT', 6379),
3 ,
null,
10
);
} }
private function clearProjectsCache(Document $project) private function clearProjectsCache(Cache $cache, Document $project)
{ {
try { try {
do { $cache->purge("cache-_{$project->getInternalId()}:*");
$iterator = null;
$pattern = "default-cache-_{$project->getInternalId()}:*";
$keys = $this->redis->scan($iterator, $pattern, 1000);
if ($keys !== false) {
foreach ($keys as $key) {
$this->redis->del($key);
}
}
} while ($iterator > 0);
} catch (\Throwable $th) { } catch (\Throwable $th) {
Console::error('Failed to clear project ("' . $project->getId() . '") cache with error: ' . $th->getMessage()); Console::error('Failed to clear project ("' . $project->getId() . '") cache with error: ' . $th->getMessage());
} }
} }
public function action(string $version, Database $dbForConsole, callable $getProjectDB, Registry $register) public function action(string $version, Cache $cache, Database $dbForConsole, callable $getProjectDB, Registry $register)
{ {
Authorization::disable(); Authorization::disable();
if (!array_key_exists($version, Migration::$versions)) { if (!array_key_exists($version, Migration::$versions)) {
@ -116,7 +87,7 @@ class Migrate extends Action
continue; continue;
} }
$this->clearProjectsCache($project); $this->clearProjectsCache($cache, $project);
try { try {
// TODO: Iterate through all project DBs // TODO: Iterate through all project DBs
@ -132,7 +103,7 @@ class Migrate extends Action
throw $th; throw $th;
} }
$this->clearProjectsCache($project); $this->clearProjectsCache($cache, $project);
} }
$sum = \count($projects); $sum = \count($projects);