欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

基于Process#waitFor()阻塞问题的解决

程序员文章站 2022-06-17 23:08:29
目录process#waitfor()阻塞问题process.waitfor()导致主线程堵塞process#waitfor()阻塞问题有时需要在程序中调用可执行程序或脚本命令:process pro...

process#waitfor()阻塞问题

有时需要在程序中调用可执行程序或脚本命令:

process process = runtime.getruntime().exec(shpath);
int exitcode = process .waitfor();

runtime.getruntime()返回当前应用程序的runtime对象,该对象的exec()方法指示java虚拟机创建一个子进程执行指定的可执行程序,并返回与该子进程对应的process对象实例。通过process可以控制该子进程的执行或获取该子进程的信息。

它的所有标准io(即stdin,stdout,stderr)操作都将通过三个流(getoutputstream(),getinputstream(),geterrorstream())重定向到父进程。父进程使用这些流来提供到子进程的输入和获得从子进程的输出。因为输入和输出流提供有限的缓冲区大小,如果读写子进程的输出流或输入流出现失败,当缓冲区满之后将无法继续写入数据,则可能导致子进程阻塞,最终造成阻塞在waifor()这里。

针对这种情况,解法是要清空getinputstream()和geterrorstream()这两个流。而且两个流的清空一定是异步的。

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来创建process对象,必须要使用processbuilder的redirecterrorstream方法。redirecterrorstream方法设置为ture的时候,会将getinputstream(),geterrorstream()两个流合并,自动会清空流,无需自己处理。如果是false,getinputstream(),geterrorstream()两个流分开,就必须自己处理,程序如下:

        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()导致主线程堵塞

今日开发的时候使用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();
}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。