packagecom.gosun.utils;importch.ethz.ssh2.ChannelCondition;importch.ethz.ssh2.Connection;importch.ethz.ssh2.Session;importch.ethz.ssh2.StreamGobbler;importjava.io.IOException;importjava.io.InputStream;importjava.nio.charset.Charset;/***
* storm 远程调用脚本..
*@authorchenwen
**/
public classShellRPC {private staticConnection conn;/**远程机器IP*/
private staticString ip;/**用户名*/
private staticString userName;/**密码*/
private staticString password;private static String charset =Charset.defaultCharset().toString();private static final int TIME_OUT = 1000 * 5 * 60;/*** 登录
*@return*@throwsIOException*/
private static boolean getConnection(String username, String password) throwsIOException {
ip= ip;
userName=userName;
password=password;
conn= newConnection(ip);
conn.connect();returnconn.authenticateWithPassword(userName, password);
}/*** 执行脚本
*
*@paramcmds
*@return*@throwsException*/
public static int exec(String cmds) throwsException {
InputStream stdOut= null;
InputStream stdErr= null;
String outStr= "";
String outErr= "";int ret = -1;try{if(getConnection()) {//Open a new {@link Session} on this connection
Session session =conn.openSession();//Execute a command on the remote machine.
session.execCommand(cmds);
stdOut= newStreamGobbler(session.getStdout());
outStr=processStream(stdOut, charset);
System.out.println("outStr" +outStr);
stdErr= newStreamGobbler(session.getStderr());
outErr=processStream(stdErr, charset);
System.out.println("outErr" +outErr);
session.waitForCondition(ChannelCondition.EXIT_STATUS, TIME_OUT);
ret=session.getExitStatus();
}else{throw new Exception("登录远程机器失败" + ip); //自定义异常类 实现略
}
}finally{if (conn != null) {
conn.close();
}
stdOut.close();
stdErr.close();
}returnret;
}/***@paramin
*@paramcharset
*@return*@throwsException*/
private static String processStream(InputStream in, String charset) throwsException {byte[] buf = new byte[1024];
StringBuilder sb= newStringBuilder();while (in.read(buf) != -1) {
sb.append(newString(buf, charset));
}returnsb.toString();
}
}
