Java - 重定向进程 I/O(管道)停滞



我正在尝试将一个进程的输出重定向到另一个进程的输入。 即管道。在Windows DOS shell中执行此操作时,它看起来像

C:> dir /s/b . | findstr dat$

但是,我正在尝试在 Java 中执行这样的命令,到目前为止它看起来像:

Stopwatch sw = Stopwatch.createStarted();
ProcessBuilder source = new ProcessBuilder("cmd", "/S/D/c", "dir", "/s/b", ".");
ProcessBuilder target = new ProcessBuilder("cmd", "/S/D/c", "findstr", "dat$");
source.directory(new File("C:/"));
target.directory(source.directory());
// I am messing with the below lines, nothing is working
source.redirectInput(target.redirectInput());
source.redirectOutput(ProcessBuilder.Redirect.PIPE);
source.redirectOutput(target.redirectInput());
source.redirectInput(target.redirectOutput());
target.redirectOutput(source.redirectInput());
Process pSource = source.start();
Process pTarget = target.start();
log.debug("Running {} | {}", source.command(), target.command());
try (BufferedReader br = new BufferedReader(new InputStreamReader(pTarget.getInputStream()))) {
String line;
while ((line = br.readLine()) != null)
log.debug("{}", line);
} finally {
log.debug("Ending process {} with exist code {} in time {}", target.command(),
pTarget.destroyForcibly().exitValue(), sw);
}

但是我发现代码在readLine上停滞不前,所以这里有些东西不起作用。如何正确使用 IO 重定向?

redirectInput()resp 接受或返回的对象redirectOutput()描述某个策略;它们不代表实际的渠道。
因此,通过声明source.redirectInput(target.redirectInput())您只是指定两个进程应具有相同的策略,您没有链接通道。

事实上,在Java 8中,直接链接两个进程的通道是不可能的。要达到类似的效果,您能做的最好的事情是启动一个后台线程,该线程将读取第一个进程的输出并将其写入第二个进程的输入:

static List<Process> doPipeJava8() throws IOException {
Process pSource = new ProcessBuilder("cmd", "/S/D/c", "dir", "/s/b", ".")
.redirectInput(ProcessBuilder.Redirect.INHERIT)
.redirectError(ProcessBuilder.Redirect.INHERIT)
.start();
Process pTarget;
try {
pTarget     = new ProcessBuilder("cmd", "/S/D/c", "findstr", "dat$")
.redirectErrorStream(true)
.redirectOutput(ProcessBuilder.Redirect.INHERIT)
.start();
} catch(Throwable t) {
pSource.destroyForcibly();
throw t;
}
new Thread(() -> {
try(InputStream srcOut = pSource.getInputStream();
OutputStream dstIn = pTarget.getOutputStream()) {
byte[] buffer = new byte[1024];
while(pSource.isAlive() && pTarget.isAlive()) {
int r = srcOut.read(buffer);
if(r > 0) dstIn.write(buffer, 0, r);
}
} catch(IOException ex) {}
}).start();
return Arrays.asList(pSource, pTarget);
}

这会将错误通道、第一个进程的输入通道和最后一个进程的输出通道配置为INHERIT以便它们使用我们的启动进程控制台。第一个进程的输出和第二个进程的输入保持默认PIPE这意味着建立到启动进程的管道,因此我们有责任将数据从一个管道复制到另一个管道。

该方法可以用作

List<Process> sub = doPipeJava8();
Process pSource = sub.get(0), pTarget = sub.get(1);
pSource.waitFor();
pTarget.waitFor();

如果我们从pTarget流程的构建器中删除.redirectOutput(ProcessBuilder.Redirect.INHERIT),我们可以读取最终输出:

List<Process> sub = doPipeJava8();
Process pSource = sub.get(0), pTarget = sub.get(1);
List<String> result = new BufferedReader(new InputStreamReader(pTarget.getInputStream()))
.lines().collect(Collectors.toList());
Java

9 是第一个支持在子进程之间建立管道的 Java 版本。它简化了解决方案

static List<Process> doPipeJava9() throws IOException {
return ProcessBuilder.startPipeline(
List.of(new ProcessBuilder("cmd", "/S/D/c", "dir", "/s/b", ".")
.redirectInput(ProcessBuilder.Redirect.INHERIT)
.redirectError(ProcessBuilder.Redirect.INHERIT),
new ProcessBuilder("cmd", "/S/D/c", "findstr", "dat$")
.redirectErrorStream(true)
.redirectOutput(ProcessBuilder.Redirect.INHERIT)) );
}

它的功能与其他解决方案¹相同;上面的示例配置为允许第一个进程从控制台读取(如果需要),最后一个进程写入控制台。同样,如果我们省略最后一个进程生成器中的.redirectOutput(ProcessBuilder.Redirect.INHERIT),我们可以读取最后一个进程的输出。

¹ 除非它将在可能的情况下使用系统的本机管道功能

相关内容

  • 没有找到相关文章

最新更新