どのように優雅なシャットダウンを実現するためにspring Boot

2533 ワード

最近は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();
            }
        }
    }
}