Option to allow devs to include the values used in the SQL query

elvanto · May 12, 2023 · 12f625c · 12f625c
1 parent b7ca8ba
commit 12f625c
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/lib/PicoDb/StatementHandler.php b/lib/PicoDb/StatementHandler.php
@@ -54,6 +54,14 @@ class StatementHandler
      */
     protected $logQueries = false;
 
+    /**
+     * Flag to combine values in the logged SQL queries.
+     *
+     * @access protected
+     * @var boolean
+     */
+    protected $logQueryValues = false;
+
     /**
      * Run explain command on each query
      *
@@ -125,10 +133,12 @@ public function __construct(Database $db)
      * Enable query logging
      *
      * @access public
+     * @param bool $includeValues
      * @return $this
      */
-    public function withLogging()
+    public function withLogging(bool $includeValues = false)
     {
+        $this->logQueryValues = $includeValues;
         $this->logQueries = true;
         return $this;
     }
@@ -229,6 +239,7 @@ public function getNbQueries()
      *
      * @access public
      * @return PDOStatement|false
+     * @throws SQLException
      */
     public function execute()
     {
@@ -285,7 +296,13 @@ protected function bindParams(PDOStatement $pdoStatement)
     protected function beforeExecute()
     {
         if ($this->logQueries) {
-            $this->db->setLogMessage($this->sql);
+            $sql = $this->sql;
+            if ($this->logQueryValues) {
+                foreach ($this->lobParams as $value) {
+                    $sql = substr_replace($sql, "'$value'", strpos($sql, '?'), 1);
+                }
+            }
+            $this->db->setLogMessage($sql);
         }
 
         if ($this->stopwatch) {
@@ -335,7 +352,6 @@ protected function cleanup()
      *
      * @access public
      * @param  PDOException $e
-     * @return bool
      * @throws SQLException
      */
     public function handleSqlError(PDOException $e)