opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-09-24 07:27:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

SSH support

Browse Source
This commit is contained in:
jipengfei-jpf
2023-07-08 20:17:38 +08:00
parent a4aa5aa9d9
commit 8e1271ee08
7 changed files with 59 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/Chat2DBContext.java
View File

@ -14,6 +14,7 @@ import ai.chat2db.spi.Plugin;
import ai.chat2db.spi.config.DBConfig;
import ai.chat2db.spi.config.DriverConfig;
import ai.chat2db.spi.model.SSHInfo;
import ai.chat2db.spi.ssh.SSHManager;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import lombok.extern.slf4j.Slf4j;

80
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/SSHManager.java
View File

@ -1,80 +0,0 @@
package ai.chat2db.spi.sql;
import java.util.concurrent.ConcurrentHashMap;
import ai.chat2db.spi.model.SSHInfo;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
/**
* @author jipengfei
* @version : SSHSessionManager.java
*/
@Slf4j
public class SSHManager {
private static final ConcurrentHashMap<SSHInfo, Session> SSH_SESSION_MAP = new ConcurrentHashMap();
public static Session getSSHSession(SSHInfo sshInfo) {
Session session = SSH_SESSION_MAP.get(sshInfo);
if (session != null && session.isConnected()) {
return session;
} else {
return createSession(sshInfo);
}
}
private static Session createSession(SSHInfo ssh) {
synchronized (ssh) {
Session session = SSH_SESSION_MAP.get(ssh);
if (session != null && session.isConnected()) {
return session;
}
try {
JSch jSch = new JSch();
session = jSch.getSession(ssh.getUserName(), ssh.getHostName(), Integer.parseInt(ssh.getPort()));
session.setPassword(ssh.getPassword());
session.setConfig("StrictHostKeyChecking", "no");
session.connect();
SSH_SESSION_MAP.put(ssh, session);
} catch (Exception e) {
throw new RuntimeException("create ssh session error", e);
}
if (StringUtils.isNotBlank(ssh.getLocalPort()) && StringUtils.isNotBlank(ssh.getRHost())
&& StringUtils.isNotBlank(ssh.getRPort())) {
try {
int port1 = session.setPortForwardingL(Integer.parseInt(ssh.getLocalPort()), ssh.getRHost(),
Integer.parseInt(ssh.getRPort()));
} catch (Exception e) {
if (session != null && session.isConnected()) {
session.disconnect();
SSH_SESSION_MAP.remove(ssh);
}
throw new RuntimeException(ssh.getLocalPort() + " port is used，please change to another port ", e);
}
}
return session;
}
}
public static void close() {
SSH_SESSION_MAP.forEach((k, v) -> {
if (v != null && v.isConnected()) {
try {
v.delPortForwardingL(Integer.parseInt(k.getLocalPort()));
} catch (Exception e) {
log.error("delPortForwardingL error", e);
}
try {
v.disconnect();
} catch (Exception e) {
log.error("disconnect error", e);
}
}
});
}
}