做自动化的时候，有时候避免不了要和服务器有互动，刚巧碰上一个项目，需要执行命令才能完成本次测试。
昨天遇到的是秘钥形式的，只有秘钥和用户名，百度找了许久也没有思路，（能账号密码登录服务器的还简单些），后来看到一篇博文，受到启发，把他的代码改了又改，结果还真行了
代码如下：
import com.jcraft.jsch.*;
import org.slf4j.LoggerFactory;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.util.Properties; /**
* @author
* 秘钥形式登录服务器执行shell脚本操作
*/
public class ExecuteShellWithKey { /**
* slf4j
*/ private final static org.slf4j.Logger logger = LoggerFactory.getLogger(ExecuteShellWithKey.class);
private Session session; /**
* 远程连接Linux 服务器 执行相关的命令
*
* @param keyFile 秘钥文件
* @param userName 用户名
* @param passphrase 重置后密码
* @param host 主机ip
* @param cmd 执行shell调用命令
* @throws JSchException
* @throws IOException
*/
private void runShell(String keyFile, String userName, String passphrase, String host, String cmd) {
if (!connect(keyFile, userName, passphrase, host)) {
return;
}
ChannelExec channelExec = null;
BufferedReader reader = null;
try {
channelExec = (ChannelExec) session.openChannel("exec");
channelExec.setCommand(cmd);
channelExec.setInputStream(null);
channelExec.setErrStream(System.err);
channelExec.connect();
InputStream in = channelExec.getInputStream();
reader = new BufferedReader(new InputStreamReader(in, Charset.forName("UTF-8")));
String buf = null;
StringBuffer sb = new StringBuffer();
while ((buf = reader.readLine()) != null) {
sb.append(buf);
// 输出每个命令执行后，返回数据
logger.info(sb.toString());
}
} catch (JSchException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
if (reader != null) {
reader.close();
}
} catch (IOException e) {
e.printStackTrace();
}
if (channelExec != null) {
channelExec.disconnect();
}
if (null != session) {
session.disconnect();
}
}
} /**
* 连接到指定的HOST
*
* @param keyFile
* @param userName
* @param passphrase
* @param host
* @throws JSchException
*/
private boolean connect(String keyFile, String userName, String passphrase, String host) {
// 创建JSch对象
JSch jsch = new JSch();
try {
// 引入秘钥文件
jsch.addIdentity(keyFile);
// 根据用户名，主机ip，端口获取一个Session对象
session = jsch.getSession(userName, host, 22);
UserInfo ui = new MyUserInfo(passphrase);
session.setUserInfo(ui);
// 设置密码
session.setPassword(passphrase);
Properties config = new Properties();
config.put("StrictHostKeyChecking", "no");
// 为Session对象设置properties
session.setConfig(config);
int timeout = 60000000;
// 设置timeout时间
session.setTimeout(timeout);
// 通过Session建立链接
session.connect();
} catch (JSchException e) {
e.printStackTrace();
logger.error("connect error !");
return false;
}
return true;
} public class MyUserInfo implements UserInfo {
private String passphrase = null; public MyUserInfo(String passphrase) {
this.passphrase = passphrase;
} public String getPassphrase() {
return passphrase;
} public String getPassword() {
return null;
} public boolean promptPassphrase(String s) {
return true;
} public boolean promptPassword(String s) {
return true;
} public boolean promptYesNo(String s) {
return true;
} public void showMessage(String s) {
logger.info(s);
}
}
}
要是觉得对你有帮助，并且解决了你的问题，麻烦请高抬你发财的贵手，在评论给我个赞即可。。。