好的，下面是使用druid实现
sql解析，
找出sql语句中select
字段的依赖
字段的Java代码：
```java
import com.alibaba.druid.
sql.
SQLUtils;
import com.alibaba.druid.
sql.ast.
SQLExpr;
import com.alibaba.druid.
sql.ast.
SQLName;
import com.alibaba.druid.
sql.ast.
SQLObject;
import com.alibaba.druid.
sql.ast.expr.*;
import com.alibaba.druid.
sql.ast.statement.*;
import com.alibaba.druid.
sql.dialect.my
sql.parser.My
SqlParser;
import com.alibaba.druid.
sql.dialect.my
sql.visitor.My
SqlSchemaStatVisitor;
import com.alibaba.druid.
sql.visitor.
SQLASTVisitorAdapter;
import com.alibaba.druid.stat.TableStat;
import java.util.*;
public class
SqlParser {
public static void main(String[] args) {
String
sql = "SELECT t1.a, t2.b, sum(t1.c) FROM table1 t1 JOIN table2 t2 ON t1.id = t2.id WHERE t1.a > 100
GROUP BY t1.a, t2.b HAVING sum(t1.c) > 100 ORDER BY t1.a DESC, t2.b ASC LIMIT 10 OFFSET 5";
List<String> dependentFields = getDependentFields(
sql);
System.out.println(dependentFields);
}
/**
* 解析
sql语句，获取select
字段的依赖
字段
*
* @param
sql sql语句
* @return select
字段的依赖
字段列表，格式为 表名.
字段名
*/
public static List<String> getDependentFields(String
sql) {
List<String> dependentFields = new ArrayList<>();
// 解析
sql语句
List<
SQLStatement> stmtList =
SQLUtils.parseStatements(
sql, My
SqlParser.DEFAULT_LCASE_OPTIONS);
for (
SQLStatement stmt : stmtList) {
stmt.accept(new
SQLASTVisitorAdapter() {
@Override
public boolean visit(
SQLSelectStatement selectStatement) {
SQLSelect select = selectStatement.getSelect();
// 获取select
字段列表
List<
SQLSelectItem> selectItems = select.getSelectList();
// 获取select
字段的依赖
字段
Set<TableStat.Column> dependencyColumns = new HashSet<>();
for (
SQLSelectItem selectItem : selectItems) {
SQLExpr expr = selectItem.getExpr();
addDependencyColumns(expr, dependencyColumns);
}
// 将依赖
字段转换为 表名.
字段名 的格式
for (TableStat.Column column : dependencyColumns) {
String tableName = column.getTable();
String columnName = column.getName();
dependentFields.add(tableName + "." + columnName);
}
return true;
}
});
}
return dependentFields;
}
/**
* 获取表达式中的依赖
字段
*
* @param expr 表达式
* @param dependencyCols 存储依赖
字段的集合
*/
private static void addDependencyColumns(
SQLExpr expr, Set<TableStat.Column> dependencyCols) {
if (expr instanceof
SQLPropertyExpr) {
SQLPropertyExpr propertyExpr = (
SQLPropertyExpr) expr;
String tableName = null;
if (propertyExpr.getOwner() instanceof
SQLIdentifierExpr) {
tableName = ((
SQLIdentifierExpr) propertyExpr.getOwner()).getName();
} else if (propertyExpr.getOwner() instanceof
SQLPropertyExpr) {
tableName = ((
SQLIdentifierExpr) ((
SQLPropertyExpr) propertyExpr.getOwner()).getOwner()).getName();
}
String columnName = propertyExpr.getName();
dependencyCols.add(new TableStat.Column(tableName, columnName));
} else if (expr instanceof
SQLAggregateExpr) {
List<
SQLExpr> arguments = ((
SQLAggregateExpr) expr).getArguments();
for (
SQLExpr argument : arguments) {
addDependencyColumns(argument, dependencyCols);
}
} else if (expr instanceof
SQLMethodInvokeExpr) {
List<
SQLExpr> arguments = ((
SQLMethodInvokeExpr) expr).getArguments();
for (
SQLExpr argument : arguments) {
addDependencyColumns(argument, dependencyCols);
}
} else if (expr instanceof
SQLQueryExpr) {
SQLSelect select = ((
SQLQueryExpr) expr).getSubQuery().getQuery();
Set<TableStat.Column> subQueryDepCols = new HashSet<>();
select.accept(new My
SqlSchemaStatVisitor(subQueryDepCols));
dependencyCols.addAll(subQueryDepCols);
}
}
}
```
该代码使用了druid的
SQL解析功能，可以解析出
sql语句中的各个部分，包括select
字段、from子句、join子句、where子句、
group by子句、having子句、order by子句、limit子句等。我们只需要解析出select
字段，然后分析select
字段中的表达式，
找出依赖的
字段即可。
该代码中的getDependentFields方法接受一个
sql语句作为参数，返回select
字段的依赖
字段列表，格式为 表名.
字段名。
445
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
