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

Merge pull request #663 from 1194570458/dev

Browse Source 
修复clickhouse无法选中数据库问题
This commit is contained in:
ji
2023-10-28 11:04:48 +08:00
committed by GitHub
parent 961d158eb0 b89a4052e1
commit 17b1edcfe8
3 changed files with 37 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,10 +4,39 @@ import java.sql.Connection;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.sql.ConnectInfo;
import ai.chat2db.spi.sql.SQLExecutor;
import org.apache.commons.lang3.StringUtils;
import javax.net.ssl.HostnameVerifier;
public class ClickHouseDBManage extends DefaultDBManage implements DBManage {
@Override
public Connection getConnection(ConnectInfo connectInfo) {
String url = setDatabaseInJdbcUrl(connectInfo);
connectInfo.setUrl(url);
return super.getConnection(connectInfo);
}
private String setDatabaseInJdbcUrl(ConnectInfo connectInfo) {
String databaseName;
String url = connectInfo.getUrl();
if (StringUtils.isBlank((databaseName = connectInfo.getDatabaseName())) && StringUtils.isBlank((databaseName = connectInfo.getSchemaName()))) {
return url;
}
String connectAddress = connectInfo.getHost() + ":" + connectInfo.getPort();
String[] addressSplit = url.split(connectAddress);
String connectParams = addressSplit[1];
if (connectParams.startsWith("/")) {
// 删除连接参数中的 /
connectParams = connectParams.substring(1);
}
// 添加数据库名
return addressSplit[0] + connectAddress + "/" + databaseName + connectParams;
}
@Override
public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {

5
chat2db-server/chat2db-server-domain/chat2db-server-domain-api/src/main/java/ai/chat2db/server/domain/api/param/datasource/DataSourceUpdateParam.java
View File

@ -53,7 +53,10 @@ public class DataSourceUpdateParam {
*/
private String envType;
/**
* 环境id
*/
private Integer environmentId;
/**
* host

5
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/data/source/request/DataSourceUpdateRequest.java
View File

@ -56,7 +56,10 @@ public class DataSourceUpdateRequest {
*/
private String envType;
/**
* 环境id
*/
private Integer environmentId;
/**
* host