Spring Boot+Hibernate全局拦截原生与托管查询并注入前置SQL方案咨询

针对你的需求——无需重构大量代码，在所有数据库查询（原生/托管）执行前设置MySQL会话变量传递用户ID，以下是两种可行的全局拦截方案：

1. Hibernate 内置的 StatementInspector（优先推荐） #

Hibernate 提供的 StatementInspector 接口，能拦截所有由Hibernate发出的SQL语句（包括HQL/JPQL、原生SQL），完美覆盖你的需求场景。

实现步骤： #

• 自定义Inspector类：实现StatementInspector，在inspect方法中拼接会话变量设置语句与原SQL：

@Component
public class UserIdSessionInspector implements StatementInspector {

    @Override
    public String inspect(String sql) {
        // 从ThreadLocal获取当前登录用户ID（需提前在登录拦截器中存入）
        Long currentUserId = UserContextHolder.getCurrentUserId();
        if (currentUserId != null) {
            return String.format("SET @CurrentUserId = %d;\n%s", currentUserId, sql);
        }
        return sql;
    }
}

• 配置Hibernate启用Inspector：在application.properties或application.yml中添加配置：

spring.jpa.properties.hibernate.session_factory.statement_inspector=com.yourpackage.UserIdSessionInspector

优势： #

• 无需修改任何业务代码，全局生效；
• 精准拦截所有Hibernate生成的查询，不会遗漏；
• 天然适配Hibernate的会话生命周期，无需处理连接池复用问题。

注意： #

如果你的应用存在直接使用JDBC（绕过Hibernate）执行的查询，这个方案无法拦截，需要结合下面的数据源层面拦截。

2. 数据源层面全局拦截 #

如果需要覆盖所有JDBC调用（包括绕过Hibernate的场景），可以通过包装数据源，在获取连接或执行语句时注入会话变量。

实现示例（基于Spring DataSource包装）： #

@Component
public class SessionVariableDataSource implements DataSource {

    @Autowired
    @Qualifier("dataSource")
    private DataSource targetDataSource;

    @Override
    public Connection getConnection() throws SQLException {
        Connection conn = targetDataSource.getConnection();
        setSessionVariable(conn);
        return conn;
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return getConnection();
    }

    private void setSessionVariable(Connection conn) throws SQLException {
        Long currentUserId = UserContextHolder.getCurrentUserId();
        if (currentUserId != null) {
            try (Statement stmt = conn.createStatement()) {
                stmt.execute(String.format("SET @CurrentUserId = %d", currentUserId));
            }
        }
    }

    // 其余DataSource接口方法全部委托给targetDataSource实现
    @Override
    public <T> T unwrap(Class<T> iface) throws SQLException {
        return targetDataSource.unwrap(iface);
    }

    @Override
    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return targetDataSource.isWrapperFor(iface);
    }

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return targetDataSource.getParentLogger();
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return targetDataSource.getLogWriter();
    }

    @Override
    public void setLogWriter(PrintWriter out) throws SQLException {
        targetDataSource.setLogWriter(out);
    }

    @Override
    public void setLoginTimeout(int seconds) throws SQLException {
        targetDataSource.setLoginTimeout(seconds);
    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return targetDataSource.getLoginTimeout();
    }
}

注意事项： #

• 若使用连接池（如HikariCP），连接会被复用，需确保每次获取连接时都重新设置会话变量（避免残留上一个用户的ID）；
• 部分连接池提供了初始化SQL配置（如HikariCP的connectionInitSql），但仅在连接创建时执行一次，无法适配用户切换场景，因此不推荐。

总结 #

• 若所有查询都通过Hibernate执行，优先使用StatementInspector，配置简单且无额外风险；
• 存在JDBC直连场景时，再结合数据源层面的拦截方案。

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