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. }