opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 02:32:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix-exportTableData

Browse Source
This commit is contained in:
zgq
2024-06-21 18:29:12 +08:00
parent 9421c3d3f4
commit 2bfbbaeb87
9 changed files with 127 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/OracleDBManage.java
View File

@ -30,29 +30,25 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
private static String TRIGGER_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('TRIGGER', trigger_name) AS ddl FROM all_triggers WHERE owner = '%s' AND trigger_name = '%s'";
private static String FUNCTION_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('FUNCTION', object_name) as ddl FROM all_procedures WHERE owner = '%s' AND object_name = '%s'";
@Override
public void exportDatabaseData(Connection connection, String databaseName, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
exportTableData(connection, tableName, asyncContext);
}
public void exportDatabase(Connection connection, String databaseName, String schemaName,AsyncContext asyncContext) throws SQLException {
exportTables(connection, schemaName, asyncContext);
exportTables(connection,databaseName, schemaName, asyncContext);
exportViews(connection, asyncContext, schemaName);
exportProcedures(connection, schemaName, asyncContext);
exportTriggers(connection, schemaName, asyncContext);
exportFunctions(connection, schemaName, asyncContext);
}
private void exportTables(Connection connection, String schemaName, AsyncContext asyncContext) throws SQLException {
private void exportTables(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) {
String tableName = resultSet.getString("TABLE_NAME");
exportTable(connection, schemaName, tableName, asyncContext);
exportTable(connection,databaseName, schemaName, tableName, asyncContext);
}
}
}
private void exportTable(Connection connection, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
private void exportTable(Connection connection, String databaseName, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
String sql = String.format(TABLE_DDL_SQL, schemaName, tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
@ -64,7 +60,7 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
exportTableComments(connection, tableName, asyncContext);
exportTableColumnsComments(connection, tableName, asyncContext);
if (asyncContext.isContainsData()) {
exportTableData(connection, tableName, asyncContext);
exportTableData(connection,databaseName,schemaName, tableName, asyncContext);
}
}
}
@ -91,36 +87,6 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
}
}
private void exportTableData(Connection connection, String tableName, AsyncContext asyncContext) throws SQLException {
String sql = String.format("SELECT * FROM %s", tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
ResultSetMetaData metaData = resultSet.getMetaData();
int columnCount = metaData.getColumnCount();
while (resultSet.next()) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("INSERT INTO ").append(tableName).append(" VALUES (");
for (int i = 1; i <= columnCount; i++) {
String columnValue = resultSet.getString(i);
if (Objects.isNull(columnValue)) {
sqlBuilder.append("NULL");
} else if (metaData.getColumnTypeName(i).equalsIgnoreCase("DATE")) {
// 处理日期值格式
columnValue = "TO_DATE('" + columnValue + "', 'YYYY-MM-DD HH24:MI:SS')";
sqlBuilder.append(columnValue);
} else {
sqlBuilder.append("'").append(columnValue).append("'");
}
if (i < columnCount) {
sqlBuilder.append(", ");
}
}
sqlBuilder.append(");");
sqlBuilder.append("\n");
asyncContext.write(sqlBuilder.toString());
}
}
}
private void exportViews(Connection connection, AsyncContext asyncContext, String schemaName) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"VIEW"})) {
while (resultSet.next()) {