Trace transaction events on PDOCollector (php-debugbar#785)

Author: erikn69

src/DebugBar/DataCollector/PDO/PDOCollector.php

@@ -149,6 +149,7 @@ protected function collectPDO(TraceablePDO $pdo, ?TimeDataCollector $timeCollect
         foreach ($pdo->getExecutedStatements() as $stmt) {
             $stmts[] = array(
                 'sql' => $this->renderSqlWithParams ? $stmt->getSqlWithParams($this->sqlQuotationChar) : $stmt->getSql(),
+                'type' => $stmt->getQueryType(),
                 'row_count' => $stmt->getRowCount(),
                 'stmt_id' => $stmt->getPreparedId(),
                 'prepared_stmt' => $stmt->getSql(),

src/DebugBar/DataCollector/PDO/TraceablePDO.php

@@ -31,6 +31,8 @@ public function __construct(PDO $pdo)
     */
     public function beginTransaction() : bool
     {
+        $this->addPdoEvent('Begin Transaction');
+
         return $this->pdo->beginTransaction();
     }
 
@@ -42,6 +44,8 @@ public function beginTransaction() : bool
      */
     public function commit() : bool
     {
+        $this->addPdoEvent('Commit Transaction');
+
         return $this->pdo->commit();
     }
 
@@ -180,6 +184,8 @@ public function quote($string, $parameter_type = PDO::PARAM_STR)
      */
     public function rollBack() : bool
     {
+        $this->addPdoEvent('Rollback Transaction');
+
         return $this->pdo->rollBack();
     }
 
@@ -241,6 +247,21 @@ public function addExecutedStatement(TracedStatement $stmt) : void
         $this->executedStatements[] = $stmt;
     }
 
+    /**
+     * Adds a PDO event
+     *
+     * @param string $event
+     */
+    public function addPdoEvent(string $event) : void
+    {
+        $stmt = new TracedStatement($event);
+        $stmt->setQueryType('transaction');
+        $stmt->start();
+        $stmt->end();
+
+        $this->executedStatements[] = $stmt;
+    }
+
     /**
      * Returns the accumulated execution time of statements
      *

src/DebugBar/DataCollector/PDO/TracedStatement.php

@@ -9,6 +9,8 @@ class TracedStatement
 {
     protected $sql;
 
+    protected $type;
+
     protected $rowCount;
 
     protected $parameters;
@@ -41,6 +43,14 @@ public function __construct(string $sql, array $params = [], ?string $preparedId
         $this->preparedId = $preparedId;
     }
 
+    /**
+     * @param string $type
+     */
+    public function setQueryType(string $type) : void
+    {
+        $this->type = $type;
+    }
+
     /**
      * @param null $startTime
      * @param null $startMemory
@@ -277,4 +287,14 @@ public function getErrorMessage() : string
     {
         return $this->exception !== null ? $this->exception->getMessage() : '';
     }
+
+    /**
+     * Returns the query type
+     *
+     * @return string
+     */
+    public function getQueryType() : string
+    {
+        return $this->type !== null ? $this->type : '';
+    }
 }