[DoctrineBridge] fix issue with missing stopwatch events

dmaicher · dmaicher · commit 65788a36c8dc · 2023-01-23T17:35:46.000+01:00
diff --git a/src/Symfony/Bridge/Doctrine/Middleware/Debug/Connection.php b/src/Symfony/Bridge/Doctrine/Middleware/Debug/Connection.php
@@ -44,7 +44,8 @@ public function prepare(string $sql): DriverStatement
             parent::prepare($sql),
             $this->debugDataHolder,
             $this->connectionName,
-            $sql
+            $sql,
+            $this->stopwatch
         );
     }
 
diff --git a/src/Symfony/Bridge/Doctrine/Middleware/Debug/Statement.php b/src/Symfony/Bridge/Doctrine/Middleware/Debug/Statement.php
@@ -15,6 +15,7 @@
 use Doctrine\DBAL\Driver\Result as ResultInterface;
 use Doctrine\DBAL\Driver\Statement as StatementInterface;
 use Doctrine\DBAL\ParameterType;
+use Symfony\Component\Stopwatch\Stopwatch;
 
 /**
  * @author Laurent VOULLEMIER <laurent.voullemier@gmail.com>
@@ -26,14 +27,16 @@ final class Statement extends AbstractStatementMiddleware
     private $debugDataHolder;
     private $connectionName;
     private $query;
+    private $stopwatch;
 
-    public function __construct(StatementInterface $statement, DebugDataHolder $debugDataHolder, string $connectionName, string $sql)
+    public function __construct(StatementInterface $statement, DebugDataHolder $debugDataHolder, string $connectionName, string $sql, ?Stopwatch $stopwatch = null)
     {
         parent::__construct($statement);
 
         $this->debugDataHolder = $debugDataHolder;
         $this->connectionName = $connectionName;
         $this->query = new Query($sql);
+        $this->stopwatch = $stopwatch;
     }
 
     public function bindParam($param, &$variable, $type = ParameterType::STRING, $length = null): bool
@@ -59,12 +62,20 @@ public function execute($params = null): ResultInterface
         // clone to prevent variables by reference to change
         $this->debugDataHolder->addQuery($this->connectionName, $query = clone $this->query);
 
+        if (null !== $this->stopwatch) {
+            $this->stopwatch->start('doctrine', 'doctrine');
+        }
+
         $query->start();
 
         try {
             $result = parent::execute($params);
         } finally {
             $query->stop();
+
+            if (null !== $this->stopwatch) {
+                $this->stopwatch->stop('doctrine');
+            }
         }
 
         return $result;
diff --git a/src/Symfony/Bridge/Doctrine/Tests/Middleware/Debug/MiddlewareTest.php b/src/Symfony/Bridge/Doctrine/Tests/Middleware/Debug/MiddlewareTest.php
@@ -161,10 +161,6 @@ public function testWithParamBound(callable $executeMethod)
         $executeMethod($stmt);
 
         // Debug data should not be affected by these changes
-        $product = 'product2';
-        $price = 13.5;
-        $stock = 4;
-
         $debug = $this->debugDataHolder->getData()['default'] ?? [];
         $this->assertCount(2, $debug);
         $this->assertSame('INSERT INTO products(name, price, stock) VALUES (?, ?, ?)', $debug[1]['sql']);
@@ -242,6 +238,14 @@ static function (Connection $conn): bool {
                     return $conn->rollBack();
                 },
             ],
+            'prepared statement' => [
+                static function (Connection $conn, string $sql): Result {
+                    return $conn->prepare($sql)->executeQuery();
+                },
+                static function (Connection $conn): bool {
+                    return $conn->commit();
+                },
+            ],
         ];
     }
 

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,8 @@ public function prepare(string $sql): DriverStatement`
`44`	`44`	`parent::prepare($sql),`
`45`	`45`	`$this->debugDataHolder,`
`46`	`46`	`$this->connectionName,`
`47`		`- $sql`
	`47`	`+ $sql,`
	`48`	`+ $this->stopwatch`
`48`	`49`	`);`
`49`	`50`	`}`
`50`	`51`