需要實(shí)現(xiàn)看門狗功能，定時(shí)檢測另外一個(gè)程序是否在運(yùn)行，使用 crontab 僅可以實(shí)現(xiàn)檢測程序是否正在運(yùn)行，無法做到擴(kuò)展，如：手動(dòng)重啟、程序升級(如果只需要實(shí)現(xiàn)自動(dòng)升級功能可以使用 inotify)等功能；最后決定使用 Spring Boot 調(diào)用 Shell 腳本來實(shí)現(xiàn)

一、腳本

1.1 啟動(dòng)腳本

#!/bin/bashps -ef | grep 'demo-app-0.0.1-SNAPSHOT.jar' | grep -v 'grep'if [ '$?' -eq 0 ]then# sleepecho $(date '+%Y-%m-%d %H:%M:%S') 'process already started!'elsenohup java -jar -server /project/watchdog/demo-app-0.0.1-SNAPSHOT.jar &echo $(date '+%Y-%m-%d %H:%M:%S') 'process has been started!'fi

1.2 重啟腳本

#!/bin/bashpid=`ps -ef | grep 'demo-app-0.0.1-SNAPSHOT.jar' | grep -v 'grep' | awk ’{print $2}’`for id in $piddo kill -9 $id echo 'killed $id'donenohup java -jar -server /project/watchdog/demo-app-0.0.1-SNAPSHOT.jar &echo $(date '+%Y-%m-%d %H:%M:%S') 'process has been restarted!'

二、功能實(shí)現(xiàn)

將腳本放置在程序的資源目錄下，每次程序啟動(dòng)時(shí)將腳本讀取到指定位置，然后再通過定時(shí)任務(wù)執(zhí)行腳本

配置內(nèi)容：

shell: directory: /project/watchdog startupFileName: startup.sh restartFileName: restart.sh@Configuration@ConfigurationProperties(prefix = 'shell')public class ShellProperties { private String directory; private String startupFileName; private String restartFileName; /* getter & setter */ public String getFullName(String fileName) { return directory + File.separator + fileName; }}

2.1 啟動(dòng)時(shí)將腳本讀取到指定位置

@Componentpublic class InitRunner implements CommandLineRunner { @Autowired private ShellProperties shellProperties; @Autowired ResourceLoader resourceLoader; @Override public void run(String... args) throws Exception { generateFile(shellProperties.getStartupFileName()); generateFile(shellProperties.getRestartFileName()); } private void generateFile(String fileName) throws IOException { String fileFullName = shellProperties.getFullName(fileName); File file = new File(fileFullName); if(file.exists()) { return; } // 如果文件已存在，F(xiàn)ileWriter 會(huì)先刪除再新建 FileWriter fileWriter = new FileWriter(fileFullName); Resource resource = resourceLoader.getResource('classpath:' + fileName); InputStream inputStream = resource.getInputStream(); InputStreamReader inputStreamReader = new InputStreamReader(inputStream); BufferedReader bufferedReader = new BufferedReader(inputStreamReader); String data; while ((data = bufferedReader.readLine()) != null) { fileWriter.write(data + 'n'); } bufferedReader.close(); inputStreamReader.close(); inputStream.close(); fileWriter.close(); // 設(shè)置權(quán)限，否則會(huì)報(bào) Permission denied file.setReadable(true); file.setWritable(true); file.setExecutable(true); }}

2.2 定時(shí)任務(wù)定時(shí)任務(wù)執(zhí)行腳本

@Componentpublic class ShellTask { private static final Logger logger = LoggerFactory.getLogger(ShellTask.class); @Autowired private ShellProperties shellProperties; @Scheduled(cron = '0/10 * * * * ? ') public void start() throws IOException { executeShell(shellProperties.getStartupFileName()); } private void executeShell(String fileName) throws IOException { String fileFullName = shellProperties.getFullName(fileName); File file = new File(fileFullName); if(!file.exists()) { logger.error('file {} not existed!', fileFullName); return; } ProcessBuilder processBuilder = new ProcessBuilder(fileFullName); processBuilder.directory(new File(shellProperties.getDirectory())); Process process = processBuilder.start();// String input;// BufferedReader stdInput = new BufferedReader(new InputStreamReader(process.getInputStream()));// BufferedReader stdError = new BufferedReader(new InputStreamReader(process.getErrorStream()));// while ((input = stdInput.readLine()) != null) {// logger.info(input);// }// while ((input = stdError.readLine()) != null) {// logger.error(input);// } int runningStatus = 0; try { runningStatus = process.waitFor(); } catch (InterruptedException e) { logger.error('shell', e); } if(runningStatus != 0) { logger.error('failed.'); }else { logger.info('success.'); } }}

2.3 擴(kuò)展

本例只實(shí)現(xiàn)了定時(shí)檢測程序是否運(yùn)行，如果沒有運(yùn)行則啟動(dòng)程序；如有需要可以添加接口，調(diào)用接口重啟程序；或者添加定時(shí)任務(wù)定時(shí)檢測程序是否有更新，如果有更新則下載新的 jar 包然后重啟程序

看門狗程序自己可以使用 crontab 定時(shí)檢測是否正在運(yùn)行，模仿上面的啟動(dòng)腳本編寫看門狗的啟動(dòng)腳本，然后添加定時(shí)任務(wù)：

crontab -e*/10 * * * * /project/watchdog/watchdog.shsudo systemctl reload crond.service

完整代碼：GitHub

參考：java去調(diào)用并執(zhí)行shell腳本以及問題總結(jié)

總結(jié)

到此這篇關(guān)于Spring Boot調(diào)用 Shell 腳本實(shí)現(xiàn)看門狗功能的文章就介紹到這了,更多相關(guān)spring boot 看門狗內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！