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

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
* <ul>
* <strong>Check root</strong>
* <li>{@link ShellUtils#checkRootPermission()}</li>
* </ul>
* <ul>
* <strong>Execte command</strong>
* <li>{@link ShellUtils#execCommand(String, boolean)}</li>
* <li>{@link ShellUtils#execCommand(String, boolean, boolean)}</li>
* <li>{@link ShellUtils#execCommand(List, boolean)}</li>
* <li>{@link ShellUtils#execCommand(List, boolean, boolean)}</li>
* <li>{@link ShellUtils#execCommand(String[], boolean)}</li>
* <li>{@link ShellUtils#execCommand(String[], boolean, boolean)}</li>
* </ul>
*
* @author <a href="http://www.trinea.cn" target="_blank">Trinea</a> 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<String> 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<String> 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 <ul>
*         <li>if isNeedResultMsg is false, {@link CommandResult#successMsg} is null and
*         {@link CommandResult#errorMsg} is null.</li>
*         <li>if {@link CommandResult#result} is -1, there maybe some excepiton.</li>
*         </ul>
*/
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
* <ul>
* <li>{@link CommandResult#result} means result of command, 0 means normal, else means error, same to excute in
* linux shell</li>
* <li>{@link CommandResult#successMsg} means success message of command result</li>
* <li>{@link CommandResult#errorMsg} means error message of command result</li>
* </ul>
*
* @author <a href="http://www.trinea.cn" target="_blank">Trinea</a> 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;
}

}