src/Eccube/EventListener/TransactionListener.php line 66

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of EC-CUBE
  5.  *
  6.  * Copyright(c) EC-CUBE CO.,LTD. All Rights Reserved.
  7.  *
  8.  * http://www.ec-cube.co.jp/
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Eccube\EventListener;
  16. use Doctrine\DBAL\Connection;
  17. use Doctrine\DBAL\TransactionIsolationLevel;
  18. use Doctrine\ORM\EntityManager;
  19. use Doctrine\ORM\EntityManagerInterface;
  20. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  21. use Symfony\Component\HttpKernel\Event\ExceptionEvent;
  22. use Symfony\Component\HttpKernel\Event\RequestEvent;
  23. use Symfony\Component\HttpKernel\Event\TerminateEvent;
  24. use Symfony\Component\HttpKernel\KernelEvents;
  26. /**
  27.  * トランザクション制御のためのListener
  28.  */
  29. class TransactionListener implements EventSubscriberInterface
  30. {
  31.     /**
  32.      * @var EntityManagerInterface
  33.      */
  34.     protected $em;
  36.     /**
  37.      * @var bool
  38.      */
  39.     protected $isEnabled = true;
  41.     /**
  42.      * TransactionListener constructor.
  43.      *
  44.      * @param EntityManager $em
  45.      * @param bool $isEnabled
  46.      */
  47.     public function __construct(EntityManagerInterface $em, $isEnabled = true)
  48.     {
  49.         $this->em = $em;
  50.         $this->isEnabled = $isEnabled;
  51.     }
  53.     /**
  54.      * Disable transaction listener.
  55.      */
  56.     public function disable()
  57.     {
  58.         $this->isEnabled = false;
  59.     }
  61.     /**
  62.      * Kernel request listener callback.
  63.      *
  64.      * @param RequestEvent $event
  65.      */
  66.     public function onKernelRequest(RequestEvent $event)
  67.     {
  68.         if (!$this->isEnabled) {
  69.             log_debug('Transaction Listener is disabled.');
  71.             return;
  72.         }
  74.         if (!$event->isMainRequest()) {
  75.             return;
  76.         }
  78.         /** @var Connection $Connection */
  79.         $Connection = $this->em->getConnection();
  80.         if (!$Connection->isConnected()) {
  81.             $Connection->connect();
  82.         }
  83.         $Connection->setAutoCommit(false);
  84.         $Connection->setTransactionIsolation(TransactionIsolationLevel::READ_COMMITTED);
  85.         $this->em->beginTransaction();
  86.         log_debug('Begin Transaction.');
  87.     }
  89.     /**
  90.      * Kernel exception listener callback.
  91.      *
  92.      * @param ExceptionEvent $event
  93.      */
  94.     public function onKernelException(ExceptionEvent $event)
  95.     {
  96.         if (!$this->isEnabled) {
  97.             log_debug('Transaction Listener is disabled.');
  99.             return;
  100.         }
  102.         if (!$event->isMainRequest()) {
  103.             return;
  104.         }
  106.         if ($this->em->getConnection()->getNativeConnection()->inTransaction()) {
  107.             if ($this->em->getConnection()->isRollbackOnly()) {
  108.                 $this->em->rollback();
  109.             }
  110.             log_debug('Rollback executed.');
  111.         } else {
  112.             log_debug('Transaction is not active. Rollback skipped.');
  113.         }
  114.     }
  116.     /**
  117.      *  Kernel terminate listener callback.
  118.      *
  119.      * @param PostResponseEvent $event
  120.      */
  121.     public function onKernelTerminate(TerminateEvent $event)
  122.     {
  123.         if (!$this->isEnabled) {
  124.             log_debug('Transaction Listener is disabled.');
  126.             return;
  127.         }
  128.         if ($this->em->getConnection()->getNativeConnection()->inTransaction()) {
  129.             if ($this->em->getConnection()->isRollbackOnly()) {
  130.                 $this->em->rollback();
  131.                 log_debug('Rollback executed.');
  132.             } else {
  133.                 $this->em->commit();
  134.                 log_debug('Commit executed.');
  135.             }
  136.         } else {
  137.             log_debug('Transaction is not active. Rollback skipped.');
  138.         }
  139.     }
  141.     /**
  142.      * Return the events to subscribe to.
  143.      *
  144.      * @return array
  145.      */
  146.     public static function getSubscribedEvents()
  147.     {
  148.         return [
  149.             KernelEvents::REQUEST => 'onKernelRequest',
  150.             KernelEvents::EXCEPTION => 'onKernelException',
  151.             KernelEvents::TERMINATE => 'onKernelTerminate',
  152.         ];
  153.     }
  154. }