package com.wgcloud.service;
|
|
import com.jcraft.jsch.ChannelExec;
|
import com.jcraft.jsch.JSch;
|
import com.jcraft.jsch.JSchException;
|
import com.jcraft.jsch.Session;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
|
import java.io.BufferedReader;
|
import java.io.InputStreamReader;
|
import java.util.Properties;
|
|
@Service
|
public class RouterConnectionService {
|
|
@Value("${router.host}")
|
private String routerHost;
|
|
@Value("${router.username}")
|
private String routerUsername;
|
|
@Value("${router.password}")
|
private String routerPassword;
|
|
private Session session;
|
|
/**
|
* 连接到路由器
|
*/
|
public boolean connect() {
|
try {
|
JSch jsch = new JSch();
|
session = jsch.getSession(routerUsername, routerHost, 22);
|
session.setPassword(routerPassword);
|
|
// 设置不进行主机密钥检查
|
Properties config = new Properties();
|
config.put("StrictHostKeyChecking", "no");
|
session.setConfig(config);
|
|
session.connect(5000); // 5秒超时
|
return true;
|
} catch (JSchException e) {
|
System.err.println("连接路由器失败: " + e.getMessage());
|
return false;
|
}
|
}
|
|
/**
|
* 断开连接
|
*/
|
public void disconnect() {
|
if (session != null && session.isConnected()) {
|
session.disconnect();
|
}
|
}
|
|
/**
|
* 执行路由器命令
|
*/
|
public String executeCommand(String command) {
|
if (!isConnected()) {
|
if (!connect()) {
|
return null;
|
}
|
}
|
|
try {
|
ChannelExec channel = (ChannelExec) session.openChannel("exec");
|
channel.setCommand(command);
|
channel.connect();
|
|
BufferedReader reader = new BufferedReader(new InputStreamReader(channel.getInputStream()));
|
StringBuilder output = new StringBuilder();
|
String line;
|
|
while ((line = reader.readLine()) != null) {
|
output.append(line).append("\n");
|
}
|
|
reader.close();
|
channel.disconnect();
|
return output.toString();
|
} catch (Exception e) {
|
System.err.println("执行命令失败: " + e.getMessage());
|
disconnect(); // 执行失败时断开连接，下次重连
|
return null;
|
}
|
}
|
|
/**
|
* 检查连接状态
|
*/
|
public boolean isConnected() {
|
return session != null && session.isConnected();
|
}
|
}
|
