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

add pg default database

Browse Source
This commit is contained in:
jipengfei-jpf
2023-07-16 18:51:48 +08:00
parent 7d7b52026e
commit b7b9069fcc
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/ssh/SSHManager.java
View File

@ -17,17 +17,23 @@ import org.apache.commons.lang3.StringUtils;
public class SSHManager { public class SSHManager {
public static Session getSSHSession(SSHInfo ssh) { public static Session getSSHSession(SSHInfo ssh) {
Session session; Session session = null;
try { try {
byte[] passphrase = StringUtils.isNotBlank(ssh.getPassphrase()) ? StringUtils.getBytes(ssh.getPassphrase(), if (StringUtils.isNotBlank(ssh.getKeyFile())) {
byte[] passphrase = StringUtils.isNotBlank(ssh.getPassphrase()) ? StringUtils.getBytes(
ssh.getPassphrase(),
"UTF-8") : null; "UTF-8") : null;
session = JschUtil.getSession(ssh.getHostName(), Integer.parseInt(ssh.getPort()), ssh.getUserName(), session = JschUtil.getSession(ssh.getHostName(), Integer.parseInt(ssh.getPort()), ssh.getUserName(),
ssh.getKeyFile(), passphrase); ssh.getKeyFile(), passphrase);
} else if (StringUtils.isNotBlank(ssh.getUserName())) {
session = JschUtil.getSession(ssh.getHostName(), Integer.parseInt(ssh.getPort()), ssh.getUserName(),
ssh.getPassword());
}
} catch (Exception e) { } catch (Exception e) {
throw new ConnectionException("connection.ssh.error", null, e); throw new ConnectionException("connection.ssh.error", null, e);
} }
if (StringUtils.isNotBlank(ssh.getRHost()) && StringUtils.isNotBlank(ssh.getRPort())) { if (session != null && StringUtils.isNotBlank(ssh.getRHost()) && StringUtils.isNotBlank(ssh.getRPort())) {
try { try {
int localPort = !StringUtils.isBlank(ssh.getLocalPort()) ? Integer.parseInt(ssh.getLocalPort()) int localPort = !StringUtils.isBlank(ssh.getLocalPort()) ? Integer.parseInt(ssh.getLocalPort())
: NetUtil.getUsableLocalPort(); : NetUtil.getUsableLocalPort();