如何獲取一條SQL語句中涉及的表名
作者：懂點IT的耿小廚
在數(shù)據(jù)庫操作和SQL查詢的開發(fā)過程中，有時候我們?yōu)榱藙討B(tài)生成查詢、進行權限控制、進行查詢優(yōu)化或者其他一些與數(shù)據(jù)庫交互相關、數(shù)據(jù)庫監(jiān)控等的需求，需要從SQL語句中提取表名。
在數(shù)據(jù)庫操作和SQL查詢的開發(fā)過程中，有時候我們?yōu)榱藙討B(tài)生成查詢、進行權限控制、進行查詢優(yōu)化或者其他一些與數(shù)據(jù)庫交互相關、數(shù)據(jù)庫監(jiān)控等的需求，需要從SQL語句中提取表名。本文分別使用正則表達式和使用SQL解析庫的方式來獲取。當然實際使用中需要進行優(yōu)化，本次只是做初步的獲取操作。
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)例如，采用如下方式獲取結(jié)果：
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)得到的結(jié)果如下：
其中關聯(lián)查詢的SQL結(jié)果有誤
select * from tb1 a , tb2 b where a.id=b.id and c1='a';2. 使用SQL解析庫
SQL解析庫能夠更全面地理解SQL語句的結(jié)構，提供了更為準確的分析。sqlparse是一個常用的SQL解析庫，它可以解析SQL語句并將其轉(zhuǎn)換成語法樹。
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)測試結(jié)果如下：
關聯(lián)查詢可以獲取到準確的表名了。
注： 以上只是簡單演示SQL解析庫的方式，對于DDL等也需要再優(yōu)化一下，否則可能獲取到錯誤信息。
3. 小結(jié)
從SQL語句中提取表名可以在數(shù)據(jù)庫操作和應用程序開發(fā)中發(fā)揮重要作用，從而使系統(tǒng)更加靈活、安全、高效。選擇合適的提取表名的方法取決于具體的需求和應用場景。例如可以在如下場景中使用：
- 動態(tài)查詢生成： 通過提取SQL語句中的表名，可以動態(tài)生成適應不同條件的查詢語句，提高代碼的靈活性
- 權限控制：根據(jù)SQL語句中涉及的表名，可以實現(xiàn)更細粒度的權限控制，確保用戶只能訪問其有權限的表
- 查詢優(yōu)化： 了解SQL語句中的表結(jié)構有助于進行查詢優(yōu)化，根據(jù)表的大小、索引情況等因素進行優(yōu)化
- 日志記錄：記錄每個查詢涉及的表名，可以用于性能分析和日志記錄，幫助理解應用程序的行為
- 數(shù)據(jù)遷移和同步：在數(shù)據(jù)遷移或同步過程中，了解SQL語句涉及的表結(jié)構有助于更好地管理數(shù)據(jù)變更，確保數(shù)據(jù)一致性
- 數(shù)據(jù)庫監(jiān)控： 可以配合監(jiān)控數(shù)據(jù)庫中對應表的使用情況監(jiān)控等
責任編輯：華軒
來源：
數(shù)據(jù)庫干貨鋪
