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

copy and delete Table

Browse Source
This commit is contained in:
SwallowGG
2024-06-27 11:50:55 +08:00
parent 0152ec4ead
commit 57f0cfa2db
10 changed files with 125 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -96,4 +96,13 @@ public class ClickHouseDBManage extends DefaultDBManage implements DBManage {
}
@Override
public void copyTable(Connection connection, String databaseName, String schemaName, String tableName, String newTableName,boolean copyData) throws SQLException {
String sql = "CREATE TABLE " + newTableName + " AS " + tableName + "";
SQLExecutor.getInstance().execute(connection, sql, resultSet -> null);
if(copyData){
sql = "INSERT INTO " + newTableName + " SELECT * FROM " + tableName;
SQLExecutor.getInstance().execute(connection, sql, resultSet -> null);
}
}
}

15
chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/ClickHouseMetaData.java
View File

@ -81,7 +81,7 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
@Override
public String tableDDL(Connection connection, @NotEmpty String databaseName, String schemaName,
@NotEmpty String tableName) {
String sql = "SHOW CREATE TABLE " + format(databaseName) + "."
String sql = "SHOW CREATE TABLE " + format(schemaName) + "."
+ format(tableName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
if (resultSet.next()) {
@ -112,7 +112,7 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
@Override
public List<Trigger> triggers(Connection connection, String databaseName, String schemaName) {
List<Trigger> triggers = new ArrayList<>();
String sql = String.format(TRIGGER_SQL_LIST, databaseName);
String sql = String.format(TRIGGER_SQL_LIST, schemaName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
while (resultSet.next()) {
Trigger trigger = new Trigger();
@ -145,7 +145,7 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
@Override
public Procedure procedure(Connection connection, @NotEmpty String databaseName, String schemaName,
String procedureName) {
String sql = String.format(ROUTINES_SQL, "PROCEDURE", databaseName, procedureName);
String sql = String.format(ROUTINES_SQL, "PROCEDURE", schemaName, procedureName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
Procedure procedure = new Procedure();
procedure.setDatabaseName(databaseName);
@ -235,7 +235,7 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
@Override
public List<TableIndex> indexes(Connection connection, String databaseName, String schemaName, String tableName) {
StringBuilder queryBuf = new StringBuilder("SHOW INDEX FROM ");
queryBuf.append("`").append(tableName).append("`");
queryBuf.append("`").append(schemaName).append("`");
queryBuf.append(" FROM ");
queryBuf.append("`").append(databaseName).append("`");
return SQLExecutor.getInstance().execute(connection, queryBuf.toString(), resultSet -> {
@ -298,11 +298,8 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
@Override
public String getMetaDataName(String... names) {
return Arrays.stream(names)
.skip(1) // 跳过第一个名称
.filter(StringUtils::isNotBlank)
.map(name -> "`" + name + "`")
.collect(Collectors.joining("."));
return Arrays.stream(names).filter(name -> StringUtils.isNotBlank(name)).map(name -> "`" + name + "`").collect(Collectors.joining("."));
}