opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 11:12:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #736 from 1194570458/dev

Browse Source 
Fix the problem of duplicate database selection.
This commit is contained in:
Chat2DB
2024-10-28 14:08:07 +08:00
committed by GitHub
parent c5afdf1f7d 37b59f4359
commit 38b03a7c11
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/ClickHouseDBManage.java
View File

@ -80,10 +80,24 @@ public class ClickHouseDBManage extends DefaultDBManage implements DBManage {
String connectAddress = connectInfo.getHost() + ":" + connectInfo.getPort();
String[] addressSplit = url.split(connectAddress);
String connectParams = addressSplit[1];
String connectParams = addressSplit.length == 2 ? addressSplit[1] : "";
if (connectParams.startsWith("/")) {
// Remove / from connection parameters
connectParams = connectParams.substring(1);
if (connectParams.startsWith(databaseName)) {
// 删除连接参数中的数据库名
connectParams = connectParams.substring(databaseName.length());
} else {
// 是否有连接参数
int beginIndex = connectParams.indexOf("?");
// 无连接参数直接设置 ""
if (beginIndex == -1) {
connectParams = "";
} else {
// 删除连接参数前的数据库名
connectParams = connectParams.substring(beginIndex);
}
}
}
// Add database name
return addressSplit[0] + connectAddress + "/" + databaseName + connectParams;