Java代码执行shell命令

Java代码执行shell命令

本文描述两种方式使用java代码执行shell命令，首先使用Runtime类调用exce方法，其次使用ProcessBuilder实例实现更灵活的方式。

1. 环境准备

执行shell命令之前，我们需要获取jvm底层操作系统，同时定义通用消费流的类。

1.1. 操作系统依赖

在创建进场执行shell命令之前，我们需要获取jvm运行在具体哪个操作系统之上。因为Windows执行shell命令是cmd.exe，而其他操作系统发布标准shell是sh：

boolean isWindows = System.getProperty("os.name").toLowerCase().startsWith("windows");

1.2. 输入和输出

此外我们还需要一种方法来连接进程的输入、输出流。直到输出流被消费进程才会返回成功，否则会挂起。下面实现通用类StreamGobbler消费InputStream：

private static class StreamGobbler implements Runnable {
    private InputStream inputStream;
    private Consumer<String> consumer;
 
    public StreamGobbler(InputStream inputStream, Consumer<String> consumer) {
        this.inputStream = inputStream;
        this.consumer = consumer;
    }
 
    @Override
    public void run() {
        new BufferedReader(new InputStreamReader(inputStream)).lines()
          .forEach(consumer);
    }
}

该类实现Runnable接口，意味着能够被任何Executor执行。

2. Runtime.exec()执行

Runtime.exec()方法是一种生成新子进程的简单方法，但不能定制。下面示例列出用户目录的目录清单并打印至控制台：

String homeDirectory = System.getProperty("user.home");
Process process;
if (isWindows) {
    process = Runtime.getRuntime()
      .exec(String.format("cmd.exe /c dir %s", homeDirectory));
} else {
    process = Runtime.getRuntime()
      .exec(String.format("sh -c ls %s", homeDirectory));
}
StreamGobbler streamGobbler = 
  new StreamGobbler(process.getInputStream(), System.out::println);
Executors.newSingleThreadExecutor().submit(streamGobbler);
int exitCode = process.waitFor();
assert exitCode == 0;

3. ProcessBuilder执行

第二种方法使用ProcessBuilder。这比Runtime方法更可取，因为能够定制一些细节。
例如：

• 改变正在运行Shell命令的工作目录，使用builder.directory()方法
• 使用builder.environment()方法，设置自定义键值对作为环境变量
• 重定向输入和输出流值自定义流
• 使用build.inheritio()方法将它们都继承到当前JVM进程的流中

ProcessBuilder builder = new ProcessBuilder();
if (isWindows) {
    builder.command("cmd.exe", "/c", "dir");
} else {
    builder.command("sh", "-c", "ls");
}
builder.directory(new File(System.getProperty("user.home")));
Process process = builder.start();
StreamGobbler streamGobbler = 
  new StreamGobbler(process.getInputStream(), System.out::println);
Executors.newSingleThreadExecutor().submit(streamGobbler);
int exitCode = process.waitFor();
assert exitCode == 0;

4. 总结

本文介绍了两种不同方法执行Shell命令。通常如果需要自定义派生流程的执行，例如更改其工作目录，则应考虑使用ProcessBuilder。