java代码执行Linux脚本,java程序执行,调用shell命令和shell脚本

坑呀！记得在start()之后， waitFor()之前把缓冲区读出来打log，否则是阻塞缓冲区，没有输出的

package com.jikexueyuancrm.util;

import java.io.BufferedReader;

import java.io.DataOutputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.PrintWriter;

import java.lang.management.ManagementFactory;

import java.util.List;

import org.apache.log4j.Logger;

import com.jikexueyuancrm.controller.AnHui;

/**

* ShellUtils

* 

* Check root

* 

{@link ShellUtils#checkRootPermission()}

* 

* 

* Execte command

* 

{@link ShellUtils#execCommand(String, boolean)}

* 

{@link ShellUtils#execCommand(String, boolean, boolean)}

* 

{@link ShellUtils#execCommand(List, boolean)}

* 

{@link ShellUtils#execCommand(List, boolean, boolean)}

* 

{@link ShellUtils#execCommand(String[], boolean)}

* 

{@link ShellUtils#execCommand(String[], boolean, boolean)}

* 

*

* @author Trinea 2013-5-16

*/

/**

* @author yuan hai

* 使用Runtime.getRuntime().exec

*/

public class ShellUtils {

private static Logger log = Logger.getLogger(ShellUtils.class);

public static final String COMMAND_SU       = "su";

public static final String COMMAND_SH       = "sh";

public static final String COMMAND_EXIT     = "exit\n";

public static final String COMMAND_LINE_END = "\n";

private ShellUtils() {

throw new AssertionError();

}

/**

* check whether has root permission

*

* @return

*/

public static boolean checkRootPermission() {

return execCommand("echo root", true, false).result == 0;

}

/**

* execute shell command, default return result msg

*

* @param command command

* @param isRoot whether need to run with root

* @return

* @see ShellUtils#execCommand(String[], boolean, boolean)

*/

public static CommandResult execCommand(String command, boolean isRoot) {

return execCommand(new String[] {command}, isRoot, true);

}

/**

* execute shell commands, default return result msg

*

* @param commands command list

* @param isRoot whether need to run with root

* @return

* @see ShellUtils#execCommand(String[], boolean, boolean)

*/

public static CommandResult execCommand(List commands, boolean isRoot) {

return execCommand(commands == null ? null : commands.toArray(new String[] {}), isRoot, true);

}

/**

* execute shell commands, default return result msg

*

* @param commands command array

* @param isRoot whether need to run with root

* @return

* @see ShellUtils#execCommand(String[], boolean, boolean)

*/

public static CommandResult execCommand(String[] commands, boolean isRoot) {

return execCommand(commands, isRoot, true);

}

/**

* execute shell command

*

* @param command command

* @param isRoot whether need to run with root

* @param isNeedResultMsg whether need result msg

* @return

* @see ShellUtils#execCommand(String[], boolean, boolean)

*/

public static CommandResult execCommand(String command, boolean isRoot, boolean isNeedResultMsg) {

return execCommand(new String[] {command}, isRoot, isNeedResultMsg);

}

/**

* execute shell commands

*

* @param commands command list

* @param isRoot whether need to run with root

* @param isNeedResultMsg whether need result msg

* @return

* @see ShellUtils#execCommand(String[], boolean, boolean)

*/

public static CommandResult execCommand(List commands, boolean isRoot, boolean isNeedResultMsg) {

return execCommand(commands == null ? null : commands.toArray(new String[] {}), isRoot, isNeedResultMsg);

}

/**

* execute shell commands

*

*

* 最基础方法

* @param commands command array

* @param isRoot whether need to run with root

* @param isNeedResultMsg whether need result msg

* @return 

*         

if isNeedResultMsg is false, {@link CommandResult#successMsg} is null and

*         {@link CommandResult#errorMsg} is null.

*         

if {@link CommandResult#result} is -1, there maybe some excepiton.

*         

*/

public static CommandResult execCommand(String[] commands, boolean isRoot, boolean isNeedResultMsg) {

int result = -1;

if (commands == null || commands.length == 0) {

return new CommandResult(result, null, null);

}

Process process = null;

BufferedReader succe***esult = null;

BufferedReader errorResult = null;

StringBuilder successMsg = null;

StringBuilder errorMsg = null;

DataOutputStream os = null;

try {

process = Runtime.getRuntime().exec(isRoot ? COMMAND_SU : COMMAND_SH);

os = new DataOutputStream(process.getOutputStream());

for (String command : commands) {

if (command == null) {

continue;

}

// donnot use os.writeBytes(commmand), avoid chinese charset error

os.write(command.getBytes());

os.writeBytes(COMMAND_LINE_END);

os.flush();

}

os.writeBytes(COMMAND_EXIT);

os.flush();

// get command result

if (isNeedResultMsg) {

successMsg = new StringBuilder();

errorMsg = new StringBuilder();

succe***esult = new BufferedReader(new InputStreamReader(process.getInputStream()));

errorResult = new BufferedReader(new InputStreamReader(process.getErrorStream()));

String s;

while ((s = succe***esult.readLine()) != null) {

successMsg.append(s).append("\n");

}

while ((s = errorResult.readLine()) != null) {

errorMsg.append(s).append("\n");

}

}

result = process.waitFor();

} catch (IOException e) {

e.printStackTrace();

} catch (Exception e) {

e.printStackTrace();

} finally {

try {

if (os != null) {

os.close();

}

if (succe***esult != null) {

succe***esult.close();

}

if (errorResult != null) {

errorResult.close();

}

} catch (IOException e) {

e.printStackTrace();

}

if (process != null) {

process.destroy();

}

}

log.info("result:"+result);

if( successMsg != null) log.info("successMsg:"+successMsg.toString());

if( errorMsg != null) log.info("errorMsg:"+errorMsg.toString());

return new CommandResult(result, successMsg == null ? null : successMsg.toString(), errorMsg == null ? null

: errorMsg.toString());

}

/**

* result of command

* 

* 

{@link CommandResult#result} means result of command, 0 means normal, else means error, same to excute in

* linux shell

* 

{@link CommandResult#successMsg} means success message of command result

* 

{@link CommandResult#errorMsg} means error message of command result

* 

*

* @author Trinea 2013-5-16

*/

public static class CommandResult {

/** result of command **/

public int    result;

/** success message of command result **/

public String successMsg;

/** error message of command result **/

public String errorMsg;

public CommandResult(int result) {

this.result = result;

}

public CommandResult(int result, String successMsg, String errorMsg) {

this.result = result;

this.successMsg = successMsg;

this.errorMsg = errorMsg;

}

}

public static void main(String[] args) throws Exception {

//

String[] cmd=new String[]{"ps aux","pwd","ls -al"};

//执行shell命令

//    execCommand(cmd,false, true);

//执行shell脚本

String[] script=new String[]{"/bin/bash /home/smgadmin/test/soyoung_comment/shell/crawler.sh start  "};

execCommand(script,false, true);

//使用ProcessBuilder

//    exeCmd("ps aux;ls -al;pwd");//执行linux命令

//    exeCmd("dir&dir");//执行windows命令一行多命令用&分隔

}

//使用ProcessBuilder

/**

* @author yuan hai 2016年12月9日

* @param shell  shell可以传入多个命令  ,例如:"ps aux;ls -al;pwd"

* @return

* @throws IOException

*/

public  static int exeCmd(String shell) throws IOException {

int success = 0;

StringBuffer sb = new StringBuffer();

BufferedReader br = null;

// get name representing the running Java virtual machine.

String name = ManagementFactory.getRuntimeMXBean().getName();

String pid = name.split("@")[0];

try {

System.out.println("Starting to exec{ " + shell + " }. PID is: " + pid);

Process process = null;

//执行脚本时，不要加-c

// 当执行系统的命令时，才会加-c

//linux命令行

ProcessBuilder pb = new ProcessBuilder("/bin/bash", "-c", shell);

//windows命令行

//  ProcessBuilder pb = new ProcessBuilder("cmd", "/c", shell);

pb.environment();

pb.redirectErrorStream(true); // merge error stream into standard stream

process = pb.start();

if (process != null) {

br = new BufferedReader(

new InputStreamReader(process.getInputStream(),"GBK"), 1024);

} else {

System.out.println("There is no PID found.");

}

sb.append("Ending exec right now, the result is：\n");

String line = null;

while (br != null && (line = br.readLine()) != null) {

sb.append(line).append("\n");

}

//记得在start()之后， waitFor()之前把缓冲区读出来打log

process.waitFor();

} catch (Exception ioe) {

sb.append("Error occured when exec cmd：\n").append(ioe.getMessage()).append("\n");

} finally {

PrintWriter writer = null;

if (br != null) {

br.close();

}

try {

writer = new PrintWriter(System.out);

writer.write(sb.toString());

} catch (Exception e) {

log.error(e.getMessage(), e);

} finally {

writer.close();

}

success = 1;

}

return success;

}

}

参考文章：

阿里大牛文章：