如何在 Netty 中的 ssl 握手之前优雅地关闭通道？

Question

我在我的应用程序中添加了一个限制功能，当传入的请求率超过阈值时，它需要在 SSL 握手之前关闭通道以减少 CPU 使用率。 现在我使用 Netty SslHandler和服务器模式来进行握手。 我当前的实现是在SslHandler之前添加一个ChannelInboundHandlerAdapter并重写channelActive方法：

@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception { 
    if (!limiter.tryAcquire()) {
        ctx.close();
        return;
    }

    ctx.pipeline().remove(this);
    ctx.fireChannelActive();
}

通过这种方式，通道可以在变为活动状态时关闭。 但是，我会在握手失败时收到SslHandshakeCompletionEvent 。 我阅读了 Netty SslHandler的源代码。 当 channelInactive 事件被触发时，它会在channelInactive方法中设置握手失败。 所以我想知道是否有更好的方法来关闭通道而不触发握手失败事件，因为当我的节流关闭通道时握手过程还没有开始。

Answer 1

你在错误的一端解决问题。 这个想法不是关闭太快被接受的通道，而是不要太快接受它们，这将在接受处理程序中完成，而不是在通道处理程序中完成。

Answer 2

你的做法是差不多吧，不过，这将是正确的覆盖channelRegistered方法，而不是channelActive 。

@Override
public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
    if (isGlobalLimitReached()) {
        ctx.close();
    } else {
        ctx.pipeline().remove(this);
        ctx.pipeline().addFirst(sslCtx.newHandler(ctx.alloc()));
        ctx.fireChannelRegistered();
    } 
}

此外，您需要在管道的开始处动态添加SslHandler并且仅当您的条件满足时（未达到限制）。

在这种情况下，您将不会获得SslHandshakeCompletionEvent 。 当SslHandler添加到管道并且channel.isActive()返回 true （你在上面的代码中的情况channel.isActive()时，握手开始。

channelRegistered是您可以拒绝/关闭连接的最新点。