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

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\DBAL3;
  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 StatementInterface;
  18. use Symfony\Bridge\Doctrine\Middleware\Debug\DebugDataHolder;
  19. use Symfony\Bridge\Doctrine\Middleware\Debug\Query;
  20. use Symfony\Component\Stopwatch\Stopwatch;
  22. /**
  23.  * @author Laurent VOULLEMIER <laurent.voullemier@gmail.com>
  24.  *
  25.  * @internal
  26.  */
  27. final class Connection extends AbstractConnectionMiddleware
  28. {
  29.     /** @var int */
  30.     private $nestingLevel = 0;
  31.     private $debugDataHolder;
  32.     private $stopwatch;
  33.     private $connectionName;
  35.     public function __construct(
  36.         ConnectionInterface $connection,
  37.         DebugDataHolder $debugDataHolder,
  38.         ?Stopwatch $stopwatch,
  39.         string $connectionName
  40.     ) {
  41.         $this->connectionName = $connectionName;
  42.         $this->stopwatch = $stopwatch;
  43.         $this->debugDataHolder = $debugDataHolder;
  45.         parent::__construct($connection);
  46.     }
  48.     public function prepare(string $sql): StatementInterface
  49.     {
  50.         return new Statement(
  51.             parent::prepare($sql),
  52.             $this->debugDataHolder,
  53.             $this->connectionName,
  54.             $sql,
  55.             $this->stopwatch,
  56.         );
  57.     }
  59.     public function query(string $sql): Result
  60.     {
  61.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  63.         if ($this->stopwatch) {
  64.             $this->stopwatch->start('doctrine', 'doctrine');
  65.         }
  66.         $query->start();
  68.         try {
  69.             return parent::query($sql);
  70.         } finally {
  71.             $query->stop();
  72.             if ($this->stopwatch) {
  73.                 $this->stopwatch->stop('doctrine');
  74.             }
  75.         }
  76.     }
  78.     public function exec(string $sql): int
  79.     {
  80.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  82.         if ($this->stopwatch) {
  83.             $this->stopwatch->start('doctrine', 'doctrine');
  84.         }
  85.         $query->start();
  87.         try {
  88.             return parent::exec($sql);
  89.         } finally {
  90.             $query->stop();
  91.             if ($this->stopwatch) {
  92.                 $this->stopwatch->stop('doctrine');
  93.             }
  94.         }
  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 ($this->stopwatch) {
  105.             $this->stopwatch->start('doctrine', 'doctrine');
  106.         }
  107.         if ($query) {
  108.             $query->start();
  109.         }
  111.         try {
  112.             return parent::beginTransaction();
  113.         } finally {
  114.             if ($query) {
  115.                 $query->stop();
  116.             }
  117.             if ($this->stopwatch) {
  118.                 $this->stopwatch->stop('doctrine');
  119.             }
  120.         }
  121.     }
  123.     public function commit(): bool
  124.     {
  125.         $query = null;
  126.         if (1 === $this->nestingLevel--) {
  127.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"COMMIT"'));
  128.         }
  130.         if ($this->stopwatch) {
  131.             $this->stopwatch->start('doctrine', 'doctrine');
  132.         }
  133.         if ($query) {
  134.             $query->start();
  135.         }
  137.         try {
  138.             return parent::commit();
  139.         } finally {
  140.             if ($query) {
  141.                 $query->stop();
  142.             }
  143.             if ($this->stopwatch) {
  144.                 $this->stopwatch->stop('doctrine');
  145.             }
  146.         }
  147.     }
  149.     public function rollBack(): bool
  150.     {
  151.         $query = null;
  152.         if (1 === $this->nestingLevel--) {
  153.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"ROLLBACK"'));
  154.         }
  156.         if ($this->stopwatch) {
  157.             $this->stopwatch->start('doctrine', 'doctrine');
  158.         }
  159.         if ($query) {
  160.             $query->start();
  161.         }
  163.         try {
  164.             return parent::rollBack();
  165.         } finally {
  166.             if ($query) {
  167.                 $query->stop();
  168.             }
  169.             if ($this->stopwatch) {
  170.                 $this->stopwatch->stop('doctrine');
  171.             }
  172.         }
  173.     }
  174. }