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

refactor(OracleDBManage): fix get tableDDL

Browse Source
This commit is contained in:
zgq
2024-07-09 14:35:13 +08:00
parent 0f23e12e50
commit ea5a7e89ff
1 changed files with 20 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -12,13 +12,10 @@ import org.apache.commons.lang3.StringUtils;
import java.sql.Connection; import java.sql.Connection;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.Objects;
@Slf4j @Slf4j
public class OracleDBManage extends DefaultDBManage implements DBManage { public class OracleDBManage extends DefaultDBManage implements DBManage {
private static String TABLE_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('TABLE', table_name) as ddl FROM all_tables WHERE owner = '%s' AND table_name = '%s'";
private static String TABLE_COMMENT_SQL = "SELECT 'COMMENT ON TABLE ' || table_name || ' IS ''' || comments || ''';' AS table_comment_ddl FROM user_tab_comments WHERE table_name = '%s'"; private static String TABLE_COMMENT_SQL = "SELECT 'COMMENT ON TABLE ' || table_name || ' IS ''' || comments || ''';' AS table_comment_ddl FROM user_tab_comments WHERE table_name = '%s'";
private static String TABLE_COLUMN_COMMENT_SQL = "SELECT 'COMMENT ON COLUMN ' || table_name || '.' || column_name || ' IS ''' || comments || ''';' AS column_comment_ddl " + private static String TABLE_COLUMN_COMMENT_SQL = "SELECT 'COMMENT ON COLUMN ' || table_name || '.' || column_name || ' IS ''' || comments || ''';' AS column_comment_ddl " +
"FROM user_col_comments " + "FROM user_col_comments " +
@ -49,20 +46,18 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
public void exportTable(Connection connection, String databaseName, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException { public void exportTable(Connection connection, String databaseName, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
String sql = String.format(TABLE_DDL_SQL, schemaName, tableName); String tableDDL = Chat2DBContext.getMetaData().tableDDL(connection, databaseName, schemaName, tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
StringBuilder sqlBuilder = new StringBuilder(); StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("DROP TABLE ").append(schemaName).append(".").append(tableName).append(";") sqlBuilder.append("DROP TABLE ").append(schemaName).append(".").append(tableName).append(";")
.append(resultSet.getString("ddl")).append(";").append("\n"); .append(tableDDL).append(";").append("\n");
asyncContext.write(sqlBuilder.toString()); asyncContext.write(sqlBuilder.toString());
}
exportTableComments(connection, tableName, asyncContext); exportTableComments(connection, tableName, asyncContext);
exportTableColumnsComments(connection, tableName, asyncContext); exportTableColumnsComments(connection, tableName, asyncContext);
if (asyncContext.isContainsData()) { if (asyncContext.isContainsData()) {
exportTableData(connection, databaseName, schemaName, tableName, asyncContext); exportTableData(connection, databaseName, schemaName, tableName, asyncContext);
} }
}
} }
private void exportTableComments(Connection connection, String tableName, AsyncContext asyncContext) throws SQLException { private void exportTableComments(Connection connection, String tableName, AsyncContext asyncContext) throws SQLException {