/** * @param InputInterface $input An InputInterface instance * @param OutputInterface $output An OutputInterface instance * * @throws \InvalidArgumentException When the number of messages to consume is less than 0 * @throws \BadFunctionCallException When the pcntl is not installed and option -s is true */ protected function initialize(InputInterface $input, OutputInterface $output) { parent::initialize($input, $output); if (defined('AMQP_WITHOUT_SIGNALS') === false) { define('AMQP_WITHOUT_SIGNALS', $input->getOption('without-signals')); } if (defined('AMQP_DEBUG') === false) { define('AMQP_DEBUG', (bool) $input->getOption('debug')); } if (($this->amount = $input->getOption('messages')) < 0) { throw new \InvalidArgumentException("The -m option should be null or greater than 0"); } $name = $input->getArgument('name'); try { $this->consumer = $this->connection->getConsumer($name); } catch (InvalidArgumentException $e) { $names = implode(', ', $this->connection->getDefinedConsumers()); throw new InvalidArgumentException("Unknown consumer {$name}, expecting one of [{$names}]\nIf you added or renamed a consumer, run 'rabbitmq:setup-fabric'"); } if (!is_null($input->getOption('memory-limit')) && ctype_digit((string) $input->getOption('memory-limit')) && $input->getOption('memory-limit') > 0) { $this->consumer->setMemoryLimit($input->getOption('memory-limit')); } if ($routingKey = $input->getOption('route')) { $this->consumer->setRoutingKey($routingKey); } if (!AMQP_WITHOUT_SIGNALS && extension_loaded('pcntl')) { if (!function_exists('pcntl_signal')) { throw new \BadFunctionCallException("Function 'pcntl_signal' is referenced in the php.ini 'disable_functions' and can't be called."); } $this->consumer->onConsume[] = function () { pcntl_signal(SIGTERM, function () { if ($this->consumer) { pcntl_signal(SIGTERM, SIG_DFL); $this->consumer->forceStopConsumer(); } }); pcntl_signal(SIGINT, function () { if ($this->consumer) { pcntl_signal(SIGINT, SIG_DFL); $this->consumer->forceStopConsumer(); } }); pcntl_signal(SIGHUP, function () { if ($this->consumer) { pcntl_signal(SIGHUP, SIG_DFL); $this->consumer->forceStopConsumer(); } // TODO: Implement restarting of consumer }); }; $this->consumer->onConsumeStop[] = function () { pcntl_signal(SIGTERM, SIG_DFL); pcntl_signal(SIGINT, SIG_DFL); pcntl_signal(SIGHUP, SIG_DFL); }; } }
/** * @internal for pcntl only */ public function signalHup() { if ($this->consumer) { pcntl_signal(SIGHUP, SIG_DFL); $this->consumer->forceStopConsumer(); } // TODO: Implement restarting of consumer }