如何使用 Java ProcessBuilder 自动化命令行脚本

Question

我是 java 的新手，正在尝试使用 java 自动化命令行。我试图在这里搜索解决方案，但找不到。

我创建了一个如下所示的简单测试 shell 脚本来测试我的程序：

---

#!/bin/bash
echo "What is your name?";
read name;
echo "Hello, $name"
echo "What is your contact number?";
read num;
echo "Saved contact number $num for $name"

---

Java 代码如下：

import java.io.*;
import java.util.*;

public class CmdLineMain {
    public static void main(String args[]) throws InterruptedException, IOException {
        List<String> command = new ArrayList<String>();
        command.add("./test.sh");

        ProcessBuilder builder = new ProcessBuilder(command);

        final Process process = builder.start();
        InputStream is = process.getInputStream();
        InputStreamReader isr = new InputStreamReader(is);
        BufferedReader br = new BufferedReader(isr);
        String line = "";
        BufferedWriter bw = null;

        while (process.isAlive()) {
            line = br.readLine();
            // since stream may be closed earlier, re-open it
            bw = new BufferedWriter(new OutputStreamWriter(process.getOutputStream()));
            System.out.println(line);

            if (line != null) {
                switch (line) {
                case "What is your name?":
                    bw.write("John Doe");
                    bw.close();
                    break;
                case "What is your contact number?":
                    bw.write("123456789");
                    bw.close();
                    break;

                }
            }

        }

        System.out.println("Program terminated!");
    }
}

---

问题：进程的第二个输入失败并出现错误：

---

    What is your name?
    Hello, John Doe
    What is your contact number?
    Exception in thread "main" java.io.IOException: Stream Closed
        at java.io.FileOutputStream.writeBytes(Native Method)
        at java.io.FileOutputStream.write(FileOutputStream.java:326)
        at java.io.BufferedOutputStream.flushBuffer(BufferedOutputStream.java:82)
        at java.io.BufferedOutputStream.flush(BufferedOutputStream.java:140)
        at java.io.FilterOutputStream.close(FilterOutputStream.java:158)
        at sun.nio.cs.StreamEncoder.implClose(StreamEncoder.java:320)
        at sun.nio.cs.StreamEncoder.close(StreamEncoder.java:149)
        at java.io.OutputStreamWriter.close(OutputStreamWriter.java:233)
        at java.io.BufferedWriter.close(BufferedWriter.java:266)
        at nkh.app.CmdLineMain.main(CmdLineMain.java:34)

---

Answer 1

关闭BufferedWriter 将关闭其底层流，这包括您通过process.getOutputStream() 获得的进程OutputStream。因此，一旦它在一个循环中关闭，在接下来的循环中，您的 BufferedWriter 将包装一个封闭的流。而是只包装一次输出流并重用它。

像这样：

BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(process.getOutputStream()));
// while the stream is open and there is something to read
// probably a better condition than `process.isAlive()`
while ((line = br.readLine()) != null) {
    switch (line) {
        case "What is your name?":
            bw.write("John Doe");
            bw.newLine();
            bw.flush();
            break;
        case "What is your contact number?":
            bw.write("123456789");
            bw.newLine();
            bw.flush();
            break;
    }
}

---

当您的新 BufferedWriter 包装封闭的底层流时，您的原始代码在 close() 而不是之前的 write() 上死掉的原因是因为 BufferedOutputStream.write() 可能实际上还没有写入底层流，因为它是缓冲的。调用flush() 应该告诉流实际写入，正如您在堆栈跟踪中看到的close() 正在调用flush()，它最终将缓冲的字节写入底层FileOutputStream，然后意识到@987654334 @ 已关闭。