opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-01 06:16:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix db2 procedure function query

Browse Source
This commit is contained in:
zgq
2024-04-15 21:46:32 +08:00
parent 81dc73c374
commit 082b212c7a
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/ClickHouseMetaData.java
View File

@ -225,7 +225,7 @@ public class ClickHouseMetaData extends DefaultMetaService implements MetaData {
table.setSchemaName(schemaName); table.setSchemaName(schemaName);
table.setName(viewName); table.setName(viewName);
if (resultSet.next()) { if (resultSet.next()) {
table.setDdl(resultSet.getString(1)); table.setDdl(resultSet.getString("create_table_query"));
} }
return table; return table;
}); });

6
chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2MetaData.java
View File

@ -115,7 +115,7 @@ public class DB2MetaData extends DefaultMetaService implements MetaData {
return table; return table;
} }
private static String ROUTINE_DDL_SQL="select TEXT from syscat.routines where ROUTINESCHEMA='%s' and ROUTINENAME='%s';"; private static String ROUTINE_DDL_SQL="select TEXT from syscat.routines where ROUTINESCHEMA='%s' and ROUTINENAME='%s' and ROUTINETYPE='%s';";
@Override @Override
public Function function(Connection connection, String databaseName, String schemaName, String functionName) { public Function function(Connection connection, String databaseName, String schemaName, String functionName) {
@ -123,7 +123,7 @@ public class DB2MetaData extends DefaultMetaService implements MetaData {
function.setDatabaseName(databaseName); function.setDatabaseName(databaseName);
function.setSchemaName(schemaName); function.setSchemaName(schemaName);
function.setFunctionName(functionName); function.setFunctionName(functionName);
String sql = String.format(ROUTINE_DDL_SQL, schemaName, functionName); String sql = String.format(ROUTINE_DDL_SQL, schemaName, functionName,'F');
SQLExecutor.getInstance().execute(connection, sql, resultSet -> { SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
if (resultSet.next()) { if (resultSet.next()) {
function.setFunctionBody(resultSet.getString("TEXT")+";"); function.setFunctionBody(resultSet.getString("TEXT")+";");
@ -138,7 +138,7 @@ public class DB2MetaData extends DefaultMetaService implements MetaData {
procedure.setDatabaseName(databaseName); procedure.setDatabaseName(databaseName);
procedure.setSchemaName(schemaName); procedure.setSchemaName(schemaName);
procedure.setProcedureName(procedureName); procedure.setProcedureName(procedureName);
String sql = String.format(ROUTINE_DDL_SQL, schemaName, procedureName); String sql = String.format(ROUTINE_DDL_SQL, schemaName, procedureName,'P');
SQLExecutor.getInstance().execute(connection, sql, resultSet -> { SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
if (resultSet.next()) { if (resultSet.next()) {
procedure.setProcedureBody(resultSet.getString("TEXT")+";"); procedure.setProcedureBody(resultSet.getString("TEXT")+";");