|
|
@@ -22,32 +22,28 @@ import java.util.concurrent.Executors;
|
|
|
@Slf4j
|
|
|
public class ShellSSH2Util {
|
|
|
|
|
|
- /**
|
|
|
- * 日志
|
|
|
- */
|
|
|
+ /** 日志 */
|
|
|
protected static Logger logger = LoggerFactory.getLogger(ShellSSH2Util.class);
|
|
|
|
|
|
-
|
|
|
- /**
|
|
|
- * 默认字符编码
|
|
|
- */
|
|
|
+ /** 默认字符编码 */
|
|
|
private static String DEFAULT_CHART = "UTF-8";
|
|
|
|
|
|
/**
|
|
|
* 登录远程服务器
|
|
|
*
|
|
|
- * @param host 远程主机
|
|
|
+ * @param host 远程主机
|
|
|
* @param username 远程主机登录用户名
|
|
|
* @param password 远程主机登录密码
|
|
|
* @return Connection
|
|
|
*/
|
|
|
- public static Connection login(String host, int port, String username, String password, StdoutListener listener) {
|
|
|
+ public static Connection login(
|
|
|
+ String host, int port, String username, String password, StdoutListener listener) {
|
|
|
Connection connection = null;
|
|
|
listener.stdout("开始连接：" + username + "@" + host);
|
|
|
try {
|
|
|
connection = new Connection(host, port);
|
|
|
- connection.connect();// 连接
|
|
|
- boolean login = connection.authenticateWithPassword(username, password);// 认证
|
|
|
+ connection.connect(); // 连接
|
|
|
+ boolean login = connection.authenticateWithPassword(username, password); // 认证
|
|
|
if (login) {
|
|
|
listener.stdout(username + "@" + host + "； 登录成功");
|
|
|
} else {
|
|
|
@@ -65,19 +61,20 @@ public class ShellSSH2Util {
|
|
|
/**
|
|
|
* 登录远程服务器
|
|
|
*
|
|
|
- * @param host 远程主机
|
|
|
+ * @param host 远程主机
|
|
|
* @param username 远程主机登录用户名
|
|
|
- * @param pemFile 登录密钥
|
|
|
+ * @param pemFile 登录密钥
|
|
|
* @param listener 执行过程监听器
|
|
|
* @return Connection
|
|
|
*/
|
|
|
- public static Connection login(String host, int port, String username, File pemFile, StdoutListener listener) {
|
|
|
+ public static Connection login(
|
|
|
+ String host, int port, String username, File pemFile, StdoutListener listener) {
|
|
|
Connection connection = null;
|
|
|
listener.stdout("开始连接：" + username + "@" + host);
|
|
|
try {
|
|
|
connection = new Connection(host, port);
|
|
|
- connection.connect();// 连接
|
|
|
- boolean login = connection.authenticateWithPublicKey(username, pemFile, null);// 认证
|
|
|
+ connection.connect(); // 连接
|
|
|
+ boolean login = connection.authenticateWithPublicKey(username, pemFile, null); // 认证
|
|
|
if (login) {
|
|
|
listener.stdout(username + "@" + host + "； 登录成功");
|
|
|
} else {
|
|
|
@@ -95,21 +92,22 @@ public class ShellSSH2Util {
|
|
|
/**
|
|
|
* 检验是否能够连接服务器
|
|
|
*
|
|
|
- * @param host 远程主机
|
|
|
- * @param port 端口
|
|
|
+ * @param host 远程主机
|
|
|
+ * @param port 端口
|
|
|
* @param username 远程主机登录用户名
|
|
|
* @param password 远程主机登录密码
|
|
|
- * @return true：可以连接 false:无法连接
|
|
|
+ * @return true：可以连接 false:无法连接
|
|
|
* @author Cyan -- 2020/4/26 10:56
|
|
|
*/
|
|
|
- public static Boolean loginCheck(String host, int port, String username, String password, StdoutListener listener) {
|
|
|
+ public static Boolean loginCheck(
|
|
|
+ String host, int port, String username, String password, StdoutListener listener) {
|
|
|
Boolean result = true;
|
|
|
Connection connection = null;
|
|
|
listener.stdout("开始连接：" + username + "@" + host);
|
|
|
try {
|
|
|
connection = new Connection(host, port);
|
|
|
- connection.connect();// 连接
|
|
|
- boolean login = connection.authenticateWithPassword(username, password);// 认证
|
|
|
+ connection.connect(); // 连接
|
|
|
+ boolean login = connection.authenticateWithPassword(username, password); // 认证
|
|
|
if (login) {
|
|
|
listener.stdout(username + "@" + host + "； 登录成功");
|
|
|
} else {
|
|
|
@@ -132,20 +130,21 @@ public class ShellSSH2Util {
|
|
|
/**
|
|
|
* 登录远程服务器
|
|
|
*
|
|
|
- * @param host 远程主机
|
|
|
+ * @param host 远程主机
|
|
|
* @param username 远程主机登录用户名
|
|
|
- * @param pemFile 登录密钥
|
|
|
+ * @param pemFile 登录密钥
|
|
|
* @param listener 执行过程监听器
|
|
|
* @return Connection
|
|
|
*/
|
|
|
- private static Connection login(String host, String username, File pemFile, StdoutListener listener) {
|
|
|
+ private static Connection login(
|
|
|
+ String host, String username, File pemFile, StdoutListener listener) {
|
|
|
Connection connection = null;
|
|
|
listener.stdout("开始连接：" + username + "@" + host);
|
|
|
try {
|
|
|
connection = new Connection(host);
|
|
|
-// connection.setClient2ServerCiphers(new String[]{"aes256-cbc"});
|
|
|
- connection.connect();// 连接
|
|
|
- boolean login = connection.authenticateWithPublicKey(username, pemFile, null);// 认证
|
|
|
+ // connection.setClient2ServerCiphers(new String[]{"aes256-cbc"});
|
|
|
+ connection.connect(); // 连接
|
|
|
+ boolean login = connection.authenticateWithPublicKey(username, pemFile, null); // 认证
|
|
|
if (login) {
|
|
|
listener.stdout(username + "@" + host + "； 登录成功");
|
|
|
} else {
|
|
|
@@ -162,20 +161,21 @@ public class ShellSSH2Util {
|
|
|
/**
|
|
|
* 登录远程服务器
|
|
|
*
|
|
|
- * @param host 远程主机
|
|
|
+ * @param host 远程主机
|
|
|
* @param username 远程主机登录用户名
|
|
|
- * @param pemFile 登录密钥
|
|
|
+ * @param pemFile 登录密钥
|
|
|
* @param listener 执行过程监听器
|
|
|
* @return Connection
|
|
|
*/
|
|
|
- public static Boolean loginCheck(String host, String username, File pemFile, StdoutListener listener) {
|
|
|
+ public static Boolean loginCheck(
|
|
|
+ String host, String username, File pemFile, StdoutListener listener) {
|
|
|
Boolean result = true;
|
|
|
Connection connection = null;
|
|
|
listener.stdout("开始连接：" + username + "@" + host);
|
|
|
try {
|
|
|
connection = new Connection(host);
|
|
|
- connection.connect();// 连接
|
|
|
- boolean login = connection.authenticateWithPublicKey(username, pemFile, null);// 认证
|
|
|
+ connection.connect(); // 连接
|
|
|
+ boolean login = connection.authenticateWithPublicKey(username, pemFile, null); // 认证
|
|
|
if (login) {
|
|
|
listener.stdout(username + "@" + host + "； 登录成功");
|
|
|
} else {
|
|
|
@@ -195,18 +195,18 @@ public class ShellSSH2Util {
|
|
|
return result;
|
|
|
}
|
|
|
|
|
|
-
|
|
|
/**
|
|
|
* 执行远程Shell命令
|
|
|
*
|
|
|
- * @param host 远程主机地址
|
|
|
- * @param username 远程主机用户名
|
|
|
+ * @param host 远程主机地址
|
|
|
+ * @param username 远程主机用户名
|
|
|
* @param pemFilePath 远程主机登录密钥路径
|
|
|
- * @param cmd 执行的命令
|
|
|
- * @param listener 标准输出监听器
|
|
|
- * @return String 执行结果
|
|
|
+ * @param cmd 执行的命令
|
|
|
+ * @param listener 标准输出监听器
|
|
|
+ * @return String 执行结果
|
|
|
*/
|
|
|
- public static String execmd(String host, String username, String pemFilePath, String cmd, StdoutListener listener) {
|
|
|
+ public static String execmd(
|
|
|
+ String host, String username, String pemFilePath, String cmd, StdoutListener listener) {
|
|
|
File pemFile = new File(pemFilePath);
|
|
|
Connection connection = login(host, username, pemFile, listener);
|
|
|
return execmd(connection, cmd, listener);
|
|
|
@@ -215,30 +215,36 @@ public class ShellSSH2Util {
|
|
|
/**
|
|
|
* 执行远程Shell命令
|
|
|
*
|
|
|
- * @param host 远程主机地址
|
|
|
+ * @param host 远程主机地址
|
|
|
* @param username 远程主机用户名
|
|
|
* @param password 远程主机密码
|
|
|
- * @param cmd 执行的命令
|
|
|
+ * @param cmd 执行的命令
|
|
|
* @param listener 标准输出监听器
|
|
|
- * @return String 执行结果
|
|
|
+ * @return String 执行结果
|
|
|
*/
|
|
|
- public static String execmdByPwd(String host, int port, String username, String password, String cmd, StdoutListener listener) {
|
|
|
+ public static String execmdByPwd(
|
|
|
+ String host,
|
|
|
+ int port,
|
|
|
+ String username,
|
|
|
+ String password,
|
|
|
+ String cmd,
|
|
|
+ StdoutListener listener) {
|
|
|
Connection connection = login(host, port, username, password, listener);
|
|
|
return execmd(connection, cmd, listener);
|
|
|
}
|
|
|
|
|
|
-
|
|
|
/**
|
|
|
* 执行远程Shell命令
|
|
|
*
|
|
|
- * @param host 远程主机地址
|
|
|
+ * @param host 远程主机地址
|
|
|
* @param username 远程主机用户名
|
|
|
- * @param pemFile 远程主机登录密钥
|
|
|
- * @param cmd 执行的命令
|
|
|
+ * @param pemFile 远程主机登录密钥
|
|
|
+ * @param cmd 执行的命令
|
|
|
* @param listener 标准输出监听器
|
|
|
- * @return String 执行结果
|
|
|
+ * @return String 执行结果
|
|
|
*/
|
|
|
- public static String execmd(String host, String username, File pemFile, String cmd, StdoutListener listener) {
|
|
|
+ public static String execmd(
|
|
|
+ String host, String username, File pemFile, String cmd, StdoutListener listener) {
|
|
|
Connection connection = login(host, username, pemFile, listener);
|
|
|
return execmd(connection, cmd, listener);
|
|
|
}
|
|
|
@@ -252,7 +258,13 @@ public class ShellSSH2Util {
|
|
|
* @param listener
|
|
|
* @return
|
|
|
*/
|
|
|
- public static String execmd(String host, int port, String username, File pemFile, String cmd, StdoutListener listener) {
|
|
|
+ public static String execmd(
|
|
|
+ String host,
|
|
|
+ int port,
|
|
|
+ String username,
|
|
|
+ File pemFile,
|
|
|
+ String cmd,
|
|
|
+ StdoutListener listener) {
|
|
|
Connection connection = login(host, port, username, pemFile, listener);
|
|
|
return execmd(connection, cmd, listener);
|
|
|
}
|
|
|
@@ -268,32 +280,42 @@ public class ShellSSH2Util {
|
|
|
Session session = null;
|
|
|
try {
|
|
|
if (connection != null) {
|
|
|
- session = connection.openSession();// 打开一个会话
|
|
|
- session.execCommand(cmd);// 执行命令
|
|
|
+ session = connection.openSession(); // 打开一个会话
|
|
|
+ session.execCommand(cmd); // 执行命令
|
|
|
|
|
|
// 使用线程池来管理输出读取线程
|
|
|
ExecutorService executorService = Executors.newFixedThreadPool(2);
|
|
|
|
|
|
// 读取标准输出
|
|
|
Session finalSession = session;
|
|
|
- executorService.submit(() -> {
|
|
|
- String stdoutResult = processStdout(finalSession.getStdout(), DEFAULT_CHART, listener);
|
|
|
- listener.stdout("命令标准输出结果：\n");
|
|
|
- listener.stdout(stdoutResult);
|
|
|
- if (StringUtils.isBlank(stdoutResult)) {
|
|
|
- listener.stderr("得到标准输出为空, 链接conn:" + connection + ",执行的命令：" + cmd);
|
|
|
-
|
|
|
- listener.stderr("得到标准输出为空, 链接conn:" + connection + ",执行的命令：" + cmd + "; 错误信息：" + stdoutResult);
|
|
|
- } else {
|
|
|
- listener.stdout("执行命令成功, 链接conn:" + connection + ",执行的命令：" + cmd);
|
|
|
- }
|
|
|
- });
|
|
|
+ executorService.submit(
|
|
|
+ () -> {
|
|
|
+ String stdoutResult =
|
|
|
+ processStdout(
|
|
|
+ finalSession.getStdout(), DEFAULT_CHART, listener);
|
|
|
+ listener.stdout("命令标准输出结果：\n");
|
|
|
+ listener.stdout(stdoutResult);
|
|
|
+ if (StringUtils.isBlank(stdoutResult)) {
|
|
|
+ listener.stderr("得到标准输出为空, 链接conn:" + connection + ",执行的命令：" + cmd);
|
|
|
+
|
|
|
+ listener.stderr(
|
|
|
+ "得到标准输出为空, 链接conn:"
|
|
|
+ + connection
|
|
|
+ + ",执行的命令："
|
|
|
+ + cmd
|
|
|
+ + "; 错误信息："
|
|
|
+ + stdoutResult);
|
|
|
+ } else {
|
|
|
+ listener.stdout("执行命令成功, 链接conn:" + connection + ",执行的命令：" + cmd);
|
|
|
+ }
|
|
|
+ });
|
|
|
|
|
|
// 读取标准错误
|
|
|
Session finalSession1 = session;
|
|
|
- executorService.submit(() -> {
|
|
|
- processStdout(finalSession1.getStderr(), DEFAULT_CHART, listener);
|
|
|
- });
|
|
|
+ executorService.submit(
|
|
|
+ () -> {
|
|
|
+ processStdout(finalSession1.getStderr(), DEFAULT_CHART, listener);
|
|
|
+ });
|
|
|
|
|
|
// 等待所有输出读取线程完成
|
|
|
executorService.shutdown();
|
|
|
@@ -320,13 +342,12 @@ public class ShellSSH2Util {
|
|
|
}
|
|
|
}
|
|
|
return result;
|
|
|
-
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* 解析脚本执行返回的结果集
|
|
|
*
|
|
|
- * @param in 输入流对象
|
|
|
+ * @param in 输入流对象
|
|
|
* @param charset 编码
|
|
|
* @return 以纯文本的格式返回
|
|
|
*/
|
|
|
@@ -362,22 +383,28 @@ public class ShellSSH2Util {
|
|
|
String username = "ubuntu";
|
|
|
File pemFile = new File("/Users/liyl/Desktop/jumpserver.pem");
|
|
|
String cmd = "./sb abc T000 test37.advichcloud.com";
|
|
|
- cmd = "mysql -h advich-wp-product-db.cqhn3c5h2mcz.us-west-2.rds.amazonaws.com -u admin -p Ew7MEm1EmJpDVvdK";
|
|
|
-// cmd = "uname -a";
|
|
|
- execmd(host, username, pemFile, cmd, new StdoutListener() {
|
|
|
- @Override
|
|
|
- public void stdout(String line) {
|
|
|
- System.out.println("标准：" + line);
|
|
|
- if ("ERROR:FILE_NAME IS EXISTS".equals(line)) {
|
|
|
- // TODO 已存在，退出
|
|
|
- }
|
|
|
- }
|
|
|
+ cmd =
|
|
|
+ "mysql -h advich-wp-product-db.cqhn3c5h2mcz.us-west-2.rds.amazonaws.com -u admin -p Ew7MEm1EmJpDVvdK";
|
|
|
+ // cmd = "uname -a";
|
|
|
+ execmd(
|
|
|
+ host,
|
|
|
+ username,
|
|
|
+ pemFile,
|
|
|
+ cmd,
|
|
|
+ new StdoutListener() {
|
|
|
+ @Override
|
|
|
+ public void stdout(String line) {
|
|
|
+ System.out.println("标准：" + line);
|
|
|
+ if ("ERROR:FILE_NAME IS EXISTS".equals(line)) {
|
|
|
+ // TODO 已存在，退出
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
- @Override
|
|
|
- public void stderr(String line) {
|
|
|
- System.out.println("错误：" + line);
|
|
|
- }
|
|
|
- });
|
|
|
+ @Override
|
|
|
+ public void stderr(String line) {
|
|
|
+ System.out.println("错误：" + line);
|
|
|
+ }
|
|
|
+ });
|
|
|
long currentTimeMillis1 = System.currentTimeMillis();
|
|
|
System.out.println("花费时间：" + (currentTimeMillis1 - currentTimeMillis));
|
|
|
}
|
|
|
@@ -403,8 +430,5 @@ public class ShellSSH2Util {
|
|
|
* @param line 每行输出内容
|
|
|
*/
|
|
|
void stderr(String line);
|
|
|
-
|
|
|
}
|
|
|
-
|
|
|
-
|
|
|
}
|
wfansh