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

remove redundant code and optimize exception throwing

Browse Source
This commit is contained in:
zgq
2024-03-28 14:49:40 +08:00
parent f425604b15
commit af467eb5fc
9 changed files with 54 additions and 228 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2DBManage.java
View File

@ -11,28 +11,20 @@ import org.apache.commons.lang3.StringUtils;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.Objects;
public class DB2DBManage extends DefaultDBManage implements DBManage {
@Override
public String exportDatabaseData(Connection connection, String databaseName, String schemaName, String tableName) throws SQLException {
StringBuilder sqlBuilder = new StringBuilder();
exportTableData(connection, tableName, sqlBuilder);
return sqlBuilder.toString();
}
@Override
public String exportDatabase(Connection connection, String databaseName, String schemaName, boolean containData) throws SQLException {
StringBuilder sqlBuilder = new StringBuilder();
exportTables(connection, schemaName, sqlBuilder, containData);
exportViews(connection, schemaName, sqlBuilder);
exportProceduresAndFunctions(connection,schemaName, sqlBuilder);
exportTriggers(connection, schemaName,sqlBuilder);
exportProceduresAndFunctions(connection, schemaName, sqlBuilder);
exportTriggers(connection, schemaName, sqlBuilder);
return sqlBuilder.toString();
}
private void exportTables(Connection connection, String schemaName, StringBuilder sqlBuilder, boolean containData) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) {
@ -53,34 +45,12 @@ public class DB2DBManage extends DefaultDBManage implements DBManage {
if (resultSet.next()) {
sqlBuilder.append(resultSet.getString("sql")).append("\n");
if (containData) {
exportTableData(connection, tableName, sqlBuilder);
exportTableData(connection, schemaName, tableName, sqlBuilder);
}
}
}
}
private void exportTableData(Connection connection, String tableName, StringBuilder sqlBuilder) throws SQLException {
String sql = String.format("select * from %s", tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
ResultSetMetaData metaData = resultSet.getMetaData();
while (resultSet.next()) {
sqlBuilder.append("INSERT INTO ").append(tableName).append(" VALUES (");
for (int i = 1; i <= metaData.getColumnCount(); i++) {
String value = resultSet.getString(i);
if (Objects.isNull(value)) {
sqlBuilder.append("NULL");
} else {
sqlBuilder.append("'").append(value).append("'");
}
if (i < metaData.getColumnCount()) {
sqlBuilder.append(", ");
}
}
sqlBuilder.append(");\n");
}
sqlBuilder.append("\n");
}
}
private void exportViews(Connection connection, String schemaName, StringBuilder sqlBuilder) throws SQLException {
String sql = String.format("select TEXT from syscat.views where VIEWSCHEMA='%s';", schemaName);
@ -93,7 +63,7 @@ public class DB2DBManage extends DefaultDBManage implements DBManage {
}
private void exportProceduresAndFunctions(Connection connection, String schemaName, StringBuilder sqlBuilder) throws SQLException {
String sql =String.format( "select TEXT from syscat.routines where ROUTINESCHEMA='%s';",schemaName);
String sql = String.format("select TEXT from syscat.routines where ROUTINESCHEMA='%s';", schemaName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) {
String ddl = resultSet.getString("TEXT");
@ -103,9 +73,8 @@ public class DB2DBManage extends DefaultDBManage implements DBManage {
}
private void exportTriggers(Connection connection, String schemaName, StringBuilder sqlBuilder) throws SQLException {
String sql = String.format("select * from SYSCAT.TRIGGERS where TRIGSCHEMA = '%s';",schemaName);
String sql = String.format("select * from SYSCAT.TRIGGERS where TRIGSCHEMA = '%s';", schemaName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) {
String ddl = resultSet.getString("TEXT");
@ -127,9 +96,10 @@ public class DB2DBManage extends DefaultDBManage implements DBManage {
e.printStackTrace();
}
}
@Override
public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
String sql = "DROP TABLE " + tableName;
SQLExecutor.getInstance().execute(connection,sql, resultSet -> null);
SQLExecutor.getInstance().execute(connection, sql, resultSet -> null);
}
}