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

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 int $nestingLevel = 0;
  29.     public function __construct(
  30.         ConnectionInterface $connection,
  31.         private DebugDataHolder $debugDataHolder,
  32.         private ?Stopwatch $stopwatch,
  33.         private string $connectionName,
  34.     ) {
  35.         parent::__construct($connection);
  36.     }
  38.     public function prepare(string $sql): DriverStatement
  39.     {
  40.         return new Statement(
  41.             parent::prepare($sql),
  42.             $this->debugDataHolder,
  43.             $this->connectionName,
  44.             $sql,
  45.         );
  46.     }
  48.     public function query(string $sql): Result
  49.     {
  50.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  52.         $this->stopwatch?->start('doctrine', 'doctrine');
  53.         $query->start();
  55.         try {
  56.             $result = parent::query($sql);
  57.         } finally {
  58.             $query->stop();
  59.             $this->stopwatch?->stop('doctrine');
  60.         }
  62.         return $result;
  63.     }
  65.     public function exec(string $sql): int
  66.     {
  67.         $this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
  69.         $this->stopwatch?->start('doctrine', 'doctrine');
  70.         $query->start();
  72.         try {
  73.             $affectedRows = parent::exec($sql);
  74.         } finally {
  75.             $query->stop();
  76.             $this->stopwatch?->stop('doctrine');
  77.         }
  79.         return $affectedRows;
  80.     }
  82.     public function beginTransaction(): bool
  83.     {
  84.         $query = null;
  85.         if (1 === ++$this->nestingLevel) {
  86.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"START TRANSACTION"'));
  87.         }
  89.         $this->stopwatch?->start('doctrine', 'doctrine');
  90.         $query?->start();
  92.         try {
  93.             $ret = parent::beginTransaction();
  94.         } finally {
  95.             $query?->stop();
  96.             $this->stopwatch?->stop('doctrine');
  97.         }
  99.         return $ret;
  100.     }
  102.     public function commit(): bool
  103.     {
  104.         $query = null;
  105.         if (1 === $this->nestingLevel--) {
  106.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"COMMIT"'));
  107.         }
  109.         $this->stopwatch?->start('doctrine', 'doctrine');
  110.         $query?->start();
  112.         try {
  113.             $ret = parent::commit();
  114.         } finally {
  115.             $query?->stop();
  116.             $this->stopwatch?->stop('doctrine');
  117.         }
  119.         return $ret;
  120.     }
  122.     public function rollBack(): bool
  123.     {
  124.         $query = null;
  125.         if (1 === $this->nestingLevel--) {
  126.             $this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"ROLLBACK"'));
  127.         }
  129.         $this->stopwatch?->start('doctrine', 'doctrine');
  130.         $query?->start();
  132.         try {
  133.             $ret = parent::rollBack();
  134.         } finally {
  135.             $query?->stop();
  136.             $this->stopwatch?->stop('doctrine');
  137.         }
  139.         return $ret;
  140.     }
  141. }