如何在Hibernate + Spring Boot中统计含IN子句的查询？

直接用StatementInspector做字符串匹配找IN关键字确实不靠谱——不仅容易误判（比如注释、字符串常量里的"IN"，或者大小写混合的"In"），遍历字符串的性能在高并发场景下也会拖后腿。下面是几个更高效精准的方案：

1. 基于SQL抽象语法树（AST）解析（首推） #

用成熟的SQL解析库直接把SQL转换成AST，遍历语法树就能精准定位所有IN子句，还能直接获取参数数量，完全避免字符串匹配的弊端。

以Java生态的JSQLParser为例，简单实现代码如下：

import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.select.PlainSelect;
import net.sf.jsqlparser.statement.select.Select;
import net.sf.jsqlparser.util.SelectVisitorAdapter;
import net.sf.jsqlparser.expression.operators.relational.InExpression;
import net.sf.jsqlparser.util.deparser.ExpressionDeParser;

public class InClauseAnalyzer {
    public static void countInClauses(String sql) throws JSQLParserException {
        Statement stmt = CCJSqlParserUtil.parse(sql);
        if (!(stmt instanceof Select)) {
            return; // 这里只处理SELECT语句，其他DML可同理扩展
        }

        Select selectStmt = (Select) stmt;
        ExpressionDeParser inClauseVisitor = new ExpressionDeParser() {
            @Override
            public void visit(InExpression inExpr) {
                super.visit(inExpr);
                // 直接从AST中获取IN子句的参数数量
                int paramCount = inExpr.getRightItemsList().getExpressions().size();
                System.out.println("检测到IN子句，参数数量：" + paramCount);
            }
        };

        // 遍历SELECT语句的WHERE子句找IN表达式
        selectStmt.getSelectBody().accept(new SelectVisitorAdapter() {
            @Override
            public void visit(PlainSelect plainSelect) {
                if (plainSelect.getWhere() != null) {
                    plainSelect.getWhere().accept(inClauseVisitor);
                }
            }
        });
    }
}

其他语言也有对应的解析库：Python用sqlparse，Go用go-sqlparser，都能实现类似逻辑。这种方法的核心优势是语法级精准匹配，不会把无关的"IN"当成子句，性能也比字符串匹配稳定。

2. ORM框架层面拦截（如果用了ORM） #

如果你的项目用了MyBatis、Hibernate这类ORM框架，可以直接在框架层面做拦截统计：

• 比如MyBatis，写一个Executor拦截器，在SQL执行前拿到BoundSql，再用AST解析工具处理生成的SQL；
• 如果是用MyBatis的<foreach>生成IN子句，甚至可以直接在Mapper的方法调用时统计传入的集合大小，不用解析SQL。

3. 数据库端全局统计（无需改应用代码） #

如果有权限操作数据库，可以直接用数据库自带的监控工具收集SQL，再批量解析统计：

• MySQL开启performance_schema，捕获所有执行的SQL语句，导出后用解析工具批量处理；
• PostgreSQL启用pg_stat_statements，收集执行过的SQL，再用pg_query函数解析AST统计IN子句。

这种方式适合做全链路的SQL统计，不用修改应用代码，对业务无侵入。

内容的提问来源于stack exchange，提问作者rohanagarwal