Add missing event listeners for database transactions

leonirlopes · Feb 18, 2017 · 506c1eb · 506c1eb
1 parent ac6adfe
commit 506c1eb
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/src/LaravelDebugbar.php b/src/LaravelDebugbar.php
@@ -337,6 +337,37 @@ function ($query, $bindings = null, $time = null, $connectionName = null) use ($
                     )
                 );
             }
+
+            try {
+                $db->getEventDispatcher()->listen([
+                    \Illuminate\Database\Events\TransactionBeginning::class,
+                    'connection.*.beganTransaction',
+                ], function ($transaction) use ($queryCollector) {
+                    $queryCollector->collectTransactionEvent('Begin Transaction', $transaction->connection);
+                });
+
+                $db->getEventDispatcher()->listen([
+                    \Illuminate\Database\Events\TransactionCommitted::class,
+                    'connection.*.committed',
+                ], function ($transaction) use ($queryCollector) {
+                    $queryCollector->collectTransactionEvent('Commit Transaction', $transaction->connection);
+                });
+
+                $db->getEventDispatcher()->listen([
+                    \Illuminate\Database\Events\TransactionRolledBack::class,
+                    'connection.*.rollingBack',
+                ], function ($transaction) use ($queryCollector) {
+                    $queryCollector->collectTransactionEvent('Rollback Transaction', $transaction->connection);
+                });
+            } catch (\Exception $e) {
+                $this->addThrowable(
+                    new Exception(
+                        'Cannot add listen transactions to Queries for Laravel Debugbar: ' . $e->getMessage(),
+                        $e->getCode(),
+                        $e
+                    )
+                );
+            }
         }
 
         if ($this->shouldCollect('mail', true) && class_exists('Illuminate\Mail\MailServiceProvider')) {