使用alibaba Druid 对sql 语句进行校验

使用druid校验sql,并格式化_黄瓜炖啤酒鸭的博客-CSDN博客_druid sql校验1,下载druid依赖<dependency> <groupId>com.alibaba</groupId> <artifactId>druid</artifactId> <version>1.1.0</version></dependency>2,测试代码如下,先...https://blog.csdn.net/qq_31866793/article/details/104289788

使用到了 Druid中的 实现了 访问者模式的 各类visitor 对 sql 进行读取与解析

<dependency>
	<groupId>com.alibaba</groupId>
	<artifactId>druid</artifactId>
        <version>1.0.17</version>
</dependency>

List<SqlDto> results = new ArrayList<SqlDto>();

		MySqlStatementParser parser = new MySqlStatementParser(sqls);
		List<SQLStatement> stmtList = parser.parseStatementList();

		// 将AST通过visitor输出
		StringBuilder out = new StringBuilder();
		MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);

		for (SQLStatement stmt : stmtList) {
			SqlDto sqlDto = new SqlDto();

			stmt.accept(visitor);
			out.append(";");

			if (stmt instanceof SQLSelectStatement) {
				sqlDto.setType(SqlType.SELECT);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLInsertStatement) {
				sqlDto.setType(SqlType.INSERT);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLUpdateStatement) {
				sqlDto.setType(SqlType.UPDATE);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLDeleteStatement) {
				sqlDto.setType(SqlType.DELETE);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLAlterTableStatement) {
				sqlDto.setType(SqlType.ALTER);
				sqlDto.setSql(out.toString());
				SQLAlterTableStatement alterStmt = (SQLAlterTableStatement) stmt;

				String tableName = alterStmt.getTableSource().toString().trim();
				int pos = tableName.indexOf('`');
				if (pos >= 0) {
					int end = tableName.lastIndexOf('`');
					tableName = tableName.substring(1, end);
				}
				sqlDto.setTableName(tableName);
			} else if (stmt instanceof SQLCreateTableStatement) {
				sqlDto.setType(SqlType.CREATE);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLExplainStatement) {
				sqlDto.setType(SqlType.EXPLAIN);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof SQLDropTableStatement) {
				sqlDto.setType(SqlType.DROP);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof MySqlRenameTableStatement) {
				sqlDto.setType(SqlType.RENAME);
				sqlDto.setSql(out.toString());
			} else if (stmt instanceof MySqlReplaceStatement) {
				sqlDto.setType(SqlType.REPLACE);
				sqlDto.setSql(out.toString());
			} else {
				sqlDto.setType(SqlType.OTHER);
				sqlDto.setSql(out.toString());
			}

			results.add(sqlDto);
			out.setLength(0);
		}

		return results;
public class SqlExplain {

    public static void main(String[] args){
        String sqls = "select ID from BCP_Prize; sel2ect name from BCP_Prize";
        MySqlStatementParser parser = new MySqlStatementParser(sqls);
        List<SQLStatement> stmtList = parser.parseStatementList();

        // 将AST通过visitor输出
        StringBuilder out = new StringBuilder();
        MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);

        for (SQLStatement stmt : stmtList) {
            stmt.accept(visitor);
            System.out.println(out + ";");
            out.setLength(0);
        }


    }
}


 

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值