49 lines
1.7 KiB
PHP
49 lines
1.7 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Bridge\Doctrine\Messenger;
|
|
|
|
use Doctrine\ORM\EntityManagerInterface;
|
|
use Symfony\Component\Messenger\Envelope;
|
|
use Symfony\Component\Messenger\Exception\HandlerFailedException;
|
|
use Symfony\Component\Messenger\Middleware\StackInterface;
|
|
use Symfony\Component\Messenger\Stamp\HandledStamp;
|
|
|
|
/**
|
|
* Wraps all handlers in a single doctrine transaction.
|
|
*
|
|
* @author Tobias Nyholm <tobias.nyholm@gmail.com>
|
|
*/
|
|
class DoctrineTransactionMiddleware extends AbstractDoctrineMiddleware
|
|
{
|
|
protected function handleForManager(EntityManagerInterface $entityManager, Envelope $envelope, StackInterface $stack): Envelope
|
|
{
|
|
$entityManager->getConnection()->beginTransaction();
|
|
try {
|
|
$envelope = $stack->next()->handle($envelope, $stack);
|
|
$entityManager->flush();
|
|
$entityManager->getConnection()->commit();
|
|
|
|
return $envelope;
|
|
} catch (\Throwable $exception) {
|
|
$entityManager->getConnection()->rollBack();
|
|
|
|
if ($exception instanceof HandlerFailedException) {
|
|
// Remove all HandledStamp from the envelope so the retry will execute all handlers again.
|
|
// When a handler fails, the queries of allegedly successful previous handlers just got rolled back.
|
|
throw new HandlerFailedException($exception->getEnvelope()->withoutAll(HandledStamp::class), $exception->getWrappedExceptions());
|
|
}
|
|
|
|
throw $exception;
|
|
}
|
|
}
|
|
}
|