Springboot项目平滑关闭及自动化关闭脚本

 package cnkj.site.utils; import org.apache.catalina.LifecycleException; import org.apache.catalina.connector.Connector; import org.apache.catalina.util.LifecycleBase; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.boot.web.embedded.tomcat.TomcatConnectorCustomizer; import org.springframework.boot.web.embedded.tomcat.TomcatServletWebServerFactory; import org.springframework.boot.web.servlet.server.ConfigurableServletWebServerFactory; import org.springframework.context.ApplicationListener; import org.springframework.context.annotation.Bean; import org.springframework.context.event.ContextClosedEvent; import java.util.concurrent.Executor; import java.util.concurrent.ThreadPoolExecutor; import java.util.concurrent.TimeUnit; /* * @version 1.0 created by Carol on 2019/4/25 16:22 */ public class GracefulShutdown implements TomcatConnectorCustomizer, ApplicationListener { private static final Logger LOGGER = LoggerFactory.getLogger(GracefulShutdown.class); private volatile Connector connector; @Override public void customize(Connector connector) { this.connector = connector; } @Override public void onApplicationEvent(ContextClosedEvent event) { try { // 指定执行的方法 shutdown(); //手动清理内存 System.gc(); LOGGER.warn("清理内存完毕，正在退出服务......"); if (this.connector == null){ return; } this.connector.pause(); LOGGER.warn("关闭全部连接......"); Executor executor = this.connector.getProtocolHandler().getExecutor(); if (executor instanceof ThreadPoolExecutor) { try { ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor; threadPoolExecutor.shutdown(); LOGGER.warn("当前服务线程池被关闭"); if (!threadPoolExecutor.awaitTermination(30, TimeUnit.SECONDS)) { LOGGER.warn("Tomcat thread pool did not shut down gracefully within 30 seconds. Proceeding with forceful shutdown"); } } catch (InterruptedException ex) { Thread.currentThread().interrupt(); } } this.connector.stop(); } catch (LifecycleException e) { e.printStackTrace(); } } @Bean public GracefulShutdown gracefulShutdown() { return new GracefulShutdown(); } @Bean public ConfigurableServletWebServerFactory webServerFactory(final GracefulShutdown gracefulShutdown) { TomcatServletWebServerFactory factory = new TomcatServletWebServerFactory(); factory.addConnectorCustomizers(gracefulShutdown); return factory; } /** * 执行服务关闭前的一些定制化操作 * 通常需要确认以下步骤 * 1.关闭kafka等数据连接 * 2.flush内存中全部的未处理数据 * 3.清理服务中全部待处理的数据 */ public void shutdown(){} }

 import cnkj.site.utils.GracefulShutdown; import org.springframework.stereotype.Component; /* * @version 1.0 created by Carol on 2019/4/25 16:39 */ @Component public class Shutdown extends GracefulShutdown { @Override public void shutdown() { // TODO 定制化关闭操作流程 // 关闭 kafka 消费 // flush全部读写流 // 清空队列 // 关闭全部文件流读写 } }

 package cn.migu.log.component; import cnkj.site.utils.HttpCommonUtil; import cnkj.site.CommonInfo; import org.springframework.boot.CommandLineRunner; import org.springframework.stereotype.Component; /* * @version 1.0 created by LXW on 2019/3/14 17:05 */ @Component public class ApplicationStarterRunner implements CommandLineRunner { @Override public void run(String... args) throws Exception { // 设置服务名 commonInfo.setSERVICE_NAME("Service-Name"); // 自动设置服务启动后的进程号 commonInfo.setSERVICE_PID(HttpCommonUtil.getCurrentPid()); } }

 package cnkj.site.utils; import lombok.Builder; import lombok.Data; import org.springframework.boot.actuate.info.Info; import org.springframework.boot.actuate.info.InfoContributor; import org.springframework.stereotype.Component; import java.util.HashMap; import java.util.Map; @Data @Component public class CommonInfo implements InfoContributor { //当前服务名 private String SERVICE_NAME="SERVICE_NAME"; //服务当前状态 private int SERVICE_PID; @Override public void contribute(Info.Builder builder) { builder.withDetail("SERVICE_NAME",SERVICE_NAME); builder.withDetail("SERVICE_PID", SERVICE_PID); } public void clearAll(){ this.SERVICE_NAME=""; this.SERVICE_PID=-1; } public Map getAll(){ Map map = new HashMap(); map.put("SERVICE_NAME", getSERVICE_NAME()); map.put("SERVICE_PID", getSERVICE_PID()); return map; } }

 package cnkj.site.utils; import javax.servlet.http.HttpServletRequest; import java.lang.management.ManagementFactory; import java.net.InetAddress; import java.net.UnknownHostException; /* * @version 1.0 created by Carol on 2018/10/25 10:04 */ public class HttpCommonUtil { /** * 获取当前服务的PID * @return PID */ public static Integer getCurrentPid(){ String name = ManagementFactory.getRuntimeMXBean().getName(); String pid = name.split("@")[0]; return Integer.valueOf(pid); } }

 #服务关闭 management.endpoint.shutdown.enabled=true #监控相关 management.endpoint.prometheus.enabled=true management.endpoints.web.exposure.include=info