vendor/symfony/doctrine-bridge/Messenger/DoctrineTransactionMiddleware.php line 31

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\Bridge\Doctrine\Messenger;
  14. use Doctrine\ORM\EntityManagerInterface;
  15. use Symfony\Component\Messenger\Envelope;
  16. use Symfony\Component\Messenger\Exception\HandlerFailedException;
  17. use Symfony\Component\Messenger\Middleware\StackInterface;
  18. use Symfony\Component\Messenger\Stamp\HandledStamp;
  20. /**
  21.  * Wraps all handlers in a single doctrine transaction.
  22.  *
  23.  * @author Tobias Nyholm <tobias.nyholm@gmail.com>
  24.  */
  25. class DoctrineTransactionMiddleware extends AbstractDoctrineMiddleware
  26. {
  27.     protected function handleForManager(EntityManagerInterface $entityManager, Envelope $envelope, StackInterface $stack): Envelope
  28.     {
  29.         $entityManager->getConnection()->beginTransaction();
  30.         try {
  31.             $envelope = $stack->next()->handle($envelope, $stack);
  32.             $entityManager->flush();
  33.             $entityManager->getConnection()->commit();
  35.             return $envelope;
  36.         } catch (\Throwable $exception) {
  37.             $entityManager->getConnection()->rollBack();
  39.             if ($exception instanceof HandlerFailedException) {
  40.                 // Remove all HandledStamp from the envelope so the retry will execute all handlers again.
  41.                 // When a handler fails, the queries of allegedly successful previous handlers just got rolled back.
  42.                 throw new HandlerFailedException($exception->getEnvelope()->withoutAll(HandledStamp::class), $exception->getNestedExceptions());
  43.             }
  45.             throw $exception;
  46.         }
  47.     }
  48. }