希望增加一个功能，获取select语句最终选出字段

获取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 语句，例如嵌套的子查询、自定义函数等。