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

feat(chat2db-oracle): fix oracle DDL

Browse Source
This commit is contained in:
zgq
2024-07-10 20:05:14 +08:00
parent c10a7fea86
commit 2560061e68
2 changed files with 72 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,11 +19,6 @@ import java.util.List;
@Slf4j @Slf4j
public class OracleDBManage extends DefaultDBManage implements DBManage { public class OracleDBManage extends DefaultDBManage implements DBManage {
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 " +
"FROM user_col_comments " +
"WHERE table_name = '%s' " +
"AND comments IS NOT NULL";
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);
@ -45,40 +40,14 @@ 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 tableDDL = Chat2DBContext.getMetaData().tableDDL(connection, databaseName, schemaName, tableName); String tableDDL = Chat2DBContext.getMetaData().tableDDL(connection, databaseName, schemaName, tableName);
StringBuilder sqlBuilder = new StringBuilder(); String sqlBuilder = "DROP TABLE " + SqlUtils.quoteObjectName(tableName) + ";" + tableDDL + "\n";
sqlBuilder.append("DROP TABLE ").append(SqlUtils.quoteObjectName(tableName)).append(";") asyncContext.write(sqlBuilder);
.append(tableDDL).append(";").append("\n");
asyncContext.write(sqlBuilder.toString());
exportTableComments(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 {
String sql = String.format(TABLE_COMMENT_SQL, tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append(resultSet.getString("table_comment_ddl")).append("\n");
asyncContext.write(sqlBuilder.toString());
}
}
}
private void exportTableColumnsComments(Connection connection, String tableName, AsyncContext asyncContext) throws SQLException {
String sql = String.format(TABLE_COLUMN_COMMENT_SQL, tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append(resultSet.getString("column_comment_ddl")).append("\n");
asyncContext.write(sqlBuilder.toString());
}
}
}
private void exportViews(Connection connection, AsyncContext asyncContext, String schemaName) throws SQLException { private void exportViews(Connection connection, AsyncContext asyncContext, String schemaName) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"VIEW"})) { try (ResultSet resultSet = connection.getMetaData().getTables(null, schemaName, null, new String[]{"VIEW"})) {

55
chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/OracleMetaData.java
View File

@ -5,6 +5,7 @@ import ai.chat2db.plugin.oracle.type.OracleColumnTypeEnum;
import ai.chat2db.plugin.oracle.type.OracleDefaultValueEnum; import ai.chat2db.plugin.oracle.type.OracleDefaultValueEnum;
import ai.chat2db.plugin.oracle.type.OracleIndexTypeEnum; import ai.chat2db.plugin.oracle.type.OracleIndexTypeEnum;
import ai.chat2db.plugin.oracle.value.OracleValueProcessor; import ai.chat2db.plugin.oracle.value.OracleValueProcessor;
import ai.chat2db.server.tools.common.util.EasyStringUtils;
import ai.chat2db.spi.MetaData; import ai.chat2db.spi.MetaData;
import ai.chat2db.spi.SqlBuilder; import ai.chat2db.spi.SqlBuilder;
import ai.chat2db.spi.ValueProcessor; import ai.chat2db.spi.ValueProcessor;
@ -30,14 +31,23 @@ import java.util.stream.Collectors;
public class OracleMetaData extends DefaultMetaService implements MetaData { public class OracleMetaData extends DefaultMetaService implements MetaData {
private static final String TABLE_DDL_SQL = "select dbms_metadata.get_ddl('TABLE','%s','%s') as sql from dual"; private static final String TABLE_DDL_SQL = "select dbms_metadata.get_ddl('TABLE','%s','%s') as sql from dual";
private static final String TABLE_COMMENT_SQL = "select owner, table_name, comments from ALL_TAB_COMMENTS where OWNER = '%s' and TABLE_NAME = '%s'";
private static final String TABLE_COLUMN_COMMENT_SQL = """
SELECT owner, table_name, column_name, comments
FROM all_col_comments
WHERE owner = '%s' and table_name = '%s' and comments is not null""";
private List<String> systemSchemas = Arrays.asList("ANONYMOUS", "APEX_030200", "APEX_PUBLIC_USER", "APPQOSSYS", "BI", "CTXSYS", "DBSNMP", "DIP", "EXFSYS", "FLOWS_FILES", "HR", "IX", "MDDATA", "MDSYS", "MGMT_VIEW", "OE", "OLAPSYS", "ORACLE_OCM", "ORDDATA", "ORDPLUGINS", "ORDSYS", "OUTLN", "OWBSYS", "OWBSYS_AUDIT", "PM", "SCOTT", "SH", "SI_INFORMTN_SCHEMA", "SPATIAL_CSW_ADMIN_USR", "SPATIAL_WFS_ADMIN_USR", "SYS", "SYSMAN", "SYSTEM", "WMSYS", "XDB", "XS$NULL"); private List<String> systemSchemas = Arrays.asList("ANONYMOUS", "APEX_030200", "APEX_PUBLIC_USER", "APPQOSSYS", "BI", "CTXSYS", "DBSNMP", "DIP", "EXFSYS", "FLOWS_FILES", "HR", "IX", "MDDATA", "MDSYS", "MGMT_VIEW", "OE", "OLAPSYS", "ORACLE_OCM", "ORDDATA", "ORDPLUGINS", "ORDSYS", "OUTLN", "OWBSYS", "OWBSYS_AUDIT", "PM", "SCOTT", "SH", "SI_INFORMTN_SCHEMA", "SPATIAL_CSW_ADMIN_USR", "SPATIAL_WFS_ADMIN_USR", "SYS", "SYSMAN", "SYSTEM", "WMSYS", "XDB", "XS$NULL");
private String PROCEDURE_LIST_DDL = """ private static final String PROCEDURE_LIST_DDL = """
SELECT OBJECT_NAME, OBJECT_TYPE SELECT OBJECT_NAME, OBJECT_TYPE
FROM ALL_OBJECTS FROM ALL_OBJECTS
WHERE OBJECT_TYPE IN ('PROCEDURE') WHERE OBJECT_TYPE IN ('PROCEDURE')
AND OWNER = '%s'"""; AND OWNER = '%s'""";
public static final String TABLE_INDEX_DDL_SQL = """
SELECT DBMS_METADATA.GET_DDL('INDEX', index_name, table_owner) AS ddl
FROM all_indexes
WHERE table_owner = '%s' AND table_name = '%s'""";
@Override @Override
public List<Procedure> procedures(Connection connection, String databaseName, String schemaName) { public List<Procedure> procedures(Connection connection, String databaseName, String schemaName) {
@ -61,18 +71,53 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
@Override @Override
public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) { public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
// TODO: only_read user can not get ddl
String sql = String.format(TABLE_DDL_SQL, tableName, schemaName); String sql = String.format(TABLE_DDL_SQL, tableName, schemaName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> { String tableCommentSql = String.format(TABLE_COMMENT_SQL, schemaName, tableName);
String tableColumnCommentSql = String.format(TABLE_COLUMN_COMMENT_SQL, schemaName, tableName);
String tableIndexSql = String.format(TABLE_INDEX_DDL_SQL, schemaName, tableName);
StringBuilder ddlBuilder = new StringBuilder();
SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
try { try {
if (resultSet.next()) { if (resultSet.next()) {
return resultSet.getString("sql"); ddlBuilder.append(resultSet.getString("sql"));
} }
} catch (SQLException e) { } catch (SQLException e) {
throw new RuntimeException(e); throw new RuntimeException(e);
} }
return null;
}); });
SQLExecutor.getInstance().execute(connection, tableCommentSql, resultSet -> {
if (resultSet.next()) {
String tableComment = resultSet.getString("comments");
if (StringUtils.isNotBlank(tableComment)) {
ddlBuilder.append("\nCOMMENT ON TABLE ").append(SqlUtils.quoteObjectName(tableName)).append(" IS ")
.append(EasyStringUtils.escapeAndQuoteString(tableComment)).append(";");
}
}
});
SQLExecutor.getInstance().execute(connection, tableColumnCommentSql, resultSet -> {
while (resultSet.next()) {
String columnName = resultSet.getString("column_name");
String columnComment = resultSet.getString("comments");
if (StringUtils.isNotBlank(columnComment)) {
ddlBuilder.append("\nCOMMENT ON COLUMN ")
.append(SqlUtils.quoteObjectName(tableName)).append(".")
.append(SqlUtils.quoteObjectName(columnName)).append(" IS ")
.append(EasyStringUtils.escapeAndQuoteString(columnComment)).append(";");
}
}
});
SQLExecutor.getInstance().execute(connection, tableIndexSql, resultSet -> {
while (resultSet.next()) {
String ddl = resultSet.getString("ddl");
if (StringUtils.isNotBlank(ddl)) {
ddlBuilder.append("\n").append(ddl).append(";");
}
}
});
return ddlBuilder.toString();
} }
private static String SELECT_TABLE_SQL = "SELECT A.OWNER, A.TABLE_NAME, B.COMMENTS " + private static String SELECT_TABLE_SQL = "SELECT A.OWNER, A.TABLE_NAME, B.COMMENTS " +