vendor/symfony/messenger/EventListener/StopWorkerOnRestartSignalListener.php line 42

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Symfony package.
  5.  *
  6.  * (c) Fabien Potencier <[email protected]>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Symfony\Component\Messenger\EventListener;
  14. use Psr\Cache\CacheItemPoolInterface;
  15. use Psr\Log\LoggerInterface;
  16. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  17. use Symfony\Component\Messenger\Event\WorkerRunningEvent;
  18. use Symfony\Component\Messenger\Event\WorkerStartedEvent;
  20. /**
  21.  * @author Ryan Weaver <[email protected]>
  22.  */
  23. class StopWorkerOnRestartSignalListener implements EventSubscriberInterface
  24. {
  25.     public const RESTART_REQUESTED_TIMESTAMP_KEY = 'workers.restart_requested_timestamp';
  27.     private $cachePool;
  28.     private $logger;
  29.     private $workerStartedAt;
  31.     public function __construct(CacheItemPoolInterface $cachePool, ?LoggerInterface $logger = null)
  32.     {
  33.         $this->cachePool = $cachePool;
  34.         $this->logger = $logger;
  35.     }
  37.     public function onWorkerStarted(): void
  38.     {
  39.         $this->workerStartedAt = microtime(true);
  40.     }
  42.     public function onWorkerRunning(WorkerRunningEvent $event): void
  43.     {
  44.         if ($this->shouldRestart()) {
  45.             $event->getWorker()->stop();
  46.             if (null !== $this->logger) {
  47.                 $this->logger->info('Worker stopped because a restart was requested.');
  48.             }
  49.         }
  50.     }
  52.     public static function getSubscribedEvents()
  53.     {
  54.         return [
  55.             WorkerStartedEvent::class => 'onWorkerStarted',
  56.             WorkerRunningEvent::class => 'onWorkerRunning',
  57.         ];
  58.     }
  60.     private function shouldRestart(): bool
  61.     {
  62.         $cacheItem = $this->cachePool->getItem(self::RESTART_REQUESTED_TIMESTAMP_KEY);
  64.         if (!$cacheItem->isHit()) {
  65.             // no restart has ever been scheduled
  66.             return false;
  67.         }
  69.         return $this->workerStartedAt < $cacheItem->get();
  70.     }
  71. }