springboot引入druid解析sql的過程
更新時間:2023年08月22日 15:15:46 作者:夜空下的星
在開發(fā)中,有時我們可能會需要獲取SQL中的表名,那么因為不同的數據源類型SQL會存在部分差異,那么我們就可以使用alibaba 的druid包實現不同的數據源類型的sql解析,需要的朋友可以參考下
一、前言
在開發(fā)中,有時我們可能會需要獲取SQL中的表名,那么因為不同的數據源類型SQL會存在部分差異,那么我們就可以使用alibaba 的druid包實現不同的數據源類型的sql解析。
二、引入相關maven依賴
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid</artifactId>
<version>1.2.8</version>
</dependency>三、通過工具類SqlUtils實現對SQL的解析。
public class QualitySqlUtils {
/**
* 根據sql及數據源類型獲取表名
*/
public static List<String> getSelectSqlTable(String sql, String dbType) {
List<String> tableList = new ArrayList<>();
List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);
SchemaStatVisitor visitor;
for (SQLStatement sqlStatement : stmtList) {
if (DbType.mysql.name().equalsIgnoreCase(dbType)) {
visitor = new MySqlSchemaStatVisitor();
} else if (DbType.hive.name().equalsIgnoreCase(dbType)) {
visitor = new HiveSchemaStatVisitor();
} else if (DbType.postgresql.name().equalsIgnoreCase(dbType)) {
visitor = new PGSchemaStatVisitor();
} else if (DbType.oracle.name().equalsIgnoreCase(dbType)) {
visitor = new OracleSchemaStatVisitor();
} else {
visitor = new SchemaStatVisitor(DbType.of(dbType));
}
sqlStatement.accept(visitor);
Map<TableStat.Name, TableStat> tables = visitor.getTables();
for (Map.Entry<TableStat.Name, TableStat> entry: tables.entrySet()){
String value = entry.getValue().toString();
if (StringUtils.isNotBlank(value)) {
tableList.add(entry.getKey().getName());
}
}
}
return tableList;
}
/**
* 根據sql獲取查詢的字段
*/
public static Map<String, String> getSelectSqlColumn(String sql, String dbType) {
Map<String, String> columnMap = new HashMap<>();
List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);
for (int i = 0; i < stmtList.size(); i++) {
SQLStatement stmt = stmtList.get(i);
if (stmt instanceof SQLSelectStatement) {
SQLSelectStatement sqlSelectStatement = (SQLSelectStatement) stmt;
SQLSelect select = sqlSelectStatement.getSelect();
SQLSelectQueryBlock query = (SQLSelectQueryBlock) select.getQuery();
List<SQLSelectItem> selectList = query.getSelectList();
selectList.forEach(item->{
columnMap.put(item.getExpr().toString(), item.getAlias());
});
}
}
return columnMap;
}
}三、測試結果
public class testDemo{
public static void main(String[] args) {
String sql = "select t.* from table1 as t left join table2 t2 on t.id = t2.id";
List<String> tableList =getSelectSqlTable(sql, "mysql");
System.out.println("獲取到的表名: "+tableList);
}
}運行結果如下:

到此這篇關于springboot引入druid解析sql的文章就介紹到這了,更多相關springboot引入druid內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
Java中的system.getProperty()的作用及使用方法
System.getProperty()?方法用于獲取系統(tǒng)屬性的值,該方法接受一個字符串參數,表示要獲取的系統(tǒng)屬性的名稱,返回值為字符串類型,表示該屬性的值,接下來通過本文給大家介紹Java中的system.getProperty()的作用及使用方法,感興趣的朋友跟隨小編一起看看吧2023-05-05
springboot?vue接口測試前后端實現模塊樹列表功能
這篇文章主要為大家介紹了springboot?vue接口測試前后端實現模塊樹列表功能,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-05-05

