主要使用了
Runtime.getRuntime().exec(new String[]{"/bin/bash", "-c", cmd});
调用shell命令的工具类:
public static String runShell(String cmd) throws Exception {
StringBuilder result = new StringBuilder();
Process process = null;
BufferedReader bufrIn = null;
BufferedReader bufrError = null;
try {
// 执行命令, 返回一个子进程对象(命令在子进程中执行)使用这种方式可以使用|管道符命令
process = Runtime.getRuntime().exec(new String[]{"/bin/bash", "-c", cmd});
// 方法阻塞, 等待命令执行完成(成功会返回0)
process.waitFor();
// 获取命令执行结果, 有两个结果: 正常的输出 和 错误的输出(PS: 子进程的输出就是主进程的输入)
bufrIn = new BufferedReader(new InputStreamReader(process.getInputStream(), StandardCharsets.UTF_8));
bufrError = new BufferedReader(new InputStreamReader(process.getErrorStream(), StandardCharsets.UTF_8));
// 读取输出
String line;
while ((line = bufrIn.readLine()) != null) {
result.append(line).append('\n');
}
while ((line = bufrError.readLine()) != null) {
result.append(line).append('\n');
}
} finally {
closeStream(bufrIn);
closeStream(bufrError);
// 销毁子进程
if (process != null) {
process.destroy();
}
}
return result.toString();
}
当有jar包上传到接口时,调用这个方法,停止正在运行的jar,并启动新jar
JAR_NAME校验自定,这里固定使用一个jar包名,方便jps时找到该进程
/**
*jar包上传及执行
*/
private ResponseVo myjarPkg(MultipartFile file) throws Exception {
//myjar目录
String myjarDir = "/data/myjar/";
File fileDir = new File(myjarDir);
fileDir.mkdirs();
//文件名校验 名称不对则不执行
if (!JAR_NAME.equals(file.getOriginalFilename())) {
return new ResponseVo(-1, "不正确的文件");
}
File myjarPkg = new File(myjarDir + file.getOriginalFilename());
//将旧的已存在的删除
if (myjarPkg.exists()) {
FileUtils.deleteQuietly(myjarPkg);
}
file.transferTo(myjarPkg);
//运行myjar程序 先停掉之前的 再启动新上传的
//由jre目录进入到父级jdk目录 不直接替换可以防止没有jre目录的情况
String javaHome = System.getProperty("java.home").replace("/jre", "") + "/bin";
final String java = javaHome + "/java";
final String jps = javaHome + "/jps";
String cmd = jps + " -l|grep " + JAR_NAME + "|awk '{print$1}'";
String pid = ShellUtil.runShell(cmd);
if (StringUtils.isNotBlank(pid)) {
cmd = "kill -9 " + pid;
String killmsg = ShellUtil.runShell(cmd);
logger.info("kill process pid:{} ,result:{}", pid, killmsg);
//杀掉之后等3s
Thread.sleep(3000);
}
cmd = "nohup " + java + " -jar " + myjarDir + JAR_NAME + " > " + myjarDir + "myjar.log 2>&1 &";
String msg = ShellUtil.runShell(cmd);
logger.info("升级程序执行结果: " + msg);
return new ResponseVo();
}
重点是Java启动的process,不能直接执行java、jps等命令,也获取不到环境变量,会报command not found
于是我使用来System.getProperty("java.home") 来获取到执行当前程序的Java路径,再把jre目录替换为jdk目录,使用jdk目录下bin目录中的java及jps命令,可以达到需求
另外需要注意命令字符串中的空格很重要,不能忽略