Spring Boot Undertow将RequestLimitingHandler添加到DeploymentInfo

Question

我正在使用Spring Boot with Undertow并尝试对Undertow接受的请求数量实施一些限制，以免在压力下过载。

我在Spring Boot Undertow看到了问题的答案在同一个应用程序中添加了阻塞处理程序和NIO处理程序 ，看起来很有希望，但我不清楚应该将HttpHandler作为参数传递给RequestLimitingHandler构造函数。

有没有一种简单的方法可以使用addDeploymentInfoCustomizers方法向UndertowEmbeddedServletContainerFactory bean添加RequestLimitingHandler？

或者，如果我深入了解Undertow所基于的Xnio代码，看起来有一个选项可以设置Options.WORKER_TASK_LIMIT，但经过进一步调查，看起来XnioWorker类在3.0.10之后忽略了这个设置。 .GA发布并简单地将taskQueue设置为无界的LinkedBlockingQueue。 我错了，这也可以选择吗？

Answer 1

回答我自己的问题，以防将来帮助其他人。 解决方案是创建一个新的Undertow HandlerWrapper并在wrap（）方法中实例化新的RequestLimitingHandler对象，如下所示：

@Bean
public UndertowEmbeddedServletContainerFactory embeddedServletContainerFactory(RootHandler rootHandler) {
    UndertowEmbeddedServletContainerFactory factory = new UndertowEmbeddedServletContainerFactory();

    factory.addDeploymentInfoCustomizers(deploymentInfo -> deploymentInfo.addInitialHandlerChainWrapper(new HandlerWrapper() {

        @Override
        public HttpHandler wrap(HttpHandler handler) {
            return new RequestLimitingHandler(maxConcurrentRequests, queueSize, handler);
        }

    }));

    return factory;
}