Skip to content

Commit 2c31fee

Browse files
dangdang01234dangdang01234
committed
support drop table
1 parent fd57226 commit 2c31fee

10 files changed

Lines changed: 89 additions & 8 deletions

File tree

‎chat2db-server/chat2db-plugins/chat2b-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,11 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
5151

5252
@Override
5353
public void dropTable(String databaseName, String schemaName, String tableName) {
54+
String sql = "DROP TABLE "+ format(tableName);
55+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
56+
}
5457

58+
public static String format(String tableName) {
59+
return "`" + tableName + "`";
5560
}
5661
}

‎chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/ClickHouseDBManage.java‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.clickhouse;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class ClickHouseDBManage implements DBManage {
67
@Override
@@ -39,6 +40,9 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
3940

4041
@Override
4142
public void dropTable(String databaseName, String schemaName, String tableName) {
42-
43+
String sql = "DROP TABLE IF EXISTS " + databaseName + "." + tableName;
44+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4345
}
46+
47+
4448
}

‎chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2DBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.db2;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class DB2DBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "DROP TABLE " + tableName;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-plugins/chat2db-dm/src/main/java/ai/chat2db/plugin/dm/DMDBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.dm;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class DMDBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "DROP TABLE IF EXISTS " +tableName;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-plugins/chat2db-h2/src/main/java/ai/chat2db/plugin/h2/H2DBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.h2;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class H2DBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "DROP TABLE " +tableName;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-plugins/chat2db-hive/src/main/java/ai/chat2db/plugin/hive/HiveDBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.hive;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class HiveDBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "drop table if exists " +tableName;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-plugins/chat2db-kingbase/src/main/java/ai/chat2db/plugin/kingbase/KingBaseDBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.kingbase;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class KingBaseDBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "drop table if exists " +tableName;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-plugins/chat2db-mariadb/src/main/java/ai/chat2db/plugin/mariadb/MariaDBManage.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package ai.chat2db.plugin.mariadb;
22

33
import ai.chat2db.spi.DBManage;
4+
import ai.chat2db.spi.sql.SQLExecutor;
45

56
public class MariaDBManage implements DBManage {
67
@Override
@@ -40,6 +41,7 @@ public void modifySchema(String databaseName, String schemaName, String newSchem
4041

4142
@Override
4243
public void dropTable(String databaseName, String schemaName, String tableName) {
43-
44+
String sql = "DROP TABLE " +tableName ;
45+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
4446
}
4547
}

‎chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/vo/TableVO.java‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,8 @@ public class TableVO {
3232
*/
3333
private List<IndexVO> indexList;
3434

35-
35+
/**
36+
* 是否已经被固定
37+
*/
3638
private boolean pinned;
3739
}

‎chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultDBManage.java‎

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
/**
2+
* alibaba.com Inc.
3+
* Copyright (c) 2004-2023 All Rights Reserved.
4+
*/
5+
package ai.chat2db.spi.jdbc;
6+
7+
import java.sql.SQLException;
8+
9+
import ai.chat2db.spi.DBManage;
10+
import ai.chat2db.spi.sql.SQLExecutor;
11+
12+
/**
13+
* @author jipengfei
14+
* @version : DefaultDBManage.java
15+
*/
16+
public class DefaultDBManage implements DBManage {
17+
18+
@Override
19+
public void connectDatabase(String database) {
20+
21+
}
22+
23+
@Override
24+
public void modifyDatabase(String databaseName, String newDatabaseName) {
25+
26+
}
27+
28+
@Override
29+
public void createDatabase(String databaseName) {
30+
31+
}
32+
33+
@Override
34+
public void dropDatabase(String databaseName) {
35+
36+
}
37+
38+
@Override
39+
public void createSchema(String databaseName, String schemaName) {
40+
41+
}
42+
43+
@Override
44+
public void dropSchema(String databaseName, String schemaName) {
45+
46+
}
47+
48+
@Override
49+
public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
50+
51+
}
52+
53+
@Override
54+
public void dropTable(String databaseName, String schemaName, String tableName) {
55+
String sql = "DROP TABLE "+ tableName ;
56+
SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
57+
}
58+
}

0 commit comments

Comments
 (0)