65 lines
2.1 KiB
PHP
65 lines
2.1 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\Messenger\Command;
|
|
|
|
use Psr\Cache\CacheItemPoolInterface;
|
|
use Symfony\Component\Console\Attribute\AsCommand;
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Output\ConsoleOutputInterface;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use Symfony\Component\Console\Style\SymfonyStyle;
|
|
use Symfony\Component\Messenger\EventListener\StopWorkerOnRestartSignalListener;
|
|
|
|
/**
|
|
* @author Ryan Weaver <ryan@symfonycasts.com>
|
|
*/
|
|
#[AsCommand(name: 'messenger:stop-workers', description: 'Stop workers after their current message')]
|
|
class StopWorkersCommand extends Command
|
|
{
|
|
public function __construct(
|
|
private CacheItemPoolInterface $restartSignalCachePool,
|
|
) {
|
|
parent::__construct();
|
|
}
|
|
|
|
protected function configure(): void
|
|
{
|
|
$this
|
|
->setDefinition([])
|
|
->setHelp(<<<'EOF'
|
|
The <info>%command.name%</info> command sends a signal to stop any <info>messenger:consume</info> processes that are running.
|
|
|
|
<info>php %command.full_name%</info>
|
|
|
|
Each worker command will finish the message they are currently processing
|
|
and then exit. Worker commands are *not* automatically restarted: that
|
|
should be handled by a process control system.
|
|
EOF
|
|
)
|
|
;
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output): int
|
|
{
|
|
$io = new SymfonyStyle($input, $output instanceof ConsoleOutputInterface ? $output->getErrorOutput() : $output);
|
|
|
|
$cacheItem = $this->restartSignalCachePool->getItem(StopWorkerOnRestartSignalListener::RESTART_REQUESTED_TIMESTAMP_KEY);
|
|
$cacheItem->set(microtime(true));
|
|
$this->restartSignalCachePool->save($cacheItem);
|
|
|
|
$io->success('Signal successfully sent to stop any running workers.');
|
|
|
|
return 0;
|
|
}
|
|
}
|