快捷導(dǎo)航

Java執(zhí)行shell命令的實(shí)現(xiàn)

更新時(shí)間：2023年01月12日 10:57:02 作者：ice&wind

本文主要介紹了Java執(zhí)行shell命令的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

前言

java執(zhí)行shell命令的方式有很多種，但是在應(yīng)用的過(guò)程中，我們可能會(huì)遇上一些特殊的情況，導(dǎo)致執(zhí)行腳本失敗，不生效的場(chǎng)景。

一、案例

場(chǎng)景

java服務(wù)，如果需要服務(wù)自動(dòng)重啟。那么我們通過(guò)java執(zhí)行shell命令，使用常用jdk的方法：Runtime.getRuntime().exec(command)的方式，重啟服務(wù)，可能會(huì)導(dǎo)致重啟失敗。

原因

java執(zhí)行本地命令啟動(dòng)的是一個(gè)子進(jìn)程處理，默認(rèn)情況下子進(jìn)程與父進(jìn)程I/O通過(guò)管道相連（ProcessBuilder.Redirect.PIPE）
當(dāng)服務(wù)執(zhí)行自身重啟命令時(shí)，父進(jìn)程關(guān)閉導(dǎo)致管道連接中斷，將導(dǎo)致子進(jìn)程也崩潰，而無(wú)法完成后續(xù)啟動(dòng)

解決方案

設(shè)置子進(jìn)程的I/O源或目標(biāo)將與當(dāng)前進(jìn)程的相同，兩者相互獨(dú)立
設(shè)置子進(jìn)程IO輸出重定向到指定文件

這里我們采用第一種解決方案

ProcessBuilder pb = new ProcessBuilder("service","java-service","restart");
pb.redirectOutput(ProcessBuilder.Redirect.INHERIT);
pb.redirectError(ProcessBuilder.Redirect.INHERIT);
pb.redirectInput(ProcessBuilder.Redirect.INHERIT);
pb.start();

ProcessBuilder 也是J2SE1.5 就有了的類(lèi)。此類(lèi)用于創(chuàng)建操作系統(tǒng)進(jìn)程，它提供一種啟動(dòng)和管理進(jìn)程（也就是應(yīng)用程序）的方法。在J2SE 1.5之前，都是由Process類(lèi)處來(lái)實(shí)現(xiàn)進(jìn)程的控制管理。

static ProcessBuilder.Redirect    DISCARD    
表示將丟棄子進(jìn)程輸出。
static ProcessBuilder.Redirect    INHERIT    
表示子進(jìn)程I / O源或目標(biāo)將與當(dāng)前進(jìn)程的相同。
static ProcessBuilder.Redirect    PIPE    
表示子進(jìn)程I / O將通過(guò)管道連接到當(dāng)前Java進(jìn)程。

ProcessBuilder 可配置執(zhí)行腳本的子進(jìn)程I / O源或目標(biāo)將與當(dāng)前進(jìn)程的相同。綁定之后，執(zhí)行重啟，就能成功，管道不會(huì)斷開(kāi)。

ProcessBuilder 可參考連接：

ProcessBuilder api
ProcessBuilder 中文文檔

二、拓展

創(chuàng)建臨時(shí)腳本，執(zhí)行shell命令

<dependency>
   <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.11.0</version>
</dependency>

可直接復(fù)制使用，注意引入maven包

public static String runAndResult(String cmd){
        StringBuilder sb = new StringBuilder();
        BufferedReader br = null;
        boolean execFlag = true;
        String uuid = UUID.randomUUID().toString().replace("-","");
        String tempFileName = "./temp" + uuid +".sh";
        try {
            String osName = System.getProperty("os.name").toUpperCase(Locale.ENGLISH);
            if (osName.matches("^(?i)LINUX.*$") || osName.contains("MAC")) {
                FileWriter execute_fw = new FileWriter(tempFileName);
                BufferedWriter execute_bw=new BufferedWriter(execute_fw);
                execute_bw.write(cmd + "\n");
                execute_bw.close();
                execute_fw.close();
                String command ="bash " + tempFileName;
                Process p = Runtime.getRuntime().exec(command);
                p.waitFor();
                br = new BufferedReader(new InputStreamReader(p.getInputStream()));
                String line;
                while ((line = br.readLine()) != null) {
                    sb.append(System.lineSeparator());
                    sb.append(line);
                }
                br.close();
                br = new BufferedReader(new InputStreamReader(p.getErrorStream()));
                while ((line = br.readLine()) != null) {
                    sb.append(System.lineSeparator());
                    sb.append(line);
                    if (line.length() > 0){
                        execFlag = false;
                    }
                }
                br.close();
                if (execFlag){
                }else {
                    throw new RuntimeException(sb.toString());
                }
            }else {
                throw new RuntimeException("不支持的操作系統(tǒng)類(lèi)型");
            }
        } catch (Exception e) {
            log.error("執(zhí)行失敗",e);
        }finally {
            if (br != null){
                try {
                    br.close();
                } catch (IOException e) {
                    log.error("io異常",e);
                }
            }
            FileUtils.deleteQuietly(new File(tempFileName));
        }
        return sb.toString();
    }