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

fix exception

Browse Source
This commit is contained in:
jipengfei-jpf
2023-07-09 21:03:29 +08:00
parent 46e2d5279e
commit e533a31f39
5 changed files with 44 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -103,8 +103,8 @@ public class Chat2DBContext {
if (session != null) {
url = url.replace(host, "127.0.0.1").replace(port, ssh.getLocalPort());
}
}catch (Exception e){
throw new ConnectionException("connection.ssh.error",null,e);
} catch (Exception e) {
throw new ConnectionException("connection.ssh.error", null, e);
}
try {
DriverConfig config = connectInfo.getDriverConfig();
@ -133,7 +133,7 @@ public class Chat2DBContext {
} catch (Exception e) {
}
}
throw new BusinessException("connection.error",null,e1);
throw new BusinessException("connection.error", null, e1);
}
connectInfo.setSession(session);
connectInfo.setConnection(connection);
@ -167,7 +167,17 @@ public class Chat2DBContext {
} catch (SQLException e) {
log.error("close connection error", e);
}
CONNECT_INFO_THREAD_LOCAL.remove();
Session session = connectInfo.getSession();
if (session != null && session.isConnected() && connectInfo.getSsh() != null
&& connectInfo.getSsh().isUse()) {
try {
session.delPortForwardingL(Integer.parseInt(connectInfo.getSsh().getLocalPort()));
} catch (JSchException e) {
}
}
}
}