Skip to content

Instantly share code, notes, and snippets.

@meker12
Last active August 22, 2019 20:24
Show Gist options
  • Save meker12/2742e5553f22d4089234694cbf11d0f3 to your computer and use it in GitHub Desktop.
Save meker12/2742e5553f22d4089234694cbf11d0f3 to your computer and use it in GitHub Desktop.
Small edits to customer-facing message string for the Start MessageQueue consumer
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\MessageQueue\Console;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Magento\Framework\MessageQueue\ConsumerFactory;
use Magento\Framework\Lock\LockManagerInterface;
/**
* Command for starting MessageQueue consumers.
*/
class StartConsumerCommand extends Command
{
const ARGUMENT_CONSUMER = 'consumer';
const OPTION_NUMBER_OF_MESSAGES = 'max-messages';
const OPTION_BATCH_SIZE = 'batch-size';
const OPTION_AREACODE = 'area-code';
const OPTION_SINGLE_THREAD = 'single-thread';
const PID_FILE_PATH = 'pid-file-path';
const COMMAND_QUEUE_CONSUMERS_START = 'queue:consumers:start';
/**
* @var ConsumerFactory
*/
private $consumerFactory;
/**
* @var \Magento\Framework\App\State
*/
private $appState;
/**
* @var LockManagerInterface
*/
private $lockManager;
/**
* StartConsumerCommand constructor.
* {@inheritdoc}
*
* @param \Magento\Framework\App\State $appState
* @param ConsumerFactory $consumerFactory
* @param string $name
* @param LockManagerInterface $lockManager
*/
public function __construct(
\Magento\Framework\App\State $appState,
ConsumerFactory $consumerFactory,
$name = null,
LockManagerInterface $lockManager = null
) {
$this->appState = $appState;
$this->consumerFactory = $consumerFactory;
$this->lockManager = $lockManager ?: \Magento\Framework\App\ObjectManager::getInstance()
->get(LockManagerInterface::class);
parent::__construct($name);
}
/**
* @inheritdoc
*/
protected function execute(InputInterface $input, OutputInterface $output)
{
$consumerName = $input->getArgument(self::ARGUMENT_CONSUMER);
$numberOfMessages = $input->getOption(self::OPTION_NUMBER_OF_MESSAGES);
$batchSize = (int)$input->getOption(self::OPTION_BATCH_SIZE);
$areaCode = $input->getOption(self::OPTION_AREACODE);
if ($input->getOption(self::PID_FILE_PATH)) {
$input->setOption(self::OPTION_SINGLE_THREAD, true);
}
$singleThread = $input->getOption(self::OPTION_SINGLE_THREAD);
if ($singleThread && $this->lockManager->isLocked(md5($consumerName))) { //phpcs:ignore
$output->writeln('<error>Consumer with the same name is running</error>');
return \Magento\Framework\Console\Cli::RETURN_FAILURE;
}
if ($singleThread) {
$this->lockManager->lock(md5($consumerName)); //phpcs:ignore
}
$this->appState->setAreaCode($areaCode ?? 'global');
$consumer = $this->consumerFactory->get($consumerName, $batchSize);
$consumer->process($numberOfMessages);
if ($singleThread) {
$this->lockManager->unlock(md5($consumerName)); //phpcs:ignore
}
return \Magento\Framework\Console\Cli::RETURN_SUCCESS;
}
/**
* @inheritdoc
*/
protected function configure()
{
$this->setName(self::COMMAND_QUEUE_CONSUMERS_START);
$this->setDescription('Start MessageQueue consumer');
$this->addArgument(
self::ARGUMENT_CONSUMER,
InputArgument::REQUIRED,
'The name of the consumer to be started.'
);
$this->addOption(
self::OPTION_NUMBER_OF_MESSAGES,
null,
InputOption::VALUE_REQUIRED,
'The number of messages to be processed by the consumer before process termination. '
. 'If not specified - terminate after processing all queued messages.'
);
$this->addOption(
self::OPTION_BATCH_SIZE,
null,
InputOption::VALUE_REQUIRED,
'The number of messages per batch. Applicable for the batch consumer only.'
);
$this->addOption(
self::OPTION_AREACODE,
null,
InputOption::VALUE_REQUIRED,
'The preferred area (global, adminhtml, etc...) '
. 'default is global.'
);
$this->addOption(
self::OPTION_SINGLE_THREAD,
null,
InputOption::VALUE_NONE,
'This option prevents running multiple copies of one consumer simultaneously.'
);
$this->addOption(
self::PID_FILE_PATH,
null,
InputOption::VALUE_REQUIRED,
'This option is deprecated. Use the `--single-thread` option instead.'
);
$this->setHelp(
<<<HELP
This command starts MessageQueue consumer by its name.
To start consumer which will process all queued messages and terminate execution:
<comment>%command.full_name% someConsumer</comment>
To specify the number of messages which should be processed by consumer before its termination:
<comment>%command.full_name% someConsumer --max-messages=50</comment>
To specify the number of messages per batch for the batch consumer:
<comment>%command.full_name% someConsumer --batch-size=500</comment>
To specify the preferred area:
<comment>%command.full_name% someConsumer --area-code='adminhtml'</comment>
To prevent multiple copies of a consumer from running at the same time, run the consumer with the `single-thread` option:
<comment>%command.full_name% someConsumer --single-thread'</comment>
HELP
);
parent::configure();
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment