一、问题描述:
Java利用Runtime.getRuntime().exec()调用Python脚本,如果不及时捕获进程的输出,会导致JAVA线程的阻塞。看似被调用的进程没有退出,但无法继续往下执行。
二、阻塞原因:
Java执行Runtime.getRuntime().exec()时,JVM会创建一个新的子进程去调用Python脚本,同时该子进程与JVM建立三个连接管道:标准输入流(getInputStream)、标准输出流(getOutputStream)、标准错误流(getErrorStream)。如果子进程不断的向标准输出流中写入数据,而主线程并没有读取,会导致读写缓冲区被填满,最终导致线程堵塞。
process = Runtime.getRuntime().exec(arg);
BufferedReader in = new BufferedReader(new InputStreamReader(process.getInputStream()));
如上,如果此时直接运行程序,发现Python脚本是被可以执行的。为什么?主线程也没有读取数据,线程不是应该阻塞吗?
因为当调用 process.getInputStream()后,只是建立了子进程与JVM之间的连接管道,并没有建立真正的缓冲区,所以不存在缓冲区被填满从而导致线程阻塞这一说。
process = Runtime.getRuntime().exec(arg);
BufferedReader in = new BufferedReader(new InputStreamReader(process.getInputStream()));
String line = in.readLine();
后面加了一行代码,in.readLine(),此时才真正的分配了缓冲区内存。如下,发现子进程已经建立,但由于此时缓冲区已经建立,且主线程中并没有持续的读取缓冲区的数据,导致缓冲区被填满,从而导致了线程阻塞。
为了解决主线程与子进程之间互相等待而造成的线程阻塞问题,在主线程中创建一个子线程去实时读取缓冲区数据,捕获Python脚本的输出。
三、源码:
public class Solution{
public static void main(String[] args) {
String fileName = "D:main.py"; //python脚本
String parser = "D:\\learning\\venv\\Scripts\\python.exe"; //python解析器
Process process;
int a = 10, b=20; //脚本参数
try {
String[] arg=new String[]{parser, fileName, String.valueOf(a), String.valueOf(b)};
process = Runtime.getRuntime().exec(arg);
StreamGobbler error= new OutStream(process.getErrorStream(),"Error");
StreamGobbler output= new OutStream(process.getInputStream(),"OutPut");
errorGobbler.start();
outputGobbler.start();
process.waitFor(); //等待子进程结束
} catch (Exception e) {
e.printStackTrace();
}
}
}
class OutStream extends Thread{
InputStream is;
String type;
public OutStream(InputStream is, String type) {
this.is = is;
this.type = type;
}
public void run() {
try {
InputStreamReader isr = new InputStreamReader(is);
BufferedReader br = new BufferedReader(isr);
String line = null;
while ((line = br.readLine()) != null) {
if (type.equals("Error")) {
System.out.println("Error :" + line);
} else {
System.out.println("Debug:" + line);
}
}
} catch (Exception ioe) {
ioe.printStackTrace();
}
}
}