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

Temporarily cancel the export trigger

Browse Source
This commit is contained in:
zgq
2024-07-11 16:40:06 +08:00
parent 493f934a2c
commit b332bd2d67
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -24,8 +24,8 @@ public class OracleDBManage extends DefaultDBManage implements DBManage {
exportTables(connection, databaseName, schemaName, asyncContext);
exportViews(connection, asyncContext, schemaName);
exportProcedures(connection, schemaName, asyncContext);
exportTriggers(connection, schemaName, asyncContext);
exportFunctions(connection, schemaName, asyncContext);
// exportTriggers(connection, schemaName, asyncContext);
}
private void exportTables(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {

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

@ -87,7 +87,7 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
try {
if (resultSet.next()) {
ddlBuilder.append(resultSet.getString("sql"));
ddlBuilder.append(resultSet.getString("sql")).append(";");
}
} catch (SQLException e) {
throw new RuntimeException(e);
@ -259,11 +259,15 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
function.setDatabaseName(databaseName);
function.setSchemaName(schemaName);
function.setFunctionName(functionName);
StringBuilder bodyBuilder = new StringBuilder("CREATE ");
StringBuilder bodyBuilder = new StringBuilder("CREATE OR REPLACE ");
while (resultSet.next()) {
bodyBuilder.append(resultSet.getString("TEXT")).append("\n");
}
function.setFunctionBody(bodyBuilder.toString());
String functionBody = bodyBuilder.toString().trim();
if (!functionBody.endsWith("/")) {
functionBody += "\n/";
}
function.setFunctionBody(functionBody);
return function;
});
@ -391,11 +395,15 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
procedure.setDatabaseName(databaseName);
procedure.setSchemaName(schemaName);
procedure.setProcedureName(procedureName);
StringBuilder bodyBuilder = new StringBuilder("CREATE ");
StringBuilder bodyBuilder = new StringBuilder("CREATE OR REPLACE ");
while (resultSet.next()) {
bodyBuilder.append(resultSet.getString("TEXT")).append("\n");
}
procedure.setProcedureBody(bodyBuilder.toString());
String procedureBody = bodyBuilder.toString().trim(); // 去掉最后的空白字符
if (!procedureBody.endsWith("/")) {
procedureBody += "\n/";
}
procedure.setProcedureBody(procedureBody);
return procedure;
});
}
@ -412,7 +420,7 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
table.setSchemaName(schemaName);
table.setName(viewName);
if (resultSet.next()) {
table.setDdl("CREATE VIEW " + viewName + " AS " + resultSet.getString("TEXT"));
table.setDdl("CREATE OR REPLACE VIEW " + viewName + " AS " + resultSet.getString("TEXT"));
}
return table;
});