快捷導(dǎo)航

基于Process#waitFor()阻塞問題的解決

更新時(shí)間：2021年12月13日 16:20:38 作者：zero__007

這篇文章主要介紹了Process#waitFor()阻塞問題的解決，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

Process#waitFor()阻塞問題

有時(shí)需要在程序中調(diào)用可執(zhí)行程序或腳本命令：

Process process = Runtime.getRuntime().exec(shPath);
int exitCode = process .waitFor();

Runtime.getRuntime()返回當(dāng)前應(yīng)用程序的Runtime對(duì)象，該對(duì)象的exec()方法指示Java虛擬機(jī)創(chuàng)建一個(gè)子進(jìn)程執(zhí)行指定的可執(zhí)行程序，并返回與該子進(jìn)程對(duì)應(yīng)的Process對(duì)象實(shí)例。通過Process可以控制該子進(jìn)程的執(zhí)行或獲取該子進(jìn)程的信息。

它的所有標(biāo)準(zhǔn)io(即stdin，stdout，stderr)操作都將通過三個(gè)流(getOutputStream()，getInputStream()，getErrorStream())重定向到父進(jìn)程。父進(jìn)程使用這些流來提供到子進(jìn)程的輸入和獲得從子進(jìn)程的輸出。因?yàn)檩斎牒洼敵隽魈峁┯邢薜木彌_區(qū)大小，如果讀寫子進(jìn)程的輸出流或輸入流出現(xiàn)失敗，當(dāng)緩沖區(qū)滿之后將無法繼續(xù)寫入數(shù)據(jù)，則可能導(dǎo)致子進(jìn)程阻塞，最終造成阻塞在waifor()這里。

針對(duì)這種情況，解法是要清空getInputStream()和getErrorStream()這兩個(gè)流。而且兩個(gè)流的清空一定是異步的。

static void drainInBackground(final InputStream is) {  
            new Thread(new Runnable(){  
                public void run(){  
                    try{  
                        while( is.read() >= 0 );  
                    } catch(IOException e){   
                        // return on IOException                  
                    }  
                }  
            }).start();  
        }

還有一種解法是用ProcessBuilder來創(chuàng)建Process對(duì)象，必須要使用ProcessBuilder的redirectErrorStream方法。redirectErrorStream方法設(shè)置為ture的時(shí)候，會(huì)將getInputStream()，getErrorStream()兩個(gè)流合并，自動(dòng)會(huì)清空流，無需自己處理。如果是false，getInputStream()，getErrorStream()兩個(gè)流分開，就必須自己處理，程序如下：

        try {  
            ProcessBuilder pbuilder=new ProcessBuilder("ping","192.168.0.125");  
            pbuilder.redirectErrorStream(true);  
            process=pbuilder.start();  
            reader=new BufferedReader(new InputStreamReader(process.getInputStream()));  
            String line=null;  
            while((line=reader.readLine())!=null){  
                System.out.println(line);  
            }     
            int result=process.waitFor();  
            System.out.println(result);  
        } catch (IOException e) {  
            // TODO Auto-generated catch block  
            e.printStackTrace();  
        } catch (InterruptedException e) {  
            // TODO Auto-generated catch block  
            e.printStackTrace();  
        }

            Process process = null;
            try {
                process = launchProcess(cmdlist, environment);
                StringBuilder sb = new StringBuilder();
                String output = getOutput(process.getInputStream());
                String errorOutput = getOutput(process.getErrorStream());
                sb.append(output);
                sb.append("\n");
                sb.append(errorOutput);
                boolean ret = process.waitFor(1L, TimeUnit.SECONDS);
                if (!ret) {
                    System.out.println(command + " is terminated abnormally. ret={}, str={}" + ret + " " + sb.toString());
                }
                return sb.toString();
            } catch (Throwable e) {
                System.out.println("Failed to run " + command + ", ");
                e.printStackTrace();
            } finally {
                if (null != process) {
                    process.destroy();
                }
            }
            return "";

注意process一定要釋放

Process.waitFor()導(dǎo)致主線程堵塞

今日開發(fā)的時(shí)候使用jdk自帶的運(yùn)行時(shí)變量 RunTime.getRunTime() 去執(zhí)行bash命令。

因?yàn)樵揵ash操作耗時(shí)比較長(zhǎng)，所以使用了Process.waitFor()去等待子線程運(yùn)行結(jié)束。

這個(gè)時(shí)候發(fā)現(xiàn)程序卡在waitFor()沒有繼續(xù)往下執(zhí)行。

看了官方解釋：

waitFor：等待子進(jìn)程執(zhí)行結(jié)束，或者已終止子進(jìn)程，此方法立即返回。

當(dāng)RunTime對(duì)象調(diào)用exec方法后，jvm會(huì)創(chuàng)建一個(gè)子進(jìn)程，該子進(jìn)程與jvm建立三個(gè)管道連接：標(biāo)準(zhǔn)輸入流、標(biāo)準(zhǔn)輸出流、標(biāo)準(zhǔn)錯(cuò)誤流。假設(shè)該子進(jìn)程不斷向標(biāo)準(zhǔn)輸入流、標(biāo)準(zhǔn)輸出流寫數(shù)據(jù)，而jvm不讀取的話，會(huì)導(dǎo)致緩沖區(qū)塞滿而無法繼續(xù)寫數(shù)據(jù)，最終堵塞在waitFor這里。

知道了問題所在就好處理了，我們只需要將子進(jìn)程返回的信息從緩沖區(qū)讀取出來，便可以避免主線程堵塞問題。

public static void main(String[] args){
    Process proc = RunTime.getRunTime().exec("sh /home/winnie/dataExtract.sh")
 
    // 標(biāo)準(zhǔn)輸入流（必須寫在 waitFor 之前）
    String inStr = consumeInputStream(proc.getInputStream());
    // 標(biāo)準(zhǔn)錯(cuò)誤流（必須寫在 waitFor 之前）
    String errStr = consumeInputStream(proc.getErrorStream());
 
    int retCode = proc.waitFor();
    if(retCode == 0){
        System.out.println("程序正常執(zhí)行結(jié)束");
    }
}
 
/**
*   消費(fèi)inputstream，并返回
*/
public static String consumeInputStream(InputStream is){
    BufferedReader br = new BufferedReader(new InputStreamReader(is));
    String s ;
    StringBuilder sb = new StringBuilder();
    while((s=br.readLine())!=null){
        System.out.println(s);
        sb.append(s);
    }
    return sb.toString();
}

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: