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