快捷導(dǎo)航

spring boot2.0實現(xiàn)優(yōu)雅停機(jī)的方法

更新時間：2018年05月04日 14:21:01 作者：cm_zdRan

這篇文章主要介紹了spring boot2.0實現(xiàn)優(yōu)雅停機(jī)的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

前期踩的坑 (spring boot 1.x)

1. 添加mavne依賴

<!-- springboot監(jiān)控 --> 
<dependency> 
  <groupId>org.springframework.boot</groupId> 
  <artifactId>spring-boot-starter-actuator</artifactId> 
</dependency>

2. 啟用shutdown

在配置文件里添加下面的配置

#啟用shutdown endpoint的HTTP訪問
endpoints.shutdown.enabled=true
#不需要驗證 
endpoints.shutdown.sensitive=false

啟動的時候可以看到下面的日志，就說明成功了

3. 優(yōu)雅停機(jī)

發(fā)送POST請求 http://localhost:8080/shutdown
如果響應(yīng)碼是404 可以嘗試POST http://localhost:8080/actuator/shutdown

spring boot 2.0

如果你使用的spring boot版本是2.x的就會發(fā)現(xiàn)，這些POST請求都會出現(xiàn)404的結(jié)果。

下面是spring boot 2.0 優(yōu)雅停機(jī)的實現(xiàn)方式。

1.修改Application啟動類

tomcat容器

@SpringBootApplication
public class ShutdownApplication {

  public static void main(String[] args) {
    SpringApplication.run(ShutdownApplication.class, args);
  }

  /**
   * 用于接受 shutdown 事件
   */
  @Bean
  public GracefulShutdown gracefulShutdown() {
    return new GracefulShutdown();
  }

  /**
   * 配置tomcat
   *
   * @return
   */
  @Bean
  public ServletWebServerFactory servletContainer() {
    TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory();
    tomcat.addConnectorCustomizers(gracefulShutdown());
    return tomcat;
  }

  /**
   * 優(yōu)雅關(guān)閉 Spring Boot。容器必須是 tomcat
   */
  private class GracefulShutdown implements TomcatConnectorCustomizer, ApplicationListener<ContextClosedEvent> {
    private final Logger log = LoggerFactory.getLogger(GracefulShutdown.class);
    private volatile Connector connector;
    private final int waitTime = 10;

    @Override
    public void customize(Connector connector) {
      this.connector = connector;
    }

    @Override
    public void onApplicationEvent(ContextClosedEvent contextClosedEvent) {
      this.connector.pause();
      Executor executor = this.connector.getProtocolHandler().getExecutor();
      if (executor instanceof ThreadPoolExecutor) {
        try {
          ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
          threadPoolExecutor.shutdown();
          if (!threadPoolExecutor.awaitTermination(waitTime, TimeUnit.SECONDS)) {
            log.warn("Tomcat 進(jìn)程在" + waitTime + " 秒內(nèi)無法結(jié)束，嘗試強(qiáng)制結(jié)束");
          }
        } catch (InterruptedException ex) {
          Thread.currentThread().interrupt();
        }
      }
    }
  }
}

Undertow容器 (沒有使用過，不保證可用)

@SpringBootApplication
public class Application {
  public static void main(String[] args) {
    SpringApplication.run(Application.class, args);
  }

  /**
   * 優(yōu)雅關(guān)閉 Spring Boot
   */
  @Component
  public class GracefulShutdown implements ApplicationListener<ContextClosedEvent> {
    @Autowired
    private GracefulShutdownWrapper gracefulShutdownWrapper;
    @Autowired
    private ServletWebServerApplicationContext context;
    @Override
    public void onApplicationEvent(ContextClosedEvent contextClosedEvent){      gracefulShutdownWrapper.getGracefulShutdownHandler().shutdown();
      try {
        UndertowServletWebServer webServer = (UndertowServletWebServer)context.getWebServer();
        Field field = webServer.getClass().getDeclaredField("undertow");
        field.setAccessible(true);
        Undertow undertow = (Undertow) field.get(webServer);
        List<Undertow.ListenerInfo> listenerInfo = undertow.getListenerInfo();
        Undertow.ListenerInfo listener = listenerInfo.get(0);
        ConnectorStatistics connectorStatistics = listener.getConnectorStatistics();
        while (connectorStatistics.getActiveConnections() > 0){}
      }catch (Exception e){
        // Application Shutdown
      }
    }
  }
  @Component
  public class GracefulShutdownWrapper implements HandlerWrapper{
    private GracefulShutdownHandler gracefulShutdownHandler;
    @Override
    public HttpHandler wrap(HttpHandler handler) {
      if(gracefulShutdownHandler == null) {
        this.gracefulShutdownHandler = new GracefulShutdownHandler(handler);
      }
      return gracefulShutdownHandler;
    }
    public GracefulShutdownHandler getGracefulShutdownHandler() {
      return gracefulShutdownHandler;
    }

  }
  @Component
  @AllArgsConstructor
  public class UndertowExtraConfiguration {
    private final GracefulShutdownWrapper gracefulShutdownWrapper;

    @Bean
    public UndertowServletWebServerFactory servletWebServerFactory() {
      UndertowServletWebServerFactory factory = new UndertowServletWebServerFactory();
      factory.addDeploymentInfoCustomizers(deploymentInfo -> deploymentInfo.addOuterHandlerChainWrapper(gracefulShutdownWrapper));
      factory.addBuilderCustomizers(builder -> builder.setServerOption(UndertowOptions.ENABLE_STATISTICS, true));
      return factory;
    }
  }
}

2. 使用 kill命令殺死進(jìn)程

使用下面的命令殺死進(jìn)程。該命令是向某個進(jìn)程發(fā)送終止信號。