feat(Framework): Add Record of execute sql and redis key hit

Rhilip · Feb 7, 2019 · 4799813 · 4799813
1 parent ffca1e7
commit 4799813
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 4 deletions.
diff --git a/framework/Database/BasePDOConnection.php b/framework/Database/BasePDOConnection.php
@@ -45,6 +45,8 @@ class BasePDOConnection extends Component
     // sql原始数据
     protected $_sqlPrepareData = [];
 
+    protected $_sqlExecuteData = [];
+
     // 默认驱动连接选项
     protected $_defaultDriverOptions = [
         \PDO::ATTR_EMULATE_PREPARES   => false,
@@ -63,6 +65,11 @@ public function onInitialize()
         $this->_driverOptions = $this->driverOptions + $this->_defaultDriverOptions;
     }
 
+    public function onRequestAfter()
+    {
+        $this->_sqlExecuteData = [];
+    }
+
     // 创建连接
     protected function createConnection()
     {
@@ -196,6 +203,7 @@ protected function prepare()
     // 清扫预处理数据
     protected function clearPrepare()
     {
+        $this->_sqlExecuteData[] = $this->getRawSql();
         $this->_sql    = '';
         $this->_params = [];
         $this->_values = [];
@@ -418,4 +426,7 @@ public function getRawSql()
         return array_shift($sqlPrepareData);
     }
 
+    public function getExecuteData() {
+        return $this->_sqlExecuteData;
+    }
 }
diff --git a/framework/Redis/BaseRedisConnection.php b/framework/Redis/BaseRedisConnection.php
@@ -210,6 +210,8 @@ class BaseRedisConnection extends Component
     /** @var \Redis */
     protected $_redis;
 
+    protected $_calledData = [];
+
     // 初始化事件
     public function onInitialize()
     {
@@ -218,6 +220,11 @@ public function onInitialize()
         $this->_driverOptions = $this->driverOptions + $this->_defaultDriverOptions;
     }
 
+    public function onRequestAfter()
+    {
+        $this->_calledData = [];
+    }
+
     // 创建连接
     protected function createConnection()
     {
@@ -259,10 +266,17 @@ protected function autoConnect()
     // 执行命令
     public function __call($name, $arguments)
     {
-        // 自动连接
-        $this->autoConnect();
-        // 执行命令
-        return call_user_func_array([$this->_redis, $name], $arguments);
+
+        $this->autoConnect();   // 自动连接
+
+        $calling = $name . ($arguments ? ' ' . $arguments[0] : '');
+        if (isset($this->_calledData[$calling])) {
+            $this->_calledData[$calling] += 1;
+        } else {
+            $this->_calledData[$calling] = 1;
+        }
+
+        return call_user_func_array([$this->_redis, $name], $arguments);  // 执行命令
     }
 
     // 扩展方法
@@ -284,4 +298,9 @@ public function typeof($key): ?string
                 return "Not Found";
         }
     }
+
+    public function getCalledData()
+    {
+        return $this->_calledData;
+    }
 }