包含子查询的sql语句,比如 select u.name, u.age from (select * from user) u; 想要获取到name和age这两个字段目前暂不支持,只能获取到子查询中的*,而且子查询的别名u这个字段暂不支持语法分析,希望以后可以得到支持,谢谢
原提问者GitHub用户Janzee
获取select语句最终选出的字段可以通过Druid SQL解析器提供的ASTVisitor来实现。具体实现步骤如下:
创建一个ASTVisitor类,用于遍历SQL语句的AST,并提取SELECT语句中的字段信息。以下是一个简单的ASTVisitor示例:
typescript
Copy
public class SelectColumnVisitor extends SQLASTVisitorAdapter {
private List columns = new ArrayList<>();
@Override
public boolean visit(SQLSelectItem selectItem) {
SQLExpr expr = selectItem.getExpr();
if (expr instanceof SQLIdentifierExpr) {
SQLIdentifierExpr identifierExpr = (SQLIdentifierExpr) expr;
String columnName = identifierExpr.getName();
columns.add(columnName);
}
return true;
}
public List<String> getColumns() {
return columns;
}
}
在上述代码中,SelectColumnVisitor继承自SQLASTVisitorAdapter,用于遍历SQL语句的AST。visit方法会在遍历到SELECT语句中的每一个SELECT项时被调用,从中提取出SELECT项中的字段名,并添加到columns列表中。getColumns方法用于获取最终选出的字段列表。
在SQL解析过程中使用ASTVisitor:在执行SQL解析时,可以将上述SelectColumnVisitor传递给Druid SQL解析器,并在解析完成后获取最终选出的字段列表。以下是一个简单的示例代码:
ini
Copy
String sql = "SELECT id, name, age FROM users WHERE age > 18";
SQLStatementParser parser = new MySqlStatementParser(sql);
SQLStatement statement = parser.parseStatement();
SQLSelectStatement selectStatement = (SQLSelectStatement) statement;
SQLSelect select = selectStatement.getSelect();
SelectColumnVisitor visitor = new SelectColumnVisitor();
select.accept(visitor);
List columns = visitor.getColumns();
在上述代码中,首先通过MySqlStatementParser将SQL语句解析为AST,然后获取SELECT语句对应的SQLSelect对象,并将SelectColumnVisitor传递给accept方法进行遍历。最后,通过getColumns方法获取最终选出的字段列表。
可以获取,提问要严谨啊,楼主很忙,把回答问题的时间留给真正需要回答的问题 MySqlStatementParser parser = new MySqlStatementParser(sql); List statementList = parser.parseStatementList(); for (SQLStatement statement : statementList) { if (statement instanceof SQLSelectStatement) { SQLSelect select = ((SQLSelectStatement) statement).getSelect(); SQLSelectQueryBlock query = (SQLSelectQueryBlock) select.getQuery(); System.out.println(query.getSelectList());//这里打印的就是name,age if(query.getFrom() instanceof SQLSubqueryTableSource){ SQLSubqueryTableSource ssts = (SQLSubqueryTableSource)query.getFrom(); MySqlSelectQueryBlock mssqb = (MySqlSelectQueryBlock) ssts.getSelect().getQuery(); System.out.println(mssqb.getSelectList());//这里打印的就是子查询的* }
console结果:
[u.name, u.age] [*]
原回答者GitHub用户boboChina
获得 SELECT 语句的查询项(即选出的字段)是 SQL 解析器中一个比较复杂的点,尤其是在涉及到子查询、别名、函数以及其他 SQL 语法成分的时候。下面介绍几种常见的方法,你可以根据自己的需求选择适合的方式:
使用 SQL 解析工具库进行语法分析,获取选出的字段。 通用的 SQL 解析器无法直接解析 SQL 语句中的 alias 和嵌套子查询等情况。 需要使用解析工具库对 SQL 语句进行更高阶段的解析。常见的 SQL 解析工具库有 Apache Calcite、ANTLR、JSqlParser、druid 等。
以 JSqlParser 为例,可以将 SELECT 查询语句解析为 PlainSelect 对象,然后从中获取查询项。 示例代码如下:
String sql = "SELECT u., u.name, u.age, (SELECT COUNT() FROM users) AS total FROM users u"; Select select = JSqlParserUtil.parse(sql);
if (select instanceof Select) { SelectBody selectBody = ((Select) select).getSelectBody(); if (selectBody instanceof PlainSelect) { PlainSelect plainSelect = (PlainSelect) selectBody; List selectItems = plainSelect.getSelectItems();
// 遍历查询项,获取选出的字段
for (SelectItem item : selectItems) {
if (item instanceof SelectExpressionItem) {
SelectExpressionItem expItem = (SelectExpressionItem) item;
Expression exp = expItem.getExpression();
System.out.println(exp);
}
}
}
}
需要注意的是,这种方式实现起来比较复杂,而且会导致性能损失。
利用正则表达式获取查询项。 一些工具类库,比如 MyBatis、Spring JDBC 等都会解析 SQL 语句,在这种情况下,通过正则表达式来提取查询项是一种常用的方式。示例代码如下:
String sql = "SELECT u., u.name, u.age, (SELECT COUNT() FROM users) AS total FROM users u"; String selectExpr = sql.substring(sql.indexOf("SELECT") + "SELECT".length(), sql.indexOf("FROM")); Matcher matcher = Pattern.compile("\b([a-zA-Z0-9_]+\.)?[a-zA-Z0-9_]+").matcher(selectExpr); while (matcher.find()) { System.out.println(matcher.group()); }
需要注意的是,这种方式无法处理一些复杂的 SQL 语句,例如嵌套的子查询、自定义函数等。
版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。