最近发现springboot内嵌的tomcat居然不支持优雅停机,一行kill - 15命令,会导致tomcat请求处理到一半就直接停止服务了。这简直是场灾难。
那么我们要怎么实现优雅停机呢,别慌,很简单,一个类轻松轻松搞定。详细说明见springBoot如何实现优雅停机
import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.apache.catalina.connector.Connector;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.web.embedded.tomcat.TomcatServletWebServerFactory;
import org.springframework.boot.web.servlet.server.ServletWebServerFactory;
import org.springframework.context.ApplicationListener;
import org.springframework.context.annotation.Bean;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.stereotype.Component;
/**
* Tomcat优雅停机
* @author LW
* @time 2019年7月13日 上午12:05:47
*/
@Component
public class GracefulShutdownTomcat implements ApplicationListener{
private final Logger log = LoggerFactory.getLogger(GracefulShutdownTomcat.class);
private final int waitTime = 30;
private static volatile Connector connector;
/**
* 用于获取tomcat连接器
* @return
*/
@Bean
public ServletWebServerFactory servletContainer() {
TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory();
tomcat.addConnectorCustomizers(connect->{
connector = connect;
});
return tomcat;
}
/**
* 监听 停止操作kill -15
*/
@Override
public void onApplicationEvent(ContextClosedEvent contextClosedEvent) {
long startTime = System.currentTimeMillis();
//tomcat暂停对外服务
connector.pause();
//获取tomcat线程池
Executor executor = connector.getProtocolHandler().getExecutor();
if (executor instanceof ThreadPoolExecutor) {
try {
ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
//线程池优雅停止(不接收新的请求,等待任务运行完成后关闭线程池)
threadPoolExecutor.shutdown();
//堵塞等待一定时间,指定时间内关闭成功则返回true,解除堵塞;否则fasle
if (threadPoolExecutor.awaitTermination(waitTime, TimeUnit.SECONDS)) {
log.info("Tomcat thread pool closed,time:{}ms",System.currentTimeMillis()-startTime);
}
} catch (InterruptedException ex) {
Thread.currentThread().interrupt();
}
}
}
}