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

18
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();
}
@ -63,7 +63,7 @@ public class Chat2DBContext {
return PLUGIN_MAP.get(getConnectInfo().getDbType()).getMetaData();
}
public static DBConfig getDBConfig(){
public static DBConfig getDBConfig() {
return PLUGIN_MAP.get(getConnectInfo().getDbType()).getDBConfig();
}
@ -103,9 +103,15 @@ public class Chat2DBContext {
if (session != null) {
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(),
connectInfo.getPassword(), connectInfo.getDbType(),
connectInfo.getDriverConfig(), ssh, connectInfo.getExtendMap());
connectInfo.getPassword(), connectInfo.getDbType(),
connectInfo.getDriverConfig(), ssh, connectInfo.getExtendMap());
} catch (Exception e1) {
log.error("getConnect error", e1);
if (connection != null) {
@ -139,8 +145,8 @@ public class Chat2DBContext {
* @return
*/
private static Connection getConnect(String url, String host, String port,
String userName, String password, String dbType,
DriverConfig jdbc, SSHInfo ssh, Map<String, Object> properties) throws SQLException {
String userName, String password, String dbType,
DriverConfig jdbc, SSHInfo ssh, Map<String, Object> properties) throws SQLException {
// 创建连接
return IDriverManager.getConnection(url, userName, password, jdbc, properties);