Armeria装饰器使用疑问：Hibernate会话提前关闭问题

嘿，你的思路并不是完全错误，只是没适配Armeria的异步响应模型——DeferredHttpResponse代表的是一个尚未完成的异步响应，delegate.serve()调用后你立刻执行了关闭会话的代码，但这时候实际的业务逻辑可能还在后台线程中运行，自然会导致会话提前关闭的问题。

正确的实现方式：绑定会话生命周期到响应完成事件 #

Armeria的HttpResponse是异步设计的，我们需要在响应完全处理完成（不管成功还是失败）后再关闭Hibernate会话。可以通过HttpResponse.whenComplete()方法来实现这个逻辑：

// 先定义一个RequestContext的属性键，用来存储会话
private static final AttributeKey<Session> SESSION_ATTR_KEY = AttributeKey.of("hibernateSession");

@Override 
public HttpResponse serve(HttpService delegate, ServiceRequestContext ctx, HttpRequest req) {
    // 1. 为当前请求打开并绑定会话到请求上下文
    Session session = sessionFactory.openSession();
    ctx.attr(SESSION_ATTR_KEY).set(session);
    
    try {
        HttpResponse response = delegate.serve(ctx, req);
        
        // 2. 绑定会话关闭逻辑到响应完成事件
        return response.whenComplete((result, cause) -> {
            // 无论响应成功还是抛出异常，都确保会话被关闭
            if (session.isOpen()) {
                session.close();
            }
        });
    } catch (Exception e) {
        // 处理delegate调用时的同步异常，避免会话泄漏
        if (session.isOpen()) {
            session.close();
        }
        throw e;
    }
}

额外优化建议 #

• 借助RequestContext存会话：业务代码中可以通过ServiceRequestContext.current().attr(SESSION_ATTR_KEY).get()直接获取会话，不用额外传递参数。
• 扩展事务支持：如果需要事务，可以在打开会话后开启事务，然后在whenComplete中根据响应结果决定提交或回滚。
• 异常全覆盖：确保任何代码路径下会话都能被关闭，避免数据库连接资源泄漏。

另外，你也可以考虑使用Armeria的DecoratingHttpService而非DecoratingHttpServiceFunction，两者核心逻辑一致，但DecoratingHttpService在结构上更适合这种绑定请求生命周期的场景。

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