实现优雅地关闭Docker中的java服务

  时至今日,Docker在项目中的应用越来越普遍了,但往往会遭遇一些麻烦,比如说,有几个请求至Docker中的服务,发起了事务处理业务,但每个事务完成可能需要1-5分钟,而此时我们正要将Docker停机准备发布新版本,那如何在不影响当前业务执行的停止服务呢?有人会说,用docker stop,默认10秒延迟关停,加个延时参数-t 300,完美!有一说一,确实,但问题又来了,当前服务中的业务处理满足了,如何防止新的请求产生?所以,如何优雅地停机成了开发者关注的问题。好了,不多BB,直接开整铁汁们。

  思路:执行docker stop -t 360,Java服务会收到容器发出的SIGTERM信号,监听到此信号并执行相关操作,如注销MQ或Eureka,关闭Controller或ThreadPool等。

  先贴一段代码,关于如何监听SIGTERM,并关闭RabbitMQ。

public class XXXApplication {

    ************
  private static boolean stopHandle = false;
public static void main(String[] args) {

        SpringApplication springApplication = new SpringApplication(XXXApplication.class);

        ************

        ApplicationContext context = springApplication.run(args);

        ApplicationParam appParam = context.getBean(ApplicationParam.class);
      // 监听信号 handleSignal(
"TERM", context);
     ************
} 

private static void handleSignal(String name, ApplicationContext context) {
   Signal signal = new Signal(name);
signal.handle(signal,Signal -> {
logger.info("signal received: "+ signal.getName());
if("TERM".equals(signal.getName())){
shutdown(context);
stopHandleApi = true;
}
});
}

private static void shutdown(ApplicationContext context) {
// 关闭rabbit mq
logger.info("shutdown rabbitMQ listener! ");
RabbitListenerEndpointRegistry rer = context.getBean(RabbitListenerEndpointRegistry.class);
rer.stop();
}
}
 

  这里将stopHandle这个静态常量设为true的目的是表明程序即将停止,这样也能在其它地方应用,比如,写一个拦截器,当接收到新的请求时拦截,判断stopHandle是否为true,如果是则说明服务要停止了,不再接受新请求,同时返回404给客户端。

  

public class RestApiInterceptor  extends HandlerInterceptorAdapter {
    private static final Logger logger = LogManager.getLogger(RestApiInterceptor.class);

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        logger.info("isStopHandleApi = "+ FintechEkycApplication.isStopHandleApi());
        if(ShortenLinkSysApplication.isStopHandleApi()){
            response.setStatus(404);
            response.setContentType("text/html");
            response.setCharacterEncoding("UTF-8");
            response.getWriter().println("application is stopping! Please wait minutes and request after it restart !");
            return false;
        }else {
            return true;
        }
    }

  测试效果如图:

  总结:如果希望代码更优雅一点可以利用事件驱动,写个继承TomcatConnectorCustomizer的类来处理像Thread Pool、Web servlet等。这里就不整活了铁汁们,今天就到这里,拜拜。

posted @ 2020-07-13 17:18  游鱼迁徙  阅读(1604)  评论(0编辑  收藏  举报