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

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

@ -12,26 +12,23 @@ 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 " +
"WHERE table_name = '%s' " + "WHERE table_name = '%s' " +
"AND comments IS NOT NULL"; "AND comments IS NOT NULL";
private static String VIEW_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('VIEW', view_name) as ddl FROM all_views WHERE owner = '%s' AND view_name = '%s'"; private static String VIEW_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('VIEW', view_name) as ddl FROM all_views WHERE owner = '%s' AND view_name = '%s'";
private String PROCEDURE_LIST_DDL ="SELECT object_name FROM all_procedures where OWNER = '%s' and OBJECT_TYPE='PROCEDURE'"; private String PROCEDURE_LIST_DDL = "SELECT object_name FROM all_procedures where OWNER = '%s' and OBJECT_TYPE='PROCEDURE'";
private static String PROCEDURE_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('PROCEDURE', object_name) as ddl FROM all_procedures WHERE owner = '%s' AND object_name = '%s'"; private static String PROCEDURE_DDL_SQL = "SELECT DBMS_METADATA.GET_DDL('PROCEDURE', object_name) as ddl FROM all_procedures WHERE owner = '%s' AND object_name = '%s'";
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 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'"; 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'";
public void exportDatabase(Connection connection, String databaseName, String schemaName,AsyncContext asyncContext) throws SQLException { public void exportDatabase(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {
exportTables(connection,databaseName, schemaName, asyncContext); exportTables(connection, databaseName, schemaName, asyncContext);
exportViews(connection, asyncContext, schemaName); exportViews(connection, asyncContext, schemaName);
exportProcedures(connection, schemaName, asyncContext); exportProcedures(connection, schemaName, asyncContext);
exportTriggers(connection, schemaName, asyncContext); exportTriggers(connection, schemaName, asyncContext);
@ -42,27 +39,25 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"TABLE", "SYSTEM TABLE"})) { try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) { while (resultSet.next()) {
String tableName = resultSet.getString("TABLE_NAME"); String tableName = resultSet.getString("TABLE_NAME");
exportTable(connection,databaseName, schemaName, tableName, asyncContext); exportTable(connection, databaseName, schemaName, tableName, asyncContext);
} }
} }
} }
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)) { StringBuilder sqlBuilder = new StringBuilder();
if (resultSet.next()) { sqlBuilder.append("DROP TABLE ").append(schemaName).append(".").append(tableName).append(";")
StringBuilder sqlBuilder = new StringBuilder(); .append(tableDDL).append(";").append("\n");
sqlBuilder.append("DROP TABLE ").append(schemaName).append(".").append(tableName).append(";") asyncContext.write(sqlBuilder.toString());
.append(resultSet.getString("ddl")).append(";").append("\n");
asyncContext.write(sqlBuilder.toString()); exportTableComments(connection, tableName, asyncContext);
} exportTableColumnsComments(connection, tableName, asyncContext);
exportTableComments(connection, tableName, asyncContext); if (asyncContext.isContainsData()) {
exportTableColumnsComments(connection, tableName, asyncContext); exportTableData(connection, databaseName, schemaName, tableName, asyncContext);
if (asyncContext.isContainsData()) {
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 {
@ -108,8 +103,8 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
} }
private void exportProcedures(Connection connection, String schemaName, AsyncContext asyncContext) throws SQLException { private void exportProcedures(Connection connection, String schemaName, AsyncContext asyncContext) throws SQLException {
String sql = String.format(PROCEDURE_LIST_DDL,schemaName); String sql = String.format(PROCEDURE_LIST_DDL, schemaName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) { try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) { while (resultSet.next()) {
String procedureName = resultSet.getString("object_name"); String procedureName = resultSet.getString("object_name");
exportProcedure(connection, schemaName, procedureName, asyncContext); exportProcedure(connection, schemaName, procedureName, asyncContext);
@ -185,11 +180,11 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
} }
@Override @Override
public void copyTable(Connection connection, String databaseName, String schemaName, String tableName, String newTableName,boolean copyData) throws SQLException { public void copyTable(Connection connection, String databaseName, String schemaName, String tableName, String newTableName, boolean copyData) throws SQLException {
String sql = ""; String sql = "";
if(copyData){ if (copyData) {
sql = "CREATE TABLE " + newTableName + " AS SELECT * FROM " + tableName; sql = "CREATE TABLE " + newTableName + " AS SELECT * FROM " + tableName;
}else { } else {
sql = "CREATE TABLE " + newTableName + " AS SELECT * FROM " + tableName + " WHERE 1=0"; sql = "CREATE TABLE " + newTableName + " AS SELECT * FROM " + tableName + " WHERE 1=0";
} }
SQLExecutor.getInstance().execute(connection, sql, resultSet -> null); SQLExecutor.getInstance().execute(connection, sql, resultSet -> null);