使用druid校验sql，并格式化

青旅半醒 2023-07-04 15:58 11阅读 0赞

1，下载druid依赖

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

2，测试代码如下，先测试格式然后格式化

>     package com.coder.flink.core.sql;
>     
>     import com.alibaba.druid.sql.ast.SQLStatement;
>     
>     import java.util.List;
>     
>     public class SqlTest {
>         public static void main(String[] args) {
>             System.out.println("1111 = " + 1111);
>     //        String  sql = "SELECT * FROM abc limit2 1"; //错误的，会报错
>             String  sql = "SELECT * FROM abc limit 1";
>             String formtStr = formatSql(sql);
>             System.out.println("formtStr = " + formtStr);
>         }
>     
>         public static boolean verifySql(String sql){
>             com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser
>                     parser=new com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser(sql);
>             boolean flag=true;
>             try{
>     
>                 List<SQLStatement> smtList=parser.parseStatementList();
>             }catch(Exception e){
>                 flag=false;
>                 e.printStackTrace();
>             }
>             return flag;
>         }
>     
>         public static String formatSql(String sql){
>             //先校验sql
>             boolean b = verifySql(sql);
>             System.out.println("b = " + b);
>             //格式化
>             String formatSql=com.alibaba.druid.sql.SQLUtils.formatMySql(sql);
>             //判断是否已经格式化过
>             if(!sql.equals(formatSql)){
>                 return formatSql;
>             }
>             return null;
>         }
>     }