From b223310a3178a90da868b971b83599686a5107cd Mon Sep 17 00:00:00 2001
From: qixinzhi <qixinzhi@easycorp.ltd>
Date: Tue, 11 Jun 2024 03:11:09 +0000
Subject: [PATCH] + [misc] add parseColumns.

---
 lib/sqlparser/sqlparser.class.php | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/lib/sqlparser/sqlparser.class.php b/lib/sqlparser/sqlparser.class.php
index f671e68e50..9160c8f8b0 100644
--- a/lib/sqlparser/sqlparser.class.php
+++ b/lib/sqlparser/sqlparser.class.php
@@ -71,6 +71,26 @@ class sqlparser
         $this->isSelect = $this->statement instanceof PhpMyAdmin\SqlParser\Statements\SelectStatement === true;
     }
 
+    /**
+     * Parse columns.
+     *
+     * @access public
+     * @return void
+     */
+    public function parseColumns()
+    {
+        $fields = array();
+        foreach($this->statement->expr as $expr)
+        {
+            /* 获取查询数据后真正展示出来的列名 */
+            $columnName = empty($expr->alias) ? $expr->column : $expr->alias;
+
+            $fields[$columnName] = array('origin' => $expr->column, 'table' => $expr->table);
+        }
+
+        return $fields;
+    }
+
     /**
      * Parse tables.
      *