Allow save debug data before throw exceptions

Author: natanfelles

src/Database.php

@@ -776,14 +776,26 @@ public function setDebugCollector(DatabaseCollector $collector) : static
     protected function addToDebug(Closure $function) : mixed
     {
         $start = \microtime(true);
-        $result = $function();
+        try {
+            $result = $function();
+        } catch (Exception $exception) {
+            $this->finalizeAddToDebug($start);
+            throw $exception;
+        }
+        $this->finalizeAddToDebug($start);
+        return $result;
+    }
+
+    protected function finalizeAddToDebug(float $start) : void
+    {
         $end = \microtime(true);
+        $rows = $this->mysqli->affected_rows;
+        $rows = $rows < 0 ? $rows . ' (failed)' : $rows;
         $this->debugCollector->addData([
             'start' => $start,
             'end' => $end,
             'statement' => $this->lastQuery(),
-            'rows' => $this->mysqli->affected_rows,
+            'rows' => $rows,
         ]);
-        return $result;
     }
 }

tests/Debug/DatabaseCollectorTest.php

@@ -120,4 +120,16 @@ protected function getHostInfo() : string
         self::assertStringContainsString('Socket:', $collector->getContents());
         self::assertStringNotContainsString('Port:', $collector->getContents());
     }
+
+    public function testFinalizeAddToDebug() : void
+    {
+        try {
+            $this->makeDatabase()->query('Foo Bar');
+        } catch (\Exception) {
+        }
+        $data = $this->collector->getData();
+        $data = $data[\array_key_last($data)];
+        self::assertSame('Foo Bar', $data['statement']);
+        self::assertSame('-1 (failed)', $data['rows']);
+    }
 }