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); 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);
exportFunctions(connection, schemaName, asyncContext); exportFunctions(connection, schemaName, asyncContext);
// exportTriggers(connection, schemaName, asyncContext);
} }
private void exportTables(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException { 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 -> { SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
try { try {
if (resultSet.next()) { if (resultSet.next()) {
ddlBuilder.append(resultSet.getString("sql")); ddlBuilder.append(resultSet.getString("sql")).append(";");
} }
} catch (SQLException e) { } catch (SQLException e) {
throw new RuntimeException(e); throw new RuntimeException(e);
@ -259,11 +259,15 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
function.setDatabaseName(databaseName); function.setDatabaseName(databaseName);
function.setSchemaName(schemaName); function.setSchemaName(schemaName);
function.setFunctionName(functionName); function.setFunctionName(functionName);
StringBuilder bodyBuilder = new StringBuilder("CREATE "); StringBuilder bodyBuilder = new StringBuilder("CREATE OR REPLACE ");
while (resultSet.next()) { while (resultSet.next()) {
bodyBuilder.append(resultSet.getString("TEXT")).append("\n"); 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; return function;
}); });
@ -391,11 +395,15 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
procedure.setDatabaseName(databaseName); procedure.setDatabaseName(databaseName);
procedure.setSchemaName(schemaName); procedure.setSchemaName(schemaName);
procedure.setProcedureName(procedureName); procedure.setProcedureName(procedureName);
StringBuilder bodyBuilder = new StringBuilder("CREATE "); StringBuilder bodyBuilder = new StringBuilder("CREATE OR REPLACE ");
while (resultSet.next()) { while (resultSet.next()) {
bodyBuilder.append(resultSet.getString("TEXT")).append("\n"); 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; return procedure;
}); });
} }
@ -412,7 +420,7 @@ public class OracleMetaData extends DefaultMetaService implements MetaData {
table.setSchemaName(schemaName); table.setSchemaName(schemaName);
table.setName(viewName); table.setName(viewName);
if (resultSet.next()) { 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; return table;
}); });