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

Merge pull request #1216 from openai0229/fix-mysql-function-ddl-query

Browse Source 
fix-mysql-function-ddl-query
This commit is contained in:
robinji0
2024-03-09 20:58:43 +08:00
committed by GitHub
parent 5fb3c21973 7882c1a870
commit 73f420ec5e
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

@ -58,19 +58,25 @@ public class MysqlMetaData extends DefaultMetaService implements MetaData {
public Function function(Connection connection, @NotEmpty String databaseName, String schemaName,
String functionName) {
String sql = String.format(ROUTINES_SQL, "FUNCTION", databaseName, functionName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
Function function = new Function();
function.setDatabaseName(databaseName);
function.setSchemaName(schemaName);
function.setFunctionName(functionName);
String functionInfoSql = String.format(ROUTINES_SQL, "FUNCTION", databaseName, functionName);
Function function = SQLExecutor.getInstance().execute(connection, functionInfoSql, resultSet -> {
Function f = new Function();
f.setDatabaseName(databaseName);
f.setSchemaName(schemaName);
f.setFunctionName(functionName);
if (resultSet.next()) {
function.setSpecificName(resultSet.getString("SPECIFIC_NAME"));
function.setRemarks(resultSet.getString("ROUTINE_COMMENT"));
function.setFunctionBody(resultSet.getString("ROUTINE_DEFINITION"));
f.setSpecificName(resultSet.getString("SPECIFIC_NAME"));
f.setRemarks(resultSet.getString("ROUTINE_COMMENT"));
}
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;
}