Merge pull request #7692 from appwrite/deletes-queue-retry

adding limit to queue retry
This commit is contained in:
Christy Jacob 2024-02-29 00:47:09 +05:30 committed by GitHub
commit ec85ec9f2a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -7,7 +7,9 @@ use Utopia\CLI\Console;
use Utopia\Platform\Action; use Utopia\Platform\Action;
use Utopia\Queue\Client; use Utopia\Queue\Client;
use Utopia\Queue\Connection; use Utopia\Queue\Connection;
use Utopia\Validator\Integer;
use Utopia\Validator\WhiteList; use Utopia\Validator\WhiteList;
use Utopia\Validator\Wildcard;
class QueueRetry extends Action class QueueRetry extends Action
{ {
@ -35,21 +37,25 @@ class QueueRetry extends Action
Event::MIGRATIONS_QUEUE_NAME, Event::MIGRATIONS_QUEUE_NAME,
Event::HAMSTER_CLASS_NAME Event::HAMSTER_CLASS_NAME
]), 'Queue name') ]), 'Queue name')
->param('limit', 0, new Wildcard(), 'jobs limit', true)
->inject('queue') ->inject('queue')
->callback(fn ($name, $queue) => $this->action($name, $queue)); ->callback(fn ($name, $limit, $queue) => $this->action($name, $limit, $queue));
} }
/** /**
* @param string $name The name of the queue to retry jobs from * @param string $name The name of the queue to retry jobs from
* @param mixed $limit
* @param Connection $queue * @param Connection $queue
*/ */
public function action(string $name, Connection $queue): void public function action(string $name, mixed $limit, Connection $queue): void
{ {
if (!$name) { if (!$name) {
Console::error('Missing required parameter $name'); Console::error('Missing required parameter $name');
return; return;
} }
$limit = (int)$limit;
$queueClient = new Client($name, $queue); $queueClient = new Client($name, $queue);
if ($queueClient->countFailedJobs() === 0) { if ($queueClient->countFailedJobs() === 0) {
@ -59,6 +65,6 @@ class QueueRetry extends Action
Console::log('Retrying failed jobs...'); Console::log('Retrying failed jobs...');
$queueClient->retry(); $queueClient->retry($limit);
} }
} }