如何獲取一條SQL語句中涉及的表名

作者：懂點IT的耿小廚 2024-01-03 17:42:32

在數據庫操作和SQL查詢的開發過程中，有時候我們為了動態生成查詢、進行權限控制、進行查詢優化或者其他一些與數據庫交互相關、數據庫監控等的需求，需要從SQL語句中提取表名。

在數據庫操作和SQL查詢的開發過程中，有時候我們為了動態生成查詢、進行權限控制、進行查詢優化或者其他一些與數據庫交互相關、數據庫監控等的需求，需要從SQL語句中提取表名。本文分別使用正則表達式和使用SQL解析庫的方式來獲取。當然實際使用中需要進行優化，本次只是做初步的獲取操作。

1. 使用正則表達式 正則表達式是一種強大的文本匹配工具，通過定義模式，可以從文本中提取需要的信息。在SQL語句中，我們可以使用正則表達式匹配關鍵字（如FROM、JOIN、UPDATE等）后面的表名，但是通常會因為SQL的復雜度的問題導致提取不夠準確。

import re


def get_table1(sql_statement):
    table_names = re.findall(r'FROM\s+(\w+)|JOIN\s+(\w+)|UPDATE\s+(\w+)|INTO\s+(\w+)|TABLE\s+(\w+)', sql_statement, re.IGNORECASE)
    for tbs in table_names:
        for tb in tbs:
            print(tb)

例如，采用如下方式獲取結果：

if __name__ == '__main__':
    sql = "select * from tb1 where c1='a';"
    get_table1(sql)
    sql = "select * from tb1 a , tb2 b  where  a.id=b.id  and   c1='a';"
    get_table1(sql)
    sql ="UPDATE  tb3  SET b = 1 WHERE c1='47d8af9d8cd1459a927327b9d548a37b' "
    get_table1(sql)

得到的結果如下：

其中關聯查詢的SQL結果有誤

select * from tb1 a , tb2 b  where  a.id=b.id  and   c1='a';

2. 使用SQL解析庫

SQL解析庫能夠更全面地理解SQL語句的結構，提供了更為準確的分析。sqlparse是一個常用的SQL解析庫，它可以解析SQL語句并將其轉換成語法樹。

import sqlparse


def get_table2(sql_statement):
    parsed = sqlparse.parse(sql_statement)
    for stmt in parsed:
        for token in stmt.tokens:
            if isinstance(token, sqlparse.sql.IdentifierList):
                for identifier in token.get_identifiers():
                    print(identifier.get_real_name())
            elif isinstance(token, sqlparse.sql.Identifier):
                print(token.get_real_name())

在使用上述SQL測試一下：

if __name__ == '__main__':
    sql = "select * from tb1 where c1='a';"
    get_table2(sql)
    sql = "select * from tb1 a , tb2 b  where  a.id=b.id  and   c1='a';"
    get_table2(sql)
    sql ="UPDATE  tb3  SET b = 1 WHERE c1='47d8af9d8cd1459a927327b9d548a37b' "
    get_table2(sql)

測試結果如下：

關聯查詢可以獲取到準確的表名了。

注：以上只是簡單演示SQL解析庫的方式，對于DDL等也需要再優化一下，否則可能獲取到錯誤信息。

3. 小結

從SQL語句中提取表名可以在數據庫操作和應用程序開發中發揮重要作用，從而使系統更加靈活、安全、高效。選擇合適的提取表名的方法取決于具體的需求和應用場景。例如可以在如下場景中使用：

動態查詢生成： 通過提取SQL語句中的表名，可以動態生成適應不同條件的查詢語句，提高代碼的靈活性
權限控制：根據SQL語句中涉及的表名，可以實現更細粒度的權限控制，確保用戶只能訪問其有權限的表
查詢優化： 了解SQL語句中的表結構有助于進行查詢優化，根據表的大小、索引情況等因素進行優化
日志記錄：記錄每個查詢涉及的表名，可以用于性能分析和日志記錄，幫助理解應用程序的行為
數據遷移和同步：在數據遷移或同步過程中，了解SQL語句涉及的表結構有助于更好地管理數據變更，確保數據一致性
數據庫監控： 可以配合監控數據庫中對應表的使用情況監控等

責任編輯：華軒來源：數據庫干貨鋪

SQL 數據庫