Question

我有一个使用带有websockets的tomcat的spring应用程序。每当tomcat创建一个新线程时，我想使用DelegatingSecurityContextRunnable，即warp tomcat线程。有谁知道这是怎么做的。可以找到问题的原因。here

也许这可以通过使用AOP和一些建议来完成？

Answer 1

在Spring启动中，您可以通过挂钩到Tomcat连接器来配置包装器。以此为例：

@Bean
    public EmbeddedServletContainerFactory servletContainerFactory() {
        TomcatEmbeddedServletContainerFactory factory = new TomcatEmbeddedServletContainerFactory();

        factory.addConnectorCustomizers(new TomcatConnectorCustomizer() {

            @Override
            public void customize(Connector connector) {
                AbstractProtocol protocolHandler = (AbstractProtocol) connector.getProtocolHandler();
                TaskQueue taskqueue = new TaskQueue() {
                    @Override
                    public boolean offer(Runnable e, long timeout, TimeUnit unit) throws InterruptedException {
                        return super.offer(new MyRunnable(e), timeout, unit);
                    }

                    @Override
                    public boolean offer(Runnable o) {
                        return super.offer(new MyRunnable(o));
                    }
                };
                TaskThreadFactory tf = new TaskThreadFactory("artur-" + "-exec-", false, 0);
                ThreadPoolExecutor e = new ThreadPoolExecutor(10, 10, 1000, TimeUnit.SECONDS, taskqueue);
                taskqueue.setParent(e);
                protocolHandler.setExecutor(e);
            }
        });
        return factory;
    }

这是我的自定义Runable（这可以是任何包装器，我没有打算完全实现你的）：

static class MyRunnable implements Runnable {

        private Runnable r;

        public MyRunnable(Runnable r) {
            this.r = r;
        }

        @Override
        public void run() {
            System.out.println("Custom runable");
            runInner();
        }

        void runInner() {
            r.run();
        }

    }

以下是我的进口商品：

import java.util.concurrent.TimeUnit;

import org.apache.catalina.connector.Connector;
import org.apache.coyote.AbstractProtocol;
import org.apache.tomcat.util.threads.TaskQueue;
import org.apache.tomcat.util.threads.TaskThreadFactory;
import org.apache.tomcat.util.threads.ThreadPoolExecutor;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.context.embedded.EmbeddedServletContainerFactory;
import org.springframework.boot.context.embedded.tomcat.TomcatConnectorCustomizer;
import org.springframework.boot.context.embedded.tomcat.TomcatEmbeddedServletContainerFactory;
import org.springframework.boot.web.support.SpringBootServletInitializer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.FilterType;
import org.springframework.context.annotation.PropertySource;

这是做什么的：

Tomcat连接器初始化自己。您可以设置要使用的执行程序，在这种情况下，Tomcat将停止创建自己的配置，而是使用您的配置。

通过覆盖队列中的商品方法，您可以将Runnable包装在任何自定义Runnable中。就我而言，为了进行测试，我只需添加一个Sysout即可确保一切正常。

我使用的Threadpool实现是tomcat默认的精确副本（减去属性）。这样，行为保持不变，除了任何Runnable现在是您的委托包装器。

当我测试时，我的控制台会打印：

 Custom runable

我希望这就是你要找的东西。

我使用spring boot，但这实际上是一个tomcat问题而不是Spring问题。您可以根据具体情况调整解决方案。

- Artur

每次tomcat创建新Thread时如何运行DelegatingSecurityContextRunnable

1 个答案: