Java调用CMD(Shell)命令

此文章最先发布于我的个人博客，CSDN为同步发布，如有需要，请访问 腿短快跑的个人博客 获取更多内容

起因

最近正在做一个视频转换平台，采用的方式是使用 Java 调用 ffmpeg 来实现视频的各种操作，因为 ffmpeg 都是使用 shell 命令来调用的，因此，我在代码中使用 Java 调用 shell 命令行

具体ffmpeg安装可以参考这篇文章：ffmpeg安装

调用方式

Java中调用 shell 或者 cmd 命令一共有两种方式：

• Runtime
  此方式历史最为悠久，使用也最广，使应用程序能够与其运行的环境相连接，但是在读取上还存在一些不便性，正常的输出流与错误流得分开读取。其他功能基本相同。在jdk8中 Runtime 底层也是通过 ProcessBuilder 实现

• ProcessBuilder
  此为jdk1.5加入的，它没有将应用程序与其运行的环境相连接。这个就需要自己设置其相关的信息。但它提供了将正常流与流程流合并在一起的解决办法，只需要设置redirectErrorStream（错误流重定向到标准数据流）属性即可。

Java实现

在这里我采用了 更为先进的 ProcessBuilder 来实现，如下为实现方式：

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author C.W
 * @date 2022/5/9 17:17
 * @desc bash工具
 */
@Slf4j
public class BashUtils {

    public static final Integer SUCCESS_CODE = 0;

    /**
     * 执行命令
     *
     * @param command
     * @return
     * @throws IOException
     * @throws InterruptedException
     */
    public static int execCommand(String command, String dir) throws IOException, InterruptedException {
        String[] commands = command.split(" ");
        List<String> commandList = new ArrayList<>(commands.length);
        for (String s : commands) {
            if (StringUtils.isBlank(s)) {
                continue;
            }
            commandList.add(s);
        }
        commands = new String[commandList.size()];
        commands = commandList.toArray(commands);
        ProcessBuilder processBuilder = new ProcessBuilder(commands);
        if (StringUtils.isNotBlank(dir)) {
            processBuilder.directory(new File(dir));
        }
        log.info("开始执行命令: {} ", command);
        Process exec = processBuilder.start();
        int code = exec.waitFor();
        log.info("命令: {} 执行结果: {}", command, code);
        return code;
    }
}

问题发现

程序写好了，当然写个简单的代码测试下：

public class BashTest {
	public static void main(String[] args) {
    	BashUtils.execCommand("ffmpeg -i https://vd4.bdstatic.com/mda-nck9kn2u9tdu76gx/sc/cae_h264_delogo/1647845979085388943/mda-nck9kn2u9tdu76gx.mp4", "");
    }
}

通过上述代码使用 ffmpeg 来下载网络视频，愉快的点击运行，等待执行完毕

1分钟。
2分钟。
3分钟。
…
10分钟。

what？视频这么大的么，10分多钟还没下载完毕，于是使用 wget 命令下载视频，打开一看，竟然只有 7.4M，什么情况？

经过一系列的测试，发现，命令执行部分出现成功的情况，部分命令就直接卡住不动了，经过一系列的资料查阅，发现可能是下面问题

因为Java程序并不是传统的命令行进程，命令行的进程会将成功、失败的信息输出在终端上，但是Java中并没有终端的概念，因此，命令执行成功和失败的信息分别写在两个内存缓存区中，这两个缓冲区非常的小，当缓冲区被写满的时候，新的信息无法写入，就导致进程被卡住。

问题解决

ok，我们已经大致确定问题所在，解决方案很简单，将缓冲区中的数据读出来就搞定了嘛，开始改写 BashUtils 工具

package cn.tdkpcw.tools.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author C.W
 * @date 2022/5/9 17:17
 * @desc bash工具
 */
@Slf4j
public class BashUtils {

    public static final Integer SUCCESS_CODE = 0;

    /**
     * 执行命令
     *
     * @param command
     * @return
     * @throws IOException
     * @throws InterruptedException
     */
    public static int execCommand(String command, String dir) throws IOException, InterruptedException {
        String[] commands = command.split(" ");
        List<String> commandList = new ArrayList<>(commands.length);
        for (String s : commands) {
            if (StringUtils.isBlank(s)) {
                continue;
            }
            commandList.add(s);
        }
        commands = new String[commandList.size()];
        commands = commandList.toArray(commands);
        ProcessBuilder processBuilder = new ProcessBuilder(commands);
        if (StringUtils.isNotBlank(dir)) {
            processBuilder.directory(new File(dir));
        }
        log.info("开始执行命令: {} ", command);
        Process exec = processBuilder.start();
        // 获取外部程序标准输出流
        new Thread(new OutputHandlerRunnable(exec.getInputStream(), false)).start();
        // 获取外部程序标准错误流
        new Thread(new OutputHandlerRunnable(exec.getErrorStream(), true)).start();
        int code = exec.waitFor();
        log.info("命令: {} 执行结果: {}", command, code);
        return code;
    }

    private static class OutputHandlerRunnable implements Runnable {
        private InputStream in;

        private boolean error;

        public OutputHandlerRunnable(InputStream in, boolean error) {
            this.in = in;
            this.error = error;
        }

        @Override
        public void run() {
            try (BufferedReader bufr = new BufferedReader(new InputStreamReader(this.in))) {
                String line = null;
                while ((line = bufr.readLine()) != null) {
                    if (error) {
                        System.out.println(line);
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}

再次运行测试代码，发现命令执行成功，问题搞定