vendor/symfony/doctrine-bridge/Middleware/Debug/Connection.php line 62

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\Middleware\Debug;
  14. use Doctrine\DBAL\Driver\Connection as ConnectionInterface;
  15. use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
  16. use Doctrine\DBAL\Driver\Result;
  17. use Doctrine\DBAL\Driver\Statement as DriverStatement;
  18. use Symfony\Component\Stopwatch\Stopwatch;
  20. /**
  21.  * @author Laurent VOULLEMIER <laurent.voullemier@gmail.com>
  22.  *
  23.  * @internal
  24.  */
  25. final class Connection extends AbstractConnectionMiddleware
  26. {
  27.     private $nestingLevel = 0;
  28.     private $debugDataHolder;
  29.     private $stopwatch;
  30.     private $connectionName;
  32.     public function __construct(ConnectionInterface $connection, DebugDataHolder $debugDataHolder, ?Stopwatch $stopwatch, string $connectionName)
  33.     {
  34.         parent::__construct($connection);
  36.         $this->debugDataHolder = $debugDataHolder;
  37.         $this->stopwatch = $stopwatch;
  38.         $this->connectionName = $connectionName;
  39.     }
  41.     public function prepare(string $sql): DriverStatement
  42.     {
  43.         return new Statement(
  44.             parent::prepare($sql),
  45.             $this->debugDataHolder,
  46.             $this->connectionName,
  47.             $sql
  48.         );
  49.     }
  51.     public function query(string $sql): Result
  52.     {
  53.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  55.         if (null !== $this->stopwatch) {
  56.             $this->stopwatch->start('doctrine', 'doctrine');
  57.         }
  59.         $query->start();
  61.         try {
  62.             $result = parent::query($sql);
  63.         } finally {
  64.             $query->stop();
  66.             if (null !== $this->stopwatch) {
  67.                 $this->stopwatch->stop('doctrine');
  68.             }
  69.         }
  71.         return $result;
  72.     }
  74.     public function exec(string $sql): int
  75.     {
  76.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  78.         if (null !== $this->stopwatch) {
  79.             $this->stopwatch->start('doctrine', 'doctrine');
  80.         }
  82.         $query->start();
  84.         try {
  85.             $affectedRows = parent::exec($sql);
  86.         } finally {
  87.             $query->stop();
  89.             if (null !== $this->stopwatch) {
  90.                 $this->stopwatch->stop('doctrine');
  91.             }
  92.         }
  94.         return $affectedRows;
  95.     }
  97.     public function beginTransaction(): bool
  98.     {
  99.         $query = null;
  100.         if (1 === ++$this->nestingLevel) {
  101.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"START TRANSACTION"'));
  102.         }
  104.         if (null !== $this->stopwatch) {
  105.             $this->stopwatch->start('doctrine', 'doctrine');
  106.         }
  108.         if (null !== $query) {
  109.             $query->start();
  110.         }
  112.         try {
  113.             $ret = parent::beginTransaction();
  114.         } finally {
  115.             if (null !== $query) {
  116.                 $query->stop();
  117.             }
  119.             if (null !== $this->stopwatch) {
  120.                 $this->stopwatch->stop('doctrine');
  121.             }
  122.         }
  124.         return $ret;
  125.     }
  127.     public function commit(): bool
  128.     {
  129.         $query = null;
  130.         if (1 === $this->nestingLevel--) {
  131.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"COMMIT"'));
  132.         }
  134.         if (null !== $this->stopwatch) {
  135.             $this->stopwatch->start('doctrine', 'doctrine');
  136.         }
  138.         if (null !== $query) {
  139.             $query->start();
  140.         }
  142.         try {
  143.             $ret = parent::commit();
  144.         } finally {
  145.             if (null !== $query) {
  146.                 $query->stop();
  147.             }
  149.             if (null !== $this->stopwatch) {
  150.                 $this->stopwatch->stop('doctrine');
  151.             }
  152.         }
  154.         return $ret;
  155.     }
  157.     public function rollBack(): bool
  158.     {
  159.         $query = null;
  160.         if (1 === $this->nestingLevel--) {
  161.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"ROLLBACK"'));
  162.         }
  164.         if (null !== $this->stopwatch) {
  165.             $this->stopwatch->start('doctrine', 'doctrine');
  166.         }
  168.         if (null !== $query) {
  169.             $query->start();
  170.         }
  172.         try {
  173.             $ret = parent::rollBack();
  174.         } finally {
  175.             if (null !== $query) {
  176.                 $query->stop();
  177.             }
  179.             if (null !== $this->stopwatch) {
  180.                 $this->stopwatch->stop('doctrine');
  181.             }
  182.         }
  184.         return $ret;
  185.     }
  186. }