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

时间:2018-10-25 11:59:40

标签: java command-line automation processbuilder

我是Java的新手,正在尝试使用Java自动执行命令行。我试图在此处搜索解决方案,但找不到它。

我创建了一个简单的测试外壳脚本,如下所示,用于测试程序:


#!/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)

1 个答案:

答案 0 :(得分:2)

关闭BufferedWriter将关闭其基础流,其中包括您通过OutputStream获得的过程process.getOutputStream()。因此,一旦在一个循环中将其关闭,在接下来的循环中,您的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;
    }
}

您的原始代码在close()而不是在新write()包装封闭的基础流时位于其前面的BufferedWriter上死亡的原因是,BufferedOutputStream.write()可能由于已缓冲,因此尚未实际写入基础流。调用flush()会告诉流实际写入流,正如您在堆栈跟踪中看到的那样,close()正在调用flush(),这最终会将缓冲的字节写入底层的FileOutputStream,然后意识到FileOutputStream已经关闭。