如何用 Java 校验 SQL 语句的合法性？有这 5 种解决方案

开篇即进入正题。

方案一：使用 JDBC API 中提供的 Statement 接口的 execute()方法

要在 Java 中校验 SQL 语句的合法性，可以使用 JDBC API 中提供的 Statement 接口的 execute()方法。这个方法会尝试执行给定的 SQL 语句，如果 SQL 语句不合法，则会抛出一个 SQLException 异常。因此，我们可以利用这个异常来判断 SQL 语句的合法性。

import java.sql.*; public class SQLValidator {     public static boolean validateSQL(String sql) {        try {            Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/mydatabase", "username", "password");            Statement stmt = conn.createStatement();            stmt.execute(sql);            return true;        } catch (SQLException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // false        System.out.println(validateSQL(sql3)); // false    }}
在这个示例代码中，validateSQL()方法接受一个 SQL 语句作为参数，然后尝试执行这个 SQL 语句。如果执行成功，返回 true，否则返回 false。在 main()方法中，我们调用了 validateSQL()方法来校验三个 SQL 语句的合法性，并打印了结果。

需要注意的是，这个方法只能判断 SQL 语句的语法是否合法，而无法判断 SQL 语句的语义是否合法。因此，如果应用程序允许用户输入 SQL 语句，一定要进行严格的输入校验和过滤，避免 SQL 注入攻击。

方案二：使用 JSqlParser 这个 Java 库

如果你不希望实际执行 SQL 语句，而只是想校验 SQL 语句的合法性，可以使用 JSqlParser 这个 Java 库。这个库可以将 SQL 语句解析成 Java 对象，然后你可以对这些 Java 对象进行检查，以判断 SQL 语句是否合法。

以下是一个简单的示例代码：

import net.sf.jsqlparser.JSQLParserException;import net.sf.jsqlparser.parser.CCJSqlParserUtil;import net.sf.jsqlparser.statement.Statement; public class SQLValidator {     public static boolean validateSQL(String sql) {        try {            Statement stmt = CCJSqlParserUtil.parse(sql);            return true;        } catch (JSQLParserException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

在这个示例代码中，validateSQL()方法使用 JSqlParser 库将 SQL 语句解析成 Java 对象。如果解析成功，返回 true，否则返回 false。在 main()方法中，我们调用了 validateSQL()方法来校验三个 SQL 语句的合法性，并打印了结果。

需要注意的是，JSqlParser 库只能检查 SQL 语句的语法是否合法，而无法检查 SQL 语句的语义是否合法。因此，同样需要进行严格的输入校验和过滤，避免 SQL 注入攻击。

方案三：使用正则表达式检查 SQL 语句的格式是否正确

使用正则表达式检查 SQL 语句的格式是否正确。例如，可以检查 SQL 语句是否以 SELECT、UPDATE、DELETE、INSERT 等关键字开头，是否包含必需的关键字和语法元素等。

import java.util.regex.Pattern; public class SQLValidator {    private static final String SELECT_PATTERN = "^\\s*SELECT.*";    private static final String UPDATE_PATTERN = "^\\s*UPDATE.*";    private static final String DELETE_PATTERN = "^\\s*DELETE.*";    private static final String INSERT_PATTERN = "^\\s*INSERT.*";     public static boolean validateSQL(String sql) {        if (Pattern.matches(SELECT_PATTERN, sql)) {            // 校验SELECT语句的合法性            return true;        } else if (Pattern.matches(UPDATE_PATTERN, sql)) {            // 校验UPDATE语句的合法性            return true;        } else if (Pattern.matches(DELETE_PATTERN, sql)) {            // 校验DELETE语句的合法性            return true;        } else if (Pattern.matches(INSERT_PATTERN, sql)) {            // 校验INSERT语句的合法性            return true;        } else {            // SQL语句格式不正确            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

方案四：使用 ANTLR 等工具生成 SQL 语法解析器，然后使用生成的解析器解析 SQL 语句，以判断 SQL 语句的合法性

ANTLR 是一种流行的解析器生成器，可以根据定义的语法规则生成解析器。

以下是一个简单的示例代码：

import org.antlr.v4.runtime.*;import org.antlr.v4.runtime.tree.*; public class SQLValidator {    public static boolean validateSQL(String sql) {        try {            CharStream input = CharStreams.fromString(sql);            SQLLexer lexer = new SQLLexer(input);            CommonTokenStream tokens = new CommonTokenStream(lexer);            SQLParser parser = new SQLParser(tokens);            ParseTree tree = parser.statement();            return true;        } catch (Exception e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

在这个示例代码中，我们使用 ANTLR 生成了一个 SQL 语法解析器，并在 validateSQL()方法中使用这个解析器来解析 SQL 语句。如果解析成功，则说明 SQL 语句格式正确，返回 true，否则返回 false。

方案五：使用 Apache Calcite 等 SQL 解析器库来解析 SQL 语句

Apache Calcite 是一个强大的 SQL 解析器和优化器，它支持大多数 SQL 语法，并能够将 SQL 语句解析为抽象语法树（AST）。

import org.apache.calcite.sql.SqlNode;import org.apache.calcite.sql.parser.SqlParseException;import org.apache.calcite.sql.parser.SqlParser;import org.apache.calcite.sql.parser.SqlParser.Config;import org.apache.calcite.sql.parser.SqlParserImplFactory; public class SQLValidator {    public static boolean validateSQL(String sql) {        try {            Config config = SqlParser.config();            SqlParserImplFactory factory = config.parserFactory();            SqlParser parser = SqlParser.create(sql, config.withParserFactory(factory));            SqlNode node = parser.parseStmt();            return true;        } catch (SqlParseException e) {            return false;        }    }     public static void main(String[] args) {        String sql1 = "SELECT * FROM mytable WHERE id = 1";        String sql2 = "SELECT * FROM mytable WHERE id = '1'";        String sql3 = "SELECT * FROM mytable WHERE id = ;DROP TABLE mytable;";         System.out.println(validateSQL(sql1)); // true        System.out.println(validateSQL(sql2)); // true        System.out.println(validateSQL(sql3)); // false    }}

在这个示例代码中，我们使用 Apache Calcite 库来解析 SQL 语句。validateSQL()方法首先创建一个 SqlParser 对象，并使用它来解析传入的 SQL 语句。如果解析成功，则返回 true，否则返回 false。

总结

总的来说，使用 JDBC API 和 JSqlParser 库、正则表达式、ANTLR 解析器生成器或 Apache Calcite 库都可以实现校验 SQL 语句的合法性。具体使用哪种方法取决于你的需求和个人喜好。