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

Support for custom drivers

Browse Source
This commit is contained in:
jipengfei-jpf
2023-06-23 15:54:53 +08:00
parent 63d002df55
commit 150d1399a6
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -46,7 +46,7 @@ public class Chat2DBContext {
} }
} }
public static DriverConfig getDefaultDriverConfig(String dbType){ public static DriverConfig getDefaultDriverConfig(String dbType) {
return PLUGIN_MAP.get(dbType).getDBConfig().getDefaultDriverConfig(); return PLUGIN_MAP.get(dbType).getDBConfig().getDefaultDriverConfig();
} }
@ -63,7 +63,7 @@ public class Chat2DBContext {
return PLUGIN_MAP.get(getConnectInfo().getDbType()).getMetaData(); return PLUGIN_MAP.get(getConnectInfo().getDbType()).getMetaData();
} }
public static DBConfig getDBConfig(){ public static DBConfig getDBConfig() {
return PLUGIN_MAP.get(getConnectInfo().getDbType()).getDBConfig(); return PLUGIN_MAP.get(getConnectInfo().getDbType()).getDBConfig();
} }
@ -103,6 +103,12 @@ public class Chat2DBContext {
if (session != null) { if (session != null) {
url = url.replace(host, "127.0.0.1").replace(port, ssh.getLocalPort()); url = url.replace(host, "127.0.0.1").replace(port, ssh.getLocalPort());
} }
DriverConfig config = connectInfo.getDriverConfig();
if (config == null) {
config = getDefaultDriverConfig(connectInfo.getDbType());
connectInfo.setDriverConfig(config);
}
connection = getConnect(url, host, port, connectInfo.getUser(), connection = getConnect(url, host, port, connectInfo.getUser(),
connectInfo.getPassword(), connectInfo.getDbType(), connectInfo.getPassword(), connectInfo.getDbType(),
connectInfo.getDriverConfig(), ssh, connectInfo.getExtendMap()); connectInfo.getDriverConfig(), ssh, connectInfo.getExtendMap());