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

Open in your IDE?
  1. <?php
  3. /*
  4. * This file is part of the Symfony package.
  5. *
  6. * (c) Fabien Potencier <fabien@symfony.com>
  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 <ryan@symfonycasts.com>
  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. }