opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 21:50:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix-mysql-function-ddl-query

Browse Source
This commit is contained in:
zgq
2024-03-09 16:24:31 +08:00
parent 1baa37820f
commit 7882c1a870
1 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlMetaData.java
View File

@ -57,19 +57,25 @@ public class MysqlMetaData extends DefaultMetaService implements MetaData {
public Function function(Connection connection, @NotEmpty String databaseName, String schemaName, public Function function(Connection connection, @NotEmpty String databaseName, String schemaName,
String functionName) { String functionName) {
String sql = String.format(ROUTINES_SQL, "FUNCTION", databaseName, functionName); String functionInfoSql = String.format(ROUTINES_SQL, "FUNCTION", databaseName, functionName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> { Function function = SQLExecutor.getInstance().execute(connection, functionInfoSql, resultSet -> {
Function function = new Function(); Function f = new Function();
function.setDatabaseName(databaseName); f.setDatabaseName(databaseName);
function.setSchemaName(schemaName); f.setSchemaName(schemaName);
function.setFunctionName(functionName); f.setFunctionName(functionName);
if (resultSet.next()) { if (resultSet.next()) {
function.setSpecificName(resultSet.getString("SPECIFIC_NAME")); f.setSpecificName(resultSet.getString("SPECIFIC_NAME"));
function.setRemarks(resultSet.getString("ROUTINE_COMMENT")); f.setRemarks(resultSet.getString("ROUTINE_COMMENT"));
function.setFunctionBody(resultSet.getString("ROUTINE_DEFINITION"));
} }
return function; return f;
}); });
String functionDDlSql =String.format("SHOW CREATE FUNCTION %s", functionName);
SQLExecutor.getInstance().execute(connection,functionDDlSql, resultSet -> {
if (resultSet.next()) {
function.setFunctionBody(resultSet.getString("Create Function"));
}
} );
return function;
} }