SpringBoot监控Tomcat活动线程数来判断是否完成请求处理

一、背景

最近项目中有一个需求，需要判断应用的请求是否已经处理完毕，想了一下，打算通过定时任务定时监控Tomcat的活动线程数来实现。

二、编码实现

新建一个SpringBoot工程，添加定时任务定时监控，代码如下：

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.embedded.tomcat.TomcatWebServer;
import org.springframework.boot.web.servlet.context.ServletWebServerApplicationContext;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

@Component
public class TomcatMonitor {
    private Logger logger = LoggerFactory.getLogger(getClass());

    @Autowired
    ServletWebServerApplicationContext applicationContext;


    @Scheduled(cron = "*/1 * * * * ?")
    public void execute() {
        TomcatWebServer webServer = (TomcatWebServer) applicationContext.getWebServer();
        String tomcatInfo = webServer.getTomcat().getConnector().getProtocolHandler().getExecutor().toString();

        System.out.println("tomcatInfo: " + tomcatInfo);
    }

}

三、测试

1. 新建一个测试请求类，代码如下：

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.TimeUnit;

@RestController
public class TestController {

    @RequestMapping("/test")
    public String test() {
        try {
            TimeUnit.SECONDS.sleep(2);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("complete OK");
        return "OK";
    }

}

2. 启动项目：

tomcatInfo: org.apache.tomcat.util.threads.ThreadPoolExecutor@49355b92[Running, pool size = 10, active threads = 0, queued tasks = 0, completed tasks = 0]
tomcatInfo: org.apache.tomcat.util.threads.ThreadPoolExecutor@49355b92[Running, pool size = 10, active threads = 0, queued tasks = 0, completed tasks = 0]

3. 访问测试接口，观察Tomcat线程池变化：

tomcatInfo: org.apache.tomcat.util.threads.ThreadPoolExecutor@49355b92[Running, pool size = 10, active threads = 1, queued tasks = 0, completed tasks = 0]
complete OK
tomcatInfo: org.apache.tomcat.util.threads.ThreadPoolExecutor@49355b92[Running, pool size = 10, active threads = 0, queued tasks = 0, completed tasks = 2]

可以看到，当有请求处理时，Tomcat的活动线程数不为0，处理完成以后回归到0，所以可以通过活动线程数判断当前Tomcat是否有请求还在处理，当然，应用到实际业务中，还有需要完善的地方。