134 lines
3.5 KiB
PHP
134 lines
3.5 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Bridge\Doctrine\Middleware\Debug\DBAL3;
|
|
|
|
use Doctrine\DBAL\Driver\Connection as ConnectionInterface;
|
|
use Doctrine\DBAL\Driver\Middleware\AbstractConnectionMiddleware;
|
|
use Doctrine\DBAL\Driver\Result;
|
|
use Symfony\Bridge\Doctrine\Middleware\Debug\DebugDataHolder;
|
|
use Symfony\Bridge\Doctrine\Middleware\Debug\Query;
|
|
use Symfony\Component\Stopwatch\Stopwatch;
|
|
|
|
/**
|
|
* @author Laurent VOULLEMIER <laurent.voullemier@gmail.com>
|
|
*
|
|
* @internal
|
|
*/
|
|
final class Connection extends AbstractConnectionMiddleware
|
|
{
|
|
private int $nestingLevel = 0;
|
|
|
|
public function __construct(
|
|
ConnectionInterface $connection,
|
|
private readonly DebugDataHolder $debugDataHolder,
|
|
private readonly ?Stopwatch $stopwatch,
|
|
private readonly string $connectionName,
|
|
) {
|
|
parent::__construct($connection);
|
|
}
|
|
|
|
public function prepare(string $sql): Statement
|
|
{
|
|
return new Statement(
|
|
parent::prepare($sql),
|
|
$this->debugDataHolder,
|
|
$this->connectionName,
|
|
$sql,
|
|
$this->stopwatch,
|
|
);
|
|
}
|
|
|
|
public function query(string $sql): Result
|
|
{
|
|
$this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
|
|
|
|
$this->stopwatch?->start('doctrine', 'doctrine');
|
|
$query->start();
|
|
|
|
try {
|
|
return parent::query($sql);
|
|
} finally {
|
|
$query->stop();
|
|
$this->stopwatch?->stop('doctrine');
|
|
}
|
|
}
|
|
|
|
public function exec(string $sql): int
|
|
{
|
|
$this->debugDataHolder->addQuery($this->connectionName, $query = new Query($sql));
|
|
|
|
$this->stopwatch?->start('doctrine', 'doctrine');
|
|
$query->start();
|
|
|
|
try {
|
|
return parent::exec($sql);
|
|
} finally {
|
|
$query->stop();
|
|
$this->stopwatch?->stop('doctrine');
|
|
}
|
|
}
|
|
|
|
public function beginTransaction(): bool
|
|
{
|
|
$query = null;
|
|
if (1 === ++$this->nestingLevel) {
|
|
$this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"START TRANSACTION"'));
|
|
}
|
|
|
|
$this->stopwatch?->start('doctrine', 'doctrine');
|
|
$query?->start();
|
|
|
|
try {
|
|
return parent::beginTransaction();
|
|
} finally {
|
|
$query?->stop();
|
|
$this->stopwatch?->stop('doctrine');
|
|
}
|
|
}
|
|
|
|
public function commit(): bool
|
|
{
|
|
$query = null;
|
|
if (1 === $this->nestingLevel--) {
|
|
$this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"COMMIT"'));
|
|
}
|
|
|
|
$this->stopwatch?->start('doctrine', 'doctrine');
|
|
$query?->start();
|
|
|
|
try {
|
|
return parent::commit();
|
|
} finally {
|
|
$query?->stop();
|
|
$this->stopwatch?->stop('doctrine');
|
|
}
|
|
}
|
|
|
|
public function rollBack(): bool
|
|
{
|
|
$query = null;
|
|
if (1 === $this->nestingLevel--) {
|
|
$this->debugDataHolder->addQuery($this->connectionName, $query = new Query('"ROLLBACK"'));
|
|
}
|
|
|
|
$this->stopwatch?->start('doctrine', 'doctrine');
|
|
$query?->start();
|
|
|
|
try {
|
|
return parent::rollBack();
|
|
} finally {
|
|
$query?->stop();
|
|
$this->stopwatch?->stop('doctrine');
|
|
}
|
|
}
|
|
}
|