spring boot2.0实现优雅停机的方法

<!-- springboot监控 --> 
<dependency> 
  <groupid>org.springframework.boot</groupid> 
  <artifactid>spring-boot-starter-actuator</artifactid> 
</dependency> 

#启用shutdown endpoint的http访问
endpoints.shutdown.enabled=true
#不需要验证 
endpoints.shutdown.sensitive=false

@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;
  }

  /**
   * 优雅关闭 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 进程在" + waittime + " 秒内无法结束，尝试强制结束");
          }
        } catch (interruptedexception ex) {
          thread.currentthread().interrupt();
        }
      }
    }
  }
}

@springbootapplication
public class application {
  public static void main(string[] args) {
    springapplication.run(application.class, args);
  }

  /**
   * 优雅关闭 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;
    }
  }
}

kill -15 [pid]