编写Java代码,使用druid实现sql解析,找出sql语句中,select字段的依赖字段,用表明.字段名表示
时间: 2024-02-27 18:57:32 浏览: 144
好的,下面是使用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.mysql.parser.MySqlParser;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
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, MySqlParser.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 MySqlSchemaStatVisitor(subQueryDepCols));
dependencyCols.addAll(subQueryDepCols);
}
}
}
```
该代码使用了druid的SQL解析功能,可以解析出sql语句中的各个部分,包括select字段、from子句、join子句、where子句、group by子句、having子句、order by子句、limit子句等。我们只需要解析出select字段,然后分析select字段中的表达式,找出依赖的字段即可。
该代码中的getDependentFields方法接受一个sql语句作为参数,返回select字段的依赖字段列表,格式为 表名.字段名。
阅读全文