此文章最先发布于我的个人博客,掘金为同步发布,如有需要,请访问 腿短快跑的个人博客 获取更多内容
起因
最近正在做一个视频转换平台,采用的方式是使用 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();
}
}
}
}
再次运行测试代码,发现命令执行成功,问题搞定