opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-28 10:12:39 +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

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;
}
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
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.setSchemaName(schemaName);
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 -> {
if (resultSet.next()) {
function.setFunctionBody(resultSet.getString("TEXT")+";");
@ -138,7 +138,7 @@ public class DB2MetaData extends DefaultMetaService implements MetaData {
procedure.setDatabaseName(databaseName);
procedure.setSchemaName(schemaName);
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 -> {
if (resultSet.next()) {
procedure.setProcedureBody(resultSet.getString("TEXT")+";");