vendor/league/tactician-doctrine/src/ORM/TransactionMiddleware.php line 36

Open in your IDE?
  1. <?php
  2. namespace League\Tactician\Doctrine\ORM;
  4. use Doctrine\ORM\EntityManagerInterface;
  5. use League\Tactician\Middleware;
  6. use Exception;
  7. use Throwable;
  9. /**
  10. * Wraps command execution inside a Doctrine ORM transaction
  11. */
  12. class TransactionMiddleware implements Middleware
  13. {
  14. /**
  15. * @var EntityManagerInterface
  16. */
  17. private $entityManager;
  19. /**
  20. * @param EntityManagerInterface $entityManager
  21. */
  22. public function __construct(EntityManagerInterface $entityManager)
  23. {
  24. $this->entityManager = $entityManager;
  25. }
  27. /**
  28. * Executes the given command and optionally returns a value
  29. *
  30. * @param object $command
  31. * @param callable $next
  32. * @return mixed
  33. * @throws Throwable
  34. * @throws Exception
  35. */
  36. public function execute($command, callable $next)
  37. {
  38. $this->entityManager->beginTransaction();
  40. try {
  41. $returnValue = $next($command);
  43. $this->entityManager->flush();
  44. $this->entityManager->commit();
  45. } catch (Exception $e) {
  46. $this->rollbackTransaction();
  48. throw $e;
  49. } catch (Throwable $e) {
  50. $this->rollbackTransaction();
  52. throw $e;
  53. }
  55. return $returnValue;
  56. }
  58. /**
  59. * Rollback the current transaction and close the entity manager when possible.
  60. */
  61. protected function rollbackTransaction()
  62. {
  63. $this->entityManager->rollback();
  65. $connection = $this->entityManager->getConnection();
  66. if (!$connection->isTransactionActive() || $connection->isRollbackOnly()) {
  67. $this->entityManager->close();
  68. }
  69. }
  70. }