From 9ff4be051a600c63ada78f69e813dd4f6d24fed2 Mon Sep 17 00:00:00 2001
From: kriss <462679766@qq.com>
Date: Tue, 19 Mar 2024 21:31:18 +0800
Subject: [PATCH] =?UTF-8?q?feature:=20collect=20=E5=90=8E=E6=B8=85?=
 =?UTF-8?q?=E7=A9=BA=20stmts?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/DataCollector/PDO/PDOCollector.php  | 24 ++++++++++++++++++++++++
 src/DataCollector/PDO/TraceablePDO.php  | 15 +++++++++++++++
 src/DataCollector/ThinkPdoCollector.php |  4 ++--
 3 files changed, 41 insertions(+), 2 deletions(-)
 create mode 100644 src/DataCollector/PDO/PDOCollector.php
 create mode 100644 src/DataCollector/PDO/TraceablePDO.php

diff --git a/src/DataCollector/PDO/PDOCollector.php b/src/DataCollector/PDO/PDOCollector.php
new file mode 100644
index 0000000..b9767a5
--- /dev/null
+++ b/src/DataCollector/PDO/PDOCollector.php
@@ -0,0 +1,24 @@
+<?php
+
+namespace WebmanTech\Debugbar\DataCollector\PDO;
+
+use DebugBar\DataCollector\PDO\TraceablePDO as DebugbarTraceablePDO;
+use DebugBar\DataCollector\TimeDataCollector;
+
+class PDOCollector extends \DebugBar\DataCollector\PDO\PDOCollector
+{
+    /**
+     * @inheritDoc
+     */
+    protected function collectPDO(DebugbarTraceablePDO $pdo, TimeDataCollector $timeCollector = null, $connectionName = null)
+    {
+        $data = parent::collectPDO($pdo, $timeCollector, $connectionName);
+
+        if ($pdo instanceof TraceablePDO) {
+            // 清除历史记录，否则在 webman 常驻内存的情况下会有历史累计
+            $pdo->cleanExecutedStatements();
+        }
+
+        return $data;
+    }
+}
\ No newline at end of file
diff --git a/src/DataCollector/PDO/TraceablePDO.php b/src/DataCollector/PDO/TraceablePDO.php
new file mode 100644
index 0000000..623e08e
--- /dev/null
+++ b/src/DataCollector/PDO/TraceablePDO.php
@@ -0,0 +1,15 @@
+<?php
+
+namespace WebmanTech\Debugbar\DataCollector\PDO;
+
+class TraceablePDO extends \DebugBar\DataCollector\PDO\TraceablePDO
+{
+    /**
+     * 清空历史
+     * @return void
+     */
+    public function cleanExecutedStatements()
+    {
+        $this->executedStatements = [];
+    }
+}
\ No newline at end of file
diff --git a/src/DataCollector/ThinkPdoCollector.php b/src/DataCollector/ThinkPdoCollector.php
index 726c484..c3038f9 100644
--- a/src/DataCollector/ThinkPdoCollector.php
+++ b/src/DataCollector/ThinkPdoCollector.php
@@ -2,11 +2,11 @@
 
 namespace WebmanTech\Debugbar\DataCollector;
 
-use DebugBar\DataCollector\PDO\PDOCollector;
-use DebugBar\DataCollector\PDO\TraceablePDO;
 use DebugBar\DataCollector\TimeDataCollector;
 use think\db\PDOConnection;
 use think\facade\Db;
+use WebmanTech\Debugbar\DataCollector\PDO\PDOCollector;
+use WebmanTech\Debugbar\DataCollector\PDO\TraceablePDO;
 
 class ThinkPdoCollector extends PDOCollector
 {