Java中通过Hibernate或JDBC向Firebird触发器传递会话用户

这个需求我之前帮朋友处理过，结合Hibernate和Firebird的会话上下文，有几个靠谱的落地方案，我给你拆解清楚：

Hibernate自带了拦截器和事件监听机制，可以在会话或事务的关键节点插入自定义逻辑。我们可以利用这个特性，在事务启动时自动执行设置用户上下文的SQL。

示例代码：自定义会话监听器 #

public class FirebirdUserContextListener implements SessionEventListener {
    // 用ThreadLocal存储当前请求的用户ID，避免多线程串用
    private static final ThreadLocal<String> CURRENT_USER_ID = new ThreadLocal<>();

    public static void setCurrentUserId(String userId) {
        CURRENT_USER_ID.set(userId);
    }

    public static void clearCurrentUserId() {
        CURRENT_USER_ID.remove();
    }

    @Override
    public void transactionStarted(TransactionStartedEvent event) {
        String userId = CURRENT_USER_ID.get();
        if (userId == null) return;

        Session session = event.getSession();
        // 执行Firebird上下文设置SQL
        session.createNativeQuery("SELECT rdb$set_context('USER_SESSION','CURRENT_USER', :userId) FROM rdb$database")
               .setParameter("userId", userId)
               .executeUpdate();
    }
}

注册监听器（Spring Boot环境） #

在LocalSessionFactoryBean中配置监听器：

@Bean
public LocalSessionFactoryBean sessionFactory(DataSource dataSource) {
    LocalSessionFactoryBean sessionFactory = new LocalSessionFactoryBean();
    sessionFactory.setDataSource(dataSource);
    sessionFactory.setPackagesToScan("com.yourpackage.entity");
    
    // 注册自定义监听器
    sessionFactory.setSessionEventListenerRegistry(registry -> {
        registry.addEventListener(EventType.TRANSACTION_STARTED, new FirebirdUserContextListener());
    });
    
    sessionFactory.setHibernateProperties(hibernateProperties());
    return sessionFactory;
}

如果想更底层地解决问题，直接在JDBC连接层面做拦截会更稳妥——毕竟Hibernate所有操作最终都依赖JDBC连接。这个方案能确保所有通过连接池的操作（包括Hibernate之外的JDBC调用）都带上正确的用户ID，还能处理连接池复用的问题。

示例代码：自定义Hikari连接拦截器 #

public class FirebirdConnectionInterceptor implements ConnectionInterceptor {
    @Override
    public void afterConnectionAcquired(Connection connection) throws SQLException {
        String userId = FirebirdUserContextListener.getCurrentUserId();
        if (userId == null) return;

        // 执行上下文设置SQL
        try (PreparedStatement stmt = connection.prepareStatement(
            "SELECT rdb$set_context('USER_SESSION','CURRENT_USER', ?) FROM rdb$database"
        )) {
            stmt.setString(1, userId);
            stmt.executeQuery();
        }
    }

    // 实现其他空方法
    @Override public void beforeConnectionReleased(Connection connection) {}
    @Override public void afterConnectionReleased(Connection connection) {}
    @Override public void beforeConnectionAcquired() {}
}

配置拦截器 #

在application.properties中添加：

spring.datasource.hikari.connection-interceptors=com.yourpackage.FirebirdConnectionInterceptor

如果你的项目用了Spring，可以用AOP切面自动给所有事务方法注入用户上下文逻辑，不用手动在业务代码里写重复代码。

示例代码：自定义切面 #

@Aspect
@Component
public class FirebirdUserContextAspect {
    @Autowired
    private EntityManager em;

    @Before("@annotation(org.springframework.transaction.annotation.Transactional)")
    public void injectUserContext() {
        String userId = FirebirdUserContextListener.getCurrentUserId();
        if (userId == null) return;

        em.createNativeQuery("SELECT rdb$set_context('USER_SESSION','CURRENT_USER', :userId) FROM rdb$database")
          .setParameter("userId", userId)
          .executeUpdate();
    }
}

• 用户ID的线程安全：一定要用ThreadLocal存储当前请求的用户ID，在请求入口（比如Filter、Spring MVC拦截器）设置，请求结束后清理，避免线程复用导致用户串用。
• 连接池复用处理：如果用了连接池，方案2是最优选择——因为连接会被不同请求复用，必须在每次获取连接时重新设置上下文，否则会残留上一个用户的信息。
• 性能影响：rdb$set_context是Firebird的轻量级操作，几乎不会有性能开销，不用过度担心。

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