如何获取一条SQL语句中涉及的表名

最新推荐文章于 2024-10-12 11:42:17 发布

gjc592

最新推荐文章于 2024-10-12 11:42:17 发布

阅读量2.4k

点赞数 1

文章标签： sql 数据库 oracle java 开发语言

本文链接：https://blog.youkuaiyun.com/gjc592/article/details/135098215

版权

本文探讨了如何使用正则表达式和SQL解析库（如sqlparse）从SQL语句中准确提取表名，以支持动态查询、权限控制、查询优化等需求。正则表达式方法存在局限性，而SQL解析库提供更精确的分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点击上方蓝字关注我

在数据库操作和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)

测试结果如下：