Springboot項(xiàng)目平滑關(guān)閉及自動(dòng)化關(guān)閉腳本

更新時(shí)間：2019年05月31日 09:16:58 作者：CarolLXW

這篇文章主要為大家詳細(xì)介紹了Springboot項(xiàng)目平滑關(guān)閉及自動(dòng)化關(guān)閉腳本，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

Springboot項(xiàng)目平滑關(guān)閉及自動(dòng)化關(guān)閉腳本，供大家參考，具體內(nèi)容如下

核心代碼

GracefulShutdown.java
Shutdown.java
ApplicationStarterRunner.java
CommonInfo.java
HttpCommonUtil.java
application.properties

操作步驟

核心代碼

GracefulShutdown.java

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<ContextClosedEvent> {
 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 {
  // 指定執(zhí)行的方法
  shutdown();
  //手動(dòng)清理內(nèi)存
  System.gc();
  LOGGER.warn("清理內(nèi)存完畢，正在退出服務(wù)......");
  if (this.connector == null){
  return;
  }
  this.connector.pause();
  LOGGER.warn("關(guān)閉全部連接......");
  Executor executor = this.connector.getProtocolHandler().getExecutor();
  if (executor instanceof ThreadPoolExecutor) {
  try {
   ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
   threadPoolExecutor.shutdown();
   LOGGER.warn("當(dāng)前服務(wù)線程池被關(guān)閉");
   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;
 }


 /**
 * 執(zhí)行服務(wù)關(guān)閉前的一些定制化操作
 * 通常需要確認(rèn)以下步驟
 * 1.關(guān)閉kafka等數(shù)據(jù)連接
 * 2.flush內(nèi)存中全部的未處理數(shù)據(jù)
 * 3.清理服務(wù)中全部待處理的數(shù)據(jù)
 */
 public void shutdown(){}
}

Shutdown.java

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 定制化關(guān)閉操作流程
 // 關(guān)閉 kafka 消費(fèi)
 // flush全部讀寫流
 // 清空隊(duì)列
 // 關(guān)閉全部文件流讀寫
 }
}

ApplicationStarterRunner.java

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 {
 // 設(shè)置服務(wù)名
 commonInfo.setSERVICE_NAME("Service-Name");
 // 自動(dòng)設(shè)置服務(wù)啟動(dòng)后的進(jìn)程號(hào)
 commonInfo.setSERVICE_PID(HttpCommonUtil.getCurrentPid());
 }
}

CommonInfo.java

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 {

 //當(dāng)前服務(wù)名
 private String SERVICE_NAME="SERVICE_NAME";
 //服務(wù)當(dāng)前狀態(tài)
 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;
 }


}

HttpCommonUtil.java

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 {

 /**
 * 獲取當(dāng)前服務(wù)的PID
 * @return PID
 */
 public static Integer getCurrentPid(){
 String name = ManagementFactory.getRuntimeMXBean().getName();
 String pid = name.split("@")[0];
 return Integer.valueOf(pid);
 }
}

application.properties

#服務(wù)關(guān)閉
management.endpoint.shutdown.enabled=true
#監(jiān)控相關(guān)
management.endpoint.prometheus.enabled=true
management.endpoints.web.exposure.include=info

操作步驟

項(xiàng)目使用步驟：

1.拷貝上面的 Shutdown.java 代碼到自己的項(xiàng)目中
2.在 Shutdown.java 文件中的shutdown 方法中寫定制化的關(guān)閉操作流程

腳本使用步驟：

1.從git獲取最新的項(xiàng)目關(guān)閉腳本地址
2.壓縮server_close 為server_close.zip
3.上傳 server_close.zip 到你服務(wù)所在服務(wù)器上的 /data/shell 路徑下
4.配置環(huán)境變量 vim /etc/profile
5.在profile文件的最下面新增 export PATH=/data/shell/server_close:$PATH
6.保存并退出 :wq
7.如果提示 /bin/bash^M: bad interpreter: No such file or directory，請(qǐng)vim serviceControll.sh，然后 :set fileformat=unix ，然后 :wq 保存并退出即可
8.cd /data/shell/server_close & ./serviceControll.sh 運(yùn)行即可使用服務(wù)關(guān)閉腳本

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: