76 lines
2.3 KiB
PHP
76 lines
2.3 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\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>
|
||
|
|
*/
|
||
|
|
class StopWorkersCommand extends Command
|
||
|
|
{
|
||
|
|
protected static $defaultName = 'messenger:stop-workers';
|
||
|
|
protected static $defaultDescription = 'Stop workers after their current message';
|
||
|
|
|
||
|
|
private $restartSignalCachePool;
|
||
|
|
|
||
|
|
public function __construct(CacheItemPoolInterface $restartSignalCachePool)
|
||
|
|
{
|
||
|
|
$this->restartSignalCachePool = $restartSignalCachePool;
|
||
|
|
|
||
|
|
parent::__construct();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* {@inheritdoc}
|
||
|
|
*/
|
||
|
|
protected function configure(): void
|
||
|
|
{
|
||
|
|
$this
|
||
|
|
->setDefinition([])
|
||
|
|
->setDescription(self::$defaultDescription)
|
||
|
|
->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
|
||
|
|
)
|
||
|
|
;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* {@inheritdoc}
|
||
|
|
*/
|
||
|
|
protected function execute(InputInterface $input, OutputInterface $output)
|
||
|
|
{
|
||
|
|
$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;
|
||
|
|
}
|
||
|
|
}
|