[Messenger] Fix exiting FailedMessagesRetryCommand

HypeMC · HypeMC · commit 2627b1b59648 · 2023-06-27T04:10:48.000+02:00
diff --git a/src/Symfony/Component/Messenger/Command/FailedMessagesRetryCommand.php b/src/Symfony/Component/Messenger/Command/FailedMessagesRetryCommand.php
@@ -19,9 +19,11 @@
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\ConsoleOutputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\SignalRegistry\SignalRegistry;
 use Symfony\Component\Console\Style\SymfonyStyle;
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 use Symfony\Component\Messenger\Event\WorkerMessageReceivedEvent;
+use Symfony\Component\Messenger\Event\WorkerStartedEvent;
 use Symfony\Component\Messenger\EventListener\StopWorkerOnMessageLimitListener;
 use Symfony\Component\Messenger\MessageBusInterface;
 use Symfony\Component\Messenger\Stamp\MessageDecodingFailedStamp;
@@ -165,6 +167,16 @@ private function runInteractive(string $failureTransportName, SymfonyStyle $io,
 
     private function runWorker(string $failureTransportName, ReceiverInterface $receiver, SymfonyStyle $io, bool $shouldForce): int
     {
+        $this->eventDispatcher->addListener(WorkerStartedEvent::class, $signalsListener = function () {
+            if (!\defined('SIGINT') || !SignalRegistry::isSupported()) {
+                return;
+            }
+
+            foreach ([\SIGINT, \SIGTERM] as $signal) {
+                pcntl_signal($signal, $this->getApplication()->getSignalRegistry()->handle(...));
+            }
+        });
+
         $count = 0;
         $listener = function (WorkerMessageReceivedEvent $messageReceivedEvent) use ($io, $receiver, $shouldForce, &$count) {
             ++$count;
@@ -197,6 +209,7 @@ private function runWorker(string $failureTransportName, ReceiverInterface $rece
         try {
             $worker->run();
         } finally {
+            $this->eventDispatcher->removeListener(WorkerStartedEvent::class, $signalsListener);
             $this->eventDispatcher->removeListener(WorkerMessageReceivedEvent::class, $listener);
         }
 
