CodeMirror/appwrite
1
0
Fork 0
mirror of synced 2024-07-03 13:41:01 +12:00
Code Issues Projects Releases Wiki Activity

Add dynamic queue setting for database worker

Browse source
This commit is contained in:
Jake Barnby 2023-10-11 16:52:58 +13:00
parent 2cf0ed1d69
commit ca5b7f5b16
No known key found for this signature in database
GPG key ID: C437A8CC85B96E9C
1 changed files with 39 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
app/workers/databases.php
View file

@ -4,6 +4,7 @@ use Appwrite\Event\Event;
use Appwrite\Extend\Exception;
use Appwrite\Messaging\Adapter\Realtime;
use Appwrite\Resque\Worker;
use Utopia\App;
use Utopia\CLI\Console;
use Utopia\Database\Database;
use Utopia\Database\Document;
@ -14,10 +15,43 @@ require_once __DIR__ . '/../init.php';
Console::title('Database V1 Worker');
Console::success(APP_NAME . ' database worker v1 has started' . "\n");
$table = new Swoole\Table(1);
$table->column('workerCount', Swoole\Table::TYPE_INT);
$table->create();
$table->set('databases', ['workerCount' => 0]);
$lock = new Swoole\Lock(SWOOLE_MUTEX);
class DatabaseV1 extends Worker
{
public function init(): void
{
global $table, $lock;
$dbQueues = App::getEnv('_APP_CONNECTIONS_DB_QUEUES');
if (empty($dbQueues)) {
$queue = 'v1-database';
} elseif (\str_contains($dbQueues, ',')) {
$dbQueues = \explode(',', $dbQueues);
$dbQueues = \array_map('trim', $dbQueues);
$dbQueues = \array_filter($dbQueues);
$dbQueues = \array_values($dbQueues);
$count = $table->get('databases', 'workerCount');
Console::log('Database worker count: ' . $count);
$queue = $dbQueues[$count];
} else {
$queue = \trim($dbQueues);
}
\putenv('QUEUE=' . $queue);
$lock->lock();
$table->incr('databases', 'workerCount');
$lock->unlock();
}
public function run(): void
@ -58,6 +92,11 @@ class DatabaseV1 extends Worker
public function shutdown(): void
{
global $table, $lock;
$lock->lock();
$table->decr('databases', 'workerCount');
$lock->unlock();
}
/**