Process.waitFor()导致主线程堵塞问题

╰半夏微凉° 2023-10-07 08:10 7阅读 0赞

今日开发的时候使用jdk自带的运行时变量 RunTime.getRunTime() 去执行bash命令。  
因为该bash操作耗时比较长，所以使用了Process.waitFor()去等待子线程运行结束。

这个时候发现程序卡在waitFor()没有继续往下执行。  
看了官方解释:

waitFor：等待子进程执行结束，或者已终止子进程，此方法立即返回。  
当RunTime对象调用exec方法后，jvm会创建一个子进程，该子进程与jvm建立三个管道连接：标准输入流、标准输出流、标准错误流。假设该子进程不断向标准输入流、标准输出流写数据，而jvm不读取的话，会导致缓冲区塞满而无法继续写数据，最终堵塞在waitFor这里。

知道了问题所在就好处理了， 我们只需要将子进程返回的信息从缓冲区读取出来，便可以避免主线程堵塞问题。

public static void main(String[] args){
        Process proc = RunTime.getRunTime().exec("sh /home/winnie/dataExtract.sh")
     
        // 标准输入流（必须写在 waitFor 之前）
        String inStr = consumeInputStream(proc.getInputStream());
        // 标准错误流（必须写在 waitFor 之前）
        String errStr = consumeInputStream(proc.getErrorStream());
     
        int retCode = proc.waitFor();
        if(retCode == 0){
            System.out.println("程序正常执行结束");
        }
    }
     
    /**
    *   消费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();
    }