Запустите подпроцесс, предоставьте вход и выход для него правильно в Java

Я использую Runtime exec() для создания подпроцесса в Java. Тем не менее, поскольку подпроцесс является интерактивной программой, мне необходимо предоставить входные данные, как и когда это требуется. Также мне нужно показать вывод подпроцесса. Как я могу сделать это самым простым способом?

Я использовал StreamGobbler, чтобы показать вывод программы, используя process.getInputStream(). Я, однако, не знаю, как определить, когда программа ожидает ввода и когда предоставлять ввод, используя proc.getOutputStream. Как я могу это сделать?

2 ответа

Решение

Вам нужно скопировать ввод и вывод между потоками подпроцесса и System потоки (System.in, System.out а также System.err). Это связано с моим недавним вопросом. Лучшее решение, которое я нашел до сих пор:

import java.io.FileInputStream;
import java.io.FilterInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.nio.ByteBuffer;
import java.nio.channels.AsynchronousCloseException;
import java.nio.channels.FileChannel;

class StreamCopier implements Runnable {
    private InputStream in;
    private OutputStream out;

    public StreamCopier(InputStream in, OutputStream out) {
        this.in = in;
        this.out = out;
    }

    public void run() {
        try {
            int n;
            byte[] buffer = new byte[4096];
            while ((n = in.read(buffer)) != -1) {
                out.write(buffer, 0, n);
                out.flush();
            }
        }
        catch (IOException e) {
            System.out.println(e);
        }
    }
}

class InputCopier implements Runnable {
    private FileChannel in;
    private OutputStream out;

    public InputCopier(FileChannel in, OutputStream out) {
        this.in = in;
        this.out = out;
    }

    public void run() {
        try {
            int n;
            ByteBuffer buffer = ByteBuffer.allocate(4096);
            while ((n = in.read(buffer)) != -1) {
                out.write(buffer.array(), 0, n);
                out.flush();
            }
            out.close();
        }
        catch (AsynchronousCloseException e) {}
        catch (IOException e) {
            System.out.println(e);
        }
    }
}

public class Test {
    private static FileChannel getChannel(InputStream in)
            throws NoSuchFieldException, IllegalAccessException {
        Field f = FilterInputStream.class.getDeclaredField("in");
        f.setAccessible(true);
        while (in instanceof FilterInputStream)
            in = (InputStream)f.get((FilterInputStream)in);
        return ((FileInputStream)in).getChannel();
    }

    public static void main(String[] args)
            throws IOException, InterruptedException,
                   NoSuchFieldException, IllegalAccessException {
        Process process = Runtime.getRuntime().exec("sh -i +m");
        Thread outThread = new Thread(new StreamCopier(
                process.getInputStream(), System.out));
        outThread.start();
        Thread errThread = new Thread(new StreamCopier(
                process.getErrorStream(), System.err));
        errThread.start();
        Thread inThread = new Thread(new InputCopier(
                getChannel(System.in), process.getOutputStream()));
        inThread.start();
        process.waitFor();
        System.in.close();
        outThread.join();
        errThread.join();
        inThread.join();
    }
}

Сложная часть здесь состоит в том, чтобы извлечь канал из System.in, Без этого вы не сможете прервать поток, который читает ввод, когда подпроцесс завершается.

Такой подход имеет серьезный недостаток: после закрытия System.in Вы больше не можете читать из него. Обходной путь, который я в настоящее время использую, состоит в том, чтобы иметь один поток перенаправления ввода, используемый для всех подпроцессов.

Задайте себе вопрос: "Как узнать, когда программе требуется ввод данных при запуске из командной строки"? Вы видите, что он запрашивает, и вводите данные на основе этого запроса. Принцип будет таким же, за исключением того, что ваш код должен будет интерпретировать вывод программы и обеспечить правильный ввод.

Чтобы не изобретать велосипед заново, взгляните на ExpectJ и / или Expect4J, которые являются реализациями Java почтенного * nix Expect инструмента, предназначенного для такого рода программных взаимодействий.

Другие вопросы по тегам