opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-01 18:53:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add custom jdbc

Browse Source
This commit is contained in:
jipengfei-jpf
2023-07-19 21:40:37 +08:00
parent 7be9d37981
commit d0879fa06f
4 changed files with 30 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/aspect/ConnectionInfoHandler.java
View File

@ -7,6 +7,7 @@ import ai.chat2db.server.tools.base.wrapper.result.DataResult;
import ai.chat2db.server.tools.common.exception.ParamBusinessException;
import ai.chat2db.server.web.api.controller.data.source.request.DataSourceBaseRequestInfo;
import ai.chat2db.server.web.api.controller.data.source.request.DataSourceConsoleRequestInfo;
import ai.chat2db.spi.config.DriverConfig;
import ai.chat2db.spi.sql.Chat2DBContext;
import ai.chat2db.spi.sql.ConnectInfo;
import lombok.extern.slf4j.Slf4j;
@ -28,7 +29,6 @@ public class ConnectionInfoHandler {
@Autowired
private DataSourceService dataSourceService;
@Around("within(@ai.chat2db.server.web.api.aspect.ConnectionInfoAspect *)")
public Object connectionInfoHandler(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
try {
@ -78,6 +78,10 @@ public class ConnectionInfoHandler {
connectInfo.setUrl(dataSource.getUrl());
connectInfo.setPort(dataSource.getPort() != null ? Integer.parseInt(dataSource.getPort()) : null);
connectInfo.setHost(dataSource.getHost());
DriverConfig driverConfig = dataSource.getDriverConfig();
if (driverConfig != null && driverConfig.notEmpty()) {
connectInfo.setDriverConfig(driverConfig);
}
return connectInfo;
}