opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-09-24 23:55:57 +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

18
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 static Session getSSHSession(SSHInfo ssh) {
Session session;
Session session = null;
try {
byte[] passphrase = StringUtils.isNotBlank(ssh.getPassphrase()) ? StringUtils.getBytes(ssh.getPassphrase(),
"UTF-8") : null;
session = JschUtil.getSession(ssh.getHostName(), Integer.parseInt(ssh.getPort()), ssh.getUserName(),
ssh.getKeyFile(), passphrase);
if (StringUtils.isNotBlank(ssh.getKeyFile())) {
byte[] passphrase = StringUtils.isNotBlank(ssh.getPassphrase()) ? StringUtils.getBytes(
ssh.getPassphrase(),
"UTF-8") : null;
session = JschUtil.getSession(ssh.getHostName(), Integer.parseInt(ssh.getPort()), ssh.getUserName(),
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) {
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 {
int localPort = !StringUtils.isBlank(ssh.getLocalPort()) ? Integer.parseInt(ssh.getLocalPort())
: NetUtil.getUsableLocalPort();